Smolagents MCPs

🚀 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 許可證，允許您自由使用、修改和分發代碼。