处理异常

new-bing/EdgeGPT.py

@@ -255,7 +255,7 @@ class ChatHub:
             extra_headers=HEADERS,
             max_size=None,
             ssl=ssl_context,
-            open_timeout=30,
+            open_timeout=45,
         )
         await self.__initial_handshake()
         # Construct a ChatHub request

new-bing/app.py

@@ -56,6 +56,7 @@ async def ws_chat(request, ws):
         logger.warn('Receive data: %s', data)
         sid = data['sid']
         q = data['q']
+        try:
             async for response in get_bot(sid).ask_stream(q, conversation_style=ConversationStyle.creative):
                 final, res = response
                 if final:
@@ -73,6 +74,23 @@ async def ws_chat(request, ws):
                         'final': final,
                         'data': res
                     }))
+        except Exception as e:
+            logger.error(e)
+            await ws.send(
+                raw_json.dumps({
+                    'final': True,
+                    'data': {
+                        'data': {
+                            'status': 'Error',
+                            'text': str(e),
+                            'suggests': [q],
+                            'message': str(e),
+                            'num_in_conversation': -1,
+                        },
+                        'cookie': os.environ.get('COOKIE_FILE'),
+                    }
+                })
+            )
 
 
 def get_bot(sid):