fix test_vertexai_model_garden_model_completion

This commit is contained in:
Ishaan Jaff 2024-12-11 12:07:22 -08:00
parent 9f32631592
commit 5885ee5e14

View file

@ -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: