fix(utils.py): fix if check

This commit is contained in:
Krrish Dholakia 2024-01-17 17:17:58 -08:00
parent 6dc39e4ed1
commit 08b409bae8
2 changed files with 1 additions and 2 deletions

View file

@ -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",

View file

@ -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