Fix the response check error based the new opanai response (#1468)

* Fix the response check error based the new opanai response The new openai response message contain a message without "message", so just continue to fix the whole programe exit. * better checks --------- Co-authored-by: Antonio Cheong <[email protected]>
acheong08 · Jul 20, 2023 · 9217c0c · 9217c0c
1 parent 8feb200
commit 9217c0c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/revChatGPT/V1.py b/src/revChatGPT/V1.py
@@ -573,7 +573,8 @@ def __send_request(
             except json.decoder.JSONDecodeError:
                 continue
             if not self.__check_fields(line):
-                raise ValueError(f"Field missing. Details: {str(line)}")
+                if line.get("is_completion"):
+                    raise ValueError(f"Field missing. Details: {str(line)}")
             if line.get("message").get("author").get("role") != "assistant":
                 continue
 
@@ -1140,8 +1141,10 @@ async def __send_request(
                     line = json.loads(line)
                 except json.decoder.JSONDecodeError:
                     continue
+
                 if not self.__check_fields(line):
-                    raise ValueError(f"Field missing. Details: {str(line)}")
+                    if line.get("is_completion"):
+                        raise ValueError(f"Field missing. Details: {str(line)}")
                 if line.get("message").get("author").get("role") != "assistant":
                     continue