UNPKG

@vapi-ai/mcp-server

Version:
52 lines (51 loc) 2.69 kB
import { CreateAssistantInputSchema, GetAssistantInputSchema, UpdateAssistantInputSchema, } from '../schemas/index.js'; import { transformAssistantInput, transformAssistantOutput, transformUpdateAssistantInput, } from '../transformers/index.js'; import { createToolHandler } from './utils.js'; export const registerAssistantTools = (server, vapiClient) => { server.tool('list_assistants', 'Lists all Vapi assistants', {}, createToolHandler(async () => { // console.log('list_assistants'); const assistants = await vapiClient.assistants.list({ limit: 10 }); // console.log('assistants', assistants); return assistants.map(transformAssistantOutput); })); server.tool('create_assistant', 'Creates a new Vapi assistant', CreateAssistantInputSchema.shape, createToolHandler(async (data) => { // console.log('create_assistant', data); const createAssistantDto = transformAssistantInput(data); const assistant = await vapiClient.assistants.create(createAssistantDto); return transformAssistantOutput(assistant); })); server.tool('get_assistant', 'Gets a Vapi assistant by ID', GetAssistantInputSchema.shape, createToolHandler(async (data) => { // console.log('get_assistant', data); const assistantId = data.assistantId; try { const assistant = await vapiClient.assistants.get(assistantId); if (!assistant) { throw new Error(`Assistant with ID ${assistantId} not found`); } return transformAssistantOutput(assistant); } catch (error) { console.error(`Error getting assistant: ${error.message}`); throw error; } })); server.tool('update_assistant', 'Updates an existing Vapi assistant', UpdateAssistantInputSchema.shape, createToolHandler(async (data) => { const assistantId = data.assistantId; try { // First check if the assistant exists const existingAssistant = await vapiClient.assistants.get(assistantId); if (!existingAssistant) { throw new Error(`Assistant with ID ${assistantId} not found`); } // Transform the update data const updateAssistantDto = transformUpdateAssistantInput(data); // Update the assistant const updatedAssistant = await vapiClient.assistants.update(assistantId, updateAssistantDto); return transformAssistantOutput(updatedAssistant); } catch (error) { console.error(`Error updating assistant: ${error.message}`); throw error; } })); };