phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): fix if check

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-17 17:17:58 -08:00
parent 94c8eab987
commit d4404fb61e
2 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_bedrock_completion.py
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_ACCESS_KEY_ID", None)
os.environ.pop("AWS_SECRET_ACCESS_KEY", None) os.environ.pop("AWS_SECRET_ACCESS_KEY", None)
os.environ.pop("AWS_REGION_NAME", None) os.environ.pop("AWS_REGION_NAME", None)
try: try:
response = completion( response = completion(
model="bedrock/anthropic.claude-v2:1", model="bedrock/anthropic.claude-v2:1",

2
litellm/utils.py
View file

@ -3198,7 +3198,7 @@ def get_optional_params(
special_params = passed_params.pop("kwargs") special_params = passed_params.pop("kwargs")
for k, v in special_params.items(): for k, v in special_params.items():
if k.startswith("aws_") and ( 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 ): # allow dynamically setting boto3 init logic
continue continue
passed_params[k] = v passed_params[k] = v