From 337f168dc0dc884a8938c52c44acdd9784d76281 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 20 Nov 2024 22:09:35 -0800 Subject: [PATCH] fix test_gemini_pass_through_endpoint --- tests/proxy_unit_tests/test_proxy_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/proxy_unit_tests/test_proxy_server.py b/tests/proxy_unit_tests/test_proxy_server.py index b1c00ce75..d70962858 100644 --- a/tests/proxy_unit_tests/test_proxy_server.py +++ b/tests/proxy_unit_tests/test_proxy_server.py @@ -1794,7 +1794,7 @@ async def test_add_callback_via_key_litellm_pre_call_utils_langsmith( async def test_gemini_pass_through_endpoint(): from starlette.datastructures import URL - from litellm.proxy.vertex_ai_endpoints.google_ai_studio_endpoints import ( + from litellm.proxy.pass_through_endpoints.llm_passthrough_endpoints import ( Request, Response, gemini_proxy_route,