Update api/utils.py

api/utils.py

@@ -103,29 +103,38 @@ async def process_streaming_response(request: ChatRequest):
                 json=json_data,
                 timeout=100,
             ) as response:
-                response.raise_for_status()
-                async for line in response.aiter_lines():
+                try:
+                    response.raise_for_status()
+                    async for line in response.aiter_lines():
+                        timestamp = int(datetime.now().timestamp())
+                        if line:
+                            content = line + "\n"
+                            if "https://www.blackbox.ai" in content:
+                                validate.getHid(True)
+                                content = "hid已刷新,重新对话即可\n"
+                                yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
+                                break
+                            if content.startswith("$@$v=undefined-rv1$@$"):
+                                yield f"data: {json.dumps(create_chat_completion_data(content[21:], request.model, timestamp))}\n\n"
+                            else:
+                                yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
+
+                    yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
+                    yield "data: [DONE]\n\n"
+                except httpx.HTTPStatusError as e:
+                    logger.error(f"HTTP error occurred: {e}")
+                    error_message = "抱歉，服务器暂时无法处理您的请求。请稍后重试。"
                     timestamp = int(datetime.now().timestamp())
-                    if line:
-                        content = line + "\n"
-                        if "https://www.blackbox.ai" in content:
-                            validate.getHid(True)
-                            content = "hid已刷新,重新对话即可\n"
-                            yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
-                            break
-                        if content.startswith("$@$v=undefined-rv1$@$"):
-                            yield f"data: {json.dumps(create_chat_completion_data(content[21:], request.model, timestamp))}\n\n"
-                        else:
-                            yield f"data: {json.dumps(create_chat_completion_data(content, request.model, timestamp))}\n\n"
-
-                yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
-                yield "data: [DONE]\n\n"
-        except httpx.HTTPStatusError as e:
-            logger.error(f"HTTP error occurred: {e}")
-            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+                    yield f"data: {json.dumps(create_chat_completion_data(error_message, request.model, timestamp))}\n\n"
+                    yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
+                    yield "data: [DONE]\n\n"
         except httpx.RequestError as e:
             logger.error(f"Error occurred during request: {e}")
-            raise HTTPException(status_code=500, detail=str(e))
+            error_message = "抱歉，连接服务器时出现错误。请检查网络连接并重试。"
+            timestamp = int(datetime.now().timestamp())
+            yield f"data: {json.dumps(create_chat_completion_data(error_message, request.model, timestamp))}\n\n"
+            yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
+            yield "data: [DONE]\n\n"
 
 
 async def process_non_streaming_response(request: ChatRequest):