@budibase/server/src/ai/tools/toAiSdkTools.ts

Budibase Web Server

import { Tool as BudibaseTool } from "@budibase/types"
import { tool, type ToolSet } from "ai"
import { z } from "zod"

export function toAiSdkTools(tools: BudibaseTool[]): ToolSet {
  const mapped: ToolSet = {}

  for (const t of tools) {
    const schema = t.parameters ?? z.object({})

    mapped[t.name] = tool({
      description: t.description,
      inputSchema: schema,
      execute: async (input: unknown) => {
        return await t.handler(input)
      },
    })
  }

  return mapped
}