Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, Request, Response
 from fastapi.responses import StreamingResponse
 from fastapi.middleware.cors import CORSMiddleware
 import httpx
+import json
 from typing import Optional
 
 app = FastAPI()
@@ -23,6 +24,13 @@ async def get_http_client():
         http2=True
     )
 
+async def is_stream_request(request: Request) -> bool:
+    # 检查请求头中的 accept 字段
+    accept = request.headers.get('accept', '')
+    # 检查查询参数
+    stream = request.query_params.get('stream', '').lower()
+    return 'text/event-stream' in accept or stream == 'true'
+
 async def proxy_handler(url: str, request: Request):
     try:
         headers = dict(request.headers)
@@ -35,25 +43,58 @@ async def proxy_handler(url: str, request: Request):
         body = await request.body()
         
         async with await get_http_client() as client:
-            response = await client.request(
-                method=request.method,
-                url=url,
-                params=params,
-                headers=headers,
-                content=body if body else None,
-            )
-            
-            response_headers = dict(response.headers)
-            response_headers.pop('transfer-encoding', None)
-            response_headers.pop('content-encoding', None)
-            response_headers.pop('content-length', None)
-            
-            return Response(
-                content=response.content,
-                status_code=response.status_code,
-                headers=response_headers,
-                media_type=response.headers.get('content-type')
-            )
+            # 判断是否需要流式传输
+            if await is_stream_request(request):
+                # 流式请求
+                response = await client.stream(
+                    method=request.method,
+                    url=url,
+                    params=params,
+                    headers=headers,
+                    content=body if body else None,
+                )
+                
+                response_headers = dict(response.headers)
+                response_headers.pop('transfer-encoding', None)
+                response_headers.pop('content-encoding', None)
+                response_headers.pop('content-length', None)
+
+                async def stream_generator():
+                    try:
+                        async for chunk in response.aiter_bytes():
+                            yield chunk
+                    except Exception as e:
+                        print(f"Streaming error: {e}")
+                    finally:
+                        await response.aclose()
+
+                return StreamingResponse(
+                    stream_generator(),
+                    status_code=response.status_code,
+                    headers=response_headers,
+                    media_type=response.headers.get('content-type')
+                )
+            else:
+                # 非流式请求
+                response = await client.request(
+                    method=request.method,
+                    url=url,
+                    params=params,
+                    headers=headers,
+                    content=body if body else None,
+                )
+                
+                response_headers = dict(response.headers)
+                response_headers.pop('transfer-encoding', None)
+                response_headers.pop('content-encoding', None)
+                response_headers.pop('content-length', None)
+                
+                return Response(
+                    content=response.content,
+                    status_code=response.status_code,
+                    headers=response_headers,
+                    media_type=response.headers.get('content-type')
+                )
             
     except httpx.TimeoutException:
         return Response(content="请求超时", status_code=504)
@@ -71,7 +112,6 @@ async def api_v1_proxy(path: str, request: Request):
 # 处理根路径和其他路径
 @app.api_route("/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "PATCH"])
 async def root_proxy(path: str, request: Request):
-    # 如果是根路径或空路径
     if not path or path == "/":
         url = TARGET_URL
     else: