Update app.py

app.py

@@ -4,6 +4,7 @@ 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
 
 app = FastAPI()
 
@@ -26,7 +27,7 @@ async def fetch_github_trending():
     headers = {}
     github_token = os.getenv("GITHUB_TOKEN")
     if github_token:
-        headers["Authorization"] = f"token {github_token}"  # 修复f-string引号问题
+        headers["Authorization"] = f"token {github_token}"
     
     try:
         response = requests.get(url, params=params, headers=headers)
@@ -51,29 +52,52 @@ 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)
+    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 等
+
 async def trending_generator():
-    """SSE 事件生成器"""
+    """SSE 事件生成器（保留原有，MCP可复用）"""
     while True:
-        # 每5分钟更新一次数据
         if not last_updated or (datetime.now() - last_updated) > timedelta(minutes=5):
-            await fetch_github_trending()  # 修复拼写错误(原trending->trending)
-        
-        # 发送当前数据
+            await fetch_github_trending()
         yield f"data: {cached_trending}\n\n"
-        
-        # 等待一段时间再检查更新
         time.sleep(30)
 
 @app.get("/trending")
 async def get_trending():
-    """获取当前 GitHub 热门项目"""
+    """原有端点"""
     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()}
 
 @app.get("/trending-sse")
 async def get_trending_sse():
-    """SSE 端点，持续推送 GitHub 热门项目更新"""
+    """原有SSE端点"""
     return StreamingResponse(
         trending_generator(),
         media_type="text/event-stream",