From 54811cf595aabb0ed003b55f80c08ae624f2263a Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 12 Feb 2025 18:14:35 -0800
Subject: [PATCH] test_async_router_context_window_fallback

---
 tests/local_testing/test_router.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py
index f4fbeaff1c..fd4922fbb8 100644
--- a/tests/local_testing/test_router.py
+++ b/tests/local_testing/test_router.py
@@ -768,7 +768,10 @@ async def test_async_router_context_window_fallback(sync_mode):
         else:
             response = router.completion(
                 model="gpt-4",
-                messages=[{"role": "user", "content": "Who was Alexander?"}],
+                messages=[
+                    {"role": "system", "content": text * 2},
+                    {"role": "user", "content": "Who was Alexander?"},
+                ],
             )
             assert "gpt-4-turbo" in response.model
     except Exception as e: