fixed bug: Additional properties are not allowed ('stream' was unexpe…

…cted)
yym68686 · Nov 14, 2023 · 40887d7 · 40887d7
1 parent 7a742a0
commit 40887d7
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/chatgpt2api/chatgpt2api.py b/chatgpt2api/chatgpt2api.py
@@ -81,7 +81,6 @@ def dall_e_3(
         json_post = {
                 "model": os.environ.get("IMAGE_MODEL_NAME") or model or self.engine,
                 "prompt": prompt,
-                "stream": True,
                 "n": 1,
                 "size": "1024x1024",
         }
@@ -96,10 +95,9 @@ def dall_e_3(
             raise t.APIConnectionError(
                 f"{response.status_code} {response.reason} {response.text}",
             )
-        for line in response.iter_lines():
-            json_data = json.loads(line.decode("utf-8"))
-            url = json_data["data"][0]["url"]
-            yield url
+        json_data = json.loads(response.text)
+        url = json_data["data"][0]["url"]
+        yield url
 
 class Chatbot:
     """