forked from lingyuzeng/agent
35 lines
919 B
Python
35 lines
919 B
Python
from fastapi import FastAPI
|
|
from mcp.server.fastmcp import FastMCP
|
|
import asyncio
|
|
from contextlib import asynccontextmanager
|
|
|
|
# 创建 FastAPI 应用
|
|
app = FastAPI()
|
|
|
|
# 创建 FastMCP 服务器
|
|
mcp = FastMCP("Weather")
|
|
|
|
# 定义工具：获取天气
|
|
@mcp.tool()
|
|
async def get_weather(location: str) -> str:
|
|
"""返回指定位置的天气情况"""
|
|
return f"The weather in {location} is sunny!"
|
|
|
|
# 使用 FastAPI 的 lifespan 来启动 MCP 服务器
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
# 在应用启动时启动 MCP 服务
|
|
loop = asyncio.get_event_loop()
|
|
task = loop.create_task(mcp.run_streamable_http_async())
|
|
yield
|
|
# 在应用关闭时停止 MCP 服务
|
|
task.cancel()
|
|
|
|
# 将 lifespan 事件添加到 FastAPI 应用
|
|
app = FastAPI(lifespan=lifespan)
|
|
|
|
# 启动 FastAPI 应用
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|