fix(utils.py): fix if check

BerriAI · Jan 18, 2024 · 08b409b · 08b409b
1 parent 6dc39e4
commit 08b409b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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