Merge pull request #377 from maxduke/patch-3

fix "the JSON object must be str, bytes or bytearray, not dict"
chatpire · Feb 7, 2024 · 5cc42b8 · 5cc42b8
2 parents 622fca5 + b0f669d
commit 5cc42b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/backend/api/sources/openai_web.py b/backend/api/sources/openai_web.py
@@ -408,7 +408,10 @@ async def complete(self, model: OpenaiWebChatModels, text_content: str, use_team
                 if "[DONE]" in line:
                     break
                 try:
-                    data = json.loads(line)
+                    if not isinstance(line, dict):
+                        data = json.loads(line)
+                    else:
+                        data = line
                     if not _check_fields(data):
                         if "error" in data:
                             logger.warning(f"error in message stream: {line}")