phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fix imports

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-26 21:36:04 -07:00
parent 4868a6cf55
commit 5aad9d2db7
4 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/vertex_ai_and_google_ai_studio/vertex_ai_anthropic.py
View file

@ -205,7 +205,9 @@ def get_vertex_client(
vertex_credentials: Optional[str], vertex_credentials: Optional[str],
) -> Tuple[Any, Optional[str]]: ) -> Tuple[Any, Optional[str]]:
args = locals() args = locals()
from litellm.llms.vertex_httpx import VertexLLM from litellm.llms.vertex_ai_and_google_ai_studio.vertex_and_google_ai_studio_gemini import (
VertexLLM,
)
try: try:
from anthropic import AnthropicVertex from anthropic import AnthropicVertex
@ -268,7 +270,9 @@ def completion(
from anthropic import AnthropicVertex from anthropic import AnthropicVertex
from litellm.llms.anthropic import AnthropicChatCompletion from litellm.llms.anthropic import AnthropicChatCompletion
from litellm.llms.vertex_httpx import VertexLLM from litellm.llms.vertex_ai_and_google_ai_studio.vertex_and_google_ai_studio_gemini import (
VertexLLM,
)
except: except:
raise VertexAIError( raise VertexAIError(
status_code=400, status_code=400,

4
litellm/llms/vertex_ai_and_google_ai_studio/vertex_ai_partner_models.py
View file

@ -126,7 +126,9 @@ class VertexAIPartnerModels(BaseLLM):
from litellm.llms.databricks import DatabricksChatCompletion from litellm.llms.databricks import DatabricksChatCompletion
from litellm.llms.openai import OpenAIChatCompletion from litellm.llms.openai import OpenAIChatCompletion
from litellm.llms.text_completion_codestral import CodestralTextCompletion from litellm.llms.text_completion_codestral import CodestralTextCompletion
from litellm.llms.vertex_httpx import VertexLLM from litellm.llms.vertex_ai_and_google_ai_studio.vertex_and_google_ai_studio_gemini import (
VertexLLM,
)
except Exception: except Exception:
raise VertexAIError( raise VertexAIError(

4
litellm/tests/test_amazing_vertex_completion.py
View file

@ -2065,7 +2065,9 @@ def test_prompt_factory_nested():
def test_get_token_url(): def test_get_token_url():
from litellm.llms.vertex_httpx import VertexLLM from litellm.llms.vertex_ai_and_google_ai_studio.vertex_and_google_ai_studio_gemini import (
VertexLLM,
)
vertex_llm = VertexLLM() vertex_llm = VertexLLM()
vertex_ai_project = "adroit-crow-413218" vertex_ai_project = "adroit-crow-413218"

4
litellm/utils.py
View file

@ -4131,7 +4131,9 @@ def get_api_base(
_optional_params.vertex_location is not None _optional_params.vertex_location is not None
and _optional_params.vertex_project is not None and _optional_params.vertex_project is not None
): ):
from litellm.llms.vertex_ai_anthropic import create_vertex_anthropic_url from litellm.llms.vertex_ai_and_google_ai_studio.vertex_ai_anthropic import (
create_vertex_anthropic_url,
)
if "claude" in model: if "claude" in model:
_api_base = create_vertex_anthropic_url( _api_base = create_vertex_anthropic_url(