🐛 fix error from cohere api

bruceunx · May 2, 2024 · 96d5d16 · 96d5d16
1 parent 0844538
commit 96d5d16
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gpt_server/routers/cohere.py b/gpt_server/routers/cohere.py
@@ -26,6 +26,8 @@ async def get_content(data):
             async for chunk in res.content:
                 if chunk:
                     tmp = json.loads(chunk.decode("utf-8"))
+                    if 'text' not in tmp:
+                        continue
                     if tmp["is_finished"]:
                         yield "[DONE]"
                     else: