@sqlrooms/vega 0.29.0-rc.1 → 0.29.0-rc.10

package/LICENSE.md

@@ -1,6 +1,7 @@
 MIT License
 
-Copyright 2025 SQLRooms Contributors
+Copyright 2024-2026 SQLRooms Contributors
+Copyright Vis.gl contributors
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 

package/README.md

@@ -10,6 +10,7 @@ npm install @sqlrooms/vega @sqlrooms/duckdb @sqlrooms/ui
 
 - `VegaLiteChart` (simple + compound component API)
 - `createVegaChartTool()` for AI tool workflows
+- `createChartImageForMarkdownTool()` for AI-generated Markdown document image assets
 - `VegaChartToolResult`
 - editor utilities/hooks (`useVegaChartEditor`, `useVegaEditorContext`)
 
@@ -104,6 +105,28 @@ These fields are supplied by the LLM when invoking the tool (not passed into
 - `vegaLiteSpec`: Vega-Lite JSON string
 - `reasoning`: explanation shown to users for why this chart/spec was chosen
 
+## Markdown document image assets
+
+`createChartImageForMarkdownTool(store)` creates an AI-only companion tool that
+renders a Vega chart to SVG or PNG, stores it as an asset on a
+`@sqlrooms/documents` Markdown artifact, and returns a ready-to-insert Markdown
+image link such as:
+
+```md
+![Revenue by week](asset://chart-revenue-week)
+```
+
+Use this alongside the existing document commands when the assistant needs a
+portable conversation summary with static chart images instead of live SQL-backed
+charts.
+
+Chart images default to the light Vega theme with an explicit background so
+exported Markdown renders predictably in GitHub, Obsidian, PDF exports, and
+other document surfaces. When the requested static theme matches the current app
+theme, the background is resolved from the app's Tailwind `--background` token
+and written into the SVG/PNG as a concrete color. The tool also accepts
+`renderTheme: "dark"` and `background` for explicit dark/static export requests.
+
 ## Example apps
 
 - Vega example: https://github.com/sqlrooms/examples/tree/main/vega

package/dist/ChartImageForMarkdownTool.d.ts

@@ -0,0 +1,84 @@
+import type { AiSliceState } from '@sqlrooms/ai-core';
+import type { DocumentsSliceState } from '@sqlrooms/documents';
+import { type DuckDbSliceState } from '@sqlrooms/duckdb';
+import { z } from 'zod';
+export declare const ChartImageForMarkdownToolParameters: z.ZodObject<{
+    documentArtifactId: z.ZodString;
+    sourceToolCallId: z.ZodOptional<z.ZodString>;
+    sqlQuery: z.ZodOptional<z.ZodString>;
+    vegaLiteSpec: z.ZodOptional<z.ZodString>;
+    format: z.ZodDefault<z.ZodOptional<z.ZodEnum<{
+        svg: "svg";
+        png: "png";
+    }>>>;
+    assetId: z.ZodOptional<z.ZodString>;
+    alt: z.ZodString;
+    title: z.ZodOptional<z.ZodString>;
+    width: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+    height: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+    pngScale: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+    renderTheme: z.ZodDefault<z.ZodOptional<z.ZodEnum<{
+        light: "light";
+        dark: "dark";
+    }>>>;
+    background: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type ChartImageForMarkdownToolParameters = z.infer<typeof ChartImageForMarkdownToolParameters>;
+export type ChartImageForMarkdownToolOutput = {
+    success: boolean;
+    details: string;
+    documentArtifactId: string;
+    assetId?: string;
+    markdown?: string;
+    mediaType?: 'image/svg+xml' | 'image/png';
+    encoding?: 'utf8' | 'base64';
+    format?: 'svg' | 'png';
+    renderTheme?: 'light' | 'dark';
+    background?: string;
+    error?: string;
+};
+type StoreLike<S> = {
+    getState: () => S;
+};
+type ChartImageToolState = AiSliceState & DuckDbSliceState & DocumentsSliceState;
+export declare function createChartImageForMarkdownTool<S extends ChartImageToolState = ChartImageToolState>(store: StoreLike<S>): import("ai").Tool<{
+    documentArtifactId: string;
+    format: "svg" | "png";
+    alt: string;
+    width: number;
+    height: number;
+    pngScale: number;
+    renderTheme: "light" | "dark";
+    sourceToolCallId?: string | undefined;
+    sqlQuery?: string | undefined;
+    vegaLiteSpec?: string | undefined;
+    assetId?: string | undefined;
+    title?: string | undefined;
+    background?: string | undefined;
+}, {
+    success: false;
+    details: string;
+    documentArtifactId: string;
+    error: string;
+    assetId?: undefined;
+    markdown?: undefined;
+    mediaType?: undefined;
+    encoding?: undefined;
+    format?: undefined;
+    renderTheme?: undefined;
+    background?: undefined;
+} | {
+    success: true;
+    details: string;
+    documentArtifactId: string;
+    assetId: string;
+    markdown: string;
+    mediaType: "image/svg+xml" | "image/png";
+    encoding: "utf8" | "base64";
+    format: "svg" | "png";
+    renderTheme: "light" | "dark";
+    background: string;
+    error?: undefined;
+}>;
+export {};
+//# sourceMappingURL=ChartImageForMarkdownTool.d.ts.map

package/dist/ChartImageForMarkdownTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChartImageForMarkdownTool.d.ts","sourceRoot":"","sources":["../src/ChartImageForMarkdownTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,YAAY,EAAC,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EAAC,mBAAmB,EAAC,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAmB,KAAK,gBAAgB,EAAC,MAAM,kBAAkB,CAAC;AAIzE,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAStB,eAAO,MAAM,mCAAmC;;;;;;;;;;;;;;;;;;;;iBA6C7C,CAAC;AAEJ,MAAM,MAAM,mCAAmC,GAAG,CAAC,CAAC,KAAK,CACvD,OAAO,mCAAmC,CAC3C,CAAC;AAEF,MAAM,MAAM,+BAA+B,GAAG;IAC5C,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,eAAe,GAAG,WAAW,CAAC;IAC1C,QAAQ,CAAC,EAAE,MAAM,GAAG,QAAQ,CAAC;IAC7B,MAAM,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC;IACvB,WAAW,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,KAAK,SAAS,CAAC,CAAC,IAAI;IAClB,QAAQ,EAAE,MAAM,CAAC,CAAC;CACnB,CAAC;AAEF,KAAK,mBAAmB,GAAG,YAAY,GACrC,gBAAgB,GAChB,mBAAmB,CAAC;AAEtB,wBAAgB,+BAA+B,CAC7C,CAAC,SAAS,mBAAmB,GAAG,mBAAmB,EACnD,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0FpB"}

package/dist/ChartImageForMarkdownTool.js

@@ -0,0 +1,287 @@
+import { arrowTableToJson } from '@sqlrooms/duckdb';
+import { tool } from 'ai';
+import { parse as vegaParse, View } from 'vega';
+import { compile } from 'vega-lite';
+import { z } from 'zod';
+import { darkTheme } from './themes/darkTheme';
+import { lightTheme } from './themes/lightTheme';
+const TAILWIND_BACKGROUND_VARIABLE = '--background';
+const FALLBACK_LIGHT_BACKGROUND = '#ffffff';
+const FALLBACK_DARK_BACKGROUND = 'oklch(0.145 0 0)';
+export const ChartImageForMarkdownToolParameters = z
+    .object({
+    documentArtifactId: z
+        .string()
+        .describe('Target Markdown document artifact ID.'),
+    sourceToolCallId: z
+        .string()
+        .optional()
+        .describe('Prior successful chart tool call ID to render.'),
+    sqlQuery: z
+        .string()
+        .optional()
+        .describe('SQL query to render when no sourceToolCallId is provided.'),
+    vegaLiteSpec: z
+        .string()
+        .optional()
+        .describe('Vega-Lite JSON spec string without inline data.'),
+    format: z.enum(['svg', 'png']).optional().default('svg'),
+    assetId: z.string().optional().describe('Stable document-local asset ID.'),
+    alt: z.string().describe('Markdown image alt text.'),
+    title: z.string().optional().describe('Optional image title.'),
+    width: z.number().positive().optional().default(960),
+    height: z.number().positive().optional().default(540),
+    pngScale: z.number().positive().optional().default(2),
+    renderTheme: z
+        .enum(['light', 'dark'])
+        .optional()
+        .default('light')
+        .describe('Static chart theme for the rendered image. Defaults to light for portable Markdown documents.'),
+    background: z
+        .string()
+        .optional()
+        .describe('CSS color used as the exported image background. Defaults to the resolved Tailwind --background token for the requested theme when available.'),
+})
+    .refine((input) => Boolean(input.sourceToolCallId) ||
+    (Boolean(input.sqlQuery) && Boolean(input.vegaLiteSpec)), {
+    message: 'Provide sourceToolCallId or both sqlQuery and vegaLiteSpec.',
+});
+export function createChartImageForMarkdownTool(store) {
+    return tool({
+        description: `Render an existing Vega chart analysis as a static SVG or PNG image asset for a Markdown document.
+Use this before embedding chart images in document artifacts. The tool stores the image in the target document and returns a Markdown image link using asset://.
+For portable Markdown documents, the default renderTheme is light with a white background. Only request dark rendering when the user explicitly wants a dark static image.`,
+        inputSchema: ChartImageForMarkdownToolParameters,
+        execute: async (params, options) => {
+            const abortSignal = options?.abortSignal;
+            try {
+                const source = resolveChartSource(store.getState(), params);
+                if (!source.success) {
+                    return {
+                        success: false,
+                        details: source.error,
+                        documentArtifactId: params.documentArtifactId,
+                        error: source.error,
+                    };
+                }
+                const connector = await store.getState().db.getConnector();
+                const queryResult = await connector.query(source.sqlQuery, {
+                    signal: abortSignal,
+                });
+                if (abortSignal?.aborted) {
+                    throw new Error('Chart image rendering was aborted');
+                }
+                const values = arrowTableToJson(queryResult);
+                const image = await renderVegaLiteImage({
+                    spec: source.vegaLiteSpec,
+                    values,
+                    format: params.format,
+                    width: params.width,
+                    height: params.height,
+                    pngScale: params.pngScale,
+                    renderTheme: params.renderTheme,
+                    background: resolveRenderBackground(params),
+                });
+                const timestamp = Date.now();
+                const assetId = params.assetId ?? createChartAssetId(params, source.sourceToolCallId);
+                const filename = `${assetId}.${params.format}`;
+                store.getState().documents.upsertAsset(params.documentArtifactId, {
+                    id: assetId,
+                    mediaType: image.mediaType,
+                    encoding: image.encoding,
+                    data: image.data,
+                    filename,
+                    alt: params.alt,
+                    title: params.title,
+                    provenance: {
+                        sourceToolCallId: source.sourceToolCallId,
+                        sqlQuery: source.sqlQuery,
+                        vegaLiteSpec: source.vegaLiteSpec,
+                        renderTheme: params.renderTheme,
+                        background: image.background,
+                        renderedAt: new Date(timestamp).toISOString(),
+                    },
+                    createdAt: timestamp,
+                    updatedAt: timestamp,
+                });
+                return {
+                    success: true,
+                    details: `Stored ${params.format.toUpperCase()} chart image asset "${assetId}" in document "${params.documentArtifactId}".`,
+                    documentArtifactId: params.documentArtifactId,
+                    assetId,
+                    markdown: `![${escapeMarkdownAlt(params.alt)}](asset://${encodeURIComponent(assetId)})`,
+                    mediaType: image.mediaType,
+                    encoding: image.encoding,
+                    format: params.format,
+                    renderTheme: params.renderTheme,
+                    background: image.background,
+                };
+            }
+            catch (error) {
+                const message = error instanceof Error ? error.message : String(error);
+                return {
+                    success: false,
+                    details: `Failed to render chart image: ${message}`,
+                    documentArtifactId: params.documentArtifactId,
+                    error: message,
+                };
+            }
+        },
+        toModelOutput: ({ output }) => ({
+            type: 'text',
+            value: JSON.stringify(output),
+        }),
+    });
+}
+function resolveChartSource(state, params) {
+    if (params.sourceToolCallId) {
+        const output = findChartToolOutput(state, params.sourceToolCallId);
+        if (!output) {
+            return {
+                success: false,
+                error: `No successful chart tool call found for "${params.sourceToolCallId}".`,
+            };
+        }
+        if (!output.sqlQuery || !output.vegaLiteSpec) {
+            return {
+                success: false,
+                error: `Chart tool call "${params.sourceToolCallId}" has no renderable SQL query or Vega-Lite spec.`,
+            };
+        }
+        return {
+            success: true,
+            sqlQuery: output.sqlQuery,
+            vegaLiteSpec: output.vegaLiteSpec,
+            sourceToolCallId: params.sourceToolCallId,
+        };
+    }
+    if (!params.sqlQuery || !params.vegaLiteSpec) {
+        return {
+            success: false,
+            error: 'Provide sourceToolCallId or both sqlQuery and vegaLiteSpec.',
+        };
+    }
+    try {
+        return {
+            success: true,
+            sqlQuery: params.sqlQuery,
+            vegaLiteSpec: JSON.parse(params.vegaLiteSpec),
+        };
+    }
+    catch (error) {
+        return {
+            success: false,
+            error: `Invalid Vega-Lite JSON: ${error instanceof Error ? error.message : String(error)}`,
+        };
+    }
+}
+function findChartToolOutput(state, sourceToolCallId) {
+    const currentSession = state.ai.getCurrentSession();
+    const messages = currentSession?.uiMessages ?? [];
+    for (const message of messages) {
+        if (message.role !== 'assistant')
+            continue;
+        for (const part of message.parts) {
+            if (!('toolCallId' in part) || part.toolCallId !== sourceToolCallId) {
+                continue;
+            }
+            if (part.state === 'output-available' &&
+                'output' in part &&
+                isVegaChartToolOutput(part.output)) {
+                return part.output;
+            }
+        }
+    }
+    return undefined;
+}
+function isVegaChartToolOutput(value) {
+    return (typeof value === 'object' &&
+        value !== null &&
+        'success' in value &&
+        value.success === true &&
+        'sqlQuery' in value &&
+        'vegaLiteSpec' in value);
+}
+async function renderVegaLiteImage({ spec, values, format, width, height, pngScale, renderTheme, background, }) {
+    const themeConfig = renderTheme === 'dark' ? darkTheme : lightTheme;
+    const specWithData = {
+        padding: 10,
+        ...spec,
+        background,
+        config: {
+            ...themeConfig,
+            ...(spec.config ?? {}),
+            background,
+        },
+        data: { values },
+        width,
+        height,
+        autosize: { contains: 'padding' },
+    };
+    const compiled = compile(specWithData).spec;
+    const view = new View(vegaParse(compiled), {
+        renderer: format === 'png' ? 'canvas' : 'none',
+    });
+    try {
+        await view.runAsync();
+        if (format === 'svg') {
+            return {
+                mediaType: 'image/svg+xml',
+                encoding: 'utf8',
+                background,
+                data: await view.toSVG(),
+            };
+        }
+        const canvas = await view.toCanvas(pngScale);
+        return {
+            mediaType: 'image/png',
+            encoding: 'base64',
+            background,
+            data: await canvasToBase64(canvas),
+        };
+    }
+    finally {
+        view.finalize();
+    }
+}
+async function canvasToBase64(canvas) {
+    const dataUrl = canvas.toDataURL('image/png');
+    return dataUrl.replace(/^data:image\/png;base64,/, '');
+}
+function createChartAssetId(params, sourceToolCallId) {
+    const base = params.title || params.alt || sourceToolCallId || 'chart';
+    const slug = base
+        .toLowerCase()
+        .trim()
+        .replace(/[^a-z0-9]+/g, '-')
+        .replace(/^-+|-+$/g, '')
+        .slice(0, 48);
+    return `${slug || 'chart'}-${Date.now().toString(36)}`;
+}
+function escapeMarkdownAlt(value) {
+    return value.replace(/]/g, '\\]');
+}
+function resolveRenderBackground(params) {
+    if (params.background)
+        return params.background;
+    const tailwindBackground = resolveTailwindBackgroundForTheme(params.renderTheme);
+    if (tailwindBackground)
+        return tailwindBackground;
+    return params.renderTheme === 'dark'
+        ? FALLBACK_DARK_BACKGROUND
+        : FALLBACK_LIGHT_BACKGROUND;
+}
+function resolveTailwindBackgroundForTheme(renderTheme) {
+    if (typeof document === 'undefined')
+        return undefined;
+    const currentTheme = document.documentElement.classList.contains('dark')
+        ? 'dark'
+        : 'light';
+    if (currentTheme !== renderTheme)
+        return undefined;
+    const value = getComputedStyle(document.documentElement)
+        .getPropertyValue(TAILWIND_BACKGROUND_VARIABLE)
+        .trim();
+    return value || undefined;
+}
+//# sourceMappingURL=ChartImageForMarkdownTool.js.map

package/dist/ChartImageForMarkdownTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ChartImageForMarkdownTool.js","sourceRoot":"","sources":["../src/ChartImageForMarkdownTool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAC,gBAAgB,EAAwB,MAAM,kBAAkB,CAAC;AACzE,OAAO,EAAC,IAAI,EAAC,MAAM,IAAI,CAAC;AACxB,OAAO,EAAC,KAAK,IAAI,SAAS,EAAE,IAAI,EAAC,MAAM,MAAM,CAAC;AAC9C,OAAO,EAAC,OAAO,EAAoB,MAAM,WAAW,CAAC;AACrD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,SAAS,EAAC,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAC,UAAU,EAAC,MAAM,qBAAqB,CAAC;AAG/C,MAAM,4BAA4B,GAAG,cAAc,CAAC;AACpD,MAAM,yBAAyB,GAAG,SAAS,CAAC;AAC5C,MAAM,wBAAwB,GAAG,kBAAkB,CAAC;AAEpD,MAAM,CAAC,MAAM,mCAAmC,GAAG,CAAC;KACjD,MAAM,CAAC;IACN,kBAAkB,EAAE,CAAC;SAClB,MAAM,EAAE;SACR,QAAQ,CAAC,uCAAuC,CAAC;IACpD,gBAAgB,EAAE,CAAC;SAChB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,gDAAgD,CAAC;IAC7D,QAAQ,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,2DAA2D,CAAC;IACxE,YAAY,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,iDAAiD,CAAC;IAC9D,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC;IACxD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IAC1E,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;IACpD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IAC9D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC;IACpD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC;IACrD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACrD,WAAW,EAAE,CAAC;SACX,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;SACvB,QAAQ,EAAE;SACV,OAAO,CAAC,OAAO,CAAC;SAChB,QAAQ,CACP,+FAA+F,CAChG;IACH,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,+IAA+I,CAChJ;CACJ,CAAC;KACD,MAAM,CACL,CAAC,KAAK,EAAE,EAAE,CACR,OAAO,CAAC,KAAK,CAAC,gBAAgB,CAAC;IAC/B,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,EAC1D;IACE,OAAO,EAAE,6DAA6D;CACvE,CACF,CAAC;AA4BJ,MAAM,UAAU,+BAA+B,CAE7C,KAAmB;IACnB,OAAO,IAAI,CAAC;QACV,WAAW,EAAE;;2KAE0J;QACvK,WAAW,EAAE,mCAAmC;QAChD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YACjC,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;YACzC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,MAAM,CAAC,CAAC;gBAC5D,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,MAAM,CAAC,KAAK;wBACrB,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;wBAC7C,KAAK,EAAE,MAAM,CAAC,KAAK;qBACsB,CAAC;gBAC9C,CAAC;gBAED,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;gBAC3D,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE;oBACzD,MAAM,EAAE,WAAW;iBACpB,CAAC,CAAC;gBACH,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;gBACvD,CAAC;gBAED,MAAM,MAAM,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;gBAC7C,MAAM,KAAK,GAAG,MAAM,mBAAmB,CAAC;oBACtC,IAAI,EAAE,MAAM,CAAC,YAAY;oBACzB,MAAM;oBACN,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,UAAU,EAAE,uBAAuB,CAAC,MAAM,CAAC;iBAC5C,CAAC,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,OAAO,GACX,MAAM,CAAC,OAAO,IAAI,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;gBACxE,MAAM,QAAQ,GAAG,GAAG,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAE/C,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,kBAAkB,EAAE;oBAChE,EAAE,EAAE,OAAO;oBACX,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,QAAQ;oBACR,GAAG,EAAE,MAAM,CAAC,GAAG;oBACf,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,UAAU,EAAE;wBACV,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;wBACzC,QAAQ,EAAE,MAAM,CAAC,QAAQ;wBACzB,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,WAAW,EAAE,MAAM,CAAC,WAAW;wBAC/B,UAAU,EAAE,KAAK,CAAC,UAAU;wBAC5B,UAAU,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE;qBAC9C;oBACD,SAAS,EAAE,SAAS;oBACpB,SAAS,EAAE,SAAS;iBACrB,CAAC,CAAC;gBAEH,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,UAAU,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,uBAAuB,OAAO,kBAAkB,MAAM,CAAC,kBAAkB,IAAI;oBAC3H,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;oBAC7C,OAAO;oBACP,QAAQ,EAAE,KAAK,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,kBAAkB,CAAC,OAAO,CAAC,GAAG;oBACvF,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,UAAU,EAAE,KAAK,CAAC,UAAU;iBACa,CAAC;YAC9C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvE,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,iCAAiC,OAAO,EAAE;oBACnD,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;oBAC7C,KAAK,EAAE,OAAO;iBAC2B,CAAC;YAC9C,CAAC;QACH,CAAC;QACD,aAAa,EAAE,CAAC,EAAC,MAAM,EAAC,EAAE,EAAE,CAAC,CAAC;YAC5B,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC9B,CAAC;KACH,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CACzB,KAA0B,EAC1B,MAA2C;IAS3C,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,mBAAmB,CAAC,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,4CAA4C,MAAM,CAAC,gBAAgB,IAAI;aAC/E,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7C,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,oBAAoB,MAAM,CAAC,gBAAgB,kDAAkD;aACrG,CAAC;QACJ,CAAC;QACD,OAAO;YACL,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;SAC1C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;QAC7C,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,6DAA6D;SACrE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,OAAO;YACL,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAiB;SAC9D,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,2BAA2B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;SAC3F,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAA0B,EAC1B,gBAAwB;IAExB,MAAM,cAAc,GAAG,KAAK,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC;IACpD,MAAM,QAAQ,GAAG,cAAc,EAAE,UAAU,IAAI,EAAE,CAAC;IAClD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,SAAS;QAC3C,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,CAAC,CAAC,YAAY,IAAI,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,KAAK,gBAAgB,EAAE,CAAC;gBACpE,SAAS;YACX,CAAC;YACD,IACE,IAAI,CAAC,KAAK,KAAK,kBAAkB;gBACjC,QAAQ,IAAI,IAAI;gBAChB,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,EAClC,CAAC;gBACD,OAAO,IAAI,CAAC,MAAM,CAAC;YACrB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,SAAS,IAAI,KAAK;QACjB,KAA6B,CAAC,OAAO,KAAK,IAAI;QAC/C,UAAU,IAAI,KAAK;QACnB,cAAc,IAAI,KAAK,CACxB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,EACjC,IAAI,EACJ,MAAM,EACN,MAAM,EACN,KAAK,EACL,MAAM,EACN,QAAQ,EACR,WAAW,EACX,UAAU,GAUX;IACC,MAAM,WAAW,GAAG,WAAW,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;IACpE,MAAM,YAAY,GAAG;QACnB,OAAO,EAAE,EAAE;QACX,GAAG,IAAI;QACP,UAAU;QACV,MAAM,EAAE;YACN,GAAG,WAAW;YACd,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;YACtB,UAAU;SACX;QACD,IAAI,EAAE,EAAC,MAAM,EAAC;QACd,KAAK;QACL,MAAM;QACN,QAAQ,EAAE,EAAC,QAAQ,EAAE,SAAS,EAAC;KAChB,CAAC;IAClB,MAAM,QAAQ,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC;IAC5C,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;QACzC,QAAQ,EAAE,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM;KAC/C,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACtB,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YACrB,OAAO;gBACL,SAAS,EAAE,eAAwB;gBACnC,QAAQ,EAAE,MAAe;gBACzB,UAAU;gBACV,IAAI,EAAE,MAAM,IAAI,CAAC,KAAK,EAAE;aACzB,CAAC;QACJ,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO;YACL,SAAS,EAAE,WAAoB;YAC/B,QAAQ,EAAE,QAAiB;YAC3B,UAAU;YACV,IAAI,EAAE,MAAM,cAAc,CAAC,MAAM,CAAC;SACnC,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,IAAI,CAAC,QAAQ,EAAE,CAAC;IAClB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,MAAyB;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC9C,OAAO,OAAO,CAAC,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,kBAAkB,CACzB,MAA2C,EAC3C,gBAAyB;IAEzB,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,GAAG,IAAI,gBAAgB,IAAI,OAAO,CAAC;IACvE,MAAM,IAAI,GAAG,IAAI;SACd,WAAW,EAAE;SACb,IAAI,EAAE;SACN,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;SAC3B,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAChB,OAAO,GAAG,IAAI,IAAI,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;AACzD,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAa;IACtC,OAAO,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,uBAAuB,CAAC,MAA2C;IAC1E,IAAI,MAAM,CAAC,UAAU;QAAE,OAAO,MAAM,CAAC,UAAU,CAAC;IAChD,MAAM,kBAAkB,GAAG,iCAAiC,CAC1D,MAAM,CAAC,WAAW,CACnB,CAAC;IACF,IAAI,kBAAkB;QAAE,OAAO,kBAAkB,CAAC;IAClD,OAAO,MAAM,CAAC,WAAW,KAAK,MAAM;QAClC,CAAC,CAAC,wBAAwB;QAC1B,CAAC,CAAC,yBAAyB,CAAC;AAChC,CAAC;AAED,SAAS,iCAAiC,CAAC,WAA6B;IACtE,IAAI,OAAO,QAAQ,KAAK,WAAW;QAAE,OAAO,SAAS,CAAC;IACtD,MAAM,YAAY,GAAG,QAAQ,CAAC,eAAe,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC;QACtE,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,OAAO,CAAC;IACZ,IAAI,YAAY,KAAK,WAAW;QAAE,OAAO,SAAS,CAAC;IAEnD,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,CAAC,eAAe,CAAC;SACrD,gBAAgB,CAAC,4BAA4B,CAAC;SAC9C,IAAI,EAAE,CAAC;IACV,OAAO,KAAK,IAAI,SAAS,CAAC;AAC5B,CAAC","sourcesContent":["import type {AiSliceState} from '@sqlrooms/ai-core';\nimport type {DocumentsSliceState} from '@sqlrooms/documents';\nimport {arrowTableToJson, type DuckDbSliceState} from '@sqlrooms/duckdb';\nimport {tool} from 'ai';\nimport {parse as vegaParse, View} from 'vega';\nimport {compile, type TopLevelSpec} from 'vega-lite';\nimport {z} from 'zod';\nimport {darkTheme} from './themes/darkTheme';\nimport {lightTheme} from './themes/lightTheme';\nimport type {VegaChartToolOutput} from './VegaChartTool';\n\nconst TAILWIND_BACKGROUND_VARIABLE = '--background';\nconst FALLBACK_LIGHT_BACKGROUND = '#ffffff';\nconst FALLBACK_DARK_BACKGROUND = 'oklch(0.145 0 0)';\n\nexport const ChartImageForMarkdownToolParameters = z\n  .object({\n    documentArtifactId: z\n      .string()\n      .describe('Target Markdown document artifact ID.'),\n    sourceToolCallId: z\n      .string()\n      .optional()\n      .describe('Prior successful chart tool call ID to render.'),\n    sqlQuery: z\n      .string()\n      .optional()\n      .describe('SQL query to render when no sourceToolCallId is provided.'),\n    vegaLiteSpec: z\n      .string()\n      .optional()\n      .describe('Vega-Lite JSON spec string without inline data.'),\n    format: z.enum(['svg', 'png']).optional().default('svg'),\n    assetId: z.string().optional().describe('Stable document-local asset ID.'),\n    alt: z.string().describe('Markdown image alt text.'),\n    title: z.string().optional().describe('Optional image title.'),\n    width: z.number().positive().optional().default(960),\n    height: z.number().positive().optional().default(540),\n    pngScale: z.number().positive().optional().default(2),\n    renderTheme: z\n      .enum(['light', 'dark'])\n      .optional()\n      .default('light')\n      .describe(\n        'Static chart theme for the rendered image. Defaults to light for portable Markdown documents.',\n      ),\n    background: z\n      .string()\n      .optional()\n      .describe(\n        'CSS color used as the exported image background. Defaults to the resolved Tailwind --background token for the requested theme when available.',\n      ),\n  })\n  .refine(\n    (input) =>\n      Boolean(input.sourceToolCallId) ||\n      (Boolean(input.sqlQuery) && Boolean(input.vegaLiteSpec)),\n    {\n      message: 'Provide sourceToolCallId or both sqlQuery and vegaLiteSpec.',\n    },\n  );\n\nexport type ChartImageForMarkdownToolParameters = z.infer<\n  typeof ChartImageForMarkdownToolParameters\n>;\n\nexport type ChartImageForMarkdownToolOutput = {\n  success: boolean;\n  details: string;\n  documentArtifactId: string;\n  assetId?: string;\n  markdown?: string;\n  mediaType?: 'image/svg+xml' | 'image/png';\n  encoding?: 'utf8' | 'base64';\n  format?: 'svg' | 'png';\n  renderTheme?: 'light' | 'dark';\n  background?: string;\n  error?: string;\n};\n\ntype StoreLike<S> = {\n  getState: () => S;\n};\n\ntype ChartImageToolState = AiSliceState &\n  DuckDbSliceState &\n  DocumentsSliceState;\n\nexport function createChartImageForMarkdownTool<\n  S extends ChartImageToolState = ChartImageToolState,\n>(store: StoreLike<S>) {\n  return tool({\n    description: `Render an existing Vega chart analysis as a static SVG or PNG image asset for a Markdown document.\nUse this before embedding chart images in document artifacts. The tool stores the image in the target document and returns a Markdown image link using asset://.\nFor portable Markdown documents, the default renderTheme is light with a white background. Only request dark rendering when the user explicitly wants a dark static image.`,\n    inputSchema: ChartImageForMarkdownToolParameters,\n    execute: async (params, options) => {\n      const abortSignal = options?.abortSignal;\n      try {\n        const source = resolveChartSource(store.getState(), params);\n        if (!source.success) {\n          return {\n            success: false,\n            details: source.error,\n            documentArtifactId: params.documentArtifactId,\n            error: source.error,\n          } satisfies ChartImageForMarkdownToolOutput;\n        }\n\n        const connector = await store.getState().db.getConnector();\n        const queryResult = await connector.query(source.sqlQuery, {\n          signal: abortSignal,\n        });\n        if (abortSignal?.aborted) {\n          throw new Error('Chart image rendering was aborted');\n        }\n\n        const values = arrowTableToJson(queryResult);\n        const image = await renderVegaLiteImage({\n          spec: source.vegaLiteSpec,\n          values,\n          format: params.format,\n          width: params.width,\n          height: params.height,\n          pngScale: params.pngScale,\n          renderTheme: params.renderTheme,\n          background: resolveRenderBackground(params),\n        });\n        const timestamp = Date.now();\n        const assetId =\n          params.assetId ?? createChartAssetId(params, source.sourceToolCallId);\n        const filename = `${assetId}.${params.format}`;\n\n        store.getState().documents.upsertAsset(params.documentArtifactId, {\n          id: assetId,\n          mediaType: image.mediaType,\n          encoding: image.encoding,\n          data: image.data,\n          filename,\n          alt: params.alt,\n          title: params.title,\n          provenance: {\n            sourceToolCallId: source.sourceToolCallId,\n            sqlQuery: source.sqlQuery,\n            vegaLiteSpec: source.vegaLiteSpec,\n            renderTheme: params.renderTheme,\n            background: image.background,\n            renderedAt: new Date(timestamp).toISOString(),\n          },\n          createdAt: timestamp,\n          updatedAt: timestamp,\n        });\n\n        return {\n          success: true,\n          details: `Stored ${params.format.toUpperCase()} chart image asset \"${assetId}\" in document \"${params.documentArtifactId}\".`,\n          documentArtifactId: params.documentArtifactId,\n          assetId,\n          markdown: `![${escapeMarkdownAlt(params.alt)}](asset://${encodeURIComponent(assetId)})`,\n          mediaType: image.mediaType,\n          encoding: image.encoding,\n          format: params.format,\n          renderTheme: params.renderTheme,\n          background: image.background,\n        } satisfies ChartImageForMarkdownToolOutput;\n      } catch (error) {\n        const message = error instanceof Error ? error.message : String(error);\n        return {\n          success: false,\n          details: `Failed to render chart image: ${message}`,\n          documentArtifactId: params.documentArtifactId,\n          error: message,\n        } satisfies ChartImageForMarkdownToolOutput;\n      }\n    },\n    toModelOutput: ({output}) => ({\n      type: 'text',\n      value: JSON.stringify(output),\n    }),\n  });\n}\n\nfunction resolveChartSource(\n  state: ChartImageToolState,\n  params: ChartImageForMarkdownToolParameters,\n):\n  | {\n      success: true;\n      sqlQuery: string;\n      vegaLiteSpec: TopLevelSpec;\n      sourceToolCallId?: string;\n    }\n  | {success: false; error: string} {\n  if (params.sourceToolCallId) {\n    const output = findChartToolOutput(state, params.sourceToolCallId);\n    if (!output) {\n      return {\n        success: false,\n        error: `No successful chart tool call found for \"${params.sourceToolCallId}\".`,\n      };\n    }\n    if (!output.sqlQuery || !output.vegaLiteSpec) {\n      return {\n        success: false,\n        error: `Chart tool call \"${params.sourceToolCallId}\" has no renderable SQL query or Vega-Lite spec.`,\n      };\n    }\n    return {\n      success: true,\n      sqlQuery: output.sqlQuery,\n      vegaLiteSpec: output.vegaLiteSpec,\n      sourceToolCallId: params.sourceToolCallId,\n    };\n  }\n\n  if (!params.sqlQuery || !params.vegaLiteSpec) {\n    return {\n      success: false,\n      error: 'Provide sourceToolCallId or both sqlQuery and vegaLiteSpec.',\n    };\n  }\n\n  try {\n    return {\n      success: true,\n      sqlQuery: params.sqlQuery,\n      vegaLiteSpec: JSON.parse(params.vegaLiteSpec) as TopLevelSpec,\n    };\n  } catch (error) {\n    return {\n      success: false,\n      error: `Invalid Vega-Lite JSON: ${error instanceof Error ? error.message : String(error)}`,\n    };\n  }\n}\n\nfunction findChartToolOutput(\n  state: ChartImageToolState,\n  sourceToolCallId: string,\n): VegaChartToolOutput | undefined {\n  const currentSession = state.ai.getCurrentSession();\n  const messages = currentSession?.uiMessages ?? [];\n  for (const message of messages) {\n    if (message.role !== 'assistant') continue;\n    for (const part of message.parts) {\n      if (!('toolCallId' in part) || part.toolCallId !== sourceToolCallId) {\n        continue;\n      }\n      if (\n        part.state === 'output-available' &&\n        'output' in part &&\n        isVegaChartToolOutput(part.output)\n      ) {\n        return part.output;\n      }\n    }\n  }\n  return undefined;\n}\n\nfunction isVegaChartToolOutput(value: unknown): value is VegaChartToolOutput {\n  return (\n    typeof value === 'object' &&\n    value !== null &&\n    'success' in value &&\n    (value as VegaChartToolOutput).success === true &&\n    'sqlQuery' in value &&\n    'vegaLiteSpec' in value\n  );\n}\n\nasync function renderVegaLiteImage({\n  spec,\n  values,\n  format,\n  width,\n  height,\n  pngScale,\n  renderTheme,\n  background,\n}: {\n  spec: TopLevelSpec;\n  values: Record<string, unknown>[];\n  format: 'svg' | 'png';\n  width: number;\n  height: number;\n  pngScale: number;\n  renderTheme: 'light' | 'dark';\n  background: string;\n}) {\n  const themeConfig = renderTheme === 'dark' ? darkTheme : lightTheme;\n  const specWithData = {\n    padding: 10,\n    ...spec,\n    background,\n    config: {\n      ...themeConfig,\n      ...(spec.config ?? {}),\n      background,\n    },\n    data: {values},\n    width,\n    height,\n    autosize: {contains: 'padding'},\n  } as TopLevelSpec;\n  const compiled = compile(specWithData).spec;\n  const view = new View(vegaParse(compiled), {\n    renderer: format === 'png' ? 'canvas' : 'none',\n  });\n\n  try {\n    await view.runAsync();\n    if (format === 'svg') {\n      return {\n        mediaType: 'image/svg+xml' as const,\n        encoding: 'utf8' as const,\n        background,\n        data: await view.toSVG(),\n      };\n    }\n    const canvas = await view.toCanvas(pngScale);\n    return {\n      mediaType: 'image/png' as const,\n      encoding: 'base64' as const,\n      background,\n      data: await canvasToBase64(canvas),\n    };\n  } finally {\n    view.finalize();\n  }\n}\n\nasync function canvasToBase64(canvas: HTMLCanvasElement) {\n  const dataUrl = canvas.toDataURL('image/png');\n  return dataUrl.replace(/^data:image\\/png;base64,/, '');\n}\n\nfunction createChartAssetId(\n  params: ChartImageForMarkdownToolParameters,\n  sourceToolCallId?: string,\n) {\n  const base = params.title || params.alt || sourceToolCallId || 'chart';\n  const slug = base\n    .toLowerCase()\n    .trim()\n    .replace(/[^a-z0-9]+/g, '-')\n    .replace(/^-+|-+$/g, '')\n    .slice(0, 48);\n  return `${slug || 'chart'}-${Date.now().toString(36)}`;\n}\n\nfunction escapeMarkdownAlt(value: string) {\n  return value.replace(/]/g, '\\\\]');\n}\n\nfunction resolveRenderBackground(params: ChartImageForMarkdownToolParameters) {\n  if (params.background) return params.background;\n  const tailwindBackground = resolveTailwindBackgroundForTheme(\n    params.renderTheme,\n  );\n  if (tailwindBackground) return tailwindBackground;\n  return params.renderTheme === 'dark'\n    ? FALLBACK_DARK_BACKGROUND\n    : FALLBACK_LIGHT_BACKGROUND;\n}\n\nfunction resolveTailwindBackgroundForTheme(renderTheme: 'light' | 'dark') {\n  if (typeof document === 'undefined') return undefined;\n  const currentTheme = document.documentElement.classList.contains('dark')\n    ? 'dark'\n    : 'light';\n  if (currentTheme !== renderTheme) return undefined;\n\n  const value = getComputedStyle(document.documentElement)\n    .getPropertyValue(TAILWIND_BACKGROUND_VARIABLE)\n    .trim();\n  return value || undefined;\n}\n"]}

package/dist/VegaChartActions.js

@@ -18,6 +18,6 @@ import { cn } from '@sqlrooms/ui';
  * ```
  */
 export const VegaChartActions = ({ children, className, }) => {
-    return (_jsx("div", { className: cn('vega-actions bg-background pointer-coarse:pointer-events-auto pointer-coarse:opacity-100 pointer-events-none absolute right-0 top-1 z-10 flex items-center gap-1 opacity-0 transition-opacity focus-within:pointer-events-auto focus-within:opacity-100 hover:pointer-events-auto hover:opacity-100 peer-focus-within:pointer-events-auto peer-focus-within:opacity-100 peer-hover:pointer-events-auto peer-hover:opacity-100', className), children: children }));
+    return (_jsx("div", { className: cn('vega-actions bg-background pointer-events-none absolute top-1 right-0 z-10 flex items-center gap-1 opacity-0 transition-opacity peer-focus-within:pointer-events-auto peer-focus-within:opacity-100 peer-hover:pointer-events-auto peer-hover:opacity-100 focus-within:pointer-events-auto focus-within:opacity-100 hover:pointer-events-auto hover:opacity-100 pointer-coarse:pointer-events-auto pointer-coarse:opacity-100', className), children: children }));
 };
 //# sourceMappingURL=VegaChartActions.js.map

package/dist/VegaChartActions.js.map

@@ -1 +1 @@
-{"version":3,"file":"VegaChartActions.js","sourceRoot":"","sources":["../src/VegaChartActions.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,EAAE,EAAC,MAAM,cAAc,CAAC;AAchC;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAoC,CAAC,EAChE,QAAQ,EACR,SAAS,GACV,EAAE,EAAE;IACH,OAAO,CACL,cACE,SAAS,EAAE,EAAE,CACX,+ZAA+Z,EAC/Z,SAAS,CACV,YAEA,QAAQ,GACL,CACP,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import {cn} from '@sqlrooms/ui';\nimport React from 'react';\n\nexport interface VegaChartActionsProps {\n  /**\n   * Action components to render in the toolbar\n   */\n  children: React.ReactNode;\n  /**\n   * Additional CSS classes for the container\n   */\n  className?: string;\n}\n\n/**\n * Container component for chart actions toolbar.\n * Positions actions as an overlay in the top-right corner of the chart.\n *\n * @example\n * ```tsx\n * <VegaLiteArrowChart spec={spec} arrowTable={data}>\n *   <VegaChartActions>\n *     <VegaExportAction />\n *     <Separator orientation=\"vertical\" className=\"h-4\" />\n *     <Button size=\"xs\" variant=\"ghost\" onClick={handleRefresh}>\n *       <RefreshCw className=\"h-4 w-4\" />\n *     </Button>\n *   </VegaChartActions>\n * </VegaLiteArrowChart>\n * ```\n */\nexport const VegaChartActions: React.FC<VegaChartActionsProps> = ({\n  children,\n  className,\n}) => {\n  return (\n    <div\n      className={cn(\n        'vega-actions bg-background pointer-coarse:pointer-events-auto pointer-coarse:opacity-100 pointer-events-none absolute right-0 top-1 z-10 flex items-center gap-1 opacity-0 transition-opacity focus-within:pointer-events-auto focus-within:opacity-100 hover:pointer-events-auto hover:opacity-100 peer-focus-within:pointer-events-auto peer-focus-within:opacity-100 peer-hover:pointer-events-auto peer-hover:opacity-100',\n        className,\n      )}\n    >\n      {children}\n    </div>\n  );\n};\n"]}
+{"version":3,"file":"VegaChartActions.js","sourceRoot":"","sources":["../src/VegaChartActions.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,EAAE,EAAC,MAAM,cAAc,CAAC;AAchC;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAoC,CAAC,EAChE,QAAQ,EACR,SAAS,GACV,EAAE,EAAE;IACH,OAAO,CACL,cACE,SAAS,EAAE,EAAE,CACX,+ZAA+Z,EAC/Z,SAAS,CACV,YAEA,QAAQ,GACL,CACP,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import {cn} from '@sqlrooms/ui';\nimport React from 'react';\n\nexport interface VegaChartActionsProps {\n  /**\n   * Action components to render in the toolbar\n   */\n  children: React.ReactNode;\n  /**\n   * Additional CSS classes for the container\n   */\n  className?: string;\n}\n\n/**\n * Container component for chart actions toolbar.\n * Positions actions as an overlay in the top-right corner of the chart.\n *\n * @example\n * ```tsx\n * <VegaLiteArrowChart spec={spec} arrowTable={data}>\n *   <VegaChartActions>\n *     <VegaExportAction />\n *     <Separator orientation=\"vertical\" className=\"h-4\" />\n *     <Button size=\"xs\" variant=\"ghost\" onClick={handleRefresh}>\n *       <RefreshCw className=\"h-4 w-4\" />\n *     </Button>\n *   </VegaChartActions>\n * </VegaLiteArrowChart>\n * ```\n */\nexport const VegaChartActions: React.FC<VegaChartActionsProps> = ({\n  children,\n  className,\n}) => {\n  return (\n    <div\n      className={cn(\n        'vega-actions bg-background pointer-events-none absolute top-1 right-0 z-10 flex items-center gap-1 opacity-0 transition-opacity peer-focus-within:pointer-events-auto peer-focus-within:opacity-100 peer-hover:pointer-events-auto peer-hover:opacity-100 focus-within:pointer-events-auto focus-within:opacity-100 hover:pointer-events-auto hover:opacity-100 pointer-coarse:pointer-events-auto pointer-coarse:opacity-100',\n        className,\n      )}\n    >\n      {children}\n    </div>\n  );\n};\n"]}

package/dist/VegaChartTool.d.ts

@@ -1,7 +1,11 @@
 import { z } from 'zod';
-import type { OpenAssistantTool } from '@openassistant/utils';
-import { EmbedOptions } from 'vega-embed';
-import { EditorMode } from './editor/types';
+import { TopLevelSpec } from 'vega-lite';
+import type { DuckDbConnector } from '@sqlrooms/duckdb';
+/**
+ * Creates a SQL validator that checks queries by executing `SELECT 1 FROM (<query>) LIMIT 1`.
+ * Returns an error message if the query fails or produces no rows.
+ */
+export declare function createSqlValidator(getConnector: () => DuckDbConnector | Promise<DuckDbConnector>): NonNullable<VegaChartToolOptions['validateSql']>;
 /**
  * Zod schema for the VegaChart tool parameters
  */
@@ -11,20 +15,13 @@ export declare const VegaChartToolParameters: z.ZodObject<{
     reasoning: z.ZodString;
 }, z.core.$strip>;
 export type VegaChartToolParameters = z.infer<typeof VegaChartToolParameters>;
-export type VegaChartToolArgs = z.ZodObject<{
-    sqlQuery: z.ZodString;
-    vegaLiteSpec: z.ZodString;
-    reasoning: z.ZodString;
-}>;
-export type VegaChartToolLlmResult = {
+export type VegaChartToolOutput = {
     success: boolean;
     details: string;
-};
-export type VegaChartToolAdditionalData = {
     sqlQuery: string;
-    vegaLiteSpec: object;
+    vegaLiteSpec: TopLevelSpec | null;
+    error?: string;
 };
-export type VegaChartToolContext = unknown;
 /**
  * Default description for the VegaChart tool
  */
@@ -38,19 +35,21 @@ export type VegaChartToolOptions = {
      */
     description?: string;
     /**
-     * Vega embed options
-     */
-    embedOptions?: EmbedOptions;
-    /**
-     * Whether editing is enabled
-     * @default true
-     */
-    editable?: boolean;
-    /**
-     * Which editors to show when editing
-     * @default 'both'
+     * Optional callback to validate the SQL query before rendering the chart.
+     * When provided, the tool will execute this function to catch SQL errors
+     * or empty results early, so the LLM can fix the query and retry.
+     *
+     * The function receives the SQL query string and an optional AbortSignal,
+     * and should return `{valid: true}` or `{valid: false, error: string}`.
+     *
+     * For performance, implementations should use `LIMIT 1` or equivalent.
      */
-    editorMode?: EditorMode;
+    validateSql?: (sqlQuery: string, abortSignal?: AbortSignal) => Promise<{
+        valid: true;
+    } | {
+        valid: false;
+        error: string;
+    }>;
 };
 /**
  * Creates a VegaLite chart visualization tool for AI assistants
@@ -60,5 +59,27 @@ export type VegaChartToolOptions = {
  * @param options.editorMode - Which editors to show ('spec', 'sql', 'both', 'none')
  * @returns A tool that can be used with the AI assistant
  */
-export declare function createVegaChartTool({ description, embedOptions, editable, editorMode, }?: VegaChartToolOptions): OpenAssistantTool<typeof VegaChartToolParameters, VegaChartToolLlmResult, VegaChartToolAdditionalData, VegaChartToolContext>;
+export declare function createVegaChartTool({ description, validateSql, }?: VegaChartToolOptions): import("ai").Tool<{
+    sqlQuery: string;
+    vegaLiteSpec: string;
+    reasoning: string;
+}, {
+    success: boolean;
+    details: string;
+    sqlQuery: string;
+    vegaLiteSpec: null;
+    error?: undefined;
+} | {
+    success: boolean;
+    details: string;
+    sqlQuery: string;
+    vegaLiteSpec: TopLevelSpec;
+    error?: undefined;
+} | {
+    success: boolean;
+    details: string;
+    error: string;
+    sqlQuery: string;
+    vegaLiteSpec: null;
+}>;
 //# sourceMappingURL=VegaChartTool.d.ts.map

package/dist/VegaChartTool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"VegaChartTool.d.ts","sourceRoot":"","sources":["../src/VegaChartTool.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAKtB,OAAO,KAAK,EAAC,iBAAiB,EAAC,MAAM,sBAAsB,CAAC;AAG5D,OAAO,EAAC,YAAY,EAAC,MAAM,YAAY,CAAC;AAExC,OAAO,EAAC,UAAU,EAAC,MAAM,gBAAgB,CAAC;AAE1C;;GAEG;AACH,eAAO,MAAM,uBAAuB;;;;iBAIlC,CAAC;AAEH,MAAM,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE9E,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,SAAS,CAAC;IAC1C,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC;IACtB,YAAY,EAAE,CAAC,CAAC,SAAS,CAAC;IAC1B,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC;CACxB,CAAC,CAAC;AAEH,MAAM,MAAM,sBAAsB,GAAG;IACnC,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG,OAAO,CAAC;AAE3C;;GAEG;AACH,eAAO,MAAM,8BAA8B,smCAY4I,CAAC;AAExL;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B;;;OAGG;IACH,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB;;;OAGG;IACH,UAAU,CAAC,EAAE,UAAU,CAAC;CACzB,CAAC;AAEF;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CAAC,EAClC,WAA4C,EAC5C,YAA2C,EAC3C,QAAe,EACf,UAAmB,GACpB,GAAE,oBAAyB,GAAG,iBAAiB,CAC9C,OAAO,uBAAuB,EAC9B,sBAAsB,EACtB,2BAA2B,EAC3B,oBAAoB,CACrB,CAwEA"}
+{"version":3,"file":"VegaChartTool.d.ts","sourceRoot":"","sources":["../src/VegaChartTool.tsx"],"names":[],"mappings":"AACA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAU,YAAY,EAAC,MAAM,WAAW,CAAC;AAEhD,OAAO,KAAK,EAAC,eAAe,EAAC,MAAM,kBAAkB,CAAC;AAEtD;;;GAGG;AACH,wBAAgB,kBAAkB,CAChC,YAAY,EAAE,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC,GAC7D,WAAW,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC,CAmBlD;AAED;;GAEG;AACH,eAAO,MAAM,uBAAuB;;;;iBAIlC,CAAC;AAEH,MAAM,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE9E,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,YAAY,GAAG,IAAI,CAAC;IAClC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,8BAA8B,smCAY4I,CAAC;AAExL;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;;;;;;OASG;IACH,WAAW,CAAC,EAAE,CACZ,QAAQ,EAAE,MAAM,EAChB,WAAW,CAAC,EAAE,WAAW,KACtB,OAAO,CAAC;QAAC,KAAK,EAAE,IAAI,CAAA;KAAC,GAAG;QAAC,KAAK,EAAE,KAAK,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAC,CAAC,CAAC;CAC7D,CAAC;AAEF;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CAAC,EAClC,WAA4C,EAC5C,WAAW,GACZ,GAAE,oBAAyB;;;;;;;;;;;;;;;;;;;;;;GAqE3B"}