使用 Python MCP SDK 构建简单的服务器

在这篇教程中，我们将使用 Model Context Protocol (MCP) Python SDK 来构建一个简单的服务器。我们将逐步引导你完成安装、设置和运行服务器的过程，并展示如何通过命令行与之交互。

1. 安装 MCP Python SDK

首先，我们需要在项目环境中安装 MCP Python SDK：

pip install mcp-sdk

2. 创建基本服务器配置文件 (server.py)

接下来，创建一个新的 Python 文件 server.py，并添加以下代码：

# server.py
from mcp import Server, StdioTransport

if __name__ == "__main__":
    # 初始化 MCP 服务器
    server = Server()
    
    @server.tool("add")
    async def add_tool(arguments):
        a = arguments["a"]
        b = arguments["b"]
        return f"{int(a) + int(b)}"
    
    @server.resource("greeting://{name}")
    async def greeting_resource(name):
        return f"Hello, {name}!"
    
    # 使用标准输入输出作为默认传输协议
    transport = StdioTransport()
    server.start(transport)

3. 运行服务器

在终端中运行以下命令启动服务器：

python server.py

此时，服务器将在后台静默运行，等待客户端连接并发送请求。

4. 创建 MCP 客户端 (client.py)

创建另一个 Python 文件 client.py，添加以下代码以连接到服务器：

# client.py
import asyncio
from mcp import Server, ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client

async def main():
    server_params = StdioServerParameters(
        command="python",
        args=["server.py"],
    )

    async with stdio_client(server_params) as (reader, writer):
        async with ClientSession(reader, writer) as session:
            await session.initialize()

            # 调用 add 工具
            result = await session.call_tool("add", arguments={"a": 3, "b": 4})
            print(f"加法工具的结果: {result}")

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

5. 运行客户端

在不同的终端窗口中运行以下命令启动客户端：

python client.py

输出结果应为：

加法工具的结果: TextContent(type='text', text='7', annotations=None)

6. 扩展服务器功能

你可以通过添加更多工具和资源来扩展服务器的功能。例如，添加一个计算器工具：

@server.tool("calculator")
async def calculator_tool(arguments):
    operation = arguments["operation"]
    a = int(arguments["a"])
    b = int(arguments["b"])
    
    if operation == "add":
        return a + b
    elif operation == "subtract":
        return a - b
    elif operation == "multiply":
        return a * b
    elif operation == "divide":
        return a / b
    else:
        raise ValueError("无效的运算操作")

7. 使用 MCP 检查工具

MCP 提供了一个命令行工具 mcp-check 来验证服务器是否正确运行。安装并使用如下：

pip install mcp-check
mcp-check http://localhost:1234

结论

通过这篇教程，我们成功地使用 MCP Python SDK 构建了一个简单的服务器，并展示了如何添加工具和资源。你可以在此基础上扩展功能，构建更复杂的系统。

附录：常见问题解答

问：为什么我的服务器没有响应？

答：请确保服务器正在运行，并且客户端连接到了正确的地址和端口。检查是否有防火墙或网络配置阻止了连接。

问：如何处理自定义资源？

答：你可以通过在服务器上注册自定义资源路径，例如 @server.resource("custom/{id}")，并返回相应的数据。

问：MCP 支持哪些传输协议？

答：除了标准输入输出之外，MCP 还支持 WebSocket 和 HTTP 等其他传输协议。你可以根据需求进行选择和配置。

通过这篇教程，你已经掌握了使用 MCP Python SDK 构建基本服务器的技能。祝你在开发过程中取得成功！