mirror of
https://github.com/supabase/supabase.git
synced 2026-10-05 09:25:06 +03:00
Charis
7798e42435
## Summary - Adds `list_notebooks` (cursor-paginated) and `get_notebook` AI tools in `lib/ai/tools/notebook-tools.ts`, modeled directly on `report-tools.ts`: server-side `getContent`/`getNotebook` with the `authorization` header forwarded, zod-validated input. - `get_notebook` resolves every cell and exposes `unchecked_sql` as a plain `sql` field for the agent to read — display only, per the `safe-sql-execution` skill; nothing here executes SQL. - Registers both tools in `lib/ai/tools/index.ts` (same platform branch as reports) and in `lib/ai/tool-filter.ts`'s `toolSetValidationSchema` + `TOOL_CATEGORY_MAP` (`SCHEMA` tier). - Adds an optional `headers` param to `content-infinite-query.ts`'s `getContent`, mirroring the sibling `content-query.ts`, so the cursor-paginated fetch can carry the `Authorization` header from a server context. - New tools are behind the Explorer feature flag. Stacked on #48907 (1.4 — notebook query and mutation hooks), per the Notebooks implementation plan (stack 2.1). Resolves FE-4081 Resolves FE-4080 ## Test plan - [x] `pnpm exec tsc --noEmit` — no new errors - [x] `pnpm exec vitest run lib/ai/tools/notebook-tools.test.ts lib/ai/tools/index.test.ts lib/ai/tools/report-tools.test.ts data/content/notebooks` — 36/36 passing - [x] `pnpm --filter studio run lint` — no new warnings - [x] `pnpm exec prettier --check` on changed files — clean <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **New Features** * Added AI tools to list project notebooks with pagination. * Added AI support for retrieving notebook markdown and resolved SQL cell content. * Notebook tools now respect project and authorization context. * Notebook features are available only when Explorer access is enabled. * Content requests can forward custom request headers. * **Tests** * Added coverage for notebook tools, Explorer access, feature flags, authorization, pagination, and error handling. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
20 lines
1.0 KiB
TypeScript
20 lines
1.0 KiB
TypeScript
import { IS_PLATFORM } from 'common'
|
|
|
|
import { getServerFlags, type TrustedUserEmail } from '@/lib/server/configcat'
|
|
|
|
// Notebooks haven't shipped yet — they're gated behind the Explorer feature flag on the
|
|
// client (`useFlag('explorer')`). The AI tools that read notebook content (list_notebooks,
|
|
// get_notebook in lib/ai/tools/notebook-tools.ts) must not be advertised to the assistant until
|
|
// this resolves true for the requesting user, otherwise the assistant will believe notebooks
|
|
// exist for users who can't yet see them in the UI.
|
|
//
|
|
// Takes TrustedUserEmail, not a plain string: callers must go through
|
|
// lib/server/configcat's trustedUserEmail() to prove this came from a verified source
|
|
// (a decoded, signature-checked JWT claim), not a raw request param.
|
|
export async function isExplorerEnabled(userEmail?: TrustedUserEmail): Promise<boolean> {
|
|
if (!IS_PLATFORM) return false
|
|
|
|
const flags = await getServerFlags(userEmail)
|
|
return flags.some((flag) => flag.settingKey === 'explorer' && flag.settingValue === true)
|
|
}
|