多模型AI代理服务器

Polybrain MCP Server是一个连接AI代理与多个LLM模型的服务器,支持对话历史管理、模型切换和无缝Claude Code集成。

  • by danielwpz
  • TypeScript
  • 本地部署
  • 2 星标
  • 更新于
在 GitHub 查看 danielwpz/polybrain-mcp

使用说明

Polybrain MCP服务器

用于将AI代理连接到多个LLM模型的MCP(模型上下文协议)服务器。支持对话历史记录、模型切换和无缝的Claude Code集成。

特性

  • 多模型支持(OpenAI、OpenRouter、自定义端点)
  • 对话历史管理
  • 对话中切换型号
  • 扩展思维/推理支持(可由提供商配置)
  • 纯MCP协议(默认为静默)
  • 自动服务器管理

安装

npm install -g polybrain
# or
pnpm add -g polybrain

快速设置

1.配置模型

选项A:YAML(推荐)

创建 ~/.polybrain.yaml:

models:
  - id: "gpt-4o"
    modelName: "gpt-4o"
    baseUrl: "https://api.openai.com/v1"
    apiKey: "${OPENAI_API_KEY}"
    provider: "openai"

  - id: "gpt-5.1"
    modelName: "openai/gpt-5.1"
    baseUrl: "https://openrouter.io/api/v1"
    apiKey: "${OPENROUTER_KEY}"
    provider: "openrouter"

设置环境变量:

export OPENAI_API_KEY="sk-..."
export OPENROUTER_KEY="sk-or-..."

选项B:环境变量

export POLYBRAIN_BASE_URL="https://api.openai.com/v1"
export POLYBRAIN_API_KEY="sk-..."
export POLYBRAIN_MODEL_NAME="gpt-4o"

2.编码代理集成

克劳德代码

运行以下命令将Polybrain添加到Claude Code中:

claude mcp add -s user -t stdio polybrain -- polybrain

OpenAI Codex

打开 ~/.codex/config.toml 并添加:

[mcpServers.polybrain]
command = "polybrain"

用法

现在,您可以让您的编码代理咨询特定的模型。例如:

“询问deepseek在mac上安装python的最佳方法是什么”

或者:

“polybrain有哪些型号?”

配置参考

环境变量

  • POLYBRAIN_BASE_URL -LLM API基础URL
  • POLYBRAIN_API_KEY -API密钥
  • POLYBRAIN_MODEL_NAME -型号名称
  • POLYBRAIN_HTTP_PORT -服务器端口(默认值:32701)
  • POLYBRAIN_LOG_LEVEL -日志级别(默认值:info)
  • POLYBRAIN_DEBUG -启用stderr的调试日志记录
  • POLYBRAIN_CONFIG_PATH -自定义配置文件路径

YAML配置字段

httpPort: 32701                    # Optional
truncateLimit: 500                 # Optional
logLevel: info                      # Optional

models:                             # Required
  - id: "model-id"                 # Internal ID
    modelName: "actual-model-name"  # API model name
    baseUrl: "https://api.url/v1"  # API endpoint
    apiKey: "key or ${ENV_VAR}"    # API key
    provider: "openai"              # Optional: provider type for reasoning support

支持的提供商

这 provider 字段支持特定于提供者的功能,如扩展思维/推理。如果未指定,推理参数将不会传递给API(安全默认值)。

提供者 推理支持 有效值
OpenAI 是 "openai"
OpenRouter 各不相同 "openrouter"

示例:

  • 使用 provider: "openai" 用于OpenAI API模型(GPT-4,o-系列)
  • 使用 provider: "openrouter" 用于OpenRouter代理服务(支持400多种型号)
  • 省略 provider 如果您的端点不支持推理参数,请输入字段

推理示例:

models:
  - id: "gpt-o1"
    modelName: "o1"
    baseUrl: "https://api.openai.com/v1"
    apiKey: "${OPENAI_API_KEY}"
    provider: "openai"           # Enables reasoning support

  - id: "gpt-5.1"
    modelName: "openai/gpt-5.1"
    baseUrl: "https://openrouter.io/api/v1"
    apiKey: "${OPENROUTER_KEY}"
    provider: "openrouter"       # Enables reasoning support

要使用推理,请设置 reasoning: true 在聊天工具调用中。如果模型和提供者支持它,您将收到响应和推理内容。

发展

设置

pnpm install

构建

pnpm build

棉绒和格式

pnpm lint
pnpm format

类型检查

pnpm type-check

发展模式

pnpm dev

项目结构

src/
├── bin/polybrain.ts    # CLI entry point
├── launcher.ts         # Server launcher & management
├── http-server.ts      # HTTP server
├── index.ts            # Main server logic
├── mcp-tools.ts        # MCP tool definitions
├── conversation-manager.ts
├── openai-client.ts
├── config.ts
├── logger.ts
└── types.ts

调试

将调试日志启用到stderr:

{
  "mcpServers": {
    "polybrain": {
      "command": "polybrain",
      "env": {
        "POLYBRAIN_DEBUG": "true"
      }
    }
  }
}

重新启动服务器

在更改配置后 ~/.polybrain.yaml,重新启动HTTP后端服务器:

polybrain --restart

这会杀死后台HTTP服务器。下次使用polybrain时,它将自动启动一个具有更新配置的新服务器。

许可证

麻省理工学院