diff --git a/litellm/utils.py b/litellm/utils.py index b95d54eb79..b5c1f4a316 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -9736,6 +9736,7 @@ class CustomStreamWrapper: or self.custom_llm_provider == "predibase" or self.custom_llm_provider == "databricks" or self.custom_llm_provider == "bedrock" + or self.custom_llm_provider == "watsonx" or self.custom_llm_provider in litellm.openai_compatible_endpoints ): async for chunk in self.completion_stream: