From 5885ee5e14276b1bbee499afc54a334e2cc4a461 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 11 Dec 2024 12:07:22 -0800 Subject: [PATCH] fix test_vertexai_model_garden_model_completion --- litellm/llms/vertex_ai/vertex_model_garden/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/vertex_ai/vertex_model_garden/main.py b/litellm/llms/vertex_ai/vertex_model_garden/main.py index 0f904ba4f4d..8d1e0c9db2f 100644 --- a/litellm/llms/vertex_ai/vertex_model_garden/main.py +++ b/litellm/llms/vertex_ai/vertex_model_garden/main.py @@ -76,7 +76,7 @@ class VertexAIModelGardenModels(VertexBase): from google.cloud import aiplatform from litellm.llms.openai_like.chat.handler import OpenAILikeChatHandler - from litellm.llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gemini import ( + from litellm.llms.vertex_ai.gemini.vertex_and_google_ai_studio_gemini import ( VertexLLM, ) except Exception: