diff --git a/litellm/tests/test_bedrock_completion.py b/litellm/tests/test_bedrock_completion.py index 07b0cb2883..873e6cc21e 100644 --- a/litellm/tests/test_bedrock_completion.py +++ b/litellm/tests/test_bedrock_completion.py @@ -185,6 +185,16 @@ def test_completion_bedrock_claude_sts_client_auth(): aws_role_name=aws_role_name, aws_session_name="my-test-session", ) + + response = completion( + model="gpt-3.5-turbo", + messages=messages, + aws_region_name="us-east-1", + aws_access_key_id=aws_access_key_id, + aws_secret_access_key=aws_secret_access_key, + aws_role_name=aws_role_name, + aws_session_name="my-test-session", + ) # Add any assertions here to check the response print(response) except RateLimitError: diff --git a/litellm/utils.py b/litellm/utils.py index 45bdca7fc6..4fcc429ee4 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3197,6 +3197,10 @@ def get_optional_params( passed_params = locals() 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" + ): # allow dynamically setting boto3 init logic + continue passed_params[k] = v default_params = { "functions": None,