diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c38eeeccf6..33eaae4725 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1866,7 +1866,7 @@ def select_data_generator(response, user_api_key_dict): # since boto3 - sagemaker does not support async calls, we should use a sync data_generator if hasattr( response, "custom_llm_provider" - ) and response.custom_llm_provider in ["sagemaker", "bedrock"]: + ) and response.custom_llm_provider in ["sagemaker"]: return data_generator( response=response, )