diff --git a/litellm/tests/test_bedrock_completion.py b/litellm/tests/test_bedrock_completion.py index 873e6cc21..d672ca780 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 4fcc429ee..5f7afc119 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