🚀 SmolAgents + MCP 服务器

本项目借助 SmolAgents 库打造了一个 AI 助手，该助手具备网络搜索以及网站抓取/爬取的能力，通过集成 Brave Search 和 Crawl4ai 模型上下文协议 (MCP) 服务器来实现。

🚀 快速开始

环境准备

在运行此项目之前，请确保安装了以下依赖项：

• smolagents
• python-dotenv
• mcp
• @modelcontextprotocol/server-brave-search（通过 npm/npx 全局安装）
• litellm

您可以使用 pip 安装 Python 包：

pip install -r requirements.txt

同时，请确保您已安装 Node.js 和 npm/npx 以及 uv，以运行 MCP 服务器。

配置环境变量

在项目根目录中创建一个 .env 文件，并添加以下变量：

BRAVE_API_KEY=your_brave_api_key
OPENAI_API_KEY=your_openai_api_key

请将 your_brave_api_key 和 your_openai_api_key 替换为您实际的 API 密钥。

运行项目

1. 克隆仓库或下载脚本 (smolagent_with_mcp.py)。
2. 在终端中导航到项目目录。
3. 运行脚本：

python smolagent_with_mcp.py

4. 通过输入搜索查询与助手交互。输入 exit 或 quit 终止会话。

✨ 主要特性

• 代码将代理的初始化和主要交互循环封装在 Python 类 (SmolAgentWithMCP) 中，这种面向对象的方法提供了模块化设计，并使扩展代理的功能变得更加容易，例如集成更多的 MCP 服务器及其工具。
• SmolAgents 中的 ToolCallingAgent 被设计为以结构化格式（如 JSON）输出工具调用，这些调用随后由环境执行（在本例中为主脚本中的主要循环使用 MCPClient 执行）。

📦 安装指南

安装依赖

pip install -r requirements.txt

安装其他必要组件

确保您已安装 Node.js 和 npm/npx 以及 uv，以运行 MCP 服务器。

💻 使用示例

基础用法

from smolagents import SmolAgentWithMCP
from modelcontextprotocol.server_brave_search import StdioServerParameters

def main():
    # 定义 MCP 服务器参数
    brave_server_params = StdioServerParameters()
    crawl4ai_server_params = StdioServerParameters()

    # 初始化代理并运行
    agent = SmolAgentWithMCP([brave_server_params, crawl4ai_server_params])
    agent.run()

if __name__ == "__main__":
    main()

高级用法

# 定义新服务器的参数（用实际命令和参数替换）
import os
new_server_params = StdioServerParameters(
    command="your_server_command",
    args=["arg1", "arg2"],
    env={"YOUR_API_KEY": os.getenv("YOUR_API_KEY")},
)

def main():
    # 定义 MCP 服务器参数列表
    server_list = [
        brave_server_params,
        crawl4ai_server_params,
        new_server_params
    ]
    
    # 初始化代理并运行
    agent = SmolAgentWithMCP(server_list)
    agent.run()

if __name__ == "__main__":
    main()

这样，您可以轻松扩展代理的功能，添加更多的 MCP 服务器和工具。

📚 详细文档

备注

• 确保 npx 已安装并可在系统 PATH 中使用，因为它是运行 Brave Search MCP 服务器所必需的。

添加更多 MCP 服务器

smolagent_with_mcp.py 文件的结构允许轻松添加更多的 MCP 服务器和其工具，以模块化的方式进行。要添加新的 MCP 服务器：

1. 定义新服务器的 StdioServerParameters。这涉及指定运行服务器的命令、其参数以及任何必要的环境变量（类似于 brave_server_params 和 crawl4ai_server_params 的定义方式）。
2. 将新定义的服务器参数对象添加到 main 函数中的 server_list 中。BraveSearchAgentApp 类会自动初始化新的服务器，并将其工具包含在代理的能力中。

📄 许可证

该项目使用 MIT 许可证，允许您自由使用、修改和分发代码。