Update app.py

app.py

@@ -4,7 +4,8 @@ import requests
 from fastapi import FastAPI, Response
 from fastapi.responses import StreamingResponse
 from datetime import datetime, timedelta
-from mcp_sdk import MCPHandler, Tool, mcp_route  # 导入MCP SDK
+from starlette.routing import Mount  # 用于挂载
+from mcp.server.fastmcp import FastMCP  # 导入MCP SDK
 
 app = FastAPI()
 
@@ -52,36 +53,21 @@ async def fetch_github_trending():
         print(f"Error fetching GitHub trending: {e}")
         return cached_trending if cached_trending else []
 
-# 定义MCP工具：暴露给AI模型使用
-def get_trending_repos(params: dict) -> dict:
-    """Tool to get top GitHub trending repositories. Params: {'num': int (default 10)}"""
-    num = params.get('num', 10)
+# 定义MCP服务器和工具
+mcp_server = FastMCP(name="GithubTrending", stateless_http=True)
+
+@mcp_server.tool()
+def get_trending_repos(num: int = 10) -> dict:
+    """Tool to get top GitHub trending repositories with stars >1000"""
     if not last_updated or (datetime.now() - last_updated) > timedelta(minutes=5):
         fetch_github_trending()
     return {"trending": cached_trending[:num]}
 
-# 注册MCP工具
-tools = [
-    Tool(
-        name="get_trending_repos",
-        description="Fetch top GitHub trending projects with stars >1000",
-        function=get_trending_repos,
-        parameters={
-            "type": "object",
-            "properties": {"num": {"type": "integer", "description": "Number of repos to return"}},
-            "required": []
-        }
-    )
-]
-
-# MCP处理器
-mcp_handler = MCPHandler(tools=tools)
-
-# 添加MCP路由到FastAPI
-app.include_router(mcp_route(mcp_handler, prefix="/mcp"))  # MCP端点：/mcp/tools, /mcp/call 等
+# 挂载MCP到FastAPI（使用/mcp路径）
+app.mount("/mcp", app=mcp_server.streamable_http_app())
 
 async def trending_generator():
-    """SSE 事件生成器（保留原有，MCP可复用）"""
+    """SSE 事件生成器"""
     while True:
         if not last_updated or (datetime.now() - last_updated) > timedelta(minutes=5):
             await fetch_github_trending()
@@ -93,7 +79,7 @@ async def get_trending():
     """原有端点"""
     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()}
+    return {"trending": cached_trending, "last_updated": last_updated.isoformat() if last_updated else None}
 
 @app.get("/trending-sse")
 async def get_trending_sse():