自定义 mcp tool 例子

## /langchain_learn/mcp学习/base_mcp_tool_study2.py
# 导入必要模块
import os
from typing import Optional
from mcp.server.fastmcp import FastMCP

# 初始化 MCP 服务器实例,指定服务器名称和版本
mcp_server = FastMCP(name="DemoServer", version="1.0.0")

#mcp_server.tool() 将函数注册为 mcp tool 工具

@mcp_server.tool()
def list_files(directory: str = "/work/langchain_learn") -> list:
    """
    获取指定目录的文件列表(默认查看桌面)
    Args:
        directory (str): 要查询的目录路径,支持 ~ 符号
    Returns:
        list: 文件名列表
    """
    try:
        # 处理跨平台路径格式
        expanded_path = os.path.expanduser(directory)
        return os.listdir(expanded_path)
    except Exception as e:
        return [f"Error: {str(e)}"]

@mcp_server.tool()
def calculate(expression: str) -> Optional[float]:
    """
    执行数学计算(支持加减乘除)
    Args:
        expression (str): 数学表达式,如 "3 + 5 * 2"
    Returns:
        float: 计算结果(保留两位小数)
    """
    try:
        # 安全计算实现(实际生产环境应使用更安全的计算方式)
        result = eval(expression)
        return round(float(result), 2)
    except:
        return None


if __name__ == "__main__":
    # 启动服务器,使用 stdio 传输协议
    mcp_server.run(
        transport='stdio',   # 标准输入输出通信
    )

cline 中 添加 下方 配置

在vscode中点击cline,然后点击下图中的模块
在这里插入图片描述
然后点击 已安装,点击下方 配置mcp 服务器,将下方的json 文件内容 放入 配置mcp 服务器中的 mcpServers 对应的 值中。
在这里插入图片描述

"myserver2": { #自定义 mcp 服务器的名字,与文件名无关
      "command": "/miniforge3/envs/langchain/bin/python", #想要使用的 conda中的某个虚拟环境下的 python解释器
      "args": [
        "/langchain_learn/mcp学习/base_mcp_tool_study2.py" #上方mcp 工具文件 的绝对路径
      ],
      "disabled": false,
      "autoApprove": [
        "multiplay"
      ],
      "description": "演示服务器(含文件查询和计算)" #当前服务器的描述,具体是可以做什么
    },
Logo

欢迎加入 MCP 技术社区!与志同道合者携手前行,一同解锁 MCP 技术的无限可能!

更多推荐