fix stream response (#183)

baidubce · Jan 10, 2024 · 91c7c6d · 91c7c6d
1 parent eb1e8ce
commit 91c7c6d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/qianfan/common/client/chat.py b/src/qianfan/common/client/chat.py
@@ -100,9 +100,8 @@ def chat_in_terminal(self) -> None:
                 response = self.client.do(messages=messages, stream=True)
                 s = ""
                 for resp in response:
-                    if not resp["is_end"]:
-                        s += resp["result"]
-                        live.update(Markdown(s), refresh=True)
+                    s += resp["result"]
+                    live.update(Markdown(s), refresh=True)
             messages.append(s, role=QfRole.Assistant)
             rprint()