# weather_server.py
# 安装: pip install mcp httpx

import json
import asyncio
import httpx
from mcp.server import Server, NotificationOptions
from mcp.server.models import InitializationCapabilities
from mcp.server.stdio import stdio_server

# 创建 MCP 服务器实例
server = Server("weather-server")

@server.list_tools()
async def list_tools():
    """告诉 AI 模型这个服务器提供了哪些工具"""
    return [
        {
            "name": "get_weather",
            "description": "获取指定城市的实时天气信息",
            "inputSchema": {
                "type": "object",
                "properties": {
                    "city": {
                        "type": "string",
                        "description": "城市名称，如 Beijing, Shanghai"
                    }
                },
                "required": ["city"]
            }
        }
    ]

@server.call_tool()
async def call_tool(name: str, arguments: dict):
    """处理 AI 模型的工具调用请求"""
    if name == "get_weather":
        city = arguments["city"]
        # 实际项目中调真实天气 API
        return [{"type": "text", "text": f"{city} 当前天气：晴，25°C，湿度 60%"}]

async def main():
    async with stdio_server() as (read_stream, write_stream):
        await server.run(
            read_stream,
            write_stream,
            InitializationCapabilities(
                sampling={},
                experimental={},
                roots={"listChanged": True}
            )
        )

if __name__ == "__main__":
    asyncio.run(main())