tool.ts

import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { VapiClient, Vapi } from '@vapi-ai/server-sdk';

import { GetToolInputSchema, CreateToolInputSchema, UpdateToolInputSchema } from '../schemas/index.js';
import { transformToolInput, transformUpdateToolInput, transformToolOutput } from '../transformers/index.js';
import { createToolHandler } from './utils.js';

export const registerToolTools = (
  server: McpServer,
  getClient: () => VapiClient
) => {
  server.tool(
    'list_tools',
    'Lists all Vapi tools',
    {},
    createToolHandler(async () => {
      const vapiClient = getClient();
      const tools = await vapiClient.tools.list({ limit: 10 });
      return tools.map(transformToolOutput);
    })
  );

  server.tool(
    'get_tool',
    'Gets details of a specific tool',
    GetToolInputSchema.shape,
    createToolHandler(async (data) => {
      const vapiClient = getClient();
      const tool = await vapiClient.tools.get(data.toolId);
      return transformToolOutput(tool);
    })
  );

  server.tool(
    'create_tool',
    'Creates a new Vapi tool',
    CreateToolInputSchema.shape,
    createToolHandler(async (data) => {
      const vapiClient = getClient();
      const createToolDto = transformToolInput(data);
      const tool = await vapiClient.tools.create(createToolDto);
      return transformToolOutput(tool);
    })
  );

  server.tool(
    'update_tool',
    'Updates an existing Vapi tool',
    UpdateToolInputSchema.shape,
    createToolHandler(async (data) => {
      const vapiClient = getClient();
      const updateToolDto = transformUpdateToolInput(data);
      const tool = await vapiClient.tools.update(data.toolId, updateToolDto);
      return transformToolOutput(tool);
    })
  );
};