菜鸟AI - 让提示词生成更简单! 全站导航 全站导航
AI工具安装 新手教程 进阶教程 辅助资源 AI提示词 热点资讯 技术资讯 产业资讯 内容生成 模型技术 AI信息库

已有账号?

首页 > AI教程 > MCP协议实战:零基础用Python开发首个AI工具
进阶教程 MCP协议实战

MCP协议实战:零基础用Python开发首个AI工具

2026-06-06
阅读 0
热度 0
作者 菜鸟AI编辑部
摘要

摘要

MCP协议赋予AI应用调用工具、读写数据等能力。文章介绍其核心概念、三大能力及stdio与SSE

大型语言模型正在获得执行能力——MCP协议让AI应用不仅能生成文本,还能真正运行代码、调用API、操作文件。本文从零开始,完整演示一个MCP Server的搭建全过程。

一、MCP协议是什么?

1.1 核心概念

MCP(模型上下文协议)定义了一套标准化规范,使AI应用(Host)能通过统一接口访问外部工具和数据。这套架构初看复杂,但拆解后逻辑清晰:AI应用作为主机,中间层是MCP客户端,外层是MCP服务端,三者通过标准通信管道协作。

MCP 协议实战:用 Python 开发你的第一个 AI 工具服务

从架构全景图可见,MCP服务端主要提供三类能力:Tools(工具调用)、Resources(数据读取)和Prompts(提示模板)。Host内部的MCP客户端充当翻译器,在AI应用和服务端之间传递消息。

1.2 MCP三大核心能力

能力说明示例
ToolsAI可调用的函数查询数据库、调用API、执行计算
ResourcesAI可读取的数据源文件内容、数据库记录、日志
Prompts可复用的提示模板代码审查模板、文档生成模板

1.3 通信方式

目前支持两种传输方式,选择依据取决于应用场景。本地开发推荐stdio模式:AI应用将MCP Server作为子进程启动,通过标准输入输出通信,实现低延迟集成。远程部署则采用SSE(Server-Sent Events),即通过HTTP进行网络通信。两者各有优劣:stdio适合快速原型,SSE适用于生产环境的多客户端并发访问。

方式一: stdio(标准输入输出)
┌──────────┐ stdin/stdout   ┌──────────┐
│ AI 应用  │◄──────────────►│MCP Server│
│ (Host)   │                │(子进程)  │
└──────────┘                └──────────┘

方式二: SSE(Server-Sent Events)
┌──────────┐ HTTP + SSE     ┌──────────┐
│ AI 应用  │◄──────────────►│MCP Server│
│ (Host)   │ (网络通信)     │(远程服务)│
└──────────┘                └──────────┘

二、开发环境搭建

2.1 安装依赖

执行命令:pip install "mcp[cli]" httpx,一行完成依赖安装。httpx作为异步HTTP客户端,功能优于requests,后续天气查询将使用它。

2.2 验证安装

运行 mcp --version,若输出版本号则表示安装成功。

2.3 项目结构

推荐采用以下目录组织方式:

my-mcp-server/
├── server.py              # MCP Server主文件
├── tools/
│   ├── weather.py         # 天气查询工具
│   ├── database.py        # 数据库查询工具
│   └── calculator.py      # 计算器工具
├── resources/
│   └── system_info.py     # 系统信息资源
├── prompts/
│   └── code_review.py     # 代码审查提示模板
├── pyproject.toml         # 项目配置
└── README.md

按功能模块拆分后,每个文件职责明确,后期维护无需面对数千行的臃肿代码库。

三、第一个MCP Server(快速上手)

3.1 最简实现

先编写一个极简示例,快速体验MCP的核心能力:

# server.py
from mcp.server.fastmcp import FastMCP

# 创建MCP Server实例
mcp = FastMCP(
    name="my-first-mcp-server",
    version="1.0.0",
)

@mcp.tool()
def add(a: int, b: int) -> int:
    """两个数字相加"""
    return a + b

@mcp.tool()
def get_current_time(timezone: str = "Asia/Shanghai") -> str:
    """获取当前时间
    Args:
        timezone: 时区名称,默认为Asia/Shanghai
    """
    from datetime import datetime
    import pytz
    tz = pytz.timezone(timezone)
    now = datetime.now(tz)
    return now.strftime("%Y-%m-%d %H:%M:%S %Z")

if __name__ == "__main__":
    mcp.run()

注意 @mcp.tool() 装饰器——这是将普通Python函数转化为AI可调用工具的关键。参数类型标注和docstring并非摆设,AI正是依赖这些信息来理解工具的调用方式。

3.2 运行测试

提供两种启动方式:mcp dev server.py 会启动可视化调试工具(MCP Inspector),在浏览器中即可测试工具。直接执行 python server.py 也可正常启动。

四、实战:开发完整的企业工具服务

下面构建一个实用的MCP Server,涵盖天气查询、数据库操作、文件管理等功能。

4.1 天气查询工具

调用OpenWeatherMap API,支持当前天气和天气预报:

# tools/weather.py
import httpx
from mcp.server.fastmcp import tool

BASE_URL = "https://api.openweathermap.org/data/2.5"

@tool()
async def get_weather(city: str, api_key: str = "") -> str:
    """查询指定城市的当前天气信息
    Args:
        city: 城市名称(中文或英文),如"北京"或"Beijing"
        api_key: OpenWeatherMap API Key(可选,默认使用内置测试key)
    """
    params = {
        "q": city,
        "appid": api_key,
        "units": "metric",
        "lang": "zh_cn",
    }
    async with httpx.AsyncClient() as client:
        resp = await client.get(f"{BASE_URL}/weather", params=params)
        data = resp.json()

    if resp.status_code != 200:
        return f"查询失败: {data.get('message', '未知错误')}"

    return (
        f"【{data['name']} 天气】\n"
        f"天气: {data['weather'][0]['description']}\n"
        f"温度: {data['main']['temp']}°C (体感 {data['main']['feels_like']}°C)\n"
        f"湿度: {data['main']['humidity']}%\n"
        f"风速: {data['wind']['speed']} m/s"
    )

@tool()
async def get_forecast(city: str, days: int = 3, api_key: str = "") -> str:
    """查询未来几天的天气预报
    Args:
        city: 城市名称
        days: 预报天数(最多5天)
        api_key: OpenWeatherMap API Key
    """
    params = {
        "q": city,
        "appid": api_key,
        "units": "metric",
        "lang": "zh_cn",
        "cnt": min(days, 5) * 8,  # 每天约8个数据点
    }
    async with httpx.AsyncClient() as client:
        resp = await client.get(f"{BASE_URL}/forecast", params=params)
        data = resp.json()

    forecasts = []
    for item in data["list"][::8]:  # 每天取一个数据点
        forecasts.append(
            f"{item['dt_txt'][:10]}: "
            f"{item['weather'][0]['description']}, "
            f"{item['main']['temp']}°C"
        )

    return f"【{data['city']['name']} 未来天气预报】\n" + "\n".join(forecasts)

4.2 SQLite数据库查询工具

工具设计注重安全性——仅允许SELECT操作,防止AI意外执行危险指令:

# tools/database.py
import sqlite3
from pathlib import Path
from mcp.server.fastmcp import tool

DB_PATH = Path("./data/app.db")

def _get_connection():
    """获取数据库连接"""
    DB_PATH.parent.mkdir(exist_ok=True)
    return sqlite3.connect(str(DB_PATH))

@tool()
def query_database(sql: str) -> str:
    """执行SQL查询语句(仅支持SELECT)
    Args:
        sql: SQL查询语句,仅允许SELECT操作
    """
    # 安全检查:只允许SELECT
    normalized = sql.strip().upper()
    if not normalized.startswith("SELECT"):
        return "错误:仅支持SELECT查询,不允许修改数据"
    
    try:
        conn = _get_connection()
        cursor = conn.execute(sql)
        columns = [desc[0] for desc in cursor.description]
        rows = cursor.fetchall()
        conn.close()

        if not rows:
            return "查询结果为空"
        
        header = "| " + " | ".join(columns) + " |"
        separator = "| " + " | ".join(["---"] * len(columns)) + " |"
        data_rows = []
        for row in rows[:50]:  # 最多返回50行
            data_rows.append("| " + " | ".join(str(v) for v in row) + " |")
        table = "\n".join([header, separator] + data_rows)
        return f"查询返回 {len(rows)} 行:\n{table}"
    
    except Exception as e:
        return f"查询出错: {str(e)}"

@tool()
def list_tables() -> str:
    """列出数据库中的所有表及其结构"""
    conn = _get_connection()
    cursor = conn.execute("SELECT name FROM sqlite_master WHERE type='table'")
    tables = cursor.fetchall()
    conn.close()

    if not tables:
        return "数据库为空,暂无表"
    
    result = []
    for (table_name,) in tables:
        result.append(f"- **{table_name}**")
    
    return "数据库中的表:\n" + "\n".join(result)

4.3 文件管理工具

文件操作需严格防范路径穿越攻击,因此实现了一个沙箱限制:

# tools/file_manager.py
import os
from pathlib import Path
from mcp.server.fastmcp import tool

SAFE_DIR = Path("./data/workspace")  # 安全沙箱目录

def _safe_path(filepath: str) -> Path:
    """确保文件路径在安全目录内"""
    full_path = (SAFE_DIR / filepath).resolve()
    if not str(full_path).startswith(str(SAFE_DIR.resolve())):
        raise ValueError("路径超出安全范围")
    return full_path

@tool()
def read_file(filepath: str) -> str:
    """读取文件内容
    Args:
        filepath: 相对于工作目录的文件路径
    """
    path = _safe_path(filepath)
    if not path.exists():
        return f"文件不存在: {filepath}"
    return path.read_text(encoding="utf-8")

@tool()
def write_file(filepath: str, content: str) -> str:
    """写入文件内容
    Args:
        filepath: 相对于工作目录的文件路径
        content: 要写入的内容
    """
    path = _safe_path(filepath)
    path.parent.mkdir(parents=True, exist_ok=True)
    path.write_text(content, encoding="utf-8")
    return f"文件写入成功: {filepath} ({len(content)} 字符)"

@tool()
def list_files(directory: str = ".") -> str:
    """列出目录下的文件
    Args:
        directory: 相对于工作目录的目录路径
    """
    path = _safe_path(directory)
    if not path.exists():
        return f"目录不存在: {directory}"
    
    items = []
    for item in sorted(path.iterdir()):
        if item.is_dir():
            items.append(f"???? {item.name}/")
        else:
            size = item.stat().st_size
            items.append(f"???? {item.name} ({size} bytes)")
    
    return "\n".join(items) if items else "目录为空"

4.4 定义Resources(数据资源)

Resources是AI可直接读取的数据源,相当于只读接口:

# resources/system_info.py
import platform
import psutil
from mcp.server.fastmcp import resource

@resource("system://info")
def get_system_info() -> str:
    """获取系统信息"""
    return f"""
系统信息:
- 操作系统: {platform.system()} {platform.release()}
- Python: {platform.python_version()}
- CPU核心: {psutil.cpu_count()}
- 内存总量: {psutil.virtual_memory().total // (1024**3)} GB
- 磁盘使用: {psutil.disk_usage('/').percent}%
"""

@resource("system://processes")
def get_top_processes() -> str:
    """获取占用资源最多的前10个进程"""
    procs = []
    for p in psutil.process_iter(["pid", "name", "cpu_percent", "memory_percent"]):
        try:
            info = p.info
            procs.append(info)
        except (psutil.NoSuchProcess, psutil.AccessDenied):
            pass
    
    procs.sort(key=lambda x: x["cpu_percent"] or 0, reverse=True)
    
    lines = ["PID    名称                CPU%   内存%"]
    lines.append("-" * 45)
    for p in procs[:10]:
        lines.append(f"{p['pid']:<8}{p['name']:<20}{p['cpu_percent']:.1f}%    {p['memory_percent']:.1f}%")
    
    return "\n".join(lines)

4.5 定义Prompts(提示模板)

Prompts本质上是预设的上下文指令,让AI按固定格式处理特定任务:

# prompts/code_review.py
from mcp.server.fastmcp import prompt

@prompt()
def code_review(code: str, language: str = "python") -> str:
    """代码审查提示模板
    Args:
        code: 需要审查的代码
        language: 编程语言
    """
    return f"""请对以下 {language} 代码进行专业审查:

```{language}
{code}
```

请从以下维度进行审查:
1. 代码质量 — 可读性、命名规范、代码风格
2. 潜在Bug — 逻辑错误、边界条件、空指针风险
3. 安全漏洞 — 注入攻击、数据泄露、权限问题
4. 性能优化 — 时间复杂度、内存使用、I/O效率
5. 最佳实践 — 设计模式、SOLID原则、错误处理

请为每个维度给出1-5分评分和具体改进建议。"""

@prompt()
def explain_code(code: str) -> str:
    """代码解释提示模板
    Args:
        code: 需要解释的代码
    """
    return f"""请用通俗易懂的语言解释以下代码的功能:

```{code}
```

要求:
1. 先用一句话概括整体功能
2. 逐段解释关键逻辑
3. 说明输入和输出
4. 指出可能的使用场景"""

4.6 组装完整Server

将上述分散的模块全部注册到主Server中:

# server.py
from mcp.server.fastmcp import FastMCP

mcp = FastMCP(
    name="enterprise-toolkit",
    version="1.0.0",
    description="企业级MCP工具服务:天气查询、数据库操作、文件管理",
)

# ── 注册Tools ──────────────────────────────────────
from tools.weather import get_weather, get_forecast
from tools.database import query_database, list_tables
from tools.file_manager import read_file, write_file, list_files

mcp.tool(get_weather)
mcp.tool(get_forecast)
mcp.tool(query_database)
mcp.tool(list_tables)
mcp.tool(read_file)
mcp.tool(write_file)
mcp.tool(list_files)

# ── 注册Resources ──────────────────────────────────
from resources.system_info import get_system_info, get_top_processes
mcp.resource("system://info")(get_system_info)
mcp.resource("system://processes")(get_top_processes)

# ── 注册Prompts ────────────────────────────────────
from prompts.code_review import code_review, explain_code
mcp.prompt(code_review)
mcp.prompt(explain_code)

# ── 启动 ────────────────────────────────────────────
if __name__ == "__main__":
    mcp.run()

五、配置与使用

5.1 在Claude Desktop中配置

编辑Claude Desktop配置文件,路径因操作系统而异:

macOS: ~/Library/Application Support/Claude/claude_desktop_config.json
Windows: %APPDATA%\Claude\claude_desktop_config.json

{
  "mcpServers": {
    "enterprise-toolkit": {
      "command": "python",
      "args": ["C:/path/to/my-mcp-server/server.py"],
      "env": {
        "OPENWEATHER_API_KEY": "your-api-key"
      }
    }
  }
}

5.2 在Cursor / VS Code中配置

项目根目录创建 .cursor/mcp.json

{
  "mcpServers": {
    "enterprise-toolkit": {
      "command": "python",
      "args": ["./my-mcp-server/server.py"]
    }
  }
}

5.3 调试流程

调试链路直观流畅:启动MCP Inspector → 浏览器打开调试界面 → 浏览已注册的各类工具 → 填入参数执行测试 → 确认结果后正式配置到AI客户端。可视化调试是MCP的突出优势,大幅减少命令行反复测试的繁琐操作。

六、进阶:SSE模式部署为远程服务

6.1 使用SSE传输

# server_sse.py
from mcp.server.fastmcp import FastMCP

mcp = FastMCP(
    name="enterprise-toolkit-remote",
    host="0.0.0.0",
    port=8080,
)

# ... 注册tools/resources/prompts(同上)...

if __name__ == "__main__":
    # SSE模式:通过HTTP暴露服务
    mcp.run(transport="sse")

6.2 Docker部署

# Dockerfile
FROM python:3.12-slim
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .
EXPOSE 8080
CMD ["python", "server_sse.py"]
# docker-compose.yml
version: "3.8"
services:
  mcp-server:
    build: .
    ports:
      - "8080:8080"
    environment:
      - OPENWEATHER_API_KEY=${OPENWEATHER_API_KEY}
    volumes:
      - ./data:/app/data
    restart: unless-stopped

6.3 客户端配置(远程SSE)

{
  "mcpServers": {
    "enterprise-toolkit": {
      "url": "http://your-server:8080/sse"
    }
  }
}

七、MCP协议消息流详解

当用户在Claude中说出“帮我查一下北京天气”,背后的完整消息流如下:用户输入 → Claude分析意图,决定调用get_weather工具 → 通过MCP发送JSON-RPC请求(method为tools/call,参数包含工具名和参数)→ MCP Server执行实际函数 → 返回结果(含天气数据)→ Claude结合结果生成自然语言回答 → 用户看到预期输出。尽管步骤较多,但由于内部采用高效的序列化协议,实际延迟极低。

八、最佳实践与安全注意事项

8.1 工具设计原则

设计MCP工具时,需牢记以下原则:单一职责——每个工具只完成一项功能;docstring是AI理解工具的关键,越清晰AI使用越准确;参数必须包含明确的类型标注和默认值;访问文件或数据库时必须实施沙箱限制;错误信息应友好可读,而非直接抛出异常;相同输入产生相同输出,保持幂等性。

8.2 安全清单

安全始终优先。SQL注入防护是基础——严格限定仅允许SELECT操作。路径穿越攻击需要通过resolve()校验前缀来防御。用户输入中的危险字符必须过滤。这些措施虽增加少量代码,但在生产环境中缺一不可。

# 安全示例:参数验证与路径限制
import re
from pathlib import Path

def validate_sql(sql: str) -> bool:
    """只允许SELECT查询"""
    forbidden = ["DROP", "DELETE", "INSERT", "UPDATE", "ALTER", "CREATE"]
    upper_sql = sql.upper()
    return all(kw not in upper_sql for kw in forbidden)

def safe_filepath(base_dir: Path, user_path: str) -> Path:
    """防止路径穿越攻击"""
    full_path = (base_dir / user_path).resolve()
    if not str(full_path).startswith(str(base_dir.resolve())):
        raise ValueError("非法路径访问")
    return full_path

def sanitize_input(text: str) -> str:
    """清理用户输入"""
    # 移除潜在的危险字符
    return re.sub(r'[<>&\']', '', text)

九、总结

从零到一,我们完整走了一遍MCP Server的开发流程。关键知识点回顾:FastMCP是Python生态中最核心的开发框架;@tool()@resource()@prompt()三个装饰器分别对应不同能力;stdio适合本地开发,SSE适合远程部署;MCP Inspector是调试利器,无需命令行手动测试。下一阶段可探索的方向包括:将MCP Server接入企业内部API、开发多用户权限管理中间件、结合RAG构建知识库服务,甚至参与开源生态发布自有工具。

MCP协议正快速成为AI工具调用的行业标准,掌握它意味着能让AI真正融入你的工作流。这不是锦上添花,而是从“对话式AI”到“行动式AI”的跨越。

来源:互联网

免责声明

本网站新闻资讯均来自公开渠道,力求准确但不保证绝对无误,内容观点仅代表作者本人,与本站无关。若涉及侵权,请联系我们处理。本站保留对声明的修改权,最终解释权归本站所有。

同类文章推荐

相关文章推荐

更多