Update app.py

app.py

@@ -1,20 +1,28 @@
 import os
-import time
 import requests
 from fastapi import FastAPI
 from datetime import datetime, timedelta
-from starlette.routing import Mount
+from fastapi.middleware.cors import CORSMiddleware
 from fastmcp.server import FastMCP
 from contextlib import asynccontextmanager
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    await fetch_github_trending()  # 启动时初始化
+    await fetch_github_trending()
     yield
 
 app = FastAPI(lifespan=lifespan)
 
-# 缓存最新获取的热门项目
+# 添加 CORS 支持
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 缓存设置
 cached_trending = []
 last_updated = None
 
@@ -31,58 +39,55 @@ async def fetch_github_trending():
     }
     
     headers = {}
-    github_token = os.getenv("GITHUB_TOKEN")
-    if github_token:
+    if github_token := os.getenv("GITHUB_TOKEN"):
         headers["Authorization"] = f"token {github_token}"
     
     try:
         response = requests.get(url, params=params, headers=headers)
         response.raise_for_status()
-        items = response.json().get("items", [])
-        
-        # 简化项目信息
-        trending_projects = []
-        for item in items:
-            trending_projects.append({
+        cached_trending = [
+            {
                 "name": item["full_name"],
                 "url": item["html_url"],
                 "stars": item["stargazers_count"],
                 "description": item["description"],
                 "language": item["language"]
-            })
-        
-        cached_trending = trending_projects
+            }
+            for item in response.json().get("items", [])
+        ]
         last_updated = datetime.now()
-        return trending_projects
+        return cached_trending
     except Exception as e:
         print(f"Error fetching GitHub trending: {e}")
         return cached_trending if cached_trending else []
 
-# 定义MCP服务器和工具
-#mcp_server = FastMCP(name="GithubTrending", stateless_http=True)
-mcp_server = FastMCP(name="GithubTrending")
-
-app.mount("/mcp", mcp_server.http_app())
-#mcp_app = mcp_server.http_app(path='/mcp')
-#app.mount("/mcp", mcp_app)
+# 初始化 MCP 服务器
+mcp_server = FastMCP(
+    name="GithubTrending",
+    stateless_http=True,
+    http_path="/mcp",
+    sse_path="/mcp/sse"
+)
 
 @mcp_server.tool()
 def get_trending_repos(num: int = 10) -> dict:
-    """Tool to get top GitHub trending repositories with stars >1000"""
+    """获取 GitHub 热门仓库"""
     if not last_updated or (datetime.now() - last_updated) > timedelta(minutes=5):
         fetch_github_trending()
     return {"trending": cached_trending[:num]}
 
+# 挂载 MCP 应用
+app.mount("/mcp", mcp_server.http_app())
+
 @app.get("/trending")
 async def get_trending():
-    """获取GitHub热门项目"""
-    # 如果缓存为空或超过5分钟未更新，则重新获取
+    """公开的 HTTP 端点"""
     if not cached_trending or (datetime.now() - last_updated) > timedelta(minutes=5):
         await fetch_github_trending()
     
     return {
         "trending": cached_trending,
         "last_updated": last_updated.isoformat() if last_updated else None,
-        "cache_expires_in": 300,  # 告诉客户端缓存有效期(秒)
-        "suggested_polling_interval": 60  # 建议客户端轮询间隔(秒)
+        "cache_expires_in": 300,
+        "suggested_polling_interval": 60
     }