Update app.py

app.py

@@ -1,6 +1,9 @@
 import os
+import json
+import asyncio
 import httpx
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import StreamingResponse
 from datetime import datetime, timedelta
 from fastapi.middleware.cors import CORSMiddleware
 from fastmcp.server import FastMCP
@@ -112,7 +115,8 @@ async def root():
         "message": "GitHub Trending API is running",
         "documentation": "/docs",
         "mcp_endpoint": "/mcp",
-        "trending_endpoint": "/trending"
+        "trending_endpoint": "/trending",
+        "sse_endpoint": "/trending/stream"
     }
 
 @app.get("/trending")
@@ -128,7 +132,6 @@ async def get_trending(num: int = 10):
         "rate_limit_info": "Consider adding a GitHub token for higher rate limits"
     }
 
-
 @app.get("/trending/stream")
 async def stream_trending(interval: int = 60, client_id: str = None):
     """增强版 SSE 端点"""
@@ -136,21 +139,19 @@ async def stream_trending(interval: int = 60, client_id: str = None):
         try:
             while True:
                 # 发送心跳
-                yield {"event": "heartbeat", "data": ""}
+                yield "event: heartbeat\ndata: \n\n"
                 
                 # 获取数据
                 if not last_updated or (datetime.now() - last_updated) > timedelta(seconds=interval):
                     await fetch_github_trending()
                 
-                yield {
-                    "event": "update",
-                    "data": json.dumps({
-                        "trending": cached_trending[:10],
-                        "last_updated": last_updated.isoformat(),
-                        "client_id": client_id,
-                        "next_update": (datetime.now() + timedelta(seconds=interval)).isoformat()
-                    })
+                data = {
+                    "trending": cached_trending[:10],
+                    "last_updated": last_updated.isoformat(),
+                    "client_id": client_id,
+                    "next_update": (datetime.now() + timedelta(seconds=interval)).isoformat()
                 }
+                yield f"event: update\ndata: {json.dumps(data)}\n\n"
                 
                 await asyncio.sleep(interval)
                 
@@ -159,13 +160,12 @@ async def stream_trending(interval: int = 60, client_id: str = None):
         except Exception as e:
             print(f"SSE 错误: {e}")
 
-    response = StreamingResponse(
+    return StreamingResponse(
         event_generator(),
         media_type="text/event-stream",
         headers={
             "Cache-Control": "no-cache",
             "Connection": "keep-alive",
-            "X-Accel-Buffering": "no"  # 针对 Nginx 的优化
+            "X-Accel-Buffering": "no"
         }
-    )
-    return response
+    )