From 08b409bae8f2f34ebdc43822fb05e92e6c2cf2ac Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 17 Jan 2024 17:17:58 -0800 Subject: [PATCH] fix(utils.py): fix if check --- litellm/tests/test_bedrock_completion.py | 1 - litellm/utils.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/tests/test_bedrock_completion.py b/litellm/tests/test_bedrock_completion.py index 873e6cc21e..d672ca7801 100644 --- a/litellm/tests/test_bedrock_completion.py +++ b/litellm/tests/test_bedrock_completion.py @@ -77,7 +77,6 @@ def test_completion_bedrock_claude_2_1_completion_auth(): os.environ.pop("AWS_ACCESS_KEY_ID", None) os.environ.pop("AWS_SECRET_ACCESS_KEY", None) os.environ.pop("AWS_REGION_NAME", None) - try: response = completion( model="bedrock/anthropic.claude-v2:1", diff --git a/litellm/utils.py b/litellm/utils.py index 4fcc429ee4..5f7afc1196 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3198,7 +3198,7 @@ def get_optional_params( special_params = passed_params.pop("kwargs") for k, v in special_params.items(): if k.startswith("aws_") and ( - custom_llm_provider != "bedrock" or custom_llm_provider != "sagemaker" + custom_llm_provider != "bedrock" and custom_llm_provider != "sagemaker" ): # allow dynamically setting boto3 init logic continue passed_params[k] = v