DataTable

import { tool } from "ai"
import { DataTableSchema } from "@/components/taw/data-table"

export const showTable = tool({
  description: "Show data in a sortable table",
  parameters: z.object({ query: z.string() }),
  outputSchema: DataTableSchema,
  execute: async ({ query }) => {
    const result = await queryDatabase(query)
    return {
      id: slugify(query),
      title: result.title,
      columns: result.columns.map(c => ({
        key: c.key,
        label: c.label,
        type: c.type,
        sortable: true,
      })),
      rows: result.rows,
      total: result.total,
      defaultSort: { key: "revenue", direction: "desc" },
      source: { label: "Database", freshness: "just now" },
    }
  },
})

import { DataTable } from "@/components/taw/data-table"
import type { TawToolPart } from "taw-ui"

function ToolOutput({ part }: { part: TawToolPart }) {
  // Handles loading, error, and success states
  return <DataTable part={part} />
}