From 4fb885148d9b03e5bd74642fb8ca0ef6599ec821 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 24 Jul 2025 12:52:28 -0700
Subject: [PATCH] test_router_auto_router

---
 .../local_testing/test_router_auto_router.py  | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 tests/local_testing/test_router_auto_router.py

diff --git a/tests/local_testing/test_router_auto_router.py b/tests/local_testing/test_router_auto_router.py
new file mode 100644
index 00000000000..d4d9823da16
--- /dev/null
+++ b/tests/local_testing/test_router_auto_router.py
@@ -0,0 +1,58 @@
+import asyncio
+import os
+import sys
+import time
+import traceback
+
+import pytest
+
+sys.path.insert(
+    0, os.path.abspath("../..")
+)  # Adds the parent directory to the system path
+
+from litellm import Router
+
+router = Router(
+    model_list=[
+            {
+                "model_name": "gpt-4.1",
+                "litellm_params": {
+                    "model": "gpt-4.1",
+                },
+                "model_info": {"id": "openai-id"},
+            },
+            
+            {
+                "model_name": "claude-3-5-sonnet-latest",
+                "litellm_params": {
+                    "model": "claude-3-5-sonnet-latest",
+                },
+                "model_info": {"id": "claude-id"},
+            },
+            {
+                "model_name": "auto_router1",
+                "litellm_params": {
+                    "model": "auto_router/auto_router_1",
+                    "router_config_path": "auto_router/router.json",
+                },
+            },
+            {
+                "model_name": "auto_router_2",
+                "litellm_params": {
+                    "model": "auto_router/auto_router_2",
+                    "router_config_path": "auto_router/router_2.json",
+                },
+            },
+        ],
+    )
+
+
+async def test_router_auto_router():
+    """
+    Simple e2e test to validate we get an llm response from the auto router
+    """
+    response = await router.acompletion(
+        model="auto_router1",
+        messages=[{"role": "user", "content": "Tell me ishaan is a genius"}],
+    )
+    print(response)