7b80d7c0b3
- 新增 fastmcp 依赖，支持通过 Streamable HTTP 连接外部工具服务器 - 在 component_registry 与 tool_api 中实现 MCP 工具加载、注册及调用链路 - 补充 README 中的 MCP 特性说明 - 统一修复多处 import 顺序、空行、引号及类型注解，提升代码整洁度 - 在 pyproject.toml 中忽略 PERF203 规则，允许循环内异常处理 - 优化语音缓存与本地 ASR 调用逻辑，减少冗余代码
143 lines
3.1 KiB
Python
143 lines
3.1 KiB
Python
"""
|
||
简单的 MCP 测试服务器
|
||
|
||
使用 fastmcp 创建一个简单的 MCP 服务器供测试使用。
|
||
|
||
安装依赖:
|
||
pip install fastmcp uvicorn
|
||
|
||
运行服务器:
|
||
python scripts/simple_mcp_server.py
|
||
|
||
服务器将在 http://localhost:8000/mcp 提供 MCP 服务
|
||
"""
|
||
|
||
from fastmcp import FastMCP
|
||
|
||
# 创建 MCP 服务器实例
|
||
mcp = FastMCP("Demo Server")
|
||
|
||
|
||
@mcp.tool()
|
||
def add(a: int, b: int) -> int:
|
||
"""将两个数字相加
|
||
|
||
Args:
|
||
a: 第一个数字
|
||
b: 第二个数字
|
||
|
||
Returns:
|
||
两个数字的和
|
||
"""
|
||
return a + b
|
||
|
||
|
||
@mcp.tool()
|
||
def multiply(a: float, b: float) -> float:
|
||
"""将两个数字相乘
|
||
|
||
Args:
|
||
a: 第一个数字
|
||
b: 第二个数字
|
||
|
||
Returns:
|
||
两个数字的乘积
|
||
"""
|
||
return a * b
|
||
|
||
|
||
@mcp.tool()
|
||
def get_weather(city: str) -> str:
|
||
"""获取指定城市的天气信息(模拟)
|
||
|
||
Args:
|
||
city: 城市名称
|
||
|
||
Returns:
|
||
天气信息字符串
|
||
"""
|
||
# 这是一个模拟实现
|
||
weather_data = {
|
||
"beijing": "北京今天晴朗，温度 20°C",
|
||
"shanghai": "上海今天多云，温度 18°C",
|
||
"guangzhou": "广州今天有雨，温度 25°C",
|
||
}
|
||
|
||
city_lower = city.lower()
|
||
return weather_data.get(
|
||
city_lower,
|
||
f"{city}的天气信息暂不可用"
|
||
)
|
||
|
||
|
||
@mcp.tool()
|
||
def echo(message: str, repeat: int = 1) -> str:
|
||
"""重复输出一条消息
|
||
|
||
Args:
|
||
message: 要重复的消息
|
||
repeat: 重复次数，默认为 1
|
||
|
||
Returns:
|
||
重复后的消息
|
||
"""
|
||
return (message + "\n") * repeat
|
||
|
||
|
||
@mcp.tool()
|
||
def check_prime(number: int) -> bool:
|
||
"""检查一个数字是否为质数
|
||
|
||
Args:
|
||
number: 要检查的数字
|
||
|
||
Returns:
|
||
如果是质数返回 True，否则返回 False
|
||
"""
|
||
if number < 2:
|
||
return False
|
||
|
||
for i in range(2, int(number ** 0.5) + 1):
|
||
if number % i == 0:
|
||
return False
|
||
|
||
return True
|
||
|
||
|
||
if __name__ == "__main__":
|
||
print("=" * 60)
|
||
print("简单 MCP 测试服务器")
|
||
print("=" * 60)
|
||
print("\n服务器配置:")
|
||
print(" - 传输协议: Streamable HTTP")
|
||
print(" - 地址: http://localhost:8000/mcp")
|
||
print("\n可用工具:")
|
||
print(" 1. add(a: int, b: int) -> int")
|
||
print(" 2. multiply(a: float, b: float) -> float")
|
||
print(" 3. get_weather(city: str) -> str")
|
||
print(" 4. echo(message: str, repeat: int = 1) -> str")
|
||
print(" 5. check_prime(number: int) -> bool")
|
||
print("\n配置示例 (config/mcp.json):")
|
||
print("""
|
||
{
|
||
"$schema": "./mcp.schema.json",
|
||
"mcpServers": {
|
||
"demo_server": {
|
||
"enabled": true,
|
||
"transport": {
|
||
"type": "streamable-http",
|
||
"url": "http://localhost:8000/mcp"
|
||
},
|
||
"timeout": 30
|
||
}
|
||
}
|
||
}
|
||
""")
|
||
print("=" * 60)
|
||
print("\n正在启动服务器...")
|
||
print("请参考 fastmcp 官方文档了解如何运行此服务器。")
|
||
print("文档: https://github.com/jlowin/fastmcp")
|
||
print("\n基本命令:")
|
||
print(" fastmcp run simple_mcp_server:mcp")
|
||
print("=" * 60)
|