fix(vertex_ai_partner_models.py): fix vertex import

This commit is contained in:
Krrish Dholakia
2024-08-28 18:08:33 -07:00
parent a857f4a8ee
commit dd9c5d10bd
@@ -83,7 +83,7 @@ class VertexAIPartnerModels(BaseLLM):
from litellm.llms.databricks import DatabricksChatCompletion
from litellm.llms.openai import OpenAIChatCompletion
from litellm.llms.text_completion_codestral import CodestralTextCompletion
from litellm.llms.vertex_ai_and_google_ai_studio.vertex_and_google_ai_studio_gemini import (
from litellm.llms.vertex_ai_and_google_ai_studio.gemini.vertex_and_google_ai_studio_gemini import (
VertexLLM,
)
except Exception: