Update main.py

main.py

@@ -276,20 +276,19 @@ async def chat_completions(
 
     logger.info(f"转发请求 → {provider_name} | 模型: {model} | 流式: {stream}")
 
-    # ── 流式响应 ──
-    if stream:
-        async def event_stream():
-            async with httpx.AsyncClient(timeout=120) as client:
-                async with client.stream(
-                    "POST", url, headers=headers, json=body
-                ) as resp:
-                    async for chunk in resp.aiter_text():
-                        yield chunk
-
-        return StreamingResponse(
-            event_stream(),
-            media_type="text/event-stream",
-        )
+ # 流式响应中增加状态码检查
+if stream:
+    async def event_stream():
+        async with httpx.AsyncClient(timeout=120) as client:
+            async with client.stream("POST", url, headers=headers, json=body) as resp:
+                if resp.status_code != 200:
+                    err = await resp.aread()
+                    logger.error(f"上游流式错误 {resp.status_code}: {err[:300]}")
+                    yield json.dumps({"error": err.decode(errors='ignore')})
+                    return
+                async for chunk in resp.aiter_text():
+                    yield chunk
+    return StreamingResponse(event_stream(), media_type="text/event-stream")
 
     # ── 普通响应 ──
     async with httpx.AsyncClient(timeout=120) as client: