phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Added watsonx to stream wrapper __anext__

Browse source
This commit is contained in:
Simon Sanchez Viloria 2024-07-07 19:21:50 +02:00
parent e2827ee28b
commit c041b75143
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/utils.py
View file

@ -9736,6 +9736,7 @@ class CustomStreamWrapper:
or self.custom_llm_provider == "predibase" or self.custom_llm_provider == "predibase"
or self.custom_llm_provider == "databricks" or self.custom_llm_provider == "databricks"
or self.custom_llm_provider == "bedrock" or self.custom_llm_provider == "bedrock"
or self.custom_llm_provider == "watsonx"
or self.custom_llm_provider in litellm.openai_compatible_endpoints or self.custom_llm_provider in litellm.openai_compatible_endpoints
): ):
async for chunk in self.completion_stream: async for chunk in self.completion_stream: