fix test_gemini_pass_through_endpoint

This commit is contained in:
Ishaan Jaff 2024-11-20 22:09:35 -08:00
parent 19ae080d3f
commit 337f168dc0

View file

@ -1794,7 +1794,7 @@ async def test_add_callback_via_key_litellm_pre_call_utils_langsmith(
async def test_gemini_pass_through_endpoint(): async def test_gemini_pass_through_endpoint():
from starlette.datastructures import URL 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, Request,
Response, Response,
gemini_proxy_route, gemini_proxy_route,