From ce57f5953137cfb5509de401d7400d2fc5ff2f33 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 17:17:12 -0700 Subject: [PATCH] test_gemini_pass_through_endpoint --- tests/proxy_unit_tests/test_proxy_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/proxy_unit_tests/test_proxy_server.py b/tests/proxy_unit_tests/test_proxy_server.py index c5aff27501..b5117a3a2a 100644 --- a/tests/proxy_unit_tests/test_proxy_server.py +++ b/tests/proxy_unit_tests/test_proxy_server.py @@ -1833,7 +1833,7 @@ async def test_gemini_pass_through_endpoint(): scope = { "type": "http", "method": "POST", - "path": "/gemini/v1beta/models/gemini-1.5-flash:countTokens", + "path": "/gemini/v1beta/models/gemini-2.5-flash:countTokens", "query_string": b"key=sk-1234", "headers": [ (b"content-type", b"application/json"), @@ -1850,7 +1850,7 @@ async def test_gemini_pass_through_endpoint(): ) resp = await gemini_proxy_route( - endpoint="v1beta/models/gemini-1.5-flash:countTokens?key=sk-1234", + endpoint="v1beta/models/gemini-2.5-flash:countTokens?key=sk-1234", request=request, fastapi_response=Response(), )