@sqlrooms/mosaic 0.29.0-rc.6 → 0.29.0-rc.8

package/dist/ai/ai.js

@@ -0,0 +1,461 @@
+/**
+ * Reusable AI helpers for Mosaic dashboards.
+ *
+ * Host apps provide a narrow adapter for their store/artifact model; Mosaic owns
+ * the dashboard tool behavior and prompts.
+ *
+ * @packageDocumentation
+ */
+import { resolveTableReference } from '@sqlrooms/duckdb';
+import { ToolLoopAgent, stepCountIs, tool, } from 'ai';
+import { z } from 'zod';
+import { createChartTools, createDefaultChartTypes, createListPanelsTool, createDataTableExplorerTool, createRemovePanelTool, } from '../charts/chart-types';
+import { DEFAULT_CHART_MAX_DATA_POINTS, } from '../chart-runtime';
+export const MAP_TOOL_KEY = 'create_dashboard_map';
+export const DASHBOARD_AI_INSTRUCTIONS = `
+Dashboard authoring:
+
+**When to use dashboard_agent vs individual tools:**
+- Use \`dashboard_agent\` for exploratory requests that require data analysis and discovery:
+  - "analyze the earthquakes dataset"
+  - "create insights dashboard for sales data"
+  - "find interesting patterns in customer behavior"
+  - Any request asking to "discover", "explore", "find insights", or "analyze"
+- Use individual chart tools for direct, specific requests:
+  - "create histogram of magnitude with 20 bins"
+  - "add a line chart showing sales over time"
+  - "update the histogram to use 30 bins"
+
+**Individual dashboard chart tools:**
+- create_dashboard_histogram, create_dashboard_line_chart, create_dashboard_box_plot, create_dashboard_scatter_plot, create_dashboard_count_plot, create_dashboard_heatmap
+- Each chart type has its own tool with specific parameters.
+- For line charts with aggregation, use yFields array with {field: string, aggregate: "sum"|"avg"|"min"|"max"}.
+- Set xInterval for temporal binning (year, month, day, hour, etc.).
+- If the host app provides \`${MAP_TOOL_KEY}\`, use it for map/geospatial/location requests and tables with longitude/latitude or geometry columns. Author its config as native Deck JSON with layer classes in \`spec.layers[].@@type\`, dataset bindings in \`_sqlroomsBinding.dataset\`, and table/query sources in \`config.datasets\`. For data-driven map colors, use color accessors such as \`getFillColor\`, \`getLineColor\`, \`getColor\`, \`getSourceColor\`, or \`getTargetColor\` with \`{"@@function":"colorScale", "field":"...", "type":"sequential"|"diverging"|"quantize"|"quantile"|"categorical", "scheme":"Viridis", "domain":"auto"}\`.
+- Use \`set_dashboard_vgplot\` with complete JSON only when no chart tool fits your needs.
+- When calling \`create_dashboard_artifact\`, \`layoutType\` may be \`grid\` or \`dock\`; omitted values default to \`grid\`.
+- Ensure specs are valid JSON objects compatible with https://idl.uw.edu/mosaic/schema/latest.json.
+- \`list_dashboard_panels\` includes runtime issues when a chart failed. Use those issues to repair panels in place: convert too-large scatter charts to heatmaps, add \`xInterval\` to too-large line charts, and inspect columns/settings for SQL errors.
+`;
+export const DASHBOARD_AGENT_INSTRUCTIONS = `You are a dashboard builder agent that creates and modifies interactive data dashboards.
+
+## Your Role
+
+You analyze data and create insightful dashboards with multiple visualizations (charts, Data Table Explorers). You can handle both direct requests ("create histogram of magnitude") and exploratory requests ("find interesting insights in earthquakes dataset").
+
+## Available Tools
+
+**Chart Tools:**
+- create_dashboard_histogram - distribution of numeric values (always safe, aggregates automatically)
+- create_dashboard_line_chart - trends over time or ordered variable (use with aggregations for >10k rows)
+- create_dashboard_box_plot - compare distributions across categories
+- create_dashboard_scatter_plot - relationship between two numeric columns (avoid for >10k rows, use heatmap instead)
+- create_dashboard_count_plot - frequency of categorical values (always safe, aggregates automatically)
+- create_dashboard_heatmap - density/patterns across two dimensions (preferred for large datasets)
+
+**Panel Tools:**
+- create_dashboard_data_table_explorer - table statistics and column summaries
+- ${MAP_TOOL_KEY} - native Deck JSON geospatial map panel (if provided by the host app)
+
+**Data Tools:**
+- query - execute SQL queries for data exploration
+
+**Management Tools:**
+- list_dashboard_panels - discover panel IDs and what's on the dashboard
+- remove_dashboard_panel - delete a panel by ID
+
+## Workflows
+
+### Direct Requests
+When user provides specific instructions:
+1. Parse intent -> identify chart type
+2. Call appropriate tool with settings
+3. Done
+
+Example: "create histogram of magnitude with 20 bins"
+-> create_dashboard_histogram(settings: {field: "magnitude", bins: 20})
+
+### Exploratory Requests
+When user asks to discover insights:
+1. Use query tool to explore data:
+   - Start with simple stats: COUNT, MIN, MAX, AVG, DISTINCT (check total row count!)
+   - Check distributions: GROUP BY with COUNT
+   - Find correlations: CORR(col1, col2)
+   - Identify outliers and patterns
+2. Create targeted charts based on discoveries:
+   - If dataset has >10k rows: avoid scatter charts and unaggregated line charts
+   - Use histogram, count plot, heatmap, or aggregated visualizations instead
+3. Stop when dashboard tells coherent story
+
+### Update Requests
+To update existing panels:
+1. Call list_dashboard_panels() to discover panel IDs
+2. Call appropriate create tool with panelId parameter
+3. Panel is updated in-place
+
+## Query Guidelines
+
+**Keep queries simple:**
+- Start with aggregations: COUNT, MIN, MAX, AVG, SUM
+- Use GROUP BY for breakdowns and distributions
+- Use LIMIT to control result size (especially with GROUP BY)
+- Avoid expensive operations: large JOINs, full table scans
+
+**Exploration strategy:**
+- Limit exploration to 5-10 queries per request
+- Start broad (overall stats), then narrow (specific patterns)
+- Stop when you have enough insights for a coherent dashboard
+
+## Best Practices
+
+- **Avoid unaggregated charts for large datasets:** For datasets >10k rows, DO NOT use scatter charts or line charts without aggregations. Use aggregated alternatives instead:
+  - For scatter plots: use heatmap or binned aggregations
+  - For line charts: use GROUP BY with time buckets or aggregations
+  - Histograms and count plots are always safe (they aggregate automatically)
+- **Check before update:** Always call list_dashboard_panels before updating/removing panels
+- **Repair broken charts:** list_dashboard_panels may return an \`issue\` per panel. For \`too-much-data\`, switch to an aggregated chart or add aggregation. For \`sql-error\`, inspect available columns/types and update the broken panel in place.
+- **Validate columns:** Query tools will validate column existence and types
+- **Handle errors gracefully:** If a query or chart creation fails, try alternative approach`;
+const DashboardCreateArtifactToolParameters = z.object({
+    title: z.string().optional(),
+    layoutType: z
+        .enum(['dock', 'grid'])
+        .optional()
+        .default('grid')
+        .describe('Dashboard layout node type to use at creation time.'),
+});
+const DashboardAgentInputSchema = z.object({
+    reasoning: z
+        .string()
+        .describe('Reasoning for why the dashboard agent is being called'),
+    prompt: z
+        .string()
+        .describe('The exploratory data analysis prompt for the agent'),
+    tableName: z
+        .string()
+        .describe('REQUIRED: The name of the table/dataset to analyze.'),
+    dashboardTitle: z
+        .string()
+        .optional()
+        .describe('Optional title for the dashboard artifact'),
+    maxSteps: z
+        .number()
+        .optional()
+        .default(20)
+        .describe('Maximum exploration steps (default: 20, range: 5-50)'),
+    temperature: z
+        .number()
+        .optional()
+        .default(0.7)
+        .describe('Model temperature for creativity vs consistency (default: 0.7, range: 0.0-1.0)'),
+});
+class DashboardAgentException extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'DashboardAgentException';
+    }
+}
+function getTablesWithColumns(state, adapter) {
+    return adapter
+        .getTables(state)
+        .map(normalizeDashboardAiTable)
+        .filter((table) => table.columns && table.columns.length > 0);
+}
+function normalizeDashboardAiTable(table) {
+    const { qualifiedName } = table;
+    const tableId = qualifiedName.toString();
+    return {
+        table: qualifiedName,
+        tableName: qualifiedName.table,
+        tableId,
+        columns: table.columns,
+        rowCount: table.rowCount,
+    };
+}
+function formatAmbiguousTableError(tableName, matches) {
+    return `Ambiguous table "${tableName}". Matching tables: ${matches
+        .map((table) => table.tableId)
+        .join(', ')}.`;
+}
+function getResolvedTable(table) {
+    if (!table.columns) {
+        throw new Error(`Table "${table.tableName}" has no column metadata.`);
+    }
+    return {
+        tableName: table.tableId,
+        qualifiedName: table.table,
+        columns: table.columns.map((column) => ({
+            name: column.name,
+            type: column.type,
+        })),
+    };
+}
+function formatAvailableTables(tables) {
+    return (tables
+        .map((table) => {
+        return table.tableId === table.tableName
+            ? table.tableName
+            : `${table.tableName} (${table.tableId})`;
+    })
+        .join(', ') || '(none)');
+}
+export function createDashboardToolDeps({ store, adapter, }) {
+    const resolveArtifact = (artifactId, createIfMissing, context) => {
+        const state = store.getState();
+        const hasRunContext = adapter.hasRunContext?.(state, context) ?? false;
+        const contextDashboardArtifactId = adapter.resolveContextDashboardArtifactId?.(state, context);
+        let targetArtifactId = artifactId ??
+            contextDashboardArtifactId ??
+            (!hasRunContext
+                ? adapter.getCurrentDashboardArtifactId(state)
+                : undefined);
+        if (!targetArtifactId && hasRunContext) {
+            throw new Error('No primary dashboard artifact is available in the current run context. Pass artifactId explicitly or use set_primary_context_artifact first.');
+        }
+        if (!targetArtifactId && createIfMissing) {
+            targetArtifactId = adapter.createDashboardArtifact(state, undefined, 'grid');
+            adapter.setCurrentArtifact(state, targetArtifactId);
+            adapter.makeDashboardPrimaryForRun?.(state, targetArtifactId, context);
+        }
+        if (!targetArtifactId) {
+            throw new Error('No dashboard artifact is available. Set createArtifactIfMissing=true or create one first.');
+        }
+        if (!adapter.isDashboardArtifact(state, targetArtifactId)) {
+            throw new Error(`Artifact "${targetArtifactId}" is not a dashboard artifact.`);
+        }
+        adapter.ensureDashboard(state, targetArtifactId);
+        return targetArtifactId;
+    };
+    const resolveTable = (artifactId, tableName) => {
+        const state = store.getState();
+        const tables = getTablesWithColumns(state, adapter);
+        const dashboard = adapter.getDashboard(state, artifactId);
+        const explicitTableName = typeof tableName === 'string' ? tableName.trim() || undefined : tableName;
+        if (explicitTableName) {
+            const { table, ambiguousMatches } = resolveTableReference(tables, explicitTableName);
+            if (ambiguousMatches) {
+                const requestedTableName = typeof explicitTableName === 'string'
+                    ? explicitTableName
+                    : explicitTableName.toString();
+                throw new Error(formatAmbiguousTableError(requestedTableName, ambiguousMatches));
+            }
+            if (!table) {
+                const requestedTableName = typeof explicitTableName === 'string'
+                    ? explicitTableName
+                    : explicitTableName.toString();
+                throw new Error(`Unknown table "${requestedTableName}". Available tables: ${formatAvailableTables(tables)}.`);
+            }
+            const resolvedTable = getResolvedTable(table);
+            adapter.setSelectedTable(state, artifactId, resolvedTable.tableName);
+            return resolvedTable;
+        }
+        if (dashboard?.selectedTable) {
+            const { table } = resolveTableReference(tables, dashboard.selectedTable);
+            if (table) {
+                return getResolvedTable(table);
+            }
+        }
+        if (tables.length === 1) {
+            const onlyTable = tables[0];
+            if (!onlyTable) {
+                throw new Error('The only available table has no column metadata.');
+            }
+            const resolvedTable = getResolvedTable(onlyTable);
+            adapter.setSelectedTable(state, artifactId, resolvedTable.tableName);
+            return resolvedTable;
+        }
+        throw new Error(`No dashboard table is selected. Provide tableName using one of: ${formatAvailableTables(tables)}.`);
+    };
+    const deps = {
+        maxDataPoints: DEFAULT_CHART_MAX_DATA_POINTS,
+        resolveArtifact,
+        resolveTable,
+        addPanel: (dashboardId, panel) => {
+            const state = store.getState();
+            return adapter.addPanel(state, dashboardId, panel);
+        },
+        updatePanel: (dashboardId, panelId, patch) => {
+            const state = store.getState();
+            adapter.updatePanel(state, dashboardId, panelId, patch);
+        },
+        getDashboard: (dashboardId) => {
+            const state = store.getState();
+            return adapter.getDashboard(state, dashboardId);
+        },
+        removePanel: (dashboardId, panelId) => {
+            const state = store.getState();
+            adapter.removePanel(state, dashboardId, panelId);
+        },
+        setCurrentArtifact: (artifactId) => {
+            const state = store.getState();
+            adapter.setCurrentArtifact(state, artifactId);
+        },
+    };
+    if (adapter.getPanelIssue) {
+        deps.getPanelIssue = (dashboardId, panelId) => {
+            const state = store.getState();
+            return adapter.getPanelIssue?.(state, dashboardId, panelId);
+        };
+    }
+    return deps;
+}
+export function createDashboardAiTools({ store, adapter, chartTypes, extraTools, }) {
+    const deps = createDashboardToolDeps({ store, adapter });
+    const resolvedChartTypes = chartTypes ?? createDefaultChartTypes({ includeCustomSpec: false });
+    const chartTools = createChartTools(resolvedChartTypes, deps);
+    const hostTools = extraTools?.(deps) ?? {};
+    const builtInTools = {
+        create_dashboard_artifact: tool({
+            description: 'Create a new dashboard artifact with a dock or grid layout and make it the active artifact. Use when no dashboard artifact exists yet.',
+            inputSchema: DashboardCreateArtifactToolParameters,
+            execute: async (params, context) => {
+                const state = store.getState();
+                const artifactId = adapter.createDashboardArtifact(state, params.title, params.layoutType);
+                adapter.setCurrentArtifact(state, artifactId);
+                adapter.makeDashboardPrimaryForRun?.(state, artifactId, context);
+                return {
+                    llmResult: {
+                        success: true,
+                        details: `Created dashboard artifact "${artifactId}".`,
+                        data: { artifactId },
+                    },
+                };
+            },
+        }),
+        ...chartTools,
+        create_dashboard_data_table_explorer: createDataTableExplorerTool(deps),
+        list_dashboard_panels: createListPanelsTool(deps),
+        remove_dashboard_panel: createRemovePanelTool(deps),
+    };
+    for (const key of Object.keys(hostTools)) {
+        if (key in builtInTools) {
+            throw new Error(`Dashboard extraTools cannot override built-in tool "${key}". Register the host tool under a unique key.`);
+        }
+    }
+    return {
+        ...builtInTools,
+        ...hostTools,
+    };
+}
+function buildAgentPrompt(userPrompt, tableName, state, adapter) {
+    const table = resolveTableReference(adapter.getTables(state).map(normalizeDashboardAiTable), tableName).table;
+    const resolvedTableName = table ? table.tableId : tableName;
+    const tableIdInfo = table && table.tableName !== resolvedTableName
+        ? `\n- Table ID for tool calls: ${resolvedTableName}`
+        : '';
+    const columnNames = table?.columns
+        ?.map((column) => column.type ? `${column.name} (${column.type})` : column.name)
+        .join(', ') || 'unknown';
+    const rowInfo = table?.rowCount !== undefined ? `Approximate rows: ${table.rowCount}` : '';
+    return `Analyze the "${table?.tableName ?? tableName}" table.
+
+Table info:
+- Columns: ${columnNames}${tableIdInfo}${rowInfo ? `\n- ${rowInfo}` : ''}
+
+User request: ${userPrompt}
+
+Focus on discovering meaningful patterns and creating visualizations that tell a clear story.`;
+}
+function validateTableExists(state, adapter, tableName) {
+    const tables = adapter.getTables(state).map(normalizeDashboardAiTable);
+    const { table, ambiguousMatches } = resolveTableReference(tables, tableName);
+    if (ambiguousMatches) {
+        throw new DashboardAgentException(formatAmbiguousTableError(tableName, ambiguousMatches));
+    }
+    if (!table) {
+        throw new DashboardAgentException(`Table "${tableName}" not found. Available tables: ${formatAvailableTables(tables)}`);
+    }
+    return table;
+}
+function getOrCreateDashboard(state, adapter, displayTableName, dashboardTitle) {
+    const dashboardId = adapter.getCurrentDashboardArtifactId(state);
+    if (dashboardId) {
+        adapter.ensureDashboard(state, dashboardId);
+        return dashboardId;
+    }
+    const suggestedTitle = dashboardTitle ||
+        `${displayTableName.charAt(0).toUpperCase() + displayTableName.slice(1)} Insights`;
+    const newDashboardId = adapter.createDashboardArtifact(state, suggestedTitle, 'grid');
+    adapter.setCurrentArtifact(state, newDashboardId);
+    return newDashboardId;
+}
+function calculateAgentMetadata(state, adapter, dashboardId, tableName, agentToolCalls) {
+    const dashboard = adapter.getDashboard(state, dashboardId);
+    return {
+        tableName,
+        panelsCreated: dashboard?.panels?.length || 0,
+        stepsExecuted: agentToolCalls.length,
+        queriesRun: agentToolCalls.filter((call) => call.toolName === 'query')
+            .length,
+    };
+}
+export function createDashboardAgentTool(options) {
+    const { store, adapter } = options;
+    return tool({
+        description: `An AI agent that explores datasets and creates comprehensive dashboards with multiple visualizations.
+
+Use this for exploratory data analysis tasks like "analyze the earthquakes dataset" or "create insights dashboard for sales data".
+
+The agent will query the data, discover patterns, and create charts and Data Table Explorers with findings.
+
+For simple tasks like "create a histogram of magnitude", use the individual chart tools instead.
+
+IMPORTANT: Always provide tableName parameter when the user mentions a specific dataset.`,
+        inputSchema: DashboardAgentInputSchema,
+        execute: async (params, toolOptions) => {
+            const { prompt, tableName, dashboardTitle, maxSteps, temperature } = params;
+            let dashboardId = '';
+            try {
+                const state = store.getState();
+                const table = validateTableExists(state, adapter, tableName);
+                const resolvedTableName = table.tableId;
+                dashboardId = getOrCreateDashboard(state, adapter, table.tableName, dashboardTitle);
+                adapter.setSelectedTable(state, dashboardId, resolvedTableName);
+                const queryTools = options.createQueryTools?.({ store });
+                const dashboardAgent = new ToolLoopAgent({
+                    model: options.getModel({ state }),
+                    tools: {
+                        ...(queryTools ?? {}),
+                        ...createDashboardAiTools({
+                            store,
+                            adapter,
+                            chartTypes: options.chartTypes,
+                            extraTools: options.extraTools,
+                        }),
+                    },
+                    temperature: Math.max(0, Math.min(1, temperature ?? 0.7)),
+                    stopWhen: [stepCountIs(Math.max(5, Math.min(50, maxSteps ?? 20)))],
+                    instructions: options.instructions ?? DASHBOARD_AGENT_INSTRUCTIONS,
+                });
+                const result = await options.runSubAgent({
+                    agent: dashboardAgent,
+                    prompt: buildAgentPrompt(prompt, resolvedTableName, state, adapter),
+                    store,
+                    parentToolCallId: toolOptions?.toolCallId || '',
+                    abortSignal: toolOptions?.abortSignal,
+                });
+                const finalState = store.getState();
+                const metadata = calculateAgentMetadata(finalState, adapter, dashboardId, resolvedTableName, result.agentToolCalls || []);
+                return {
+                    success: true,
+                    finalOutput: result.finalOutput || 'Dashboard created successfully.',
+                    dashboardId,
+                    metadata,
+                };
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                const friendlyMessage = error instanceof DashboardAgentException
+                    ? errorMessage
+                    : 'Dashboard agent execution failed.';
+                return {
+                    success: false,
+                    finalOutput: friendlyMessage,
+                    dashboardId,
+                    error: errorMessage,
+                };
+            }
+        },
+    });
+}
+//# sourceMappingURL=ai.js.map

package/dist/ai/ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.js","sourceRoot":"","sources":["../../src/ai/ai.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAC,qBAAqB,EAA0B,MAAM,kBAAkB,CAAC;AAChF,OAAO,EAGL,aAAa,EACb,WAAW,EACX,IAAI,GACL,MAAM,IAAI,CAAC;AACZ,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EACL,gBAAgB,EAChB,uBAAuB,EACvB,oBAAoB,EACpB,2BAA2B,EAC3B,qBAAqB,GAMtB,MAAM,uBAAuB,CAAC;AAM/B,OAAO,EACL,6BAA6B,GAE9B,MAAM,kBAAkB,CAAC;AAgJ1B,MAAM,CAAC,MAAM,YAAY,GAAG,sBAAsB,CAAC;AAEnD,MAAM,CAAC,MAAM,yBAAyB,GAAG;;;;;;;;;;;;;;;;;;;+BAmBV,YAAY;;;;;CAK1C,CAAC;AAEF,MAAM,CAAC,MAAM,4BAA4B,GAAG;;;;;;;;;;;;;;;;;;IAkBxC,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6FA4D6E,CAAC;AAE9F,MAAM,qCAAqC,GAAG,CAAC,CAAC,MAAM,CAAC;IACrD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,UAAU,EAAE,CAAC;SACV,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;SACtB,QAAQ,EAAE;SACV,OAAO,CAAC,MAAM,CAAC;SACf,QAAQ,CAAC,qDAAqD,CAAC;CACnE,CAAC,CAAC;AAKH,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CAAC,uDAAuD,CAAC;IACpE,MAAM,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,CAAC,oDAAoD,CAAC;IACjE,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CAAC,qDAAqD,CAAC;IAClE,cAAc,EAAE,CAAC;SACd,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,2CAA2C,CAAC;IACxD,QAAQ,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,CAAC,sDAAsD,CAAC;IACnE,WAAW,EAAE,CAAC;SACX,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,OAAO,CAAC,GAAG,CAAC;SACZ,QAAQ,CACP,gFAAgF,CACjF;CACJ,CAAC,CAAC;AAGH,MAAM,uBAAwB,SAAQ,KAAK;IACzC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,yBAAyB,CAAC;IACxC,CAAC;CACF;AAED,SAAS,oBAAoB,CAC3B,KAAa,EACb,OAAmC;IAEnC,OAAO,OAAO;SACX,SAAS,CAAC,KAAK,CAAC;SAChB,GAAG,CAAC,yBAAyB,CAAC;SAC9B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,yBAAyB,CAChC,KAAuB;IAEvB,MAAM,EAAC,aAAa,EAAC,GAAG,KAAK,CAAC;IAC9B,MAAM,OAAO,GAAG,aAAa,CAAC,QAAQ,EAAE,CAAC;IAEzC,OAAO;QACL,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,aAAa,CAAC,KAAK;QAC9B,OAAO;QACP,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;KACzB,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB,CAChC,SAAiB,EACjB,OAAmC;IAEnC,OAAO,oBAAoB,SAAS,uBAAuB,OAAO;SAC/D,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;SAC7B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;AACnB,CAAC;AAED,SAAS,gBAAgB,CAAC,KAA+B;IACvD,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,UAAU,KAAK,CAAC,SAAS,2BAA2B,CAAC,CAAC;IACxE,CAAC;IACD,OAAO;QACL,SAAS,EAAE,KAAK,CAAC,OAAO;QACxB,aAAa,EAAE,KAAK,CAAC,KAAK;QAC1B,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACtC,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAkC;IAC/D,OAAO,CACL,MAAM;SACH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACb,OAAO,KAAK,CAAC,OAAO,KAAK,KAAK,CAAC,SAAS;YACtC,CAAC,CAAC,KAAK,CAAC,SAAS;YACjB,CAAC,CAAC,GAAG,KAAK,CAAC,SAAS,KAAK,KAAK,CAAC,OAAO,GAAG,CAAC;IAC9C,CAAC,CAAC;SACD,IAAI,CAAC,IAAI,CAAC,IAAI,QAAQ,CAC1B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAS,EAC9C,KAAK,EACL,OAAO,GACgC;IACvC,MAAM,eAAe,GAAG,CACtB,UAAmB,EACnB,eAAyB,EACzB,OAAmC,EAC3B,EAAE;QACV,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,KAAK,CAAC;QACvE,MAAM,0BAA0B,GAC9B,OAAO,CAAC,iCAAiC,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,gBAAgB,GAClB,UAAU;YACV,0BAA0B;YAC1B,CAAC,CAAC,aAAa;gBACb,CAAC,CAAC,OAAO,CAAC,6BAA6B,CAAC,KAAK,CAAC;gBAC9C,CAAC,CAAC,SAAS,CAAC,CAAC;QAEjB,IAAI,CAAC,gBAAgB,IAAI,aAAa,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CACb,8IAA8I,CAC/I,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,gBAAgB,IAAI,eAAe,EAAE,CAAC;YACzC,gBAAgB,GAAG,OAAO,CAAC,uBAAuB,CAChD,KAAK,EACL,SAAS,EACT,MAAM,CACP,CAAC;YACF,OAAO,CAAC,kBAAkB,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;YACpD,OAAO,CAAC,0BAA0B,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,OAAO,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,2FAA2F,CAC5F,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,EAAE,gBAAgB,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CACb,aAAa,gBAAgB,gCAAgC,CAC9D,CAAC;QACJ,CAAC;QAED,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;QACjD,OAAO,gBAAgB,CAAC;IAC1B,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,CACnB,UAAkB,EAClB,SAAuC,EACvC,EAAE;QACF,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,oBAAoB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAC1D,MAAM,iBAAiB,GACrB,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAE5E,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,EAAC,KAAK,EAAE,gBAAgB,EAAC,GAAG,qBAAqB,CACrD,MAAM,EACN,iBAAiB,CAClB,CAAC;YACF,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,kBAAkB,GACtB,OAAO,iBAAiB,KAAK,QAAQ;oBACnC,CAAC,CAAC,iBAAiB;oBACnB,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CACb,yBAAyB,CAAC,kBAAkB,EAAE,gBAAgB,CAAC,CAChE,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,kBAAkB,GACtB,OAAO,iBAAiB,KAAK,QAAQ;oBACnC,CAAC,CAAC,iBAAiB;oBACnB,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CACb,kBAAkB,kBAAkB,wBAAwB,qBAAqB,CAAC,MAAM,CAAC,GAAG,CAC7F,CAAC;YACJ,CAAC;YACD,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC9C,OAAO,CAAC,gBAAgB,CAAC,KAAK,EAAE,UAAU,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC;YACrE,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,IAAI,SAAS,EAAE,aAAa,EAAE,CAAC;YAC7B,MAAM,EAAC,KAAK,EAAC,GAAG,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;YACvE,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;YACtE,CAAC;YACD,MAAM,aAAa,GAAG,gBAAgB,CAAC,SAAS,CAAC,CAAC;YAClD,OAAO,CAAC,gBAAgB,CAAC,KAAK,EAAE,UAAU,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC;YACrE,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,MAAM,IAAI,KAAK,CACb,mEAAmE,qBAAqB,CAAC,MAAM,CAAC,GAAG,CACpG,CAAC;IACJ,CAAC,CAAC;IAEF,MAAM,IAAI,GAAsB;QAC9B,aAAa,EAAE,6BAA6B;QAC5C,eAAe;QACf,YAAY;QACZ,QAAQ,EAAE,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;YAC/B,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;QACrD,CAAC;QACD,WAAW,EAAE,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;YAC3C,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1D,CAAC;QACD,YAAY,EAAE,CAAC,WAAW,EAAE,EAAE;YAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAClD,CAAC;QACD,WAAW,EAAE,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE;YACpC,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QACnD,CAAC;QACD,kBAAkB,EAAE,CAAC,UAAU,EAAE,EAAE;YACjC,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,kBAAkB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAChD,CAAC;KACF,CAAC;IAEF,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE;YAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC/B,OAAO,OAAO,CAAC,aAAa,EAAE,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QAC9D,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAS,EAC7C,KAAK,EACL,OAAO,EACP,UAAU,EACV,UAAU,GAC4B;IACtC,MAAM,IAAI,GAAG,uBAAuB,CAAC,EAAC,KAAK,EAAE,OAAO,EAAC,CAAC,CAAC;IACvD,MAAM,kBAAkB,GACtB,UAAU,IAAI,uBAAuB,CAAC,EAAC,iBAAiB,EAAE,KAAK,EAAC,CAAC,CAAC;IACpE,MAAM,UAAU,GAAG,gBAAgB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;IAE3C,MAAM,YAAY,GAAG;QACnB,yBAAyB,EAAE,IAAI,CAAC;YAC9B,WAAW,EACT,wIAAwI;YAC1I,WAAW,EAAE,qCAAqC;YAClD,OAAO,EAAE,KAAK,EACZ,MAA6C,EAC7C,OAAO,EACP,EAAE;gBACF,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC/B,MAAM,UAAU,GAAG,OAAO,CAAC,uBAAuB,CAChD,KAAK,EACL,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,UAAU,CAClB,CAAC;gBACF,OAAO,CAAC,kBAAkB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;gBAC9C,OAAO,CAAC,0BAA0B,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;gBACjE,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,IAAI;wBACb,OAAO,EAAE,+BAA+B,UAAU,IAAI;wBACtD,IAAI,EAAE,EAAC,UAAU,EAAC;qBACnB;iBACF,CAAC;YACJ,CAAC;SACF,CAAC;QACF,GAAG,UAAU;QACb,oCAAoC,EAAE,2BAA2B,CAAC,IAAI,CAAC;QACvE,qBAAqB,EAAE,oBAAoB,CAAC,IAAI,CAAC;QACjD,sBAAsB,EAAE,qBAAqB,CAAC,IAAI,CAAC;KACpD,CAAC;IAEF,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QACzC,IAAI,GAAG,IAAI,YAAY,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,uDAAuD,GAAG,+CAA+C,CAC1G,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,GAAG,YAAY;QACf,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,UAAkB,EAClB,SAAiB,EACjB,KAAa,EACb,OAAmC;IAEnC,MAAM,KAAK,GAAG,qBAAqB,CACjC,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,yBAAyB,CAAC,EACvD,SAAS,CACV,CAAC,KAAK,CAAC;IACR,MAAM,iBAAiB,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,MAAM,WAAW,GACf,KAAK,IAAI,KAAK,CAAC,SAAS,KAAK,iBAAiB;QAC5C,CAAC,CAAC,gCAAgC,iBAAiB,EAAE;QACrD,CAAC,CAAC,EAAE,CAAC;IACT,MAAM,WAAW,GACf,KAAK,EAAE,OAAO;QACZ,EAAE,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACf,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAC9D;SACA,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC;IAC7B,MAAM,OAAO,GACX,KAAK,EAAE,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,qBAAqB,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAE7E,OAAO,gBAAgB,KAAK,EAAE,SAAS,IAAI,SAAS;;;aAGzC,WAAW,GAAG,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE;;gBAExD,UAAU;;8FAEoE,CAAC;AAC/F,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAa,EACb,OAAmC,EACnC,SAAiB;IAEjB,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;IACvE,MAAM,EAAC,KAAK,EAAE,gBAAgB,EAAC,GAAG,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC3E,IAAI,gBAAgB,EAAE,CAAC;QACrB,MAAM,IAAI,uBAAuB,CAC/B,yBAAyB,CAAC,SAAS,EAAE,gBAAgB,CAAC,CACvD,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,uBAAuB,CAC/B,UAAU,SAAS,kCAAkC,qBAAqB,CAAC,MAAM,CAAC,EAAE,CACrF,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAC3B,KAAa,EACb,OAAmC,EACnC,gBAAwB,EACxB,cAAuB;IAEvB,MAAM,WAAW,GAAG,OAAO,CAAC,6BAA6B,CAAC,KAAK,CAAC,CAAC;IACjE,IAAI,WAAW,EAAE,CAAC;QAChB,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAC5C,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,cAAc,GAClB,cAAc;QACd,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC;IACrF,MAAM,cAAc,GAAG,OAAO,CAAC,uBAAuB,CACpD,KAAK,EACL,cAAc,EACd,MAAM,CACP,CAAC;IACF,OAAO,CAAC,kBAAkB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IAClD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,SAAS,sBAAsB,CAC7B,KAAa,EACb,OAAmC,EACnC,WAAmB,EACnB,SAAiB,EACjB,cAAwC;IAExC,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC3D,OAAO;QACL,SAAS;QACT,aAAa,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;QAC7C,aAAa,EAAE,cAAc,CAAC,MAAM;QACpC,UAAU,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,OAAO,CAAC;aACnE,MAAM;KACV,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,OAAgD;IAEhD,MAAM,EAAC,KAAK,EAAE,OAAO,EAAC,GAAG,OAAO,CAAC;IAEjC,OAAO,IAAI,CAAC;QACV,WAAW,EAAE;;;;;;;;yFAQwE;QACrF,WAAW,EAAE,yBAAyB;QACtC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAiC,EAAE;YACpE,MAAM,EAAC,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,QAAQ,EAAE,WAAW,EAAC,GAAG,MAAM,CAAC;YAE1E,IAAI,WAAW,GAAG,EAAE,CAAC;YAErB,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC/B,MAAM,KAAK,GAAG,mBAAmB,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;gBAC7D,MAAM,iBAAiB,GAAG,KAAK,CAAC,OAAO,CAAC;gBAExC,WAAW,GAAG,oBAAoB,CAChC,KAAK,EACL,OAAO,EACP,KAAK,CAAC,SAAS,EACf,cAAc,CACf,CAAC;gBACF,OAAO,CAAC,gBAAgB,CAAC,KAAK,EAAE,WAAW,EAAE,iBAAiB,CAAC,CAAC;gBAChE,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,EAAC,KAAK,EAAC,CAAC,CAAC;gBAEvD,MAAM,cAAc,GAAG,IAAI,aAAa,CAAC;oBACvC,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAC,KAAK,EAAC,CAAC;oBAChC,KAAK,EAAE;wBACL,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC;wBACrB,GAAG,sBAAsB,CAAC;4BACxB,KAAK;4BACL,OAAO;4BACP,UAAU,EAAE,OAAO,CAAC,UAAU;4BAC9B,UAAU,EAAE,OAAO,CAAC,UAAU;yBAC/B,CAAC;qBACH;oBACD,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,IAAI,GAAG,CAAC,CAAC;oBACzD,QAAQ,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;oBAClE,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,4BAA4B;iBACnE,CAAC,CAAC;gBAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC;oBACvC,KAAK,EAAE,cAAc;oBACrB,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE,iBAAiB,EAAE,KAAK,EAAE,OAAO,CAAC;oBACnE,KAAK;oBACL,gBAAgB,EAAE,WAAW,EAAE,UAAU,IAAI,EAAE;oBAC/C,WAAW,EAAE,WAAW,EAAE,WAAW;iBACtC,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,sBAAsB,CACrC,UAAU,EACV,OAAO,EACP,WAAW,EACX,iBAAiB,EACjB,MAAM,CAAC,cAAc,IAAI,EAAE,CAC5B,CAAC;gBAEF,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,iCAAiC;oBACpE,WAAW;oBACX,QAAQ;iBACT,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACzD,MAAM,eAAe,GACnB,KAAK,YAAY,uBAAuB;oBACtC,CAAC,CAAC,YAAY;oBACd,CAAC,CAAC,mCAAmC,CAAC;gBAE1C,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,WAAW,EAAE,eAAe;oBAC5B,WAAW;oBACX,KAAK,EAAE,YAAY;iBACpB,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["/**\n * Reusable AI helpers for Mosaic dashboards.\n *\n * Host apps provide a narrow adapter for their store/artifact model; Mosaic owns\n * the dashboard tool behavior and prompts.\n *\n * @packageDocumentation\n */\n\nimport {resolveTableReference, type QualifiedTableName} from '@sqlrooms/duckdb';\nimport {\n  type LanguageModel,\n  type Tool,\n  ToolLoopAgent,\n  stepCountIs,\n  tool,\n} from 'ai';\nimport {z} from 'zod';\nimport {\n  createChartTools,\n  createDefaultChartTypes,\n  createListPanelsTool,\n  createDataTableExplorerTool,\n  createRemovePanelTool,\n  type ChartBuilderColumn,\n  type ChartToolExecutionContext,\n  type DashboardToolDeps,\n  type PanelPatch,\n  type ChartTypeDefinition,\n} from '../charts/chart-types';\nimport type {\n  MosaicDashboardEntry,\n  MosaicDashboardPanelConfig,\n} from '../dashboard/dashboard-types';\nimport type {MosaicDashboardLayoutType} from '../dashboard/core-types';\nimport {\n  DEFAULT_CHART_MAX_DATA_POINTS,\n  type ChartRuntimeIssue,\n} from '../chart-runtime';\n\nexport type {ChartToolExecutionContext} from '../charts/chart-types';\n\nexport type DashboardAiStore<TState> = {\n  getState: () => TState;\n};\n\nexport type DashboardAiTable = {\n  /**\n   * Canonical table identity used for routing after natural-language table\n   * selection resolves to a concrete table.\n   */\n  qualifiedName: QualifiedTableName;\n  columns?: ChartBuilderColumn[];\n  rowCount?: number;\n};\n\ntype ResolvedDashboardAiTable = {\n  table: QualifiedTableName;\n  tableName: string;\n  tableId: string;\n  columns?: ChartBuilderColumn[];\n  rowCount?: number;\n};\n\nexport type DashboardAiAdapter<TState> = {\n  getTables: (state: TState) => DashboardAiTable[];\n  hasRunContext?: (\n    state: TState,\n    context?: ChartToolExecutionContext,\n  ) => boolean;\n  resolveContextDashboardArtifactId?: (\n    state: TState,\n    context?: ChartToolExecutionContext,\n  ) => string | undefined;\n  makeDashboardPrimaryForRun?: (\n    state: TState,\n    dashboardId: string,\n    context?: ChartToolExecutionContext,\n  ) => void;\n  getCurrentDashboardArtifactId: (state: TState) => string | undefined;\n  createDashboardArtifact: (\n    state: TState,\n    title?: string,\n    layoutType?: MosaicDashboardLayoutType,\n  ) => string;\n  isDashboardArtifact: (state: TState, artifactId: string) => boolean;\n  setCurrentArtifact: (state: TState, artifactId: string) => void;\n  ensureDashboard: (\n    state: TState,\n    dashboardId: string,\n    title?: string,\n    layoutType?: MosaicDashboardLayoutType,\n  ) => void;\n  getDashboard: (\n    state: TState,\n    dashboardId: string,\n  ) => MosaicDashboardEntry | undefined;\n  getPanelIssue?: (\n    state: TState,\n    dashboardId: string,\n    panelId: string,\n  ) => ChartRuntimeIssue | undefined;\n  setSelectedTable: (\n    state: TState,\n    dashboardId: string,\n    tableName: string,\n  ) => void;\n  addPanel: (\n    state: TState,\n    dashboardId: string,\n    panel: MosaicDashboardPanelConfig,\n  ) => string;\n  updatePanel: (\n    state: TState,\n    dashboardId: string,\n    panelId: string,\n    patch: Partial<PanelPatch>,\n  ) => void;\n  removePanel: (state: TState, dashboardId: string, panelId: string) => void;\n};\n\nexport type CreateDashboardToolDepsOptions<TState> = {\n  store: DashboardAiStore<TState>;\n  adapter: DashboardAiAdapter<TState>;\n};\n\nexport type CreateDashboardAiToolsOptions<TState> =\n  CreateDashboardToolDepsOptions<TState> & {\n    chartTypes?: ChartTypeDefinition<any>[];\n    /**\n     * Host-provided dashboard tools keyed by their registered tool name.\n     * Register geospatial map tools under MAP_TOOL_KEY so prompts and tools\n     * stay in sync.\n     */\n    extraTools?: (deps: DashboardToolDeps) => Record<string, Tool>;\n  };\n\nexport type DashboardAgentToolCall = {\n  toolName: string;\n};\n\nexport type DashboardAgentRunResult = {\n  finalOutput?: string;\n  agentToolCalls?: DashboardAgentToolCall[];\n};\n\nexport type DashboardAgentResult = {\n  success: boolean;\n  finalOutput: string;\n  dashboardId: string;\n  error?: string;\n  metadata?: {\n    tableName: string;\n    panelsCreated: number;\n    stepsExecuted: number;\n    queriesRun: number;\n  };\n};\n\nexport type CreateDashboardAgentToolOptions<TState> =\n  CreateDashboardToolDepsOptions<TState> & {\n    getModel: (args: {state: TState}) => LanguageModel;\n    createQueryTools?: (args: {store: DashboardAiStore<TState>}) => {\n      query: Tool;\n    } & Record<string, Tool>;\n    runSubAgent: (args: {\n      agent: ToolLoopAgent<any, any, any>;\n      prompt: string;\n      store: DashboardAiStore<TState>;\n      parentToolCallId: string;\n      abortSignal?: AbortSignal;\n    }) => Promise<DashboardAgentRunResult>;\n    instructions?: string;\n    chartTypes?: ChartTypeDefinition<any>[];\n    /**\n     * Host-provided dashboard tools keyed by their registered tool name.\n     * Register geospatial map tools under MAP_TOOL_KEY so prompts and tools\n     * stay in sync.\n     */\n    extraTools?: (deps: DashboardToolDeps) => Record<string, Tool>;\n  };\n\nexport const MAP_TOOL_KEY = 'create_dashboard_map';\n\nexport const DASHBOARD_AI_INSTRUCTIONS = `\nDashboard authoring:\n\n**When to use dashboard_agent vs individual tools:**\n- Use \\`dashboard_agent\\` for exploratory requests that require data analysis and discovery:\n  - \"analyze the earthquakes dataset\"\n  - \"create insights dashboard for sales data\"\n  - \"find interesting patterns in customer behavior\"\n  - Any request asking to \"discover\", \"explore\", \"find insights\", or \"analyze\"\n- Use individual chart tools for direct, specific requests:\n  - \"create histogram of magnitude with 20 bins\"\n  - \"add a line chart showing sales over time\"\n  - \"update the histogram to use 30 bins\"\n\n**Individual dashboard chart tools:**\n- create_dashboard_histogram, create_dashboard_line_chart, create_dashboard_box_plot, create_dashboard_scatter_plot, create_dashboard_count_plot, create_dashboard_heatmap\n- Each chart type has its own tool with specific parameters.\n- For line charts with aggregation, use yFields array with {field: string, aggregate: \"sum\"|\"avg\"|\"min\"|\"max\"}.\n- Set xInterval for temporal binning (year, month, day, hour, etc.).\n- If the host app provides \\`${MAP_TOOL_KEY}\\`, use it for map/geospatial/location requests and tables with longitude/latitude or geometry columns. Author its config as native Deck JSON with layer classes in \\`spec.layers[].@@type\\`, dataset bindings in \\`_sqlroomsBinding.dataset\\`, and table/query sources in \\`config.datasets\\`. For data-driven map colors, use color accessors such as \\`getFillColor\\`, \\`getLineColor\\`, \\`getColor\\`, \\`getSourceColor\\`, or \\`getTargetColor\\` with \\`{\"@@function\":\"colorScale\", \"field\":\"...\", \"type\":\"sequential\"|\"diverging\"|\"quantize\"|\"quantile\"|\"categorical\", \"scheme\":\"Viridis\", \"domain\":\"auto\"}\\`.\n- Use \\`set_dashboard_vgplot\\` with complete JSON only when no chart tool fits your needs.\n- When calling \\`create_dashboard_artifact\\`, \\`layoutType\\` may be \\`grid\\` or \\`dock\\`; omitted values default to \\`grid\\`.\n- Ensure specs are valid JSON objects compatible with https://idl.uw.edu/mosaic/schema/latest.json.\n- \\`list_dashboard_panels\\` includes runtime issues when a chart failed. Use those issues to repair panels in place: convert too-large scatter charts to heatmaps, add \\`xInterval\\` to too-large line charts, and inspect columns/settings for SQL errors.\n`;\n\nexport const DASHBOARD_AGENT_INSTRUCTIONS = `You are a dashboard builder agent that creates and modifies interactive data dashboards.\n\n## Your Role\n\nYou analyze data and create insightful dashboards with multiple visualizations (charts, Data Table Explorers). You can handle both direct requests (\"create histogram of magnitude\") and exploratory requests (\"find interesting insights in earthquakes dataset\").\n\n## Available Tools\n\n**Chart Tools:**\n- create_dashboard_histogram - distribution of numeric values (always safe, aggregates automatically)\n- create_dashboard_line_chart - trends over time or ordered variable (use with aggregations for >10k rows)\n- create_dashboard_box_plot - compare distributions across categories\n- create_dashboard_scatter_plot - relationship between two numeric columns (avoid for >10k rows, use heatmap instead)\n- create_dashboard_count_plot - frequency of categorical values (always safe, aggregates automatically)\n- create_dashboard_heatmap - density/patterns across two dimensions (preferred for large datasets)\n\n**Panel Tools:**\n- create_dashboard_data_table_explorer - table statistics and column summaries\n- ${MAP_TOOL_KEY} - native Deck JSON geospatial map panel (if provided by the host app)\n\n**Data Tools:**\n- query - execute SQL queries for data exploration\n\n**Management Tools:**\n- list_dashboard_panels - discover panel IDs and what's on the dashboard\n- remove_dashboard_panel - delete a panel by ID\n\n## Workflows\n\n### Direct Requests\nWhen user provides specific instructions:\n1. Parse intent -> identify chart type\n2. Call appropriate tool with settings\n3. Done\n\nExample: \"create histogram of magnitude with 20 bins\"\n-> create_dashboard_histogram(settings: {field: \"magnitude\", bins: 20})\n\n### Exploratory Requests\nWhen user asks to discover insights:\n1. Use query tool to explore data:\n   - Start with simple stats: COUNT, MIN, MAX, AVG, DISTINCT (check total row count!)\n   - Check distributions: GROUP BY with COUNT\n   - Find correlations: CORR(col1, col2)\n   - Identify outliers and patterns\n2. Create targeted charts based on discoveries:\n   - If dataset has >10k rows: avoid scatter charts and unaggregated line charts\n   - Use histogram, count plot, heatmap, or aggregated visualizations instead\n3. Stop when dashboard tells coherent story\n\n### Update Requests\nTo update existing panels:\n1. Call list_dashboard_panels() to discover panel IDs\n2. Call appropriate create tool with panelId parameter\n3. Panel is updated in-place\n\n## Query Guidelines\n\n**Keep queries simple:**\n- Start with aggregations: COUNT, MIN, MAX, AVG, SUM\n- Use GROUP BY for breakdowns and distributions\n- Use LIMIT to control result size (especially with GROUP BY)\n- Avoid expensive operations: large JOINs, full table scans\n\n**Exploration strategy:**\n- Limit exploration to 5-10 queries per request\n- Start broad (overall stats), then narrow (specific patterns)\n- Stop when you have enough insights for a coherent dashboard\n\n## Best Practices\n\n- **Avoid unaggregated charts for large datasets:** For datasets >10k rows, DO NOT use scatter charts or line charts without aggregations. Use aggregated alternatives instead:\n  - For scatter plots: use heatmap or binned aggregations\n  - For line charts: use GROUP BY with time buckets or aggregations\n  - Histograms and count plots are always safe (they aggregate automatically)\n- **Check before update:** Always call list_dashboard_panels before updating/removing panels\n- **Repair broken charts:** list_dashboard_panels may return an \\`issue\\` per panel. For \\`too-much-data\\`, switch to an aggregated chart or add aggregation. For \\`sql-error\\`, inspect available columns/types and update the broken panel in place.\n- **Validate columns:** Query tools will validate column existence and types\n- **Handle errors gracefully:** If a query or chart creation fails, try alternative approach`;\n\nconst DashboardCreateArtifactToolParameters = z.object({\n  title: z.string().optional(),\n  layoutType: z\n    .enum(['dock', 'grid'])\n    .optional()\n    .default('grid')\n    .describe('Dashboard layout node type to use at creation time.'),\n});\ntype DashboardCreateArtifactToolParameters = z.infer<\n  typeof DashboardCreateArtifactToolParameters\n>;\n\nconst DashboardAgentInputSchema = z.object({\n  reasoning: z\n    .string()\n    .describe('Reasoning for why the dashboard agent is being called'),\n  prompt: z\n    .string()\n    .describe('The exploratory data analysis prompt for the agent'),\n  tableName: z\n    .string()\n    .describe('REQUIRED: The name of the table/dataset to analyze.'),\n  dashboardTitle: z\n    .string()\n    .optional()\n    .describe('Optional title for the dashboard artifact'),\n  maxSteps: z\n    .number()\n    .optional()\n    .default(20)\n    .describe('Maximum exploration steps (default: 20, range: 5-50)'),\n  temperature: z\n    .number()\n    .optional()\n    .default(0.7)\n    .describe(\n      'Model temperature for creativity vs consistency (default: 0.7, range: 0.0-1.0)',\n    ),\n});\ntype DashboardAgentInputSchema = z.infer<typeof DashboardAgentInputSchema>;\n\nclass DashboardAgentException extends Error {\n  constructor(message: string) {\n    super(message);\n    this.name = 'DashboardAgentException';\n  }\n}\n\nfunction getTablesWithColumns<TState>(\n  state: TState,\n  adapter: DashboardAiAdapter<TState>,\n): ResolvedDashboardAiTable[] {\n  return adapter\n    .getTables(state)\n    .map(normalizeDashboardAiTable)\n    .filter((table) => table.columns && table.columns.length > 0);\n}\n\nfunction normalizeDashboardAiTable(\n  table: DashboardAiTable,\n): ResolvedDashboardAiTable {\n  const {qualifiedName} = table;\n  const tableId = qualifiedName.toString();\n\n  return {\n    table: qualifiedName,\n    tableName: qualifiedName.table,\n    tableId,\n    columns: table.columns,\n    rowCount: table.rowCount,\n  };\n}\n\nfunction formatAmbiguousTableError(\n  tableName: string,\n  matches: ResolvedDashboardAiTable[],\n): string {\n  return `Ambiguous table \"${tableName}\". Matching tables: ${matches\n    .map((table) => table.tableId)\n    .join(', ')}.`;\n}\n\nfunction getResolvedTable(table: ResolvedDashboardAiTable) {\n  if (!table.columns) {\n    throw new Error(`Table \"${table.tableName}\" has no column metadata.`);\n  }\n  return {\n    tableName: table.tableId,\n    qualifiedName: table.table,\n    columns: table.columns.map((column) => ({\n      name: column.name,\n      type: column.type,\n    })),\n  };\n}\n\nfunction formatAvailableTables(tables: ResolvedDashboardAiTable[]): string {\n  return (\n    tables\n      .map((table) => {\n        return table.tableId === table.tableName\n          ? table.tableName\n          : `${table.tableName} (${table.tableId})`;\n      })\n      .join(', ') || '(none)'\n  );\n}\n\nexport function createDashboardToolDeps<TState>({\n  store,\n  adapter,\n}: CreateDashboardToolDepsOptions<TState>): DashboardToolDeps {\n  const resolveArtifact = (\n    artifactId?: string,\n    createIfMissing?: boolean,\n    context?: ChartToolExecutionContext,\n  ): string => {\n    const state = store.getState();\n    const hasRunContext = adapter.hasRunContext?.(state, context) ?? false;\n    const contextDashboardArtifactId =\n      adapter.resolveContextDashboardArtifactId?.(state, context);\n    let targetArtifactId =\n      artifactId ??\n      contextDashboardArtifactId ??\n      (!hasRunContext\n        ? adapter.getCurrentDashboardArtifactId(state)\n        : undefined);\n\n    if (!targetArtifactId && hasRunContext) {\n      throw new Error(\n        'No primary dashboard artifact is available in the current run context. Pass artifactId explicitly or use set_primary_context_artifact first.',\n      );\n    }\n\n    if (!targetArtifactId && createIfMissing) {\n      targetArtifactId = adapter.createDashboardArtifact(\n        state,\n        undefined,\n        'grid',\n      );\n      adapter.setCurrentArtifact(state, targetArtifactId);\n      adapter.makeDashboardPrimaryForRun?.(state, targetArtifactId, context);\n    }\n\n    if (!targetArtifactId) {\n      throw new Error(\n        'No dashboard artifact is available. Set createArtifactIfMissing=true or create one first.',\n      );\n    }\n\n    if (!adapter.isDashboardArtifact(state, targetArtifactId)) {\n      throw new Error(\n        `Artifact \"${targetArtifactId}\" is not a dashboard artifact.`,\n      );\n    }\n\n    adapter.ensureDashboard(state, targetArtifactId);\n    return targetArtifactId;\n  };\n\n  const resolveTable = (\n    artifactId: string,\n    tableName?: string | QualifiedTableName,\n  ) => {\n    const state = store.getState();\n    const tables = getTablesWithColumns(state, adapter);\n    const dashboard = adapter.getDashboard(state, artifactId);\n    const explicitTableName =\n      typeof tableName === 'string' ? tableName.trim() || undefined : tableName;\n\n    if (explicitTableName) {\n      const {table, ambiguousMatches} = resolveTableReference(\n        tables,\n        explicitTableName,\n      );\n      if (ambiguousMatches) {\n        const requestedTableName =\n          typeof explicitTableName === 'string'\n            ? explicitTableName\n            : explicitTableName.toString();\n        throw new Error(\n          formatAmbiguousTableError(requestedTableName, ambiguousMatches),\n        );\n      }\n      if (!table) {\n        const requestedTableName =\n          typeof explicitTableName === 'string'\n            ? explicitTableName\n            : explicitTableName.toString();\n        throw new Error(\n          `Unknown table \"${requestedTableName}\". Available tables: ${formatAvailableTables(tables)}.`,\n        );\n      }\n      const resolvedTable = getResolvedTable(table);\n      adapter.setSelectedTable(state, artifactId, resolvedTable.tableName);\n      return resolvedTable;\n    }\n\n    if (dashboard?.selectedTable) {\n      const {table} = resolveTableReference(tables, dashboard.selectedTable);\n      if (table) {\n        return getResolvedTable(table);\n      }\n    }\n\n    if (tables.length === 1) {\n      const onlyTable = tables[0];\n      if (!onlyTable) {\n        throw new Error('The only available table has no column metadata.');\n      }\n      const resolvedTable = getResolvedTable(onlyTable);\n      adapter.setSelectedTable(state, artifactId, resolvedTable.tableName);\n      return resolvedTable;\n    }\n\n    throw new Error(\n      `No dashboard table is selected. Provide tableName using one of: ${formatAvailableTables(tables)}.`,\n    );\n  };\n\n  const deps: DashboardToolDeps = {\n    maxDataPoints: DEFAULT_CHART_MAX_DATA_POINTS,\n    resolveArtifact,\n    resolveTable,\n    addPanel: (dashboardId, panel) => {\n      const state = store.getState();\n      return adapter.addPanel(state, dashboardId, panel);\n    },\n    updatePanel: (dashboardId, panelId, patch) => {\n      const state = store.getState();\n      adapter.updatePanel(state, dashboardId, panelId, patch);\n    },\n    getDashboard: (dashboardId) => {\n      const state = store.getState();\n      return adapter.getDashboard(state, dashboardId);\n    },\n    removePanel: (dashboardId, panelId) => {\n      const state = store.getState();\n      adapter.removePanel(state, dashboardId, panelId);\n    },\n    setCurrentArtifact: (artifactId) => {\n      const state = store.getState();\n      adapter.setCurrentArtifact(state, artifactId);\n    },\n  };\n\n  if (adapter.getPanelIssue) {\n    deps.getPanelIssue = (dashboardId, panelId) => {\n      const state = store.getState();\n      return adapter.getPanelIssue?.(state, dashboardId, panelId);\n    };\n  }\n\n  return deps;\n}\n\nexport function createDashboardAiTools<TState>({\n  store,\n  adapter,\n  chartTypes,\n  extraTools,\n}: CreateDashboardAiToolsOptions<TState>): Record<string, Tool> {\n  const deps = createDashboardToolDeps({store, adapter});\n  const resolvedChartTypes =\n    chartTypes ?? createDefaultChartTypes({includeCustomSpec: false});\n  const chartTools = createChartTools(resolvedChartTypes, deps);\n  const hostTools = extraTools?.(deps) ?? {};\n\n  const builtInTools = {\n    create_dashboard_artifact: tool({\n      description:\n        'Create a new dashboard artifact with a dock or grid layout and make it the active artifact. Use when no dashboard artifact exists yet.',\n      inputSchema: DashboardCreateArtifactToolParameters,\n      execute: async (\n        params: DashboardCreateArtifactToolParameters,\n        context,\n      ) => {\n        const state = store.getState();\n        const artifactId = adapter.createDashboardArtifact(\n          state,\n          params.title,\n          params.layoutType,\n        );\n        adapter.setCurrentArtifact(state, artifactId);\n        adapter.makeDashboardPrimaryForRun?.(state, artifactId, context);\n        return {\n          llmResult: {\n            success: true,\n            details: `Created dashboard artifact \"${artifactId}\".`,\n            data: {artifactId},\n          },\n        };\n      },\n    }),\n    ...chartTools,\n    create_dashboard_data_table_explorer: createDataTableExplorerTool(deps),\n    list_dashboard_panels: createListPanelsTool(deps),\n    remove_dashboard_panel: createRemovePanelTool(deps),\n  };\n\n  for (const key of Object.keys(hostTools)) {\n    if (key in builtInTools) {\n      throw new Error(\n        `Dashboard extraTools cannot override built-in tool \"${key}\". Register the host tool under a unique key.`,\n      );\n    }\n  }\n\n  return {\n    ...builtInTools,\n    ...hostTools,\n  };\n}\n\nfunction buildAgentPrompt<TState>(\n  userPrompt: string,\n  tableName: string,\n  state: TState,\n  adapter: DashboardAiAdapter<TState>,\n): string {\n  const table = resolveTableReference(\n    adapter.getTables(state).map(normalizeDashboardAiTable),\n    tableName,\n  ).table;\n  const resolvedTableName = table ? table.tableId : tableName;\n  const tableIdInfo =\n    table && table.tableName !== resolvedTableName\n      ? `\\n- Table ID for tool calls: ${resolvedTableName}`\n      : '';\n  const columnNames =\n    table?.columns\n      ?.map((column) =>\n        column.type ? `${column.name} (${column.type})` : column.name,\n      )\n      .join(', ') || 'unknown';\n  const rowInfo =\n    table?.rowCount !== undefined ? `Approximate rows: ${table.rowCount}` : '';\n\n  return `Analyze the \"${table?.tableName ?? tableName}\" table.\n\nTable info:\n- Columns: ${columnNames}${tableIdInfo}${rowInfo ? `\\n- ${rowInfo}` : ''}\n\nUser request: ${userPrompt}\n\nFocus on discovering meaningful patterns and creating visualizations that tell a clear story.`;\n}\n\nfunction validateTableExists<TState>(\n  state: TState,\n  adapter: DashboardAiAdapter<TState>,\n  tableName: string,\n): ResolvedDashboardAiTable {\n  const tables = adapter.getTables(state).map(normalizeDashboardAiTable);\n  const {table, ambiguousMatches} = resolveTableReference(tables, tableName);\n  if (ambiguousMatches) {\n    throw new DashboardAgentException(\n      formatAmbiguousTableError(tableName, ambiguousMatches),\n    );\n  }\n  if (!table) {\n    throw new DashboardAgentException(\n      `Table \"${tableName}\" not found. Available tables: ${formatAvailableTables(tables)}`,\n    );\n  }\n  return table;\n}\n\nfunction getOrCreateDashboard<TState>(\n  state: TState,\n  adapter: DashboardAiAdapter<TState>,\n  displayTableName: string,\n  dashboardTitle?: string,\n): string {\n  const dashboardId = adapter.getCurrentDashboardArtifactId(state);\n  if (dashboardId) {\n    adapter.ensureDashboard(state, dashboardId);\n    return dashboardId;\n  }\n\n  const suggestedTitle =\n    dashboardTitle ||\n    `${displayTableName.charAt(0).toUpperCase() + displayTableName.slice(1)} Insights`;\n  const newDashboardId = adapter.createDashboardArtifact(\n    state,\n    suggestedTitle,\n    'grid',\n  );\n  adapter.setCurrentArtifact(state, newDashboardId);\n  return newDashboardId;\n}\n\nfunction calculateAgentMetadata<TState>(\n  state: TState,\n  adapter: DashboardAiAdapter<TState>,\n  dashboardId: string,\n  tableName: string,\n  agentToolCalls: DashboardAgentToolCall[],\n): DashboardAgentResult['metadata'] {\n  const dashboard = adapter.getDashboard(state, dashboardId);\n  return {\n    tableName,\n    panelsCreated: dashboard?.panels?.length || 0,\n    stepsExecuted: agentToolCalls.length,\n    queriesRun: agentToolCalls.filter((call) => call.toolName === 'query')\n      .length,\n  };\n}\n\nexport function createDashboardAgentTool<TState>(\n  options: CreateDashboardAgentToolOptions<TState>,\n): Tool {\n  const {store, adapter} = options;\n\n  return tool({\n    description: `An AI agent that explores datasets and creates comprehensive dashboards with multiple visualizations.\n\nUse this for exploratory data analysis tasks like \"analyze the earthquakes dataset\" or \"create insights dashboard for sales data\".\n\nThe agent will query the data, discover patterns, and create charts and Data Table Explorers with findings.\n\nFor simple tasks like \"create a histogram of magnitude\", use the individual chart tools instead.\n\nIMPORTANT: Always provide tableName parameter when the user mentions a specific dataset.`,\n    inputSchema: DashboardAgentInputSchema,\n    execute: async (params, toolOptions): Promise<DashboardAgentResult> => {\n      const {prompt, tableName, dashboardTitle, maxSteps, temperature} = params;\n\n      let dashboardId = '';\n\n      try {\n        const state = store.getState();\n        const table = validateTableExists(state, adapter, tableName);\n        const resolvedTableName = table.tableId;\n\n        dashboardId = getOrCreateDashboard(\n          state,\n          adapter,\n          table.tableName,\n          dashboardTitle,\n        );\n        adapter.setSelectedTable(state, dashboardId, resolvedTableName);\n        const queryTools = options.createQueryTools?.({store});\n\n        const dashboardAgent = new ToolLoopAgent({\n          model: options.getModel({state}),\n          tools: {\n            ...(queryTools ?? {}),\n            ...createDashboardAiTools({\n              store,\n              adapter,\n              chartTypes: options.chartTypes,\n              extraTools: options.extraTools,\n            }),\n          },\n          temperature: Math.max(0, Math.min(1, temperature ?? 0.7)),\n          stopWhen: [stepCountIs(Math.max(5, Math.min(50, maxSteps ?? 20)))],\n          instructions: options.instructions ?? DASHBOARD_AGENT_INSTRUCTIONS,\n        });\n\n        const result = await options.runSubAgent({\n          agent: dashboardAgent,\n          prompt: buildAgentPrompt(prompt, resolvedTableName, state, adapter),\n          store,\n          parentToolCallId: toolOptions?.toolCallId || '',\n          abortSignal: toolOptions?.abortSignal,\n        });\n\n        const finalState = store.getState();\n        const metadata = calculateAgentMetadata(\n          finalState,\n          adapter,\n          dashboardId,\n          resolvedTableName,\n          result.agentToolCalls || [],\n        );\n\n        return {\n          success: true,\n          finalOutput: result.finalOutput || 'Dashboard created successfully.',\n          dashboardId,\n          metadata,\n        };\n      } catch (error) {\n        const errorMessage =\n          error instanceof Error ? error.message : String(error);\n        const friendlyMessage =\n          error instanceof DashboardAgentException\n            ? errorMessage\n            : 'Dashboard agent execution failed.';\n\n        return {\n          success: false,\n          finalOutput: friendlyMessage,\n          dashboardId,\n          error: errorMessage,\n        };\n      }\n    },\n  });\n}\n"]}

package/dist/ai/data-table-explorer-tool.d.ts

@@ -0,0 +1,40 @@
+import { z } from 'zod';
+import type { DashboardToolDeps } from '../charts/chart-types/base-types';
+export declare const DataTableExplorerToolParameters: z.ZodObject<{
+    artifactId: z.ZodOptional<z.ZodString>;
+    tableName: z.ZodOptional<z.ZodString>;
+    createArtifactIfMissing: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+    panelId: z.ZodOptional<z.ZodString>;
+    title: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    config: z.ZodObject<{
+        pageSize: z.ZodOptional<z.ZodNumber>;
+    }, z.core.$strip>;
+    reasoning: z.ZodString;
+}, z.core.$strip>;
+export type DataTableExplorerToolParams = z.infer<typeof DataTableExplorerToolParameters>;
+export declare function createDataTableExplorerTool(deps: DashboardToolDeps): import("ai").Tool<{
+    createArtifactIfMissing: boolean;
+    title: string;
+    config: {
+        pageSize?: number | undefined;
+    };
+    reasoning: string;
+    artifactId?: string | undefined;
+    tableName?: string | undefined;
+    panelId?: string | undefined;
+}, {
+    llmResult: {
+        success: boolean;
+        details: string;
+        data: import("./tool-helpers").PanelResult;
+        errorMessage?: undefined;
+    };
+} | {
+    llmResult: {
+        success: boolean;
+        errorMessage: string;
+        details?: undefined;
+        data?: undefined;
+    };
+}>;
+//# sourceMappingURL=data-table-explorer-tool.d.ts.map

package/dist/ai/data-table-explorer-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"data-table-explorer-tool.d.ts","sourceRoot":"","sources":["../../src/ai/data-table-explorer-tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,KAAK,EAAC,iBAAiB,EAAC,MAAM,kCAAkC,CAAC;AAIxE,eAAO,MAAM,+BAA+B;;;;;;;;;;iBA6B1C,CAAC;AAEH,MAAM,MAAM,2BAA2B,GAAG,CAAC,CAAC,KAAK,CAC/C,OAAO,+BAA+B,CACvC,CAAC;AAEF,wBAAgB,2BAA2B,CAAC,IAAI,EAAE,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;GA8ClE"}