Added frontend messages for common errors.

Author: wehos

main_helper/core.py

@@ -250,7 +250,14 @@ async def send_lanlan_response(self, text: str, is_first_chunk: bool = False):
         except Exception as e:
             logger.error(f"💥 WS Send Lanlan Response Error: {e}")
 
-    async def handle_connection_error(self):
+    async def handle_connection_error(self, message=None):
+        if message:
+            if '欠费' in message:
+                await self.send_status("💥 智谱API触发欠费bug。请考虑充值1元。")
+            elif 'standing' in message:
+                await self.send_status("💥 阿里API已欠费。")
+            else:
+                await self.send_status(message)
         logger.info("💥 Session closed by API Server.")
         await self.disconnected_by_server()
 
@@ -358,6 +365,10 @@ async def start_session(self, websocket: WebSocket, new=False):
             logger.error(f"💥 {error_message}")
             traceback.print_exc()
             await self.send_status(error_message)
+            if 'actively refused it' in str(e):
+                await self.send_status("💥 记忆服务器已崩溃。请检查API Key是否正确。")
+            elif '401' in str(e):
+                await self.send_status("💥 API Key被服务器拒绝。请检查API Key是否与所选模型匹配。")
             await self.cleanup()
 
     async def send_user_activity(self):

main_helper/omni_realtime_client.py

@@ -65,7 +65,7 @@ def __init__(
         on_interrupt: Optional[Callable[[], Awaitable[None]]] = None,
         on_input_transcript: Optional[Callable[[str], Awaitable[None]]] = None,
         on_output_transcript: Optional[Callable[[str, bool], Awaitable[None]]] = None,
-        on_connection_error: Optional[Callable[[], Awaitable[None]]] = None,
+        on_connection_error: Optional[Callable[[str], Awaitable[None]]] = None,
         on_response_done: Optional[Callable[[], Awaitable[None]]] = None,
         extra_event_handlers: Optional[Dict[str, Callable[[Dict[str, Any]], Awaitable[None]]]] = None
     ):
@@ -247,6 +247,10 @@ async def handle_messages(self) -> None:
                 #     print(f"Event type: {event_type}")
                 if event_type == "error":
                     logger.error(f"API Error: {event['error']}")
+                    if '欠费' in event['error'] or 'standing' in event['error']:
+                        if self.handle_connection_error:
+                            await self.handle_connection_error(event['error'])
+                        await self.close()
                     continue
                 elif event_type == "response.done":
                     self._is_responding = False