fixed bug in AsyncUnify streaming.

unifyai · Sep 13, 2024 · 0571034 · 0571034
1 parent 54dbda2
commit 0571034
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/unify/chat/clients/uni_llm.py b/unify/chat/clients/uni_llm.py
@@ -613,7 +613,8 @@ async def _generate_stream(
                 self.set_provider(chunk.model.split("@")[-1])
                 if message_content_only:
                     yield chunk.choices[0].delta.content or ""
-                yield chunk
+                else:
+                    yield chunk
         except openai.APIStatusError as e:
             raise Exception(e.message)