scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

fix: fix imports

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-14 18:21:42 -07:00
parent de9bd9de61
commit edcda3170c
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -3552,7 +3552,7 @@ def embedding(
client=client,
timeout=timeout,
aembedding=aembedding,
litellm_params=litellm_params,
litellm_params={},
api_base=api_base,
print_verbose=print_verbose,
extra_headers=extra_headers,

2
litellm/tests/test_bedrock_completion.py
View file

@ -27,7 +27,7 @@ from litellm import (
completion_cost,
embedding,
)
from litellm.llms.bedrock.chat.invoke_handler import BedrockLLM, ToolBlock
from litellm.llms.bedrock.chat import BedrockLLM
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
from litellm.llms.prompt_templates.factory import _bedrock_tools_pt

2
litellm/tests/test_secret_manager.py
View file

@ -17,7 +17,7 @@ sys.path.insert(
import pytest
from litellm.llms.AzureOpenAI.azure import get_azure_ad_token_from_oidc
from litellm.llms.bedrock.chat.invoke_handler import BedrockConverseLLM, BedrockLLM
from litellm.llms.bedrock.chat import BedrockConverseLLM, BedrockLLM
from litellm.secret_managers.aws_secret_manager import load_aws_secret_manager
from litellm.secret_managers.main import get_secret