From b8386ec01e60229e096724eaf44571d94938849b Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 24 Jul 2025 16:20:00 -0700
Subject: [PATCH] test_router_auto_router

---
 .../local_testing/test_router_auto_router.py  | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/tests/local_testing/test_router_auto_router.py b/tests/local_testing/test_router_auto_router.py
index c936c420ac9..6792242babe 100644
--- a/tests/local_testing/test_router_auto_router.py
+++ b/tests/local_testing/test_router_auto_router.py
@@ -15,7 +15,16 @@ from litellm import Router
 current_path = os.path.dirname(os.path.abspath(__file__))
 router_json_path = os.path.join(current_path, "auto_router", "router.json")
 
-router = Router(
+
+@pytest.mark.asyncio
+async def test_router_auto_router():
+    """
+    Simple e2e test to validate we get an llm response from the auto router
+    """
+    import litellm
+    litellm._turn_on_debug()
+
+    router = Router(
     model_list=[
             {
                 "model_name": "custom-text-embedding-model",
@@ -68,14 +77,6 @@ router = Router(
     )
 
 
-@pytest.mark.asyncio
-async def test_router_auto_router():
-    """
-    Simple e2e test to validate we get an llm response from the auto router
-    """
-    import litellm
-    litellm._turn_on_debug()
-
     # this goes to gpt-4.1
     # these are the utterances in the router.json file
     response = await router.acompletion(