Snippet for creating tools for AI function calling

ai-tools.code-snippets

"AI Tool Template": {
    "prefix": "aiTool",
    "body": [
      "import { ToolResult } from './AITools';",
      "import { tool } from 'ai';",
      "import { z } from 'zod';",
      "",
      "export interface ${TM_FILENAME_BASE/(.*)/${1:/pascalcase}/}Params {",
      "  input: string;",
      "}",
      "",
      "type ${TM_FILENAME_BASE/(.*)/${1:/pascalcase}/}Result = ToolResult<{",
      "  output: string;",
      "}>;",
      "",
      "export async function ${TM_FILENAME_BASE/(.*)/${1:/camelcase}/}({",
      "  input,",
      "}: ${TM_FILENAME_BASE/(.*)/${1:/pascalcase}/}Params): Promise<${TM_FILENAME_BASE/(.*)/${1:/pascalcase}/}Result> {",
      "    return {",
      "      data: {",
      "        output: 'success',",
      "      },",
      "      saveToMemory: true,",
      "      error: null,",
      "    };",
      "}",
      "",
      "export const ${TM_FILENAME_BASE/(.*)/${1:/camelcase}/}Tool = tool({",
      "  description: 'Tool call',",
      "  parameters: z.object({",
      "    input: z",
      "      .string()",
      "      .describe('The output'),",
      "  }),",
      "  execute: async ({ input }) => {",
      "    return await ${TM_FILENAME_BASE/(.*)/${1:/camelcase}/}({ input });",
      "  },",
      "});"
    ],
    "description": "Creates an AI Tool for function calling"
  }