mirror of
https://github.com/supabase/supabase.git
synced 2026-10-19 08:09:11 +03:00
Charis
9ae8432dc7
## Summary Migrates the create, delete, and run notebook assistant tools to the tool contract + server pattern used by `update_notebook`. Each tool now defines its schema once in a shared contract, with server-only execution split into a `.server.ts` module. This removes duplicated client-side schemas and gives the three tools one shared notebook mutation output schema. ## Changes - Add `create-notebook`, `delete-notebook`, and `run-notebook` tool contracts, each with a `.server.ts` implementation. - Remove the duplicated client-side schemas for these tools from `notebook-tools.ts` and the registry. - Add `notebook-mutation-output.ts` as the single output schema shared by the notebook mutation tools. - Add `lib/ai/opt-in-level.ts` with the opt-in level helpers, extracted from the existing module to break an import cycle. - Add optional `ToolContract.outputOnInvalid` so `run_notebook` history replay can reuse a stored output when input fails validation. - Update the AI assistant panel renderers (proposal, run, opt-in modal and request) and the tool filter, sanitizer, and cache invalidation to use the new contracts. ## Deviations worth reviewing - **`outputOnInvalid` on `ToolContract`.** This is a new optional hook. It exists so `run_notebook` history replay can return a valid output when stored input no longer passes validation. Reviewers should confirm this is the right level for the hook rather than handling it inside the tool. - **`opt-in-level.ts` extraction.** The helpers were moved into their own module to break a circular import between the tool modules and the opt-in components. This adds one file, but no behavior change is intended. - **`create-notebook` contract imports the data-layer `notebook-schema`.** This matches `update-notebook`. It should be moved behind the contract before the planned ESLint rule step, which restricts contract modules from importing data-layer code. ## Testing - `vitest` for `lib/ai`, `components/ui/AIAssistantPanel`, and `state`: 920 tests pass across 71 files. - `tsc`: one error remains in `packages/ui-patterns` (`InstructionBlocks.tsx`), which is unrelated to this change. I have not verified whether it also fails on the base branch. - ESLint reports no errors on the changed files (9 `no-explicit-any` warnings in `tool-filter.ts` and its test); Prettier is clean. - Not run: knip and the lint ratchet.
25 lines
1.0 KiB
TypeScript
25 lines
1.0 KiB
TypeScript
import { z } from 'zod'
|
|
|
|
import { notebookMutationOutputSchema } from './notebook-mutation-output'
|
|
import type { ToolContract } from './types'
|
|
import { agentNotebookSchema } from '@/data/content/notebooks/notebook-schema'
|
|
|
|
const inputSchema = z.object({
|
|
name: z.string().describe('A short, descriptive name for the notebook.'),
|
|
description: z.string().optional().describe('A short description of what the notebook is for.'),
|
|
content: agentNotebookSchema.describe(
|
|
'The notebook content: a schema version and an ordered list of cells (markdown, database, or log). Cells must not include an id — one is assigned when the notebook is saved.'
|
|
),
|
|
})
|
|
|
|
export const contract = {
|
|
name: 'create_notebook',
|
|
description:
|
|
'Asks the user to create a new notebook with the given cells. Requires user approval before creating.',
|
|
inputSchema,
|
|
outputSchema: notebookMutationOutputSchema,
|
|
minOptInLevel: 'schema',
|
|
needsApproval: true,
|
|
toModelOutput: ({ id, name }) => ({ id, name }),
|
|
} as const satisfies ToolContract
|