Update app.py

app.py

@@ -135,11 +135,10 @@ async def access_control(request: Request, call_next):
 # 流式响应生成器
 async def stream_generator(response):
     try:
-        async for chunk in response.aiter_bytes():
-            if chunk:
-                yield chunk
-                await asyncio.sleep(0.01)  # 小延迟保持平滑
-                
+        async for line in response.aiter_lines():
+            if line:
+                yield f"data: {line}\n\n"
+                await asyncio.sleep(0.01)
     except Exception as e:
         print(f"Stream error: {str(e)}")
         error_msg = json.dumps({"error": str(e)})
@@ -321,29 +320,30 @@ async def chat_completions(request: Request):
         
         print("Starting chat completion request...")
         
-        async with httpx.AsyncClient(timeout=None) as client:
+        async with httpx.AsyncClient(timeout=httpx.Timeout(60.0)) as client:
             key = get_chat_key()
             headers["Authorization"] = f"Bearer {key}"
             
-            async with client.stream(
-                method="POST",
+            response = await client.post(
                 url=f"{Config.OPENAI_API_BASE}/chat/completions",
                 headers=headers,
-                json=body_json
-            ) as response:
-                if response.status_code != 200:
-                    raise HTTPException(status_code=response.status_code, detail="API request failed")
-                
-                return StreamingResponse(
-                    stream_generator(response),
-                    media_type="text/event-stream",
-                    headers={
-                        "Cache-Control": "no-cache",
-                        "Connection": "keep-alive",
-                        "Content-Type": "text/event-stream",
-                        "X-Accel-Buffering": "no"
-                    }
-                )
+                json=body_json,
+                timeout=60.0
+            )
+            
+            if response.status_code != 200:
+                raise HTTPException(status_code=response.status_code, detail="API request failed")
+            
+            return StreamingResponse(
+                stream_generator(response),
+                media_type="text/event-stream",
+                headers={
+                    "Cache-Control": "no-cache",
+                    "Connection": "keep-alive",
+                    "Content-Type": "text/event-stream",
+                    "X-Accel-Buffering": "no"
+                }
+            )
         
     except Exception as e:
         print(f"Chat completion error: {str(e)}")