CSToolbox (ChatSpeed Toolbox)

简体中文 | English

CSToolbox 是 ChatSpeed 的扩展工具集，通过 MCP 协议提供网络搜索、网页内容爬取和图表生成等功能。

功能特性

• 🔍 网络搜索 - 支持多种搜索引擎（Google、Bing、百度等）
• 🕷️ 网页爬取 - 从网页提取结构化内容（支持 Markdown/HTML 格式）
• 📊 图表生成 - 快速生成各类数据可视化图表，支持曲线图、柱状图和饼图
• 📄 PDF处理 - 从PDF URL 下载文档并提取文本内容

如何使用

mcp 客户端配置

{
  "mcpServers": {
    "cstoolbox": {
      "command": "uvx",
      "args": [
        "cstoolbox"
      ],
      "env": {
        "CS_LOG_LEVEL": "DEBUG",
        "CS_LOG_DIR": "logs",
        "CS_PROXY": "http://localhost:15154",
        "CS_BROWSER_TZ": "Asia/Shanghai",
        "CS_BROWSER_LANG": "zh-CN",
        "CS_REGION": "com",
        "CS_HEADLESS": "true",
        "CS_BROWSER_TYPE": "chromium",
        "CS_EXECUTABLE_PATH": "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
        "CS_USER_DATA_DIR": "~/Library/Application Support/Google/Chrome"
      }
    }
  }
}

环境变量说明

• CS_LOG_LEVEL：日志级别，可选值为 DEBUG、INFO、WARNING、ERROR、CRITICAL，默认为 INFO
• CS_LOG_DIR：日志目录，默认为 logs
• CS_PROXY：代理服务器地址，有些地区无法访问 google.com 或者 bing.com 搜索引擎，因此需要设置代理，另外，有些网站对用户所在地区有限制，如果没有代理也是访问不了的
• CS_BROWSER_TZ：时区，默认为 Etc/UTC
• CS_BROWSER_LANG：浏览器语言，默认为 en-US
• CS_REGION：搜索引擎区域，可选值为 com、cn、us、uk等，默认为 com
• CS_HEADLESS：是否启用无头模式，默认为 true
• CS_BROWSER_TYPE：浏览器类型，可选值为 chromium、firefox、webkit，默认为 chromium
• CS_EXECUTABLE_PATH：浏览器可执行文件路径，默认为空。你可以用它来指定系统已安装的浏览器路径，这样就可以利用你系统的浏览器的状态数据（如登录状态、cookies等）。如果你指定了CS_EXECUTABLE_PATH，则徐注意CS_BROWSER_TYPE类型匹配
• CS_USER_DATA_DIR：用户数据目录，如果你指定了CS_EXECUTABLE_PATH，建议将CS_USER_DATA_DIR设置为浏览器的「个人资料路径」的上一级。

如何获得 chrome 路径和个人资料路径

1. 打开 chrome 浏览器
2. 在地址栏输入 chrome://version/
3. 在界面上你可以看到类似「可执行文件路径」，路径类似/Applications/Google Chrome.app/Contents/MacOS/Google Chrome，这个路径就是你要设置的CS_EXECUTABLE_PATH
4. 找到「个人资料路径」，路径类似/Users/xxx/Library/Application Support/Google/Chrome/Default，CS_USER_DATA_DIR的值就是/Users/xxx/Library/Application Support/Google/Chrome（注意路径不包含最后的Default）。

⚠️注意事项

• 部分地区使用 google 或者 bing 进行网络搜索时需确保代理设置正确
• CS_EXECUTABLE_PATH最好不要设置成你当前正在使用的浏览器，那样会冲突，你如果习惯了使用 google chrome，则可以安装 edge、brave 等 chromium 内核的浏览器，反之如果你习惯使用 edge 浏览器，则强烈建议你安装个 google chrome。

MCP 配置推荐

{
  "mcpServers": {
    "cstoolbox": {
      "command": "uvx",
      "args": [
        "cstoolbox"
      ],
      "env": {
        "CS_PROXY": "http://{your-proxy-server}",
        "CS_HEADLESS": "false",
        "CS_BROWSER_TYPE": "chromium",
        "CS_EXECUTABLE_PATH": "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
        "CS_USER_DATA_DIR": "~/Library/Application Support/Google/Chrome"
      }
    }
  }
}

python 调用示例

更多 python 调用示例请参考tests/mcp_client.py文件

from pathlib import Path

from mcp import ClientSession, StdioServerParameters, types
from mcp.client.stdio import stdio_client

project_root = Path(__file__).resolve().parent.parent

# Create server parameters for stdio connection
server_params = StdioServerParameters(
    command="python",
    args=[f"{project_root}/src/cstoolbox/main.py"],  # Path to cstoolbox mcp main.py
    env={
        "CS_PROXY": "http://localhost:15154",
        "CS_BROWSER_TZ": "Asia/Shanghai",
        "CS_BROWSER_LANG": "zh-CN"
    },
)


# Optional: create a sampling callback
async def handle_sampling_message(
    message: types.CreateMessageRequestParams,
) -> types.CreateMessageResult:
    return types.CreateMessageResult(
        role="assistant",
        content=types.TextContent(
            type="text",
            text="Hello, world! from model",
        ),
        model="gpt-3.5-turbo",
        stopReason="endTurn",
    )


async def run():
    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write, sampling_callback=handle_sampling_message) as session:
            # Initialize the connection
            await session.initialize()

            # List available tools
            tools = await session.list_tools()

            # Call a web search tool
            result = await session.call_tool(
                "web_search",
                arguments={"provider": "bing", "kw": "deepseek r2", "number": 10, "page": 1, "time_period": "month"},
            )
            print(result)

if __name__ == "__main__":
    import asyncio

    asyncio.run(run())

开发

1. 克隆源代码仓库:

git clone https://github.com/aidyou/cstoolbox.git
cd cstoolbox

1. 安装 uv
   Macos/Linux

curl -LsSf https://astral.sh/uv/install.sh | sh

Windows使用irm下载并安装

powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | iex"

1. 创建并激活venv环境:

uv venv --python=python3.12
source .venv/bin/activate

1. 安装依赖:

uv pip install .

1. 启动测试:

mcp dev src/cstoolbox/main.py

现在你可以通过http://127.0.0.1:6274/#tools 进行功能测试

1. http 接口测试
   在.vscode/launch.json文件中添加以下配置

{
    "version": "0.2.0",
    "configurations": [
        {
            "name": "http dev",
            "type": "debugpy",
            "request": "launch",
            "module": "cstoolbox.http_api",
            "args": [
            ],
            "console": "integratedTerminal",
            "env": {
                "PYTHONPATH": "${workspaceFolder}/src",
                "CS_BROWSER_TZ": "Asia/Shanghai",
                "CS_BROWSER_LANG": "zh-CN",
                "CS_LOG_LEVEL": "DEBUG",
                "CS_PROXY": "http://localhost:15154"
            },
            "python": "${workspaceFolder}/.venv/bin/python"
        }
    ]
}

env配置中CS_*请根据实际情况进行调整，在 vscode 中启动调试后即可通过如下接口进行测试：

• 搜索： curl http://localhost:12321/chp/web_search?provider=google&kw=deepseek+r2&number=10&page=1
• 内容抓取：curl http://localhost:12321/chp/web_crawler?url=https://medium.com/@lbq999/deepseek-r2-is-around-the-corner-c449a41bfec6

协议

本项目采用 MIT 协议 开源，您可以自由使用、修改和分发本软件。