MCP Workers Ai

🚀 MCP 工人 AI

Cloudflare Workers 用の MCP サーバー SDKです。

🚀 クイックスタート

📦 インストール

以下のコマンドを使用してインストールします。

yarn add mcp-workers-ai
# または
npm install -S mcp-workers-ai

💻 使用例

基本的な使用法

MCP サーバーツールをロードするには、以下のコードを使用します。

import { loadTools } from "mcp-workers-ai"

const tools = await loadTools([
  import("@modelcontextprotocol/server-gitlab"),
  import("@modelcontextprotocol/server-slack"),
  ...
]);

// `tools` をLLM推論リクエストに渡します。

ツールを呼び出すには、以下のコードを使用します。

import { callTool } from "mcp-workers-ai"

// 通常、LLMが自動的に使用するツールを選択します。
const selected_tool = {
  arguments: {
    project_id: 'svensauleau/test',
    branch: 'main',
    files: [ ... ],
    commit_message: 'added unit tests'
  },
  name: 'push_files'
};

const res = await callTool(selected_tool)

// `res` を再度LLM推論リクエストに渡します。

高度な使用法

Wrangler の設定例は以下の通りです。

name = "test"
main = "src/index.ts"

[ai]
binding = "AI"

[vars]
GITLAB_PERSONAL_ACCESS_TOKEN = "glpat-aaaaaaaaaaaaaaaaaaaa"

[alias]
"@modelcontextprotocol/sdk/server/index.js" = "mcp-workers-ai/sdk/server/index.js"
"@modelcontextprotocol/sdk/server/stdio.js" = "mcp-workers-ai/sdk/server/stdio.js"

以下は、ワーカースクリプトの例です。

import { loadTools, callTool } from "mcp-workers-ai"

export default {
  async fetch(request: Request, env: any): Promise<Response> {
    // トークンを設定する前にツールをインポートすることを確認します。
    process.env.GITLAB_PERSONAL_ACCESS_TOKEN = env.GITLAB_PERSONAL_ACCESS_TOKEN;

    const tools = await loadTools([
      import("@modelcontextprotocol/server-gitlab/dist/"),
    ]);

    const prompt = await request.text();

    const response = await env.AI.run(
      "@hf/nousresearch/hermes-2-pro-mistral-7b",
      {
        messages: [{ role: "user", content: prompt }],
        tools,
      },
    );

    if (response.tool_calls && response.tool_calls.length > 0) {
      const selected_tool = response.tool_calls[0];
      const res = await callTool(selected_tool)

      if (res.content.length > 1) {
        throw new Error("too many responses")
      }

      const finalResponse = await env.AI.run(
        "@hf/nousresearch/hermes-2-pro-mistral-7b",
        {
          messages: [
            {
              role: "user",
              content: prompt,
            },
            {
              role: "assistant",
              content: "",
              tool_call: selected_tool.name,
            },
            {
              role: "tool",
              name: selected_tool.name,
              content: res.content[0].text,
            },
          ],
          tools,
        },
      );
      return new Response(finalResponse.response);

    } else {
      return new Response(response.response);
    }
  }
};

AIを呼び出すには、以下のコマンドを使用します。

$ curl http://example.com \
  -d "创建名为 'joke.txt' 的文件并存入我的 svensauleau/test 项目中，添加你最喜欢的笑话到主分支。使用提交信息 'added unit tests'"

我已经成功地在你的项目 'svensauleau/test' 主分支中创建了一个名为 'joke.txt' 的文件，并添加了我的 favorite 笑话。提交信息是 'added unit tests'。您可以在项目的仓库中查看该提交和文件。

結果