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

package/dist/ai/createDataTableExplorerTool.js

@@ -0,0 +1,68 @@
+import { tool } from 'ai';
+import { z } from 'zod';
+import { ensureTable } from './tool-helpers';
+/**
+ * Parameters for creating a Data Table Explorer panel.
+ */
+export const DataTableExplorerParameters = z.object({
+    tableName: z
+        .string()
+        .min(1, 'tableName cannot be empty')
+        .optional()
+        .describe('Optional table name. Use when no table is selected yet.'),
+    title: z
+        .string()
+        .optional()
+        .default('Data Table Explorer')
+        .describe('Title for the Data Table Explorer panel.'),
+    intent: z
+        .string()
+        .optional()
+        .describe('Optional natural-language objective this Data Table Explorer should satisfy.'),
+});
+/**
+ * Tool input schema including reasoning and explorer parameters.
+ */
+export const DataTableExplorerToolInput = z.object({
+    reasoning: z
+        .string()
+        .describe('Brief rationale for creating the Data Table Explorer.'),
+    ...DataTableExplorerParameters.shape,
+});
+/**
+ * Creates an AI tool for generating Data Table Explorer panels with schema and column statistics.
+ */
+export function createDataTableExplorerTool({ databaseAdapter, addDataTable, }) {
+    return tool({
+        description: `Data Table Explorer: displays table schema and statistics for all columns (count, distinct values, min/max, etc.).
+
+Use when: user asks to "profile the data", "show table statistics", "what's in this table", "summarize columns", "give me an overview of the data".
+
+Useful for: quick data exploration, understanding data quality, finding missing values, identifying column types.`,
+        inputSchema: DataTableExplorerToolInput,
+        execute: async (params) => {
+            try {
+                if (params.tableName) {
+                    ensureTable(databaseAdapter, params.tableName);
+                }
+                const dataTable = await addDataTable(params);
+                return {
+                    llmResult: {
+                        success: true,
+                        details: `Created Data Table Explorer "${params.title}".`,
+                        data: dataTable ? { ...params, dataTable } : params,
+                    },
+                };
+            }
+            catch (error) {
+                return {
+                    llmResult: {
+                        success: false,
+                        errorMessage: error instanceof Error ? error.message : String(error),
+                    },
+                };
+            }
+        },
+    });
+}
+//# sourceMappingURL=createDataTableExplorerTool.js.map

package/dist/ai/createDataTableExplorerTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDataTableExplorerTool.js","sourceRoot":"","sources":["../../src/ai/createDataTableExplorerTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,IAAI,EAAC,MAAM,IAAI,CAAC;AACxB,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,WAAW,EAAC,MAAM,gBAAgB,CAAC;AAG3C;;GAEG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAClD,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,EAAE,2BAA2B,CAAC;SACnC,QAAQ,EAAE;SACV,QAAQ,CAAC,yDAAyD,CAAC;IACtE,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,OAAO,CAAC,qBAAqB,CAAC;SAC9B,QAAQ,CAAC,0CAA0C,CAAC;IACvD,MAAM,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,8EAA8E,CAC/E;CACJ,CAAC,CAAC;AASH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CAAC,uDAAuD,CAAC;IACpE,GAAG,2BAA2B,CAAC,KAAK;CACrC,CAAC,CAAC;AAiBH;;GAEG;AACH,MAAM,UAAU,2BAA2B,CAAC,EAC1C,eAAe,EACf,YAAY,GACuB;IACnC,OAAO,IAAI,CAAC;QACV,WAAW,EAAE;;;;kHAIiG;QAC9G,WAAW,EAAE,0BAA0B;QACvC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACxB,IAAI,CAAC;gBACH,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;oBACrB,WAAW,CAAC,eAAe,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;gBAE7C,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,IAAI;wBACb,OAAO,EAAE,gCAAgC,MAAM,CAAC,KAAK,IAAI;wBACzD,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,EAAC,GAAG,MAAM,EAAE,SAAS,EAAC,CAAC,CAAC,CAAC,MAAM;qBAClD;iBACF,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,KAAK;wBACd,YAAY,EACV,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBACzD;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC","sourcesContent":["import {tool} from 'ai';\nimport {z} from 'zod';\nimport {ensureTable} from './tool-helpers';\nimport {DatabaseAiAdapter} from './database-types';\n\n/**\n * Parameters for creating a Data Table Explorer panel.\n */\nexport const DataTableExplorerParameters = z.object({\n  tableName: z\n    .string()\n    .min(1, 'tableName cannot be empty')\n    .optional()\n    .describe('Optional table name. Use when no table is selected yet.'),\n  title: z\n    .string()\n    .optional()\n    .default('Data Table Explorer')\n    .describe('Title for the Data Table Explorer panel.'),\n  intent: z\n    .string()\n    .optional()\n    .describe(\n      'Optional natural-language objective this Data Table Explorer should satisfy.',\n    ),\n});\n\n/**\n * Inferred type for Data Table Explorer parameters.\n */\nexport type DataTableExplorerParameters = z.infer<\n  typeof DataTableExplorerParameters\n>;\n\n/**\n * Tool input schema including reasoning and explorer parameters.\n */\nexport const DataTableExplorerToolInput = z.object({\n  reasoning: z\n    .string()\n    .describe('Brief rationale for creating the Data Table Explorer.'),\n  ...DataTableExplorerParameters.shape,\n});\n\n/**\n * Inferred type for Data Table Explorer tool input.\n */\nexport type DataTableExplorerToolInput = z.infer<\n  typeof DataTableExplorerToolInput\n>;\n\n/**\n * Options for creating the Data Table Explorer tool.\n */\nexport type CreateDataTableExplorerToolOptions = {\n  databaseAdapter: DatabaseAiAdapter;\n  addDataTable(params: DataTableExplorerParameters): unknown | Promise<unknown>;\n};\n\n/**\n * Creates an AI tool for generating Data Table Explorer panels with schema and column statistics.\n */\nexport function createDataTableExplorerTool({\n  databaseAdapter,\n  addDataTable,\n}: CreateDataTableExplorerToolOptions) {\n  return tool({\n    description: `Data Table Explorer: displays table schema and statistics for all columns (count, distinct values, min/max, etc.).\n\nUse when: user asks to \"profile the data\", \"show table statistics\", \"what's in this table\", \"summarize columns\", \"give me an overview of the data\".\n\nUseful for: quick data exploration, understanding data quality, finding missing values, identifying column types.`,\n    inputSchema: DataTableExplorerToolInput,\n    execute: async (params) => {\n      try {\n        if (params.tableName) {\n          ensureTable(databaseAdapter, params.tableName);\n        }\n\n        const dataTable = await addDataTable(params);\n\n        return {\n          llmResult: {\n            success: true,\n            details: `Created Data Table Explorer \"${params.title}\".`,\n            data: dataTable ? {...params, dataTable} : params,\n          },\n        };\n      } catch (error) {\n        return {\n          llmResult: {\n            success: false,\n            errorMessage:\n              error instanceof Error ? error.message : String(error),\n          },\n        };\n      }\n    },\n  });\n}\n"]}

package/dist/ai/dashboard/constants.d.ts

@@ -0,0 +1,6 @@
+export declare const DASHBOARD_CHART_TOOL_PREFIX = "create_dashboard_panel_";
+export declare const KnownDashboardTools: {
+    readonly list_dashboard_panels: "list_dashboard_panels";
+    readonly create_dashboard_panel_data_table_explorer: "create_dashboard_panel_data_table_explorer";
+};
+//# sourceMappingURL=constants.d.ts.map

package/dist/ai/dashboard/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../src/ai/dashboard/constants.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,2BAA2B,4BAA4B,CAAC;AAErE,eAAO,MAAM,mBAAmB;;;CAItB,CAAC"}

package/dist/ai/dashboard/constants.js

@@ -0,0 +1,6 @@
+export const DASHBOARD_CHART_TOOL_PREFIX = 'create_dashboard_panel_';
+export const KnownDashboardTools = {
+    list_dashboard_panels: 'list_dashboard_panels',
+    create_dashboard_panel_data_table_explorer: 'create_dashboard_panel_data_table_explorer',
+};
+//# sourceMappingURL=constants.js.map

package/dist/ai/dashboard/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/ai/dashboard/constants.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,2BAA2B,GAAG,yBAAyB,CAAC;AAErE,MAAM,CAAC,MAAM,mBAAmB,GAAG;IACjC,qBAAqB,EAAE,uBAAuB;IAC9C,0CAA0C,EACxC,4CAA4C;CACtC,CAAC","sourcesContent":["export const DASHBOARD_CHART_TOOL_PREFIX = 'create_dashboard_panel_';\n\nexport const KnownDashboardTools = {\n  list_dashboard_panels: 'list_dashboard_panels',\n  create_dashboard_panel_data_table_explorer:\n    'create_dashboard_panel_data_table_explorer',\n} as const;\n"]}

package/dist/ai/dashboard/createDashboardAgentTool.d.ts

@@ -0,0 +1,17 @@
+import { type Tool } from 'ai';
+import type { CreateDashboardAgentToolOptions } from './dashboard-types';
+import { MosaicDashboardStoreState } from '../../dashboard/MosaicDashboardSlice';
+/**
+ * Creates an AI agent tool for populating dashboards with charts and interactive panels.
+ * The agent explores data through queries and creates multiple panels showing different aspects.
+ *
+ * @template TState - Store state type extending MosaicDashboardStoreState
+ * @param options - Configuration options for the dashboard agent
+ * @param options.store - Zustand store instance for state management
+ * @param options.databaseAdapter - Adapter for database operations and queries
+ * @param options.chartToolsOptions - Optional chart configuration and type restrictions
+ * @param options.extraTools - Optional factory for additional custom tools (e.g., map panels)
+ * @returns Tool instance that orchestrates multi-turn dashboard creation via a ToolLoopAgent
+ */
+export declare function createDashboardAgentTool<TState extends MosaicDashboardStoreState>(options: CreateDashboardAgentToolOptions<TState>): Tool;
+//# sourceMappingURL=createDashboardAgentTool.d.ts.map

package/dist/ai/dashboard/createDashboardAgentTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardAgentTool.d.ts","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardAgentTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,KAAK,IAAI,EAAmC,MAAM,IAAI,CAAC;AAG/D,OAAO,KAAK,EACV,+BAA+B,EAEhC,MAAM,mBAAmB,CAAC;AAI3B,OAAO,EAAC,yBAAyB,EAAC,MAAM,sCAAsC,CAAC;AAqH/E;;;;;;;;;;;GAWG;AACH,wBAAgB,wBAAwB,CACtC,MAAM,SAAS,yBAAyB,EACxC,OAAO,EAAE,+BAA+B,CAAC,MAAM,CAAC,GAAG,IAAI,CAyGxD"}

package/dist/ai/dashboard/createDashboardAgentTool.js

@@ -0,0 +1,205 @@
+import { ToolLoopAgent, stepCountIs, tool } from 'ai';
+import { z } from 'zod';
+import { MAP_TOOL_KEY } from '../constants';
+import { AiAgentError } from '../errors';
+import { ensureTable, calculateAgentResultMetadata } from '../tool-helpers';
+import { createDashboardAiTools } from './createDashboardAiTools';
+import { createDashboardAiAdapter } from './createDashboardAiAdapter';
+import { resolveChartTypes } from '../../charts/chart-types/resolveChartTypes';
+import { createChartToolsInstructions } from '../../charts/chart-types/createChartInstructions';
+import { DASHBOARD_CHART_TOOL_PREFIX, KnownDashboardTools } from './constants';
+import { AgentIntentSchemaFields } from '../agentIntent';
+import { getMosaicTableIdentity } from '../../mosaicTableReference';
+function getDashboardAgentInstructions(options) {
+    const chartTools = resolveChartTypes(options.chartToolsOptions?.chartTypes);
+    const chartToolsInstructions = createChartToolsInstructions(chartTools, DASHBOARD_CHART_TOOL_PREFIX);
+    return `
+You are a dashboard builder AI agent that creates dashboard.
+
+## Your Role
+
+You create DASHBOARD contains multiple panel including:
+- Chart panel
+- Data Table Explorer panel
+- Map panel (if provided by the host app)
+
+You can handle both direct requests ("create histogram of magnitude") and exploratory requests ("find interesting insights in earthquakes dataset").
+
+## Dataset
+
+Dashboard uses the **tableName** parameter to specify the dataset to analyze. If tableName is omitted for an existing dashboard, the agent uses the dashboard's selected table.
+You can use the query tool to explore the data and discover patterns, then create panels based on your findings.
+
+## Available Tools
+
+**Chart Tools:**
+${chartToolsInstructions}
+
+**Panel Tools:**
+- ${KnownDashboardTools.list_dashboard_panels} - inspect existing panel IDs, the selected table, and runtime issues
+- ${KnownDashboardTools.create_dashboard_panel_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
+
+## Workflows
+
+### Direct Requests
+When user provides specific instructions:
+1. Parse intent -> identify panel type
+2. If the user asks to change, update, edit, replace, or fix an existing panel, call ${KnownDashboardTools.list_dashboard_panels} first and pass the target panelId to the appropriate chart or map tool.
+3. Call appropriate tool with settings
+4. Done
+
+### 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 panels based on discoveries:
+   - Create 3-5 panels showing different aspects of the data
+   - 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
+
+## 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)
+- **Validate columns:** Query tools will validate column existence and types
+- **Update existing panels:** Chart tools accept panelId. Pass panelId when the user wants to modify an existing chart; omit it only when creating a new panel.
+- **Handle errors gracefully:** If a query or chart creation fails, try alternative approach`;
+}
+const DashboardAgentInputSchema = z.object({
+    reasoning: z
+        .string()
+        .describe('Reasoning for why the dashboard agent is being called'),
+    ...AgentIntentSchemaFields,
+    tableName: z
+        .string()
+        .optional()
+        .describe('The name of the table/dataset to analyze. Optional when the target dashboard already has a selected table.'),
+    dashboardId: z
+        .string()
+        .describe('The ID of an existing dashboard to update.'),
+    maxSteps: z
+        .number()
+        .optional()
+        .default(20)
+        .describe('Maximum exploration steps (default: 20, range: 5-50)'),
+});
+/**
+ * Creates an AI agent tool for populating dashboards with charts and interactive panels.
+ * The agent explores data through queries and creates multiple panels showing different aspects.
+ *
+ * @template TState - Store state type extending MosaicDashboardStoreState
+ * @param options - Configuration options for the dashboard agent
+ * @param options.store - Zustand store instance for state management
+ * @param options.databaseAdapter - Adapter for database operations and queries
+ * @param options.chartToolsOptions - Optional chart configuration and type restrictions
+ * @param options.extraTools - Optional factory for additional custom tools (e.g., map panels)
+ * @returns Tool instance that orchestrates multi-turn dashboard creation via a ToolLoopAgent
+ */
+export function createDashboardAgentTool(options) {
+    const { store, databaseAdapter, chartToolsOptions, extraTools } = options;
+    return tool({
+        description: `An AI agent that populates a dashboard with charts and interactive panels.
+
+This agent:
+- Queries the data to discover patterns and insights
+- Creates charts (histograms, scatter plots, heatmaps, box plots, etc.)
+- Adds Data Table Explorer panels for tabular analysis
+- Builds interactive dashboards for data exploration
+
+REQUIRED PARAMETERS:
+- dashboardId: The ID of the dashboard to populate
+- intent: What insights, patterns, or specific charts to create
+
+OPTIONAL PARAMETERS:
+- tableName: The dataset to analyze. If omitted, the dashboard's selected table is used.
+
+The agent will explore the data and create 3-5 panels showing different aspects based on the intent.
+
+IMPORTANT: IF primary artefact in run context is a dashboard, prioritize using this tool for any queries or data analysis tasks.`,
+        inputSchema: DashboardAgentInputSchema,
+        execute: async (params, toolOptions) => {
+            const { dashboardId, maxSteps } = params;
+            const { intent } = params;
+            const dashboardAdapter = createDashboardAiAdapter(store, dashboardId);
+            try {
+                const state = store.getState();
+                const tableName = params.tableName ?? dashboardAdapter.getSelectedTable?.();
+                if (!tableName) {
+                    throw new AiAgentError('Dashboard tableName was not provided and the dashboard has no selected table.');
+                }
+                const dataTable = ensureTable(databaseAdapter, tableName);
+                await dashboardAdapter.setSelectedTable(dataTable.table ? getMosaicTableIdentity(dataTable.table) : tableName);
+                const dataTools = options.createDataTools?.({ store }) ?? {};
+                const agent = new ToolLoopAgent({
+                    model: options.getModel({ state }),
+                    tools: {
+                        ...dataTools,
+                        ...createDashboardAiTools({
+                            dashboardAdapter,
+                            databaseAdapter,
+                            chartToolsOptions,
+                            extraTools,
+                        }),
+                    },
+                    stopWhen: [stepCountIs(Math.max(5, Math.min(50, maxSteps ?? 20)))],
+                    instructions: [
+                        options.instructions ?? getDashboardAgentInstructions(options),
+                        options.additionalInstructions,
+                    ]
+                        .filter(Boolean)
+                        .join('\n\n'),
+                });
+                const result = await options.runSubAgent({
+                    agent,
+                    prompt: intent,
+                    store,
+                    parentToolCallId: toolOptions?.toolCallId || '',
+                    abortSignal: toolOptions?.abortSignal,
+                });
+                const metadata = calculateAgentResultMetadata(tableName, 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 AiAgentError
+                    ? errorMessage
+                    : 'Dashboard agent execution failed.';
+                return {
+                    success: false,
+                    finalOutput: friendlyMessage,
+                    dashboardId,
+                    error: errorMessage,
+                };
+            }
+        },
+    });
+}
+//# sourceMappingURL=createDashboardAgentTool.js.map

package/dist/ai/dashboard/createDashboardAgentTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardAgentTool.js","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardAgentTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,aAAa,EAAE,WAAW,EAAE,IAAI,EAAC,MAAM,IAAI,CAAC;AAC/D,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAK1C,OAAO,EAAC,YAAY,EAAC,MAAM,WAAW,CAAC;AACvC,OAAO,EAAC,WAAW,EAAE,4BAA4B,EAAC,MAAM,iBAAiB,CAAC;AAC1E,OAAO,EAAC,sBAAsB,EAAC,MAAM,0BAA0B,CAAC;AAEhE,OAAO,EAAC,wBAAwB,EAAC,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAC,iBAAiB,EAAC,MAAM,4CAA4C,CAAC;AAC7E,OAAO,EAAC,4BAA4B,EAAC,MAAM,kDAAkD,CAAC;AAC9F,OAAO,EAAC,2BAA2B,EAAE,mBAAmB,EAAC,MAAM,aAAa,CAAC;AAC7E,OAAO,EAAC,uBAAuB,EAAC,MAAM,gBAAgB,CAAC;AACvD,OAAO,EAAC,sBAAsB,EAAC,MAAM,4BAA4B,CAAC;AAElE,SAAS,6BAA6B,CACpC,OAAgD;IAEhD,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,CAAC,iBAAiB,EAAE,UAAU,CAAC,CAAC;IAE5E,MAAM,sBAAsB,GAAG,4BAA4B,CACzD,UAAU,EACV,2BAA2B,CAC5B,CAAC;IAEF,OAAO;;;;;;;;;;;;;;;;;;;;EAoBP,sBAAsB;;;IAGpB,mBAAmB,CAAC,qBAAqB;IACzC,mBAAmB,CAAC,0CAA0C;IAC9D,YAAY;;;;;;;;;;uFAUuE,mBAAmB,CAAC,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;6FAsCnC,CAAC;AAC9F,CAAC;AAED,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CAAC,uDAAuD,CAAC;IACpE,GAAG,uBAAuB;IAC1B,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,4GAA4G,CAC7G;IACH,WAAW,EAAE,CAAC;SACX,MAAM,EAAE;SACR,QAAQ,CAAC,4CAA4C,CAAC;IACzD,QAAQ,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,CAAC,sDAAsD,CAAC;CACpE,CAAC,CAAC;AAIH;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,wBAAwB,CAEtC,OAAgD;IAChD,MAAM,EAAC,KAAK,EAAE,eAAe,EAAE,iBAAiB,EAAE,UAAU,EAAC,GAAG,OAAO,CAAC;IAExE,OAAO,IAAI,CAAC;QACV,WAAW,EAAE;;;;;;;;;;;;;;;;;iIAiBgH;QAC7H,WAAW,EAAE,yBAAyB;QACtC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAiC,EAAE;YACpE,MAAM,EAAC,WAAW,EAAE,QAAQ,EAAC,GAAG,MAAM,CAAC;YACvC,MAAM,EAAC,MAAM,EAAC,GAAG,MAAM,CAAC;YAExB,MAAM,gBAAgB,GAAG,wBAAwB,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAEtE,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC/B,MAAM,SAAS,GACb,MAAM,CAAC,SAAS,IAAI,gBAAgB,CAAC,gBAAgB,EAAE,EAAE,CAAC;gBAE5D,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,MAAM,IAAI,YAAY,CACpB,+EAA+E,CAChF,CAAC;gBACJ,CAAC;gBAED,MAAM,SAAS,GAAG,WAAW,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;gBAE1D,MAAM,gBAAgB,CAAC,gBAAgB,CACrC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CACtE,CAAC;gBAEF,MAAM,SAAS,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,EAAC,KAAK,EAAC,CAAC,IAAI,EAAE,CAAC;gBAE3D,MAAM,KAAK,GAAG,IAAI,aAAa,CAAC;oBAC9B,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAC,KAAK,EAAC,CAAC;oBAChC,KAAK,EAAE;wBACL,GAAG,SAAS;wBACZ,GAAG,sBAAsB,CAAC;4BACxB,gBAAgB;4BAChB,eAAe;4BACf,iBAAiB;4BACjB,UAAU;yBACX,CAAC;qBACH;oBACD,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;wBACZ,OAAO,CAAC,YAAY,IAAI,6BAA6B,CAAC,OAAO,CAAC;wBAC9D,OAAO,CAAC,sBAAsB;qBAC/B;yBACE,MAAM,CAAC,OAAO,CAAC;yBACf,IAAI,CAAC,MAAM,CAAC;iBAChB,CAAC,CAAC;gBAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC;oBACvC,KAAK;oBACL,MAAM,EAAE,MAAM;oBACd,KAAK;oBACL,gBAAgB,EAAE,WAAW,EAAE,UAAU,IAAI,EAAE;oBAC/C,WAAW,EAAE,WAAW,EAAE,WAAW;iBACtC,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG,4BAA4B,CAC3C,SAAS,EACT,MAAM,CAAC,cAAc,CACtB,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;gBAEzD,MAAM,eAAe,GACnB,KAAK,YAAY,YAAY;oBAC3B,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":["import {type Tool, ToolLoopAgent, stepCountIs, tool} from 'ai';\nimport {z} from 'zod';\nimport {MAP_TOOL_KEY} from '../constants';\nimport type {\n  CreateDashboardAgentToolOptions,\n  DashboardAgentResult,\n} from './dashboard-types';\nimport {AiAgentError} from '../errors';\nimport {ensureTable, calculateAgentResultMetadata} from '../tool-helpers';\nimport {createDashboardAiTools} from './createDashboardAiTools';\nimport {MosaicDashboardStoreState} from '../../dashboard/MosaicDashboardSlice';\nimport {createDashboardAiAdapter} from './createDashboardAiAdapter';\nimport {resolveChartTypes} from '../../charts/chart-types/resolveChartTypes';\nimport {createChartToolsInstructions} from '../../charts/chart-types/createChartInstructions';\nimport {DASHBOARD_CHART_TOOL_PREFIX, KnownDashboardTools} from './constants';\nimport {AgentIntentSchemaFields} from '../agentIntent';\nimport {getMosaicTableIdentity} from '../../mosaicTableReference';\n\nfunction getDashboardAgentInstructions<TState>(\n  options: CreateDashboardAgentToolOptions<TState>,\n): string {\n  const chartTools = resolveChartTypes(options.chartToolsOptions?.chartTypes);\n\n  const chartToolsInstructions = createChartToolsInstructions(\n    chartTools,\n    DASHBOARD_CHART_TOOL_PREFIX,\n  );\n\n  return `\nYou are a dashboard builder AI agent that creates dashboard.\n\n## Your Role\n\nYou create DASHBOARD contains multiple panel including:\n- Chart panel\n- Data Table Explorer panel\n- Map panel (if provided by the host app)\n\nYou can handle both direct requests (\"create histogram of magnitude\") and exploratory requests (\"find interesting insights in earthquakes dataset\").\n\n## Dataset\n\nDashboard uses the **tableName** parameter to specify the dataset to analyze. If tableName is omitted for an existing dashboard, the agent uses the dashboard's selected table.\nYou can use the query tool to explore the data and discover patterns, then create panels based on your findings.\n\n## Available Tools\n\n**Chart Tools:**\n${chartToolsInstructions}\n\n**Panel Tools:**\n- ${KnownDashboardTools.list_dashboard_panels} - inspect existing panel IDs, the selected table, and runtime issues\n- ${KnownDashboardTools.create_dashboard_panel_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## Workflows\n\n### Direct Requests\nWhen user provides specific instructions:\n1. Parse intent -> identify panel type\n2. If the user asks to change, update, edit, replace, or fix an existing panel, call ${KnownDashboardTools.list_dashboard_panels} first and pass the target panelId to the appropriate chart or map tool.\n3. Call appropriate tool with settings\n4. Done\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 panels based on discoveries:\n   - Create 3-5 panels showing different aspects of the data\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## 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- **Validate columns:** Query tools will validate column existence and types\n- **Update existing panels:** Chart tools accept panelId. Pass panelId when the user wants to modify an existing chart; omit it only when creating a new panel.\n- **Handle errors gracefully:** If a query or chart creation fails, try alternative approach`;\n}\n\nconst DashboardAgentInputSchema = z.object({\n  reasoning: z\n    .string()\n    .describe('Reasoning for why the dashboard agent is being called'),\n  ...AgentIntentSchemaFields,\n  tableName: z\n    .string()\n    .optional()\n    .describe(\n      'The name of the table/dataset to analyze. Optional when the target dashboard already has a selected table.',\n    ),\n  dashboardId: z\n    .string()\n    .describe('The ID of an existing dashboard to update.'),\n  maxSteps: z\n    .number()\n    .optional()\n    .default(20)\n    .describe('Maximum exploration steps (default: 20, range: 5-50)'),\n});\n\ntype DashboardAgentInputSchema = z.infer<typeof DashboardAgentInputSchema>;\n\n/**\n * Creates an AI agent tool for populating dashboards with charts and interactive panels.\n * The agent explores data through queries and creates multiple panels showing different aspects.\n *\n * @template TState - Store state type extending MosaicDashboardStoreState\n * @param options - Configuration options for the dashboard agent\n * @param options.store - Zustand store instance for state management\n * @param options.databaseAdapter - Adapter for database operations and queries\n * @param options.chartToolsOptions - Optional chart configuration and type restrictions\n * @param options.extraTools - Optional factory for additional custom tools (e.g., map panels)\n * @returns Tool instance that orchestrates multi-turn dashboard creation via a ToolLoopAgent\n */\nexport function createDashboardAgentTool<\n  TState extends MosaicDashboardStoreState,\n>(options: CreateDashboardAgentToolOptions<TState>): Tool {\n  const {store, databaseAdapter, chartToolsOptions, extraTools} = options;\n\n  return tool({\n    description: `An AI agent that populates a dashboard with charts and interactive panels.\n\nThis agent:\n- Queries the data to discover patterns and insights\n- Creates charts (histograms, scatter plots, heatmaps, box plots, etc.)\n- Adds Data Table Explorer panels for tabular analysis\n- Builds interactive dashboards for data exploration\n\nREQUIRED PARAMETERS:\n- dashboardId: The ID of the dashboard to populate\n- intent: What insights, patterns, or specific charts to create\n\nOPTIONAL PARAMETERS:\n- tableName: The dataset to analyze. If omitted, the dashboard's selected table is used.\n\nThe agent will explore the data and create 3-5 panels showing different aspects based on the intent.\n\nIMPORTANT: IF primary artefact in run context is a dashboard, prioritize using this tool for any queries or data analysis tasks.`,\n    inputSchema: DashboardAgentInputSchema,\n    execute: async (params, toolOptions): Promise<DashboardAgentResult> => {\n      const {dashboardId, maxSteps} = params;\n      const {intent} = params;\n\n      const dashboardAdapter = createDashboardAiAdapter(store, dashboardId);\n\n      try {\n        const state = store.getState();\n        const tableName =\n          params.tableName ?? dashboardAdapter.getSelectedTable?.();\n\n        if (!tableName) {\n          throw new AiAgentError(\n            'Dashboard tableName was not provided and the dashboard has no selected table.',\n          );\n        }\n\n        const dataTable = ensureTable(databaseAdapter, tableName);\n\n        await dashboardAdapter.setSelectedTable(\n          dataTable.table ? getMosaicTableIdentity(dataTable.table) : tableName,\n        );\n\n        const dataTools = options.createDataTools?.({store}) ?? {};\n\n        const agent = new ToolLoopAgent({\n          model: options.getModel({state}),\n          tools: {\n            ...dataTools,\n            ...createDashboardAiTools({\n              dashboardAdapter,\n              databaseAdapter,\n              chartToolsOptions,\n              extraTools,\n            }),\n          },\n          stopWhen: [stepCountIs(Math.max(5, Math.min(50, maxSteps ?? 20)))],\n          instructions: [\n            options.instructions ?? getDashboardAgentInstructions(options),\n            options.additionalInstructions,\n          ]\n            .filter(Boolean)\n            .join('\\n\\n'),\n        });\n\n        const result = await options.runSubAgent({\n          agent,\n          prompt: intent,\n          store,\n          parentToolCallId: toolOptions?.toolCallId || '',\n          abortSignal: toolOptions?.abortSignal,\n        });\n\n        const metadata = calculateAgentResultMetadata(\n          tableName,\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\n        const friendlyMessage =\n          error instanceof AiAgentError\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/dashboard/createDashboardAiAdapter.d.ts

@@ -0,0 +1,14 @@
+import { DashboardAiAdapter } from './dashboard-types';
+import { MosaicDashboardStoreState } from '../../dashboard/MosaicDashboardSlice';
+import { AiStore } from '../types';
+/**
+ * Creates a dashboard adapter for AI operations scoped to a specific dashboard.
+ * The adapter provides methods for managing panels, tables, and dashboard state.
+ *
+ * @template TState - Store state type extending MosaicDashboardStoreState
+ * @param store - Zustand store instance with dashboard management capabilities
+ * @param dashboardId - ID of the dashboard to scope adapter operations to
+ * @returns Dashboard adapter instance with panel and table management methods
+ */
+export declare function createDashboardAiAdapter<TState extends MosaicDashboardStoreState>(store: AiStore<TState>, dashboardId: string): DashboardAiAdapter;
+//# sourceMappingURL=createDashboardAiAdapter.d.ts.map

package/dist/ai/dashboard/createDashboardAiAdapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardAiAdapter.d.ts","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardAiAdapter.ts"],"names":[],"mappings":"AAIA,OAAO,EAAC,kBAAkB,EAAC,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAC,yBAAyB,EAAC,MAAM,sCAAsC,CAAC;AAE/E,OAAO,EAAC,OAAO,EAAC,MAAM,UAAU,CAAC;AAsBjC;;;;;;;;GAQG;AACH,wBAAgB,wBAAwB,CACtC,MAAM,SAAS,yBAAyB,EACxC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,MAAM,GAAG,kBAAkB,CAwEjE"}

package/dist/ai/dashboard/createDashboardAiAdapter.js

@@ -0,0 +1,75 @@
+import { hasCommandSliceState, } from '@sqlrooms/room-store';
+import { MOSAIC_DASHBOARD_COMMAND_IDS } from '../../dashboard/MosaicDashboardCommands';
+function getCommands(store) {
+    const state = store.getState();
+    return hasCommandSliceState(state) ? state.commands : undefined;
+}
+function hasCommand(store, commandId) {
+    return Boolean(getCommands(store)?.getCommand(commandId));
+}
+function assertCommandSuccess(result, commandId) {
+    if (!result.success) {
+        throw new Error(result.error ?? result.message ?? `Command "${commandId}" failed.`);
+    }
+}
+/**
+ * Creates a dashboard adapter for AI operations scoped to a specific dashboard.
+ * The adapter provides methods for managing panels, tables, and dashboard state.
+ *
+ * @template TState - Store state type extending MosaicDashboardStoreState
+ * @param store - Zustand store instance with dashboard management capabilities
+ * @param dashboardId - ID of the dashboard to scope adapter operations to
+ * @returns Dashboard adapter instance with panel and table management methods
+ */
+export function createDashboardAiAdapter(store, dashboardId) {
+    return {
+        getSelectedTable: () => store.getState().mosaicDashboard.getDashboard(dashboardId)?.selectedTable,
+        getPanels: () => store.getState().mosaicDashboard.getDashboard(dashboardId)?.panels ?? [],
+        setSelectedTable: async (tableName) => {
+            const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.setSelectedTable;
+            if (!hasCommand(store, commandId)) {
+                store
+                    .getState()
+                    .mosaicDashboard.setSelectedTable(dashboardId, tableName);
+                return;
+            }
+            const result = await getCommands(store).invokeCommand(commandId, { dashboardId, tableName }, { surface: 'ai', actor: 'dashboard-ai-adapter' });
+            assertCommandSuccess(result, commandId);
+        },
+        addPanel: async (panel) => {
+            const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.addPanel;
+            if (!hasCommand(store, commandId)) {
+                return store.getState().mosaicDashboard.addPanel(dashboardId, panel);
+            }
+            const result = await getCommands(store).invokeCommand(commandId, { dashboardId, panel }, { surface: 'ai', actor: 'dashboard-ai-adapter' });
+            assertCommandSuccess(result, commandId);
+            return (result.data?.panelId ?? panel.id);
+        },
+        updatePanel: async (panelId, patch) => {
+            const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.updatePanel;
+            if (!hasCommand(store, commandId)) {
+                store
+                    .getState()
+                    .mosaicDashboard.updatePanel(dashboardId, panelId, patch);
+                return;
+            }
+            const result = await getCommands(store).invokeCommand(commandId, { dashboardId, panelId, patch }, { surface: 'ai', actor: 'dashboard-ai-adapter' });
+            assertCommandSuccess(result, commandId);
+        },
+        removePanel: async (panelId) => {
+            const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.removePanel;
+            if (!hasCommand(store, commandId)) {
+                store.getState().mosaicDashboard.removePanel(dashboardId, panelId);
+                return;
+            }
+            const result = await getCommands(store).invokeCommand(commandId, { dashboardId, panelId }, { surface: 'ai', actor: 'dashboard-ai-adapter' });
+            assertCommandSuccess(result, commandId);
+        },
+        getPanel: (panelId) => store
+            .getState()
+            .mosaicDashboard.getDashboard(dashboardId)
+            ?.panels.find((panel) => panel.id === panelId),
+        getPanelIssue: (panelId) => store.getState().mosaicDashboard.getPanelIssue(dashboardId, panelId),
+    };
+}
+//# sourceMappingURL=createDashboardAiAdapter.js.map

package/dist/ai/dashboard/createDashboardAiAdapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardAiAdapter.js","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardAiAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,GAErB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EAAC,4BAA4B,EAAC,MAAM,yCAAyC,CAAC;AAGrF,SAAS,WAAW,CAAC,KAAyC;IAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;IAC/B,OAAO,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;AAClE,CAAC;AAED,SAAS,UAAU,CACjB,KAAyC,EACzC,SAAiB;IAEjB,OAAO,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAyB,EAAE,SAAiB;IACxE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CACb,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,IAAI,YAAY,SAAS,WAAW,CACnE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,wBAAwB,CAEtC,KAAsB,EAAE,WAAmB;IAC3C,OAAO;QACL,gBAAgB,EAAE,GAAG,EAAE,CACrB,KAAK,CAAC,QAAQ,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,aAAa;QAC3E,SAAS,EAAE,GAAG,EAAE,CACd,KAAK,CAAC,QAAQ,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,MAAM,IAAI,EAAE;QAC1E,gBAAgB,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE;YACpC,MAAM,SAAS,GAAG,4BAA4B,CAAC,gBAAgB,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;gBAClC,KAAK;qBACF,QAAQ,EAAE;qBACV,eAAe,CAAC,gBAAgB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;gBAC5D,OAAO;YACT,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,KAAK,CAAE,CAAC,aAAa,CACpD,SAAS,EACT,EAAC,WAAW,EAAE,SAAS,EAAC,EACxB,EAAC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,sBAAsB,EAAC,CAC/C,CAAC;YACF,oBAAoB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACxB,MAAM,SAAS,GAAG,4BAA4B,CAAC,QAAQ,CAAC;YACxD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;gBAClC,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YACvE,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,KAAK,CAAE,CAAC,aAAa,CACpD,SAAS,EACT,EAAC,WAAW,EAAE,KAAK,EAAC,EACpB,EAAC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,sBAAsB,EAAC,CAC/C,CAAC;YACF,oBAAoB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YACxC,OAAO,CACJ,MAAM,CAAC,IAAuC,EAAE,OAAO,IAAI,KAAK,CAAC,EAAE,CACrE,CAAC;QACJ,CAAC;QACD,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;YACpC,MAAM,SAAS,GAAG,4BAA4B,CAAC,WAAW,CAAC;YAC3D,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;gBAClC,KAAK;qBACF,QAAQ,EAAE;qBACV,eAAe,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;gBAC5D,OAAO;YACT,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,KAAK,CAAE,CAAC,aAAa,CACpD,SAAS,EACT,EAAC,WAAW,EAAE,OAAO,EAAE,KAAK,EAAC,EAC7B,EAAC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,sBAAsB,EAAC,CAC/C,CAAC;YACF,oBAAoB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC;QACD,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC7B,MAAM,SAAS,GAAG,4BAA4B,CAAC,WAAW,CAAC;YAC3D,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;gBAClC,KAAK,CAAC,QAAQ,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBACnE,OAAO;YACT,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,KAAK,CAAE,CAAC,aAAa,CACpD,SAAS,EACT,EAAC,WAAW,EAAE,OAAO,EAAC,EACtB,EAAC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,sBAAsB,EAAC,CAC/C,CAAC;YACF,oBAAoB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC1C,CAAC;QACD,QAAQ,EAAE,CAAC,OAAO,EAAE,EAAE,CACpB,KAAK;aACF,QAAQ,EAAE;aACV,eAAe,CAAC,YAAY,CAAC,WAAW,CAAC;YAC1C,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,KAAK,OAAO,CAAC;QAClD,aAAa,EAAE,CAAC,OAAO,EAAE,EAAE,CACzB,KAAK,CAAC,QAAQ,EAAE,CAAC,eAAe,CAAC,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC;KACvE,CAAC;AACJ,CAAC","sourcesContent":["import {\n  hasCommandSliceState,\n  type RoomCommandResult,\n} from '@sqlrooms/room-store';\nimport {DashboardAiAdapter} from './dashboard-types';\nimport {MosaicDashboardStoreState} from '../../dashboard/MosaicDashboardSlice';\nimport {MOSAIC_DASHBOARD_COMMAND_IDS} from '../../dashboard/MosaicDashboardCommands';\nimport {AiStore} from '../types';\n\nfunction getCommands(store: AiStore<MosaicDashboardStoreState>) {\n  const state = store.getState();\n  return hasCommandSliceState(state) ? state.commands : undefined;\n}\n\nfunction hasCommand(\n  store: AiStore<MosaicDashboardStoreState>,\n  commandId: string,\n) {\n  return Boolean(getCommands(store)?.getCommand(commandId));\n}\n\nfunction assertCommandSuccess(result: RoomCommandResult, commandId: string) {\n  if (!result.success) {\n    throw new Error(\n      result.error ?? result.message ?? `Command \"${commandId}\" failed.`,\n    );\n  }\n}\n\n/**\n * Creates a dashboard adapter for AI operations scoped to a specific dashboard.\n * The adapter provides methods for managing panels, tables, and dashboard state.\n *\n * @template TState - Store state type extending MosaicDashboardStoreState\n * @param store - Zustand store instance with dashboard management capabilities\n * @param dashboardId - ID of the dashboard to scope adapter operations to\n * @returns Dashboard adapter instance with panel and table management methods\n */\nexport function createDashboardAiAdapter<\n  TState extends MosaicDashboardStoreState,\n>(store: AiStore<TState>, dashboardId: string): DashboardAiAdapter {\n  return {\n    getSelectedTable: () =>\n      store.getState().mosaicDashboard.getDashboard(dashboardId)?.selectedTable,\n    getPanels: () =>\n      store.getState().mosaicDashboard.getDashboard(dashboardId)?.panels ?? [],\n    setSelectedTable: async (tableName) => {\n      const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.setSelectedTable;\n      if (!hasCommand(store, commandId)) {\n        store\n          .getState()\n          .mosaicDashboard.setSelectedTable(dashboardId, tableName);\n        return;\n      }\n      const result = await getCommands(store)!.invokeCommand(\n        commandId,\n        {dashboardId, tableName},\n        {surface: 'ai', actor: 'dashboard-ai-adapter'},\n      );\n      assertCommandSuccess(result, commandId);\n    },\n    addPanel: async (panel) => {\n      const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.addPanel;\n      if (!hasCommand(store, commandId)) {\n        return store.getState().mosaicDashboard.addPanel(dashboardId, panel);\n      }\n      const result = await getCommands(store)!.invokeCommand(\n        commandId,\n        {dashboardId, panel},\n        {surface: 'ai', actor: 'dashboard-ai-adapter'},\n      );\n      assertCommandSuccess(result, commandId);\n      return (\n        (result.data as {panelId?: string} | undefined)?.panelId ?? panel.id\n      );\n    },\n    updatePanel: async (panelId, patch) => {\n      const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.updatePanel;\n      if (!hasCommand(store, commandId)) {\n        store\n          .getState()\n          .mosaicDashboard.updatePanel(dashboardId, panelId, patch);\n        return;\n      }\n      const result = await getCommands(store)!.invokeCommand(\n        commandId,\n        {dashboardId, panelId, patch},\n        {surface: 'ai', actor: 'dashboard-ai-adapter'},\n      );\n      assertCommandSuccess(result, commandId);\n    },\n    removePanel: async (panelId) => {\n      const commandId = MOSAIC_DASHBOARD_COMMAND_IDS.removePanel;\n      if (!hasCommand(store, commandId)) {\n        store.getState().mosaicDashboard.removePanel(dashboardId, panelId);\n        return;\n      }\n      const result = await getCommands(store)!.invokeCommand(\n        commandId,\n        {dashboardId, panelId},\n        {surface: 'ai', actor: 'dashboard-ai-adapter'},\n      );\n      assertCommandSuccess(result, commandId);\n    },\n    getPanel: (panelId) =>\n      store\n        .getState()\n        .mosaicDashboard.getDashboard(dashboardId)\n        ?.panels.find((panel) => panel.id === panelId),\n    getPanelIssue: (panelId) =>\n      store.getState().mosaicDashboard.getPanelIssue(dashboardId, panelId),\n  };\n}\n"]}

package/dist/ai/dashboard/createDashboardAiTools.d.ts

@@ -0,0 +1,26 @@
+import type { Tool } from 'ai';
+import { DashboardAiAdapter, ExtraDashboardAiToolsFactory } from './dashboard-types';
+import { DatabaseAiAdapter } from '../database-types';
+import { ChartToolsOptions } from '../types';
+/**
+ * Options for creating dashboard AI tools.
+ *
+ * @property databaseAdapter - Adapter for database operations and table queries
+ * @property dashboardAdapter - Adapter for dashboard-specific operations like adding panels
+ * @property chartToolsOptions - Optional configuration for chart tool behavior
+ * @property extraTools - Optional factory function for registering additional custom tools
+ */
+export type CreateDashboardAiToolsOptions = {
+    databaseAdapter: DatabaseAiAdapter;
+    dashboardAdapter: DashboardAiAdapter;
+    chartToolsOptions?: ChartToolsOptions;
+    extraTools?: ExtraDashboardAiToolsFactory;
+};
+/**
+ * Creates the complete set of AI tools for dashboard operations, including chart tools and data table explorer.
+ *
+ * @param options - Configuration options for creating dashboard tools
+ * @returns Record of tool names to Tool instances, including built-in and any extra tools
+ */
+export declare function createDashboardAiTools({ dashboardAdapter, databaseAdapter, chartToolsOptions, extraTools, }: CreateDashboardAiToolsOptions): Record<string, Tool>;
+//# sourceMappingURL=createDashboardAiTools.d.ts.map

package/dist/ai/dashboard/createDashboardAiTools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardAiTools.d.ts","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardAiTools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,IAAI,EAAC,MAAM,IAAI,CAAC;AAG7B,OAAO,EACL,kBAAkB,EAClB,4BAA4B,EAC7B,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EAAC,iBAAiB,EAAC,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAC,iBAAiB,EAAC,MAAM,UAAU,CAAC;AAI3C;;;;;;;GAOG;AACH,MAAM,MAAM,6BAA6B,GAAG;IAC1C,eAAe,EAAE,iBAAiB,CAAC;IACnC,gBAAgB,EAAE,kBAAkB,CAAC;IACrC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,UAAU,CAAC,EAAE,4BAA4B,CAAC;CAC3C,CAAC;AAEF;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,EACrC,gBAAgB,EAChB,eAAe,EACf,iBAAiB,EACjB,UAAU,GACX,EAAE,6BAA6B,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAmCtD"}

package/dist/ai/dashboard/createDashboardAiTools.js

@@ -0,0 +1,40 @@
+import { ensureNoOverride } from '../tool-helpers';
+import { createDashboardChartTools } from './createDashboardChartTools';
+import { createDashboardDataTableExplorerTool } from './createDashboardDataTableExplorerTool';
+import { KnownDashboardTools } from './constants';
+import { createListDashboardPanelsTool } from './createListDashboardPanelsTool';
+/**
+ * Creates the complete set of AI tools for dashboard operations, including chart tools and data table explorer.
+ *
+ * @param options - Configuration options for creating dashboard tools
+ * @returns Record of tool names to Tool instances, including built-in and any extra tools
+ */
+export function createDashboardAiTools({ dashboardAdapter, databaseAdapter, chartToolsOptions, extraTools, }) {
+    const chartTools = createDashboardChartTools({
+        databaseAdapter,
+        dashboardAdapter,
+        chartToolsOptions,
+    });
+    const dataTableExplorerTool = createDashboardDataTableExplorerTool({
+        databaseAdapter,
+        dashboardAdapter,
+    });
+    const listDashboardPanelsTool = createListDashboardPanelsTool({
+        dashboardAdapter,
+    });
+    const builtInTools = {
+        [KnownDashboardTools.list_dashboard_panels]: listDashboardPanelsTool,
+        ...chartTools,
+        [KnownDashboardTools.create_dashboard_panel_data_table_explorer]: dataTableExplorerTool,
+    };
+    const additionalTools = extraTools?.({
+        databaseAdapter,
+        dashboardAdapter,
+    }) ?? {};
+    ensureNoOverride(builtInTools, additionalTools);
+    return {
+        ...builtInTools,
+        ...additionalTools,
+    };
+}
+//# sourceMappingURL=createDashboardAiTools.js.map

package/dist/ai/dashboard/createDashboardAiTools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardAiTools.js","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardAiTools.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,gBAAgB,EAAC,MAAM,iBAAiB,CAAC;AAMjD,OAAO,EAAC,yBAAyB,EAAC,MAAM,6BAA6B,CAAC;AACtE,OAAO,EAAC,oCAAoC,EAAC,MAAM,wCAAwC,CAAC;AAG5F,OAAO,EAAC,mBAAmB,EAAC,MAAM,aAAa,CAAC;AAChD,OAAO,EAAC,6BAA6B,EAAC,MAAM,iCAAiC,CAAC;AAiB9E;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CAAC,EACrC,gBAAgB,EAChB,eAAe,EACf,iBAAiB,EACjB,UAAU,GACoB;IAC9B,MAAM,UAAU,GAAG,yBAAyB,CAAC;QAC3C,eAAe;QACf,gBAAgB;QAChB,iBAAiB;KAClB,CAAC,CAAC;IAEH,MAAM,qBAAqB,GAAG,oCAAoC,CAAC;QACjE,eAAe;QACf,gBAAgB;KACjB,CAAC,CAAC;IAEH,MAAM,uBAAuB,GAAG,6BAA6B,CAAC;QAC5D,gBAAgB;KACjB,CAAC,CAAC;IAEH,MAAM,YAAY,GAAyB;QACzC,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,EAAE,uBAAuB;QACpE,GAAG,UAAU;QACb,CAAC,mBAAmB,CAAC,0CAA0C,CAAC,EAC9D,qBAAqB;KACxB,CAAC;IAEF,MAAM,eAAe,GACnB,UAAU,EAAE,CAAC;QACX,eAAe;QACf,gBAAgB;KACjB,CAAC,IAAI,EAAE,CAAC;IAEX,gBAAgB,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;IAEhD,OAAO;QACL,GAAG,YAAY;QACf,GAAG,eAAe;KACnB,CAAC;AACJ,CAAC","sourcesContent":["import type {Tool} from 'ai';\nimport {ensureNoOverride} from '../tool-helpers';\n\nimport {\n  DashboardAiAdapter,\n  ExtraDashboardAiToolsFactory,\n} from './dashboard-types';\nimport {createDashboardChartTools} from './createDashboardChartTools';\nimport {createDashboardDataTableExplorerTool} from './createDashboardDataTableExplorerTool';\nimport {DatabaseAiAdapter} from '../database-types';\nimport {ChartToolsOptions} from '../types';\nimport {KnownDashboardTools} from './constants';\nimport {createListDashboardPanelsTool} from './createListDashboardPanelsTool';\n\n/**\n * Options for creating dashboard AI tools.\n *\n * @property databaseAdapter - Adapter for database operations and table queries\n * @property dashboardAdapter - Adapter for dashboard-specific operations like adding panels\n * @property chartToolsOptions - Optional configuration for chart tool behavior\n * @property extraTools - Optional factory function for registering additional custom tools\n */\nexport type CreateDashboardAiToolsOptions = {\n  databaseAdapter: DatabaseAiAdapter;\n  dashboardAdapter: DashboardAiAdapter;\n  chartToolsOptions?: ChartToolsOptions;\n  extraTools?: ExtraDashboardAiToolsFactory;\n};\n\n/**\n * Creates the complete set of AI tools for dashboard operations, including chart tools and data table explorer.\n *\n * @param options - Configuration options for creating dashboard tools\n * @returns Record of tool names to Tool instances, including built-in and any extra tools\n */\nexport function createDashboardAiTools({\n  dashboardAdapter,\n  databaseAdapter,\n  chartToolsOptions,\n  extraTools,\n}: CreateDashboardAiToolsOptions): Record<string, Tool> {\n  const chartTools = createDashboardChartTools({\n    databaseAdapter,\n    dashboardAdapter,\n    chartToolsOptions,\n  });\n\n  const dataTableExplorerTool = createDashboardDataTableExplorerTool({\n    databaseAdapter,\n    dashboardAdapter,\n  });\n\n  const listDashboardPanelsTool = createListDashboardPanelsTool({\n    dashboardAdapter,\n  });\n\n  const builtInTools: Record<string, Tool> = {\n    [KnownDashboardTools.list_dashboard_panels]: listDashboardPanelsTool,\n    ...chartTools,\n    [KnownDashboardTools.create_dashboard_panel_data_table_explorer]:\n      dataTableExplorerTool,\n  };\n\n  const additionalTools =\n    extraTools?.({\n      databaseAdapter,\n      dashboardAdapter,\n    }) ?? {};\n\n  ensureNoOverride(builtInTools, additionalTools);\n\n  return {\n    ...builtInTools,\n    ...additionalTools,\n  };\n}\n"]}

package/dist/ai/dashboard/createDashboardChartTools.d.ts

@@ -0,0 +1,27 @@
+import type { Tool } from 'ai';
+import { DatabaseAiAdapter } from '../database-types';
+import { DashboardAiAdapter } from './dashboard-types';
+import { ChartToolsOptions } from '../types';
+/**
+ * Parameters for creating dashboard chart tools.
+ *
+ * @property databaseAdapter - Adapter for database operations and table queries
+ * @property dashboardAdapter - Adapter for adding panels to the dashboard
+ * @property chartToolsOptions - Optional configuration for chart types and data point limits
+ */
+export type CreateDashboardChartToolsParams = {
+    databaseAdapter: DatabaseAiAdapter;
+    dashboardAdapter: DashboardAiAdapter;
+    chartToolsOptions?: ChartToolsOptions;
+};
+/**
+ * Creates AI tools for generating chart panels in dashboards.
+ *
+ * @param params - Configuration parameters for chart tool creation
+ * @param params.databaseAdapter - Adapter for database operations
+ * @param params.dashboardAdapter - Adapter for dashboard panel management
+ * @param params.chartToolsOptions - Optional chart configuration options
+ * @returns Record of chart tool names to Tool instances, prefixed with dashboard-specific identifiers
+ */
+export declare function createDashboardChartTools({ databaseAdapter, dashboardAdapter, chartToolsOptions, }: CreateDashboardChartToolsParams): Record<string, Tool>;
+//# sourceMappingURL=createDashboardChartTools.d.ts.map

package/dist/ai/dashboard/createDashboardChartTools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createDashboardChartTools.d.ts","sourceRoot":"","sources":["../../../src/ai/dashboard/createDashboardChartTools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,IAAI,EAAC,MAAM,IAAI,CAAC;AAI7B,OAAO,EAAC,iBAAiB,EAAC,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAC,kBAAkB,EAAC,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAC,iBAAiB,EAAC,MAAM,UAAU,CAAC;AAO3C;;;;;;GAMG;AACH,MAAM,MAAM,+BAA+B,GAAG;IAC5C,eAAe,EAAE,iBAAiB,CAAC;IACnC,gBAAgB,EAAE,kBAAkB,CAAC;IACrC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACvC,CAAC;AAEF;;;;;;;;GAQG;AACH,wBAAgB,yBAAyB,CAAC,EACxC,eAAe,EACf,gBAAgB,EAChB,iBAAiB,GAClB,EAAE,+BAA+B,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAoDxD"}