close #632

acheong08 · Aug 2, 2023 · 7126349 · 7126349
1 parent 5e37e6f
commit 7126349
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/src/EdgeGPT/chathub.py b/src/EdgeGPT/chathub.py
@@ -154,15 +154,19 @@ async def ask_stream(
                             )
                             == "GenerateContentQuery"
                         ):
-                            async with ImageGenAsync(
-                                all_cookies=self.cookies,
-                            ) as image_generator:
-                                images = await image_generator.get_images(
-                                    response["arguments"][0]["messages"][0]["text"],
-                                )
-                            for i, image in enumerate(images):
-                                resp_txt = f"{resp_txt}\n![image{i}]({image})"
-                            draw = True
+                            try:
+                                async with ImageGenAsync(
+                                    all_cookies=self.cookies,
+                                ) as image_generator:
+                                    images = await image_generator.get_images(
+                                        response["arguments"][0]["messages"][0]["text"],
+                                    )
+                                for i, image in enumerate(images):
+                                    resp_txt = f"{resp_txt}\n![image{i}]({image})"
+                                draw = True
+                            except Exception as e:
+                                print(e)
+                                continue
                         if (
                             (
                                 response["arguments"][0]["messages"][0]["contentOrigin"]