#!/usr/bin/env node
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { z } from "zod";
import {
getGenAIResponse,
getGoogleSearchResponse,
getRandomGreeting,
} from "./tools/index.ts";
import process from "node:process";
// MCPサーバーの初期化
const server = new McpServer({
name: "GenAI MCP Server",
version: "0.1.1",
capabilities: {
resource: {},
tools: {},
},
});
server.tool(
"call_gemini_or_vertex_ai",
"Calls either the Gemini Developer API or Vertex AI with a user prompt and options.",
{
userMessage: z.string().describe("The message/prompt to send to the Generative AI."),
options: z.object({
useVertexAI: z.boolean().describe("Set to true to use Vertex AI, false for Gemini Developer API."),
model: z.string().describe("The model name (e.g., 'gemini-1.5-flash-001') or full model path for Vertex AI."),
project: z.string().optional().describe("Google Cloud Project ID (required if useVertexAI is true)."),
location: z.string().optional().describe("Google Cloud Project Location (required if useVertexAI is true)."),
}).describe("Options for configuring the AI call."),
},
async ({ userMessage, options }: { userMessage: string; options: { useVertexAI: boolean; model: string; project?: string; location?: string; } }) => {
return await getGenAIResponse(userMessage, options);
},
);
server.tool(
"call_google_search",
"ユーザープロンプトを受け取り、Google検索を使用してAIから応答を生成します。",
{
userMessage: z.string().describe("検索および生成AIに送信するメッセージ/プロンプト。"),
options: z.object({
useVertexAI: z.boolean().describe("Vertex AIを使用する場合はtrue、Gemini Developer APIを使用する場合はfalseに設定。"),
model: z.string().describe("モデル名（例：'gemini-1.5-flash-001'）またはVertex AI用の完全なモデルパス。"),
project: z.string().optional().describe("Google Cloud Project ID（useVertexAIがtrueの場合は必須）。"),
location: z.string().optional().describe("Google Cloud Project Location（useVertexAIがtrueの場合は必須）。"),
apiVersion: z.string().optional().describe("API バージョン（デフォルトは 'v1'）。"),
}).describe("AI呼び出しを設定するためのオプション。"),
},
async ({ userMessage, options }: { userMessage: string; options: { useVertexAI: boolean; model: string; project?: string; location?: string; apiVersion?: string; } }) => {
return await getGoogleSearchResponse(userMessage, options);
},
);
server.tool(
"get_random_greeting",
"ユーザーからのプロンプトを受け取り、ランダムな挨拶を返します。",
{
userPrompt: z.string().describe("ユーザーからのメッセージやプロンプト。"),
},
async ({ userPrompt }: { userPrompt: string }) => {
return await getRandomGreeting(userPrompt);
},
);
// 起動
async function setMCPServer() {
const transport = new StdioServerTransport();
await server.connect(transport);
console.error("GenAI MCP Server running on stdio");
}
setMCPServer().catch((error) => {
console.error("Fatal error in setMCPServer():", error);
process.exit(1);
});
