From c041b751439a197e19c9c9e3f570843e1b7cb30c Mon Sep 17 00:00:00 2001
From: Simon Sanchez Viloria <simon.s.viloria@ibm.com>
Date: Sun, 7 Jul 2024 19:21:50 +0200
Subject: [PATCH] Added watsonx to stream wrapper __anext__

---
 litellm/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/utils.py b/litellm/utils.py
index b95d54eb7..b5c1f4a31 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: