Ollamaで実装する無料 MCP Agent
ローカル MCP Agent の作成
今回は Ollama を経由してローカル LLM で、MCP Agent を動作させます。API は OpenAI 互換のものを使っています。この構成なら開発時には電気代以外の費用がかからないというメリットがあります。
今回のサンプルコードと動作画面
https://github.com/SoraKumo001/mcp-agent

カスタム Transport を作って、McpServer を直接 McpClient に接続する
MCP のやり取りで@modelcontextprotocol/sdkを使った場合、McpServer を McpClient で扱うには、標準だと stdio,websocket,http を経由する Transport しか用意されていません。別プロセスにしたり、WebServer を立ち上げたりすると開発が面倒になるので、カスタムの Transport を作成して、直接やりとりをガッチンコさせます。
libs/direct-transport.ts
1import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";2import type { JSONRPCMessage } from "@modelcontextprotocol/sdk/types.js";34class DirectClientTransport implements Transport {5 onclose?: () => void;6 onmessage?: (message: JSONRPCMessage) => void;7 constructor(private serverTransport: Transport) {}8 async start() {}9 async close() {}10 async send(message: JSONRPCMessage) {11 this.serverTransport.onmessage?.(message);12 }13}1415export class DirectServerTransport implements Transport {16 onclose?: () => void;17 onmessage?: (message: JSONRPCMessage) => void;18 clientTransport: DirectClientTransport;19 constructor() {20 this.clientTransport = new DirectClientTransport(this);21 }22 async start() {}23 async close() {}24 async send(message: JSONRPCMessage) {25 this.clientTransport.onmessage?.(message);26 }27 getClientTransport() {28 return this.clientTransport;29 }30}
MCP Server を作成
現在の時刻を返す McpServer と指定した地点の天気予報を返す McpServer を作成します。
mcp-servers/get-current-time.ts
時間を返すだけです
1import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";23const server = new McpServer({4 name: "時間表示サーバー",5 version: "1.0.0",6});78server.tool("get-current-time", "現在の時刻を返す", async () => {9 return {10 content: [11 {12 type: "text",13 text: new Date().toLocaleString("ja-JP", {14 year: "numeric",15 month: "long",16 day: "numeric",17 weekday: "long",18 hour: "2-digit",19 minute: "2-digit",20 second: "2-digit",21 }),22 },23 ],24 };25});2627export const TimeServer = server;
mcp-servers/get-weather.ts
気象庁のサイトから天気情報の文章を取得します
1import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";2import { z } from "zod";34interface Center {5 name: string;6 enName: string;7 officeName?: string;8 children?: string[];9 parent?: string;10 kana?: string;11}12interface Centers {13 [key: string]: Center;14}15interface Area {16 centers: Centers;17 offices: Centers;18 class10s: Centers;19 class15s: Centers;20 class20s: Centers;21}2223interface Weather {24 publishingOffice: string;25 reportDatetime: Date;26 targetArea: string;27 headlineText: string;28 text: string;29}3031const server = new McpServer({32 name: "天気予報サーバー",33 version: "1.0.0",34});3536server.tool(37 "get-weather",38 `指定した都道府県の天気予報を返す`,39 {40 name: z.string({41 description: "都道府県名の漢字、例「東京」",42 }),43 },44 async ({ name: areaName }) => {45 const result = await fetch(46 "https://www.jma.go.jp/bosai/common/const/area.json"47 )48 .then((v) => v.json())49 .then((v: Area) => v.offices)50 .then((v: Centers) =>51 Object.entries(v).flatMap(([id, { name }]) =>52 name.includes(areaName) ? [id] : []53 )54 );55 const weathers = await Promise.all(56 result.map((id) =>57 fetch(58 `https://www.jma.go.jp/bosai/forecast/data/overview_forecast/${id}.json`59 )60 .then((v) => v.json())61 .then((v: Weather) => v.text)62 )63 );64 return {65 content: [66 {67 type: "text",68 text: weathers.join("---"),69 },70 ],71 };72 }73);7475export const WeatherServer = server;
MCP Agent の作成
MCP Agent で必要なのは、McpServer からツール情報を収集する部分です。getMcpToolsで McpClient を作成し、McpServer と接続してツールを取り出しています。
次に、取り出したツールをopenai.chat.completions.createで渡して、応答するために必要なツールの呼び出しを列挙させます。content.message.tool_callsに入っている情報に基づいて、mcp.callToolを呼び出していきます。
ツールの実行結果を message に格納したら、openai.chat.completions.createで再度応答を生成します。
時々、頓珍漢な応答が生成されるので、結果の正当性を確認してダメそうならリトライするような構造も必要そうです。
Ollama でエージェントに使う LLM を探すときは、toolsのタグがついているものが必要になります。また、tools 対応でも LLM によって、ツール利用の失敗率が異常に高いものがあるので、最適なものを探してくる必要があります。
index.ts
1import { Client } from "@modelcontextprotocol/sdk/client/index.js";2import OpenAI from "openai";3import { DirectServerTransport } from "./libs/direct-transport.js";4import { TimeServer } from "./mcp-servers/get-current-time.js";5import { WeatherServer } from "./mcp-servers/get-weather.js";67import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";8import type {9 ChatCompletionContentPartText,10 ChatCompletionMessageParam,11 ChatCompletionTool,12} from "openai/resources.mjs";1314const getMcpTools = async (servers: McpServer[]) => {15 const tools: ChatCompletionTool[] = [];16 const functionMap: Record<string, Client> = {};17 const clients: Client[] = [];18 for (const server of servers) {19 const mcpClient = new Client({20 name: "mcp-client-cli",21 version: "1.0.0",22 });23 // Connecting McpServer directly to McpClient24 const transport = new DirectServerTransport();25 server.connect(transport);26 await mcpClient.connect(transport.getClientTransport());2728 clients.push(mcpClient);29 const toolsResult = await mcpClient.listTools();30 tools.push(31 ...toolsResult.tools.map((tool): ChatCompletionTool => {32 functionMap[tool.name] = mcpClient;33 return {34 type: "function",35 function: {36 name: tool.name,37 description: tool.description,38 parameters: tool.inputSchema,39 },40 };41 })42 );43 }44 const close = () => {45 return Promise.all(46 clients.map(async (v) => {47 await v.close();48 })49 );50 };51 return { tools, functionMap, close };52};5354const query = async (55 openai: OpenAI,56 model: string,57 mcpTools: Awaited<ReturnType<typeof getMcpTools>>,58 query: string59) => {60 console.log(`\n[question] ${query}`);61 const messages: ChatCompletionMessageParam[] = [62 {63 role: "system",64 content: "日本語を使用する,タグを出力しない,plain/textで回答する",65 },66 {67 role: "user",68 content: query,69 },70 ];7172 const response = await openai.chat.completions.create({73 model,74 messages: messages,75 tools: mcpTools.tools,76 });7778 for (const content of response.choices) {79 if (content.finish_reason === "tool_calls" && content.message.tool_calls) {80 await Promise.all(81 content.message.tool_calls.map(async (toolCall) => {82 const toolName = toolCall.function.name;83 const toolArgs = toolCall.function.arguments;84 const mcp = mcpTools.functionMap[toolName];85 console.info(`[tool] ${toolName} ${toolArgs}`);86 if (!mcp) {87 throw new Error(`Tool ${toolName} not found`);88 }8990 const toolResult = await mcp.callTool({91 name: toolName,92 arguments: JSON.parse(toolArgs),93 });94 messages.push({95 role: "tool",96 tool_call_id: toolCall.id,97 content: toolResult.content as Array<ChatCompletionContentPartText>,98 });99 })100 );101102 const response = await openai.chat.completions.create({103 model,104 messages,105 max_completion_tokens: 512,106 stream: true,107 });108 console.log("[answer]");109 for await (const message of response) {110 process.stdout.write(message.choices[0].delta.content!);111 }112 console.log();113 } else {114 console.log(content.message.content);115 }116 }117};118119async function main() {120 const openai = new OpenAI({121 baseURL: "http://localhost:11434/v1",122 apiKey: "ollama",123 });124 const mcpTools = await getMcpTools([TimeServer, WeatherServer]);125 const model = "qwen2.5-coder:14b";126 await query(openai, model, mcpTools, "東京の天気は?");127 await query(openai, model, mcpTools, "今日の青森と千葉の天気は?");128 await query(openai, model, mcpTools, "今日は何曜日?");129 await mcpTools.close();130}131132main();
実行結果
正常応答
1[question] 東京の天気は?2[tool] get-weather {"name":"東京"}3[answer]4現在、東京は曇っているようですね。5631日にかけては、たまに雨が降る可能性がありますから、傘を忘れないでくださいね。78また、4月1日には、伊豆諸島を中心に雨か雪が降る予定なので、そこの場合は特に注意が必要です。910[question] 今日の青森と千葉の天気は?11[tool] get-weather {"name":"青森"}12[tool] get-weather {"name":"千葉"}13[answer]14青森県は、曇りや晴れで、雪が降っている所があります。31日には雨が降る可能性があり、4月1日には晴れですが、午後は曇りになる見込みです。1516千葉県は、曇りで雨が降っている所があります。31日には曇りと雨が続く可能性あり、4月1日には雨となる見込みです。また、千葉県の太平洋沿岸では、31日には波が高いので注意が必要です。1718[question] 今日は何曜日?19[tool] get-current-time {}20[answer]21月曜日
正常応答
1[question] 東京の天気は?2[tool] get-weather {"name":"東京"}3[answer]4東京現在の天気は曇りです。5631日に高気圧の影响を受けますが、気圧の谷や湿った空気の影響があり、昼過ぎから雨が降る可能性があります。784月1日には、千島に中心持つ高気圧が北東へ移動し、雨となるでしょう。伊豆諸島では雷を伴う雨が降る可能性も高いため、外出する際は予防が必要です。910関東甲信地方全体では、曇りや晴れで、雨や雪の降る所もあります。海上では、31日は波が高い、4月1日はしけるため船舶注意が必要です。1112[question] 今日の青森と千葉の天気は?13[tool] get-weather {"name":"青森"}14[tool] get-weather {"name":"千葉"}15[answer]16青森県:17- 今日:曇りや晴れ、雪の降っている所がある18- 明日:晴れますが、午後よりは曇りとなる可能性あり1920千葉県:21- 今日:曇りで雨が降っている所がある22- 明日:雨となる見込み2324[question] 今日は何曜日?25[tool] get-current-time {}26[answer]27今日は月曜日です。
ツールの呼び出しの内容が、何故か content 側に出てしまう
1[question] 東京の天気は?2[tool] get-weather {"name":"東京"}3[answer]4現在、東京の天気は曇りしています。5631日には、曇りで昼過ぎから雨が降る可能性があります。784月1日には、千島の東側に中心を持つ高気圧が移動すると予想され、結果として雨となるでしょう。伊豆諸島では雷を伴う雨も考えられます。910[question] 今日の青森と千葉の天気は?11[tool] get-weather {"name":"青森"}12[tool] get-weather {"name":"千葉"}13[answer]14青森県と千葉県の天気予報は以下の通りです:1516### 青森県17- **31日**: 高気圧に覆われているため、晴れや曂りですが、昼前まで雪の降る可能性があります。18- **4月1日**: 高気圧に覆われたため、午後は曇りとなる見込みです。1920### 千葉県21- **31日**: 気圧の谷や湿った空気の影響を受けるため、曇りで雨が降る所があるでしょう。22- **4月1日**: 千島の東に中心を持つ高気圧が北東へ移動し、雨となる見込みです。2324千葉県の太平洋沿岸の海上では:25- **31日**: うねりを伴い、波が高い可能性があります。26- **4月1日**: うねりを伴いしけとなる見込みです。船舶は高波に注意してください。2728[question] 今日は何曜日?29{30 "name": "get-current-time",31 "arguments": null32}
まとめ
これから、様々なサービスで API が McpServer 化される機会が増えてくると思います。そのお鉢が自分に回ってきたときに、ローカルでサクッと動作確認が出来る環境を用意しておくと便利です。