mirror of
https://github.com/supabase/supabase.git
synced 2026-10-19 16:19:10 +03:00
## 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.
20 lines
795 B
TypeScript
20 lines
795 B
TypeScript
import { getAuthHeaders } from './auth-headers'
|
|
import { contract } from './delete-notebook'
|
|
import { toolFromContract } from './tool-from-contract'
|
|
import type { ToolFactory } from './types'
|
|
import { deleteContents } from '@/data/content/content-delete-mutation'
|
|
import { getNotebook } from '@/data/content/notebooks/notebook-query'
|
|
|
|
export const createTool: ToolFactory<typeof contract> = (ctx) => {
|
|
const { projectRef } = ctx
|
|
const authHeaders = getAuthHeaders(ctx.authorization)
|
|
|
|
return toolFromContract(contract, ctx, async ({ id }, { abortSignal }) => {
|
|
const notebook = await getNotebook({ projectRef, id }, abortSignal, authHeaders)
|
|
|
|
await deleteContents({ projectRef: projectRef ?? '', ids: [id] }, abortSignal, authHeaders)
|
|
|
|
return { id, name: notebook.name }
|
|
})
|
|
}
|