Update app.py

app.py

@@ -134,20 +134,21 @@ async def access_control(request: Request, call_next):
 
 # 流式响应生成器
 async def stream_generator(response):
-    async for chunk in response.aiter_bytes():
-        try:
-            # 解码字节为文本
+    try:
+        async for chunk in response.aiter_bytes():
             chunk_str = chunk.decode('utf-8')
-            # 处理多行数据
             for line in chunk_str.split('\n'):
-                if line.strip():  # 忽略空行
+                line = line.strip()
+                if line:
                     if line.startswith('data: '):
                         yield f"{line}\n\n"
                     else:
                         yield f"data: {line}\n\n"
-        except Exception as e:
-            print(f"Stream error: {str(e)}")
-            continue
+            # 添加小延迟以保持连接
+            await asyncio.sleep(0.01)
+    except Exception as e:
+        print(f"Stream error: {str(e)}")
+        yield f"data: [ERROR] {str(e)}\n\n"
 
 # 处理API请求
 async def handle_api_request(url: str, headers: dict, method: str = "GET", body: dict = None, for_chat: bool = False):
@@ -167,7 +168,8 @@ async def handle_api_request(url: str, headers: dict, method: str = "GET", body:
                 
             headers["Authorization"] = f"Bearer {key}"
             
-            async with httpx.AsyncClient(timeout=60.0) as client:
+            # 修改超时设置
+            async with httpx.AsyncClient(timeout=httpx.Timeout(30.0, read=None)) as client:
                 response = await client.request(
                     method=method,
                     url=url,
@@ -308,7 +310,6 @@ async def chat_completions(request: Request):
         body = await request.body()
         body_json = json.loads(body)
         
-        # 获取headers
         headers = {
             "Content-Type": "application/json",
             "Accept": "text/event-stream" if body_json.get("stream") else "application/json"