From edcda3170c9d5466c971c54f19a5b6c3c24bc6df Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 14 Sep 2024 18:21:42 -0700 Subject: [PATCH] fix: fix imports --- litellm/main.py | 2 +- litellm/tests/test_bedrock_completion.py | 2 +- litellm/tests/test_secret_manager.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 3f1d57f6e25..53c60213b9c 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -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, diff --git a/litellm/tests/test_bedrock_completion.py b/litellm/tests/test_bedrock_completion.py index f80ace9fa4f..d14363983e1 100644 --- a/litellm/tests/test_bedrock_completion.py +++ b/litellm/tests/test_bedrock_completion.py @@ -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 diff --git a/litellm/tests/test_secret_manager.py b/litellm/tests/test_secret_manager.py index 0ca3a6cff22..397128ecb06 100644 --- a/litellm/tests/test_secret_manager.py +++ b/litellm/tests/test_secret_manager.py @@ -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