@budibase/server/src/api/routes/ai.ts

Budibase Web Server

import * as ai from "../controllers/ai"
import {
  chatAgentValidator,
  createToolSourceValidator,
  updateToolSourceValidator,
} from "./utils/validators/agent"
import { middleware } from "@budibase/pro"
import recaptcha from "../../middleware/recaptcha"
import { builderAdminRoutes, endpointGroupList } from "./endpointGroups"

export const licensedRoutes = endpointGroupList.group(
  middleware.licenseAuth,
  recaptcha
)

builderAdminRoutes
  .post("/api/ai/tables", ai.generateTables)
  .post("/api/agent/chat", chatAgentValidator(), ai.agentChat)
  .post("/api/agent/chat", ai.agentChat)
  .post("/api/agent/chat/stream", ai.agentChatStream)
  .delete("/api/agent/history/:historyId", ai.remove)
  .get("/api/agent/history", ai.fetchHistory)
  .post(
    "/api/agent/toolsource",
    createToolSourceValidator(),
    ai.createToolSource
  )
  .put(
    "/api/agent/toolsource",

    updateToolSourceValidator(),
    ai.updateToolSource
  )
  .delete(
    "/api/agent/toolsource/:toolSourceId",

    ai.deleteToolSource
  )
  .get("/api/agent/toolsource", ai.fetchToolSources)
  .post("/api/ai/cron", ai.generateCronExpression)
  .post("/api/ai/js", ai.generateJs)

licensedRoutes
  .post("/api/ai/chat", ai.chatCompletion)
  .post("/api/ai/upload-file", ai.uploadFile)