AI导哈

数学计算代理

一个使用多MCP服务器进行数学计算的代理系统，支持基础和高阶数学运算，具备智能工具选择和错误处理功能。

  • by abhihash01
  • Python
  • 本地部署
  • 更新于
在 GitHub 查看 abhihash01/math_agent_mcp

使用说明

使用MCP的数学代理

构建一个可以使用不同mcp服务器的数学代理。

🚀 特性

  • 多MCP集成：使用多个MCP服务器进行不同的数学运算
  • 会话上下文：维护多个查询的对话历史记录
  • 智能刀具选择：根据用户查询自动选择合适的工具
  • 错误处理：稳健的错误处理和适当的验证
  • 可扩展架构：易于添加新的MCP服务器和数学运算

🏗️ 建筑

该系统由以下部分组成：

  1. 主要代理人 (client.py)：基于LangGraph的代理，协调多个MCP服务器
  2. 基础数学服务器 (basic_math_server.py)：基本算术运算
  3. 高级数学服务器 (advanced_math_server.py)：电力操作和高等数学
┌─────────────────┐    
│   Math Agent    │
└─────────┬───────┘
          │
          ├── Basic Math MCP Server
          │   ├── Addition
          │   ├── Subtraction  
          │   ├── Multiplication
          │   └── Division
          │
          └── Advanced Math MCP Server
              ├── Power/Exponentiation
              ├── Square
              ├── Cube
              └── Square Root

📋 先决条件

  • Python 3.8+
  • 谷歌云凭据（适用于Vertex AI Gemini模型）
  • MCP服务器运行能力

🛠️ 安装

  1. 克隆存储库:

    git clone <your-repo-url>
cd Math_Agent_MCP

  2. 安装依赖项:

    pip install -r requirements.txt

  3. 设置Google Cloud凭据:

    # Option 1: Using gcloud CLI
gcloud auth application-default login

# Option 2: Using service account key
export GOOGLE_APPLICATION_CREDENTIALS="path/to/your/service-account-key.json"

🚀 用法

基本用法

运行主演示：

python client.py

这将启动数学代理，并演示跨多个查询的各种数学操作。

交互示例

演示包括几个示例查询：

  1. 基本算术: "What's (3 + 5) × 12?"
  2. 背景跟进: "Now I want to get the square of that result"
  3. 高级操作: "What's 2^8 + 15?"

样本输出

🧮 LangGraph Multi-MCP Math Agent Demo
========================================

🔍 Query 1: What's (3 + 5) × 12?
----------------------------------------
Elementary Math: 3 + 5 = 8
Elementary Math: 8 × 12 = 96
AI: The result of (3 + 5) × 12 is 96.

🔍 Query 2: Now I want to get the square of that result
----------------------------------------
Advanced Math: 96² = 9216
AI: The square of 96 is 9,216.

🔍 Query 3: What's 2^8 + 15?
----------------------------------------
Advanced Math: 2^8 = 256
Elementary Math: 256 + 15 = 271
AI: 2^8 + 15 equals 271.

📁 项目结构

├── client.py                           # Main LangGraph agent
├── basic_math_server.py           # Basic math operations MCP server
├── advanced_math_server.py       # Advanced math operations MCP server
├── requirements.txt                    # Python dependencies
└── README.md                          # This file

🔧 可用操作

初等数学服务器

  • 加法: add(a, b) -将两个整数相加
  • 减法: subtract(a, b) -减去两个整数
  • 乘法: multiply(a, b) -将两个整数相乘
  • 部门: divide(a, b) -除以商和余数

高级数学服务器

  • 力量: power(base, exponent) -计算基数指数
  • 方形: square(number) -计算数字²
  • 立方体: cube(number) -计算数量³
  • 平方根: square_root(number) -计算√数

🎯 关键组件

MathAgent类

地心末日 MathAgent 类提供：

  • 初始化：设置MCP客户端和工具
  • 图形构建：创建LangGraph工作流
  • 查询处理：根据上下文处理单个查询
  • 消息管理：格式化和显示对话历史记录

LangGraph工作流

代理使用简单但有效的工作流程：

  1. 模型调用：LLM分析查询并决定工具的使用
  2. 工具执行：如果需要工具，请通过MCP服务器执行它们
  3. 响应生成：根据工具结果生成最终响应
  4. 连续逻辑：确定是否需要更多的工具调用

🔄 扩展系统

添加新的MCP服务器

  1. 创建新的MCP服务器:

    from mcp.server.fastmcp import FastMCP

mcp = FastMCP("Your Server Name")

@mcp.tool()
def your_function(param: int) -> int:
    """Your function description"""
    return param * 2

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

  2. 添加到客户端配置:

    server_config = {
    "your_server": {
        "command": "python",
        "args": ["your_server.py"],
        "transport": "stdio",
    },
    # ... existing servers
}

添加新操作

只需添加新 @mcp.tool() 将装饰函数添加到相应的服务器文件中。

🐛 故障排除

常见问题

  1. 谷歌云身份验证:

    # Verify authentication
gcloud auth list
gcloud config list

  2. MCP服务器连接问题:

    • 确保Python脚本可执行
    • 检查MCP服务器是否正常启动
    • 验证stdio传输是否正常工作

  3. 工具未找到错误:

    • 检查所有MCP服务器是否配置正确
    • 验证服务器和客户端之间的工具名称是否匹配

调试模式

通过修改日志级别启用详细日志记录：

logging.basicConfig(level=logging.DEBUG)

📊 性能注意事项

  • 工具缓存：初始化期间加载一次MCP工具
  • 消息上下文：对话历史记录随着每次查询而增长
  • 并行操作：MCP服务器异步处理请求

相关 MCP