feat: add update_tool

Author: steven-diaz-vapi

src/schemas/index.ts

@@ -363,10 +363,8 @@ const BackoffPlanSchema = z.object({
   baseDelaySeconds: z.number().default(1).describe('Base delay between retries in seconds'),
 });
 
-export const CreateToolInputSchema = z.object({
-  type: z.enum(['sms', 'transferCall', 'function', 'apiRequest'])
-    .describe('Type of the tool to create'),
-  
+// Base tool configuration schema (reusable for both create and update)
+const BaseToolConfigSchema = z.object({
   // Common fields for all tools
   name: z.string().optional().describe('Name of the function/tool'),
   description: z.string().optional().describe('Description of what the function/tool does'),
@@ -400,10 +398,21 @@ export const CreateToolInputSchema = z.object({
   }).optional().describe('API Request tool configuration - for HTTP API integration'),
 });
 
+export const CreateToolInputSchema = BaseToolConfigSchema.extend({
+  type: z.enum(['sms', 'transferCall', 'function', 'apiRequest'])
+    .describe('Type of the tool to create'),
+});
+
+export const UpdateToolInputSchema = BaseToolConfigSchema.extend({
+  toolId: z.string().describe('ID of the tool to update'),
+});
+
 export const ToolOutputSchema = BaseResponseSchema.extend({
   type: z
     .string()
     .describe('Type of the tool (dtmf, function, mcp, query, etc.)'),
   name: z.string().describe('Name of the tool'),
   description: z.string().describe('Description of the tool'),
+  parameters: z.record(z.any()).describe('Parameters of the tool'),
+  server: ServerSchema.describe('Server of the tool'),
 });

src/tools/tool.ts

@@ -1,8 +1,8 @@
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { VapiClient, Vapi } from '@vapi-ai/server-sdk';
 
-import { GetToolInputSchema, CreateToolInputSchema } from '../schemas/index.js';
-import { transformToolInput, transformToolOutput } from '../transformers/index.js';
+import { GetToolInputSchema, CreateToolInputSchema, UpdateToolInputSchema } from '../schemas/index.js';
+import { transformToolInput, transformUpdateToolInput, transformToolOutput } from '../transformers/index.js';
 import { createToolHandler } from './utils.js';
 
 export const registerToolTools = (
@@ -39,4 +39,15 @@ export const registerToolTools = (
       return transformToolOutput(tool);
     })
   );
+
+  server.tool(
+    'update_tool',
+    'Updates an existing Vapi tool',
+    UpdateToolInputSchema.shape,
+    createToolHandler(async (data) => {
+      const updateToolDto = transformUpdateToolInput(data);
+      const tool = await vapiClient.tools.update(data.toolId, updateToolDto);
+      return transformToolOutput(tool);
+    })
+  );
 };

src/transformers/index.ts

@@ -9,6 +9,7 @@ import {
   ToolOutputSchema,
   UpdateAssistantInputSchema,
   CreateToolInputSchema,
+  UpdateToolInputSchema,
 } from '../schemas/index.js';
 
 // ===== Assistant Transformers =====
@@ -310,8 +311,61 @@ export function transformToolInput(
   return toolDto;
 }
 
+export function transformUpdateToolInput(
+  input: z.infer<typeof UpdateToolInputSchema>
+): any {
+  let updateDto: any = {};
+
+  // Add function definition if name and description are provided
+  if (input.name || input.description) {
+    updateDto.function = {
+      ...(input.name && { name: input.name }),
+      ...(input.description && { description: input.description }),
+    };
+  }
+
+  // Handle SMS tool configuration
+  if (input.sms?.metadata) {
+    updateDto.metadata = input.sms.metadata;
+  }
+
+  // Handle Transfer call tool configuration
+  if (input.transferCall?.destinations) {
+    updateDto.destinations = input.transferCall.destinations;
+  }
+
+  // Handle Function tool configuration
+  if (input.function?.parameters && input.function?.server) {
+    // For function tools, add parameters to the existing function object
+    if (updateDto.function) {
+      updateDto.function.parameters = input.function.parameters;
+    } else {
+      updateDto.function = {
+        parameters: input.function.parameters,
+      };
+    }
+    
+    updateDto.server = {
+      url: input.function.server.url,
+      ...(input.function.server.headers && { headers: input.function.server.headers }),
+    };
+  }
+
+  // Handle API Request tool configuration
+  if (input.apiRequest) {
+    if (input.apiRequest.url) updateDto.url = input.apiRequest.url;
+    if (input.apiRequest.method) updateDto.method = input.apiRequest.method;
+    if (input.apiRequest.headers) updateDto.headers = input.apiRequest.headers;
+    if (input.apiRequest.body) updateDto.body = input.apiRequest.body;
+    if (input.apiRequest.backoffPlan) updateDto.backoffPlan = input.apiRequest.backoffPlan;
+    if (input.apiRequest.timeoutSeconds) updateDto.timeoutSeconds = input.apiRequest.timeoutSeconds;
+  }
+
+  return updateDto;
+}
+
 export function transformToolOutput(
-  tool: any
+  tool: Vapi.ToolsGetResponse
 ): z.infer<typeof ToolOutputSchema> {
   return {
     id: tool.id,
@@ -320,5 +374,10 @@ export function transformToolOutput(
     type: tool.type || '',
     name: tool.function?.name || '',
     description: tool.function?.description || '',
+    parameters: tool.function?.parameters || {},
+    server: {
+      url: tool.server?.url || '',
+      headers: tool.server?.headers as Record<string, string> || {},
+    }
   };
 }