@lightdash/common 0.2083.0 → 0.2084.0

package/dist/cjs/ee/AiAgent/schemas/tools/toolRunQueryArgs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolRunQueryArgs.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolRunQueryArgs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAyG3C,eAAO,MAAM,0BAA0B,2vDA0BtC,CAAC;AAEF,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAYvB,CAAC;AAEb,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAEtE,eAAO,MAAM,iCAAiC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EASvC,CAAC;AAER,MAAM,MAAM,2BAA2B,GAAG,CAAC,CAAC,KAAK,CAC7C,OAAO,iCAAiC,CAC3C,CAAC;AAEF,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;EAGnC,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC"}

package/dist/cjs/ee/AiAgent/schemas/tools/toolRunQueryArgs.js

@@ -0,0 +1,137 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.toolRunQueryOutputSchema = exports.toolRunQueryArgsSchemaTransformed = exports.toolRunQueryArgsSchema = exports.TOOL_RUN_QUERY_DESCRIPTION = void 0;
+const tslib_1 = require("tslib");
+const zod_1 = require("zod");
+const types_1 = require("../../types");
+const customMetrics_1 = require("../customMetrics");
+const fieldId_1 = require("../fieldId");
+const filters_1 = require("../filters");
+const outputMetadata_1 = require("../outputMetadata");
+const sortField_1 = tslib_1.__importDefault(require("../sortField"));
+const tableCalcs_1 = require("../tableCalcs/tableCalcs");
+const toolSchemaBuilder_1 = require("../toolSchemaBuilder");
+const visualizationMetadata_1 = tslib_1.__importDefault(require("../visualizationMetadata"));
+// Query configuration schema - what data to fetch
+const queryConfigSchema = zod_1.z.object({
+    exploreName: zod_1.z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for the chart.'),
+    dimensions: zod_1.z
+        .array((0, fieldId_1.getFieldIdSchema)({ additionalDescription: null }))
+        .describe('The field ids for the dimensions to group the metrics by. dimensions[0] is the primary grouping (x-axis for charts). dimensions[1+] create additional grouping levels.'),
+    metrics: zod_1.z
+        .array((0, fieldId_1.getFieldIdSchema)({ additionalDescription: null }))
+        .describe('The field ids of the metrics to be calculated. They will be grouped by the dimensions.'),
+    sorts: zod_1.z
+        .array(sortField_1.default)
+        .describe('Sort configuration for the query, it can use a combination of metrics and dimensions.'),
+    limit: zod_1.z
+        .number()
+        .nullable()
+        .describe('The total number of data points / rows allowed on the chart.'),
+});
+// Chart-specific configuration for rendering hints
+const chartConfigSchema = zod_1.z
+    .object({
+    defaultVizType: zod_1.z
+        .enum([
+        'table',
+        'bar',
+        'horizontal',
+        'line',
+        'scatter',
+        'pie',
+        'funnel',
+    ])
+        .describe('The default visualization type to render'),
+    // Series creation control
+    groupBy: zod_1.z
+        .array((0, fieldId_1.getFieldIdSchema)({ additionalDescription: null }))
+        .nullable()
+        .describe('Dimensions to split metrics into separate series (e.g., one line per region, one bar per status). IMPORTANT: Do NOT include the x-axis dimension in groupBy - only include dimensions you want to use for breaking down the data into multiple series. Example: dimensions=["order_date", "status"], groupBy=["status"] creates separate series for each status value. Leave null for simple single-series charts.'),
+    // Bar and horizontal bar chart specific
+    xAxisType: zod_1.z
+        .enum(['category', 'time'])
+        .nullable()
+        .describe('The x-axis type can be categorical for string value or time if the dimension is a date or timestamp. Applies to bar, horizontal, and scatter charts.'),
+    stackBars: zod_1.z
+        .boolean()
+        .nullable()
+        .describe('If groupBy is provided then this will stack the bars on top of each other instead of side by side. Applies to bar and horizontal charts.'),
+    // Line chart specific
+    lineType: zod_1.z
+        .enum(['line', 'area'])
+        .nullable()
+        .describe('default line. The type of line to display. If area then the area under the line will be filled in.'),
+    // Funnel chart specific
+    funnelDataInput: zod_1.z
+        .enum(['row', 'column'])
+        .nullable()
+        .describe('How to interpret funnel data. Use "row" when each row represents a funnel stage (most common). Use "column" when comparing multiple funnels side-by-side.'),
+    // Common display properties
+    xAxisLabel: zod_1.z
+        .string()
+        .nullable()
+        .describe('A helpful label to explain the x-axis'),
+    yAxisLabel: zod_1.z
+        .string()
+        .nullable()
+        .describe('A helpful label to explain the y-axis'),
+})
+    .nullable();
+exports.TOOL_RUN_QUERY_DESCRIPTION = `Tool: runQuery
+
+Purpose:
+Execute a metric query and create a chart artifact. The results can be viewed as a table, bar, horizontal bar, line, scatter, pie, or funnel chart.
+
+Chart Type Selection Guide:
+- 'bar': Vertical bars for categorical comparisons (e.g., sales by product)
+- 'horizontal': Horizontal bars for long category names or ranking (e.g., top 10 customers)
+- 'line': Time series trends (e.g., revenue over months)
+- 'scatter': Correlation between two metrics (e.g., ad spend vs revenue)
+- 'pie': Part-to-whole proportions (e.g., market share by segment)
+- 'funnel': Sequential conversion flows (e.g., sales funnel stages)
+- 'table': Raw data display with all fields
+
+Configuration Tips:
+- Specify exploreName, dimensions (for grouping/x-axis), and metrics (for y-axis values)
+- First dimension is the x-axis; additional dimensions can be used for series breakdown via groupBy
+- At least one metric is required for all chart types except table
+- chartConfig.groupBy: Use to split data into multiple series (e.g., one line per region). Do NOT include the x-axis dimension. Only include dimensions for series breakdown. Leave null for simple single-series charts.
+- For bar/horizontal charts: use xAxisType 'category' for strings or 'time' for dates/timestamps
+- For bar/horizontal charts: stackBars (when groupBy is provided) stacks bars instead of placing them side by side
+- For line charts: use lineType 'area' to fill the area under the line
+- For scatter charts: each point represents one row of data
+- For funnel charts: set funnelDataInput to 'row' (each row = stage) or 'column' (multiple funnels)
+- Users can switch between visualization types in the UI after creation
+- xAxisLabel and yAxisLabel provide helpful context for chart axes
+`;
+exports.toolRunQueryArgsSchema = (0, toolSchemaBuilder_1.createToolSchema)({
+    type: types_1.AiResultType.QUERY_RESULT,
+    description: exports.TOOL_RUN_QUERY_DESCRIPTION,
+})
+    .extend({
+    ...visualizationMetadata_1.default.shape,
+    queryConfig: queryConfigSchema,
+    chartConfig: chartConfigSchema,
+    customMetrics: customMetrics_1.customMetricsSchema,
+    tableCalculations: tableCalcs_1.tableCalcsSchema,
+    filters: filters_1.filtersSchemaV2.nullable(),
+})
+    .build();
+exports.toolRunQueryArgsSchemaTransformed = exports.toolRunQueryArgsSchema
+    .extend({
+    customMetrics: customMetrics_1.customMetricsSchema.default(null),
+    tableCalculations: tableCalcs_1.tableCalcsSchema.default(null),
+    chartConfig: chartConfigSchema.default(null),
+})
+    .transform((data) => ({
+    ...data,
+    filters: filters_1.filtersSchemaTransformed.parse(data.filters),
+}));
+exports.toolRunQueryOutputSchema = zod_1.z.object({
+    result: zod_1.z.string(),
+    metadata: outputMetadata_1.baseOutputMetadataSchema,
+});
+//# sourceMappingURL=toolRunQueryArgs.js.map

package/dist/cjs/ee/AiAgent/schemas/tools/toolRunQueryArgs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolRunQueryArgs.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolRunQueryArgs.ts"],"names":[],"mappings":";;;;AAAA,6BAAwB;AACxB,uCAA2C;AAC3C,oDAAuD;AACvD,wCAA8C;AAC9C,wCAAuE;AACvE,sDAA6D;AAC7D,qEAA2C;AAC3C,yDAA4D;AAC5D,4DAAwD;AACxD,6FAAmE;AAEnE,kDAAkD;AAClD,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC/B,WAAW,EAAE,OAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,mFAAmF,CACtF;IACL,UAAU,EAAE,OAAC;SACR,KAAK,CAAC,IAAA,0BAAgB,EAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,QAAQ,CACL,wKAAwK,CAC3K;IACL,OAAO,EAAE,OAAC;SACL,KAAK,CAAC,IAAA,0BAAgB,EAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,QAAQ,CACL,wFAAwF,CAC3F;IACL,KAAK,EAAE,OAAC;SACH,KAAK,CAAC,mBAAe,CAAC;SACtB,QAAQ,CACL,uFAAuF,CAC1F;IACL,KAAK,EAAE,OAAC;SACH,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACL,8DAA8D,CACjE;CACR,CAAC,CAAC;AAEH,mDAAmD;AACnD,MAAM,iBAAiB,GAAG,OAAC;KACtB,MAAM,CAAC;IACJ,cAAc,EAAE,OAAC;SACZ,IAAI,CAAC;QACF,OAAO;QACP,KAAK;QACL,YAAY;QACZ,MAAM;QACN,SAAS;QACT,KAAK;QACL,QAAQ;KACX,CAAC;SACD,QAAQ,CAAC,0CAA0C,CAAC;IAEzD,0BAA0B;IAC1B,OAAO,EAAE,OAAC;SACL,KAAK,CAAC,IAAA,0BAAgB,EAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,QAAQ,EAAE;SACV,QAAQ,CACL,oZAAoZ,CACvZ;IAEL,wCAAwC;IACxC,SAAS,EAAE,OAAC;SACP,IAAI,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;SAC1B,QAAQ,EAAE;SACV,QAAQ,CACL,sJAAsJ,CACzJ;IACL,SAAS,EAAE,OAAC;SACP,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,QAAQ,CACL,0IAA0I,CAC7I;IAEL,sBAAsB;IACtB,QAAQ,EAAE,OAAC;SACN,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;SACtB,QAAQ,EAAE;SACV,QAAQ,CACL,oGAAoG,CACvG;IAEL,wBAAwB;IACxB,eAAe,EAAE,OAAC;SACb,IAAI,CAAC,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;SACvB,QAAQ,EAAE;SACV,QAAQ,CACL,2JAA2J,CAC9J;IAEL,4BAA4B;IAC5B,UAAU,EAAE,OAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;IACtD,UAAU,EAAE,OAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;CACzD,CAAC;KACD,QAAQ,EAAE,CAAC;AAEH,QAAA,0BAA0B,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;CA0BzC,CAAC;AAEW,QAAA,sBAAsB,GAAG,IAAA,oCAAgB,EAAC;IACnD,IAAI,EAAE,oBAAY,CAAC,YAAY;IAC/B,WAAW,EAAE,kCAA0B;CAC1C,CAAC;KACG,MAAM,CAAC;IACJ,GAAG,+BAA2B,CAAC,KAAK;IACpC,WAAW,EAAE,iBAAiB;IAC9B,WAAW,EAAE,iBAAiB;IAC9B,aAAa,EAAE,mCAAmB;IAClC,iBAAiB,EAAE,6BAAgB;IACnC,OAAO,EAAE,yBAAe,CAAC,QAAQ,EAAE;CACtC,CAAC;KACD,KAAK,EAAE,CAAC;AAIA,QAAA,iCAAiC,GAAG,8BAAsB;KAClE,MAAM,CAAC;IACJ,aAAa,EAAE,mCAAmB,CAAC,OAAO,CAAC,IAAI,CAAC;IAChD,iBAAiB,EAAE,6BAAgB,CAAC,OAAO,CAAC,IAAI,CAAC;IACjD,WAAW,EAAE,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC;CAC/C,CAAC;KACD,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClB,GAAG,IAAI;IACP,OAAO,EAAE,kCAAwB,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;CACxD,CAAC,CAAC,CAAC;AAMK,QAAA,wBAAwB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7C,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE;IAClB,QAAQ,EAAE,yCAAwB;CACrC,CAAC,CAAC"}

package/dist/cjs/ee/AiAgent/schemas/visualizations/index.d.ts

@@ -2,11 +2,11 @@ import { z } from 'zod';
 export * from './tableViz';
 export * from './timeSeriesViz';
 export * from './verticalBarViz';
-export declare const ToolNameSchema: z.ZodEnum<["generateBarVizConfig", "generateTableVizConfig", "generateTimeSeriesVizConfig", "generateDashboard", "findExplores", "findFields", "searchFieldValues", "findDashboards", "findCharts", "improveContext", "proposeChange"]>;
+export declare const ToolNameSchema: z.ZodEnum<["generateBarVizConfig", "generateTableVizConfig", "generateTimeSeriesVizConfig", "generateDashboard", "findExplores", "findFields", "searchFieldValues", "findDashboards", "findCharts", "improveContext", "proposeChange", "runQuery"]>;
 export type ToolName = z.infer<typeof ToolNameSchema>;
 export declare const isToolName: (toolName: string) => toolName is ToolName;
-export declare const ToolDisplayMessagesSchema: z.ZodRecord<z.ZodEnum<["generateBarVizConfig", "generateTableVizConfig", "generateTimeSeriesVizConfig", "generateDashboard", "findExplores", "findFields", "searchFieldValues", "findDashboards", "findCharts", "improveContext", "proposeChange"]>, z.ZodString>;
-export declare const TOOL_DISPLAY_MESSAGES: Partial<Record<"generateBarVizConfig" | "generateTableVizConfig" | "generateTimeSeriesVizConfig" | "generateDashboard" | "findExplores" | "findFields" | "searchFieldValues" | "findDashboards" | "findCharts" | "improveContext" | "proposeChange", string>>;
-export declare const TOOL_DISPLAY_MESSAGES_AFTER_TOOL_CALL: Partial<Record<"generateBarVizConfig" | "generateTableVizConfig" | "generateTimeSeriesVizConfig" | "generateDashboard" | "findExplores" | "findFields" | "searchFieldValues" | "findDashboards" | "findCharts" | "improveContext" | "proposeChange", string>>;
+export declare const ToolDisplayMessagesSchema: z.ZodRecord<z.ZodEnum<["generateBarVizConfig", "generateTableVizConfig", "generateTimeSeriesVizConfig", "generateDashboard", "findExplores", "findFields", "searchFieldValues", "findDashboards", "findCharts", "improveContext", "proposeChange", "runQuery"]>, z.ZodString>;
+export declare const TOOL_DISPLAY_MESSAGES: Partial<Record<"generateBarVizConfig" | "generateTableVizConfig" | "generateTimeSeriesVizConfig" | "generateDashboard" | "findExplores" | "findFields" | "searchFieldValues" | "findDashboards" | "findCharts" | "improveContext" | "proposeChange" | "runQuery", string>>;
+export declare const TOOL_DISPLAY_MESSAGES_AFTER_TOOL_CALL: Partial<Record<"generateBarVizConfig" | "generateTableVizConfig" | "generateTimeSeriesVizConfig" | "generateDashboard" | "findExplores" | "findFields" | "searchFieldValues" | "findDashboards" | "findCharts" | "improveContext" | "proposeChange" | "runQuery", string>>;
 export declare const AVAILABLE_VISUALIZATION_TYPES: readonly ["generateBarVizConfig", "generateTableVizConfig", "generateTimeSeriesVizConfig"];
 //# sourceMappingURL=index.d.ts.map

package/dist/cjs/ee/AiAgent/schemas/visualizations/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,cAAc,YAAY,CAAC;AAC3B,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AASjC,eAAO,MAAM,cAAc,yOAUzB,CAAC;AAEH,MAAM,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAEtD,eAAO,MAAM,UAAU,aAAc,MAAM,KAAG,QAAQ,IAAI,QACZ,CAAC;AAG/C,eAAO,MAAM,yBAAyB,mQAAuC,CAAC;AAE9E,eAAO,MAAM,qBAAqB,+PAWhC,CAAC;AAGH,eAAO,MAAM,qCAAqC,+PAY5C,CAAC;AAEP,eAAO,MAAM,6BAA6B,4FAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,cAAc,YAAY,CAAC;AAC3B,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AASjC,eAAO,MAAM,cAAc,qPAWzB,CAAC;AAEH,MAAM,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAEtD,eAAO,MAAM,UAAU,aAAc,MAAM,KAAG,QAAQ,IAAI,QACZ,CAAC;AAG/C,eAAO,MAAM,yBAAyB,+QAAuC,CAAC;AAE9E,eAAO,MAAM,qBAAqB,4QAYhC,CAAC;AAGH,eAAO,MAAM,qCAAqC,4QAa5C,CAAC;AAEP,eAAO,MAAM,6BAA6B,4FAAqB,CAAC"}

package/dist/cjs/ee/AiAgent/schemas/visualizations/index.js

@@ -22,6 +22,7 @@ exports.ToolNameSchema = zod_1.z.enum([
     'findCharts',
     'improveContext',
     'proposeChange',
+    'runQuery',
 ]);
 const isToolName = (toolName) => exports.ToolNameSchema.safeParse(toolName).success;
 exports.isToolName = isToolName;
@@ -38,6 +39,7 @@ exports.TOOL_DISPLAY_MESSAGES = exports.ToolDisplayMessagesSchema.parse({
     generateDashboard: 'Generating a dashboard',
     findCharts: 'Finding relevant charts',
     improveContext: 'Improving context',
+    runQuery: 'Running query',
 });
 // after-tool-call messages
 exports.TOOL_DISPLAY_MESSAGES_AFTER_TOOL_CALL = exports.ToolDisplayMessagesSchema.parse({
@@ -51,6 +53,7 @@ exports.TOOL_DISPLAY_MESSAGES_AFTER_TOOL_CALL = exports.ToolDisplayMessagesSchem
     generateDashboard: 'Generated a dashboard',
     findCharts: 'Found relevant charts',
     improveContext: 'Improved context',
+    runQuery: 'Ran query',
 });
 exports.AVAILABLE_VISUALIZATION_TYPES = VisualizationTools;
 //# sourceMappingURL=index.js.map

package/dist/cjs/ee/AiAgent/schemas/visualizations/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/index.ts"],"names":[],"mappings":";;;;AAAA,6BAAwB;AAExB,qDAA2B;AAC3B,0DAAgC;AAChC,2DAAiC;AAEjC,MAAM,kBAAkB,GAAG;IACvB,sBAAsB;IACtB,wBAAwB;IACxB,6BAA6B;CACvB,CAAC;AAEX,oBAAoB;AACP,QAAA,cAAc,GAAG,OAAC,CAAC,IAAI,CAAC;IACjC,GAAG,kBAAkB;IACrB,mBAAmB;IACnB,cAAc;IACd,YAAY;IACZ,mBAAmB;IACnB,gBAAgB;IAChB,YAAY;IACZ,gBAAgB;IAChB,eAAe;CAClB,CAAC,CAAC;AAII,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAwB,EAAE,CACjE,sBAAc,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;AADlC,QAAA,UAAU,cACwB;AAE/C,0BAA0B;AACb,QAAA,yBAAyB,GAAG,OAAC,CAAC,MAAM,CAAC,sBAAc,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;AAEjE,QAAA,qBAAqB,GAAG,iCAAyB,CAAC,KAAK,CAAC;IACjE,YAAY,EAAE,2BAA2B;IACzC,cAAc,EAAE,6BAA6B;IAC7C,UAAU,EAAE,yBAAyB;IACrC,iBAAiB,EAAE,wBAAwB;IAC3C,oBAAoB,EAAE,wBAAwB;IAC9C,sBAAsB,EAAE,oBAAoB;IAC5C,2BAA2B,EAAE,yBAAyB;IACtD,iBAAiB,EAAE,wBAAwB;IAC3C,UAAU,EAAE,yBAAyB;IACrC,cAAc,EAAE,mBAAmB;CACtC,CAAC,CAAC;AAEH,2BAA2B;AACd,QAAA,qCAAqC,GAC9C,iCAAyB,CAAC,KAAK,CAAC;IAC5B,YAAY,EAAE,yBAAyB;IACvC,cAAc,EAAE,2BAA2B;IAC3C,UAAU,EAAE,uBAAuB;IACnC,iBAAiB,EAAE,oBAAoB;IACvC,oBAAoB,EAAE,uBAAuB;IAC7C,sBAAsB,EAAE,mBAAmB;IAC3C,2BAA2B,EAAE,wBAAwB;IACrD,iBAAiB,EAAE,uBAAuB;IAC1C,UAAU,EAAE,uBAAuB;IACnC,cAAc,EAAE,kBAAkB;CACrC,CAAC,CAAC;AAEM,QAAA,6BAA6B,GAAG,kBAAkB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/index.ts"],"names":[],"mappings":";;;;AAAA,6BAAwB;AAExB,qDAA2B;AAC3B,0DAAgC;AAChC,2DAAiC;AAEjC,MAAM,kBAAkB,GAAG;IACvB,sBAAsB;IACtB,wBAAwB;IACxB,6BAA6B;CACvB,CAAC;AAEX,oBAAoB;AACP,QAAA,cAAc,GAAG,OAAC,CAAC,IAAI,CAAC;IACjC,GAAG,kBAAkB;IACrB,mBAAmB;IACnB,cAAc;IACd,YAAY;IACZ,mBAAmB;IACnB,gBAAgB;IAChB,YAAY;IACZ,gBAAgB;IAChB,eAAe;IACf,UAAU;CACb,CAAC,CAAC;AAII,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAwB,EAAE,CACjE,sBAAc,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;AADlC,QAAA,UAAU,cACwB;AAE/C,0BAA0B;AACb,QAAA,yBAAyB,GAAG,OAAC,CAAC,MAAM,CAAC,sBAAc,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;AAEjE,QAAA,qBAAqB,GAAG,iCAAyB,CAAC,KAAK,CAAC;IACjE,YAAY,EAAE,2BAA2B;IACzC,cAAc,EAAE,6BAA6B;IAC7C,UAAU,EAAE,yBAAyB;IACrC,iBAAiB,EAAE,wBAAwB;IAC3C,oBAAoB,EAAE,wBAAwB;IAC9C,sBAAsB,EAAE,oBAAoB;IAC5C,2BAA2B,EAAE,yBAAyB;IACtD,iBAAiB,EAAE,wBAAwB;IAC3C,UAAU,EAAE,yBAAyB;IACrC,cAAc,EAAE,mBAAmB;IACnC,QAAQ,EAAE,eAAe;CAC5B,CAAC,CAAC;AAEH,2BAA2B;AACd,QAAA,qCAAqC,GAC9C,iCAAyB,CAAC,KAAK,CAAC;IAC5B,YAAY,EAAE,yBAAyB;IACvC,cAAc,EAAE,2BAA2B;IAC3C,UAAU,EAAE,uBAAuB;IACnC,iBAAiB,EAAE,oBAAoB;IACvC,oBAAoB,EAAE,uBAAuB;IAC7C,sBAAsB,EAAE,mBAAmB;IAC3C,2BAA2B,EAAE,wBAAwB;IACrD,iBAAiB,EAAE,uBAAuB;IAC1C,UAAU,EAAE,uBAAuB;IACnC,cAAc,EAAE,kBAAkB;IAClC,QAAQ,EAAE,WAAW;CACxB,CAAC,CAAC;AAEM,QAAA,6BAA6B,GAAG,kBAAkB,CAAC"}

package/dist/cjs/ee/AiAgent/types.d.ts

@@ -1,10 +1,11 @@
 import type { Filters } from '../../types/filter';
 import type { AdditionalMetric, MetricQuery } from '../../types/metricQuery';
-import type { ToolTableVizArgs, ToolTimeSeriesArgs, ToolVerticalBarArgs } from './schemas/tools';
+import type { ToolRunQueryArgs, ToolTableVizArgs, ToolTimeSeriesArgs, ToolVerticalBarArgs } from './schemas/tools';
 export declare enum AiResultType {
     TIME_SERIES_RESULT = "time_series_chart",
     VERTICAL_BAR_RESULT = "vertical_bar_chart",
     TABLE_RESULT = "table",
+    QUERY_RESULT = "query_result",
     DASHBOARD_RESULT = "dashboard",
     IMPROVE_CONTEXT = "improve_context",
     PROPOSE_CHANGE = "propose_change"
@@ -24,5 +25,8 @@ export type AiAgentVizConfig = {
 } | {
     type: 'table';
     config: ToolTableVizArgs;
+} | {
+    type: 'query_result';
+    config: ToolRunQueryArgs;
 };
 //# sourceMappingURL=types.d.ts.map

package/dist/cjs/ee/AiAgent/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/ee/AiAgent/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC7E,OAAO,KAAK,EACR,gBAAgB,EAChB,kBAAkB,EAClB,mBAAmB,EACtB,MAAM,iBAAiB,CAAC;AAEzB,oBAAY,YAAY;IACpB,kBAAkB,sBAAsB;IACxC,mBAAmB,uBAAuB;IAC1C,YAAY,UAAU;IACtB,gBAAgB,cAAc;IAC9B,eAAe,oBAAoB;IACnC,cAAc,mBAAmB;CACpC;AAED,MAAM,MAAM,aAAa,GAAG,IAAI,CAC5B,WAAW,EACT,SAAS,GACT,YAAY,GACZ,OAAO,GACP,OAAO,GACP,aAAa,GACb,mBAAmB,CACxB,GAAG;IACA,iBAAiB,EAAE,IAAI,CAAC,gBAAgB,EAAE,KAAK,CAAC,EAAE,CAAC;CACtD,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG,aAAa,GAAG;IACnD,OAAO,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,gBAAgB,GACtB;IACI,IAAI,EAAE,oBAAoB,CAAC;IAC3B,MAAM,EAAE,mBAAmB,CAAC;CAC/B,GACD;IACI,IAAI,EAAE,mBAAmB,CAAC;IAC1B,MAAM,EAAE,kBAAkB,CAAC;CAC9B,GACD;IACI,IAAI,EAAE,OAAO,CAAC;IACd,MAAM,EAAE,gBAAgB,CAAC;CAC5B,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/ee/AiAgent/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC7E,OAAO,KAAK,EACR,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAClB,mBAAmB,EACtB,MAAM,iBAAiB,CAAC;AAEzB,oBAAY,YAAY;IACpB,kBAAkB,sBAAsB;IACxC,mBAAmB,uBAAuB;IAC1C,YAAY,UAAU;IACtB,YAAY,iBAAiB;IAC7B,gBAAgB,cAAc;IAC9B,eAAe,oBAAoB;IACnC,cAAc,mBAAmB;CACpC;AAED,MAAM,MAAM,aAAa,GAAG,IAAI,CAC5B,WAAW,EACT,SAAS,GACT,YAAY,GACZ,OAAO,GACP,OAAO,GACP,aAAa,GACb,mBAAmB,CACxB,GAAG;IACA,iBAAiB,EAAE,IAAI,CAAC,gBAAgB,EAAE,KAAK,CAAC,EAAE,CAAC;CACtD,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG,aAAa,GAAG;IACnD,OAAO,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,gBAAgB,GACtB;IACI,IAAI,EAAE,oBAAoB,CAAC;IAC3B,MAAM,EAAE,mBAAmB,CAAC;CAC/B,GACD;IACI,IAAI,EAAE,mBAAmB,CAAC;IAC1B,MAAM,EAAE,kBAAkB,CAAC;CAC9B,GACD;IACI,IAAI,EAAE,OAAO,CAAC;IACd,MAAM,EAAE,gBAAgB,CAAC;CAC5B,GACD;IACI,IAAI,EAAE,cAAc,CAAC;IACrB,MAAM,EAAE,gBAAgB,CAAC;CAC5B,CAAC"}

package/dist/cjs/ee/AiAgent/types.js

@@ -6,6 +6,7 @@ var AiResultType;
     AiResultType["TIME_SERIES_RESULT"] = "time_series_chart";
     AiResultType["VERTICAL_BAR_RESULT"] = "vertical_bar_chart";
     AiResultType["TABLE_RESULT"] = "table";
+    AiResultType["QUERY_RESULT"] = "query_result";
     AiResultType["DASHBOARD_RESULT"] = "dashboard";
     AiResultType["IMPROVE_CONTEXT"] = "improve_context";
     AiResultType["PROPOSE_CHANGE"] = "propose_change";

package/dist/cjs/ee/AiAgent/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/types.ts"],"names":[],"mappings":";;;AAQA,IAAY,YAOX;AAPD,WAAY,YAAY;IACpB,wDAAwC,CAAA;IACxC,0DAA0C,CAAA;IAC1C,sCAAsB,CAAA;IACtB,8CAA8B,CAAA;IAC9B,mDAAmC,CAAA;IACnC,iDAAiC,CAAA;AACrC,CAAC,EAPW,YAAY,4BAAZ,YAAY,QAOvB"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/types.ts"],"names":[],"mappings":";;;AASA,IAAY,YAQX;AARD,WAAY,YAAY;IACpB,wDAAwC,CAAA;IACxC,0DAA0C,CAAA;IAC1C,sCAAsB,CAAA;IACtB,6CAA6B,CAAA;IAC7B,8CAA8B,CAAA;IAC9B,mDAAmC,CAAA;IACnC,iDAAiC,CAAA;AACrC,CAAC,EARW,YAAY,4BAAZ,YAAY,QAQvB"}

package/dist/cjs/ee/AiAgent/utils/chartConfigUtils.d.ts

@@ -0,0 +1,26 @@
+import type { ItemsMap } from '../../../types/field';
+import type { MetricQuery } from '../../../types/metricQuery';
+import type { ChartConfig, PivotReference } from '../../../types/savedCharts';
+import type { ToolRunQueryArgsTransformed } from '../schemas/tools/toolRunQueryArgs';
+export type ChartTypeOption = 'table' | 'bar' | 'horizontal' | 'line' | 'scatter' | 'pie' | 'funnel';
+/**
+ * Determines if a query can be rendered as a specific chart type
+ */
+export declare const canRenderAsChart: (chartType: ChartTypeOption, metricQuery: MetricQuery) => boolean;
+/**
+ * Gets all available chart types for a query
+ */
+export declare const getAvailableChartTypes: (metricQuery: MetricQuery) => ChartTypeOption[];
+export declare const formatFieldLabel: (fieldId: string, fieldsMap: ItemsMap) => string;
+export declare const formatPivotValueLabel: (pivotReference: PivotReference, fieldsMap: ItemsMap) => string;
+/**
+ * Converts runQuery tool result to echarts config
+ * This is the main function used for chart type switching
+ */
+export declare const getChartConfigFromRunQuery: ({ queryTool, metricQuery, fieldsMap, overrideChartType, }: {
+    queryTool: ToolRunQueryArgsTransformed;
+    metricQuery: MetricQuery;
+    fieldsMap: ItemsMap;
+    overrideChartType?: ChartTypeOption;
+}) => ChartConfig;
+//# sourceMappingURL=chartConfigUtils.d.ts.map

package/dist/cjs/ee/AiAgent/utils/chartConfigUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chartConfigUtils.d.ts","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/utils/chartConfigUtils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAErD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,KAAK,EAER,WAAW,EAGX,cAAc,EACjB,MAAM,4BAA4B,CAAC;AAQpC,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AASrF,MAAM,MAAM,eAAe,GACrB,OAAO,GACP,KAAK,GACL,YAAY,GACZ,MAAM,GACN,SAAS,GACT,KAAK,GACL,QAAQ,CAAC;AAEf;;GAEG;AACH,eAAO,MAAM,gBAAgB,cACd,eAAe,eACb,WAAW,KACzB,OAuBF,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,sBAAsB,gBAClB,WAAW,KACzB,eAAe,EAQjB,CAAC;AAEF,eAAO,MAAM,gBAAgB,YAChB,MAAM,aACJ,QAAQ,KACpB,MAMF,CAAC;AAEF,eAAO,MAAM,qBAAqB,mBACd,cAAc,aACnB,QAAQ,KACpB,MAcF,CAAC;AA4XF;;;GAGG;AACH,eAAO,MAAM,0BAA0B,8DAKpC;IACC,SAAS,EAAE,2BAA2B,CAAC;IACvC,WAAW,EAAE,WAAW,CAAC;IACzB,SAAS,EAAE,QAAQ,CAAC;IACpB,iBAAiB,CAAC,EAAE,eAAe,CAAC;CACvC,KAAG,WAsEH,CAAC"}