From afd0d7cb41718f7b058e98ca59af14a7979f6514 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 10 Feb 2024 18:29:01 -0800 Subject: [PATCH] (fix) undo today's mistake where bedrock was async --- litellm/proxy/proxy_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c38eeeccf61..33eaae47251 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, )