@lightdash/common 0.2104.7 → 0.2104.8

package/dist/types/ee/AiAgent/schemas/tools/toolSearchFieldValuesArgs.js

@@ -0,0 +1,46 @@
+import { z } from 'zod';
+import { getFieldIdSchema } from '../fieldId';
+import { filtersSchemaTransformed, filtersSchemaV2 } from '../filters';
+import { baseOutputMetadataSchema } from '../outputMetadata';
+import { createToolSchema } from '../toolSchemaBuilder';
+export const TOOL_SEARCH_FIELD_VALUES_DESCRIPTION = `Tool: searchFieldValues
+
+Purpose:
+Search for unique values of a specific field in a table. Returns all unique values by default, or use the query parameter to narrow down results. This is useful for finding suggestions for field values when building filters or exploring data.
+
+Usage Tips:
+- Specify the table and field you want to search values for
+- Query parameter: use it to narrow down the search to matching values
+- Optionally add filters to further restrict the results
+- Results are returned as a list of unique field values (limited to 100)
+`;
+export const toolSearchFieldValuesArgsSchema = createToolSchema({
+    type: 'search_field_values',
+    description: TOOL_SEARCH_FIELD_VALUES_DESCRIPTION,
+})
+    .extend({
+    table: z.string().describe('The table to search in.'),
+    fieldId: getFieldIdSchema({
+        additionalDescription: 'The ID of the field to search values for',
+    }),
+    query: z
+        .string()
+        .describe('Query string to filter field values')
+        .nullable(),
+    filters: filtersSchemaV2
+        .nullable()
+        .describe('Filters to apply to the query. Filtered fields must exist in the selected explore or should be referenced from the custom metrics.'),
+})
+    .build();
+export const toolSearchFieldValuesArgsSchemaTransformed = toolSearchFieldValuesArgsSchema.transform((data) => ({
+    ...data,
+    filters: data.filters
+        ? filtersSchemaTransformed.parse(data.filters)
+        : undefined,
+    query: data.query ?? '',
+}));
+export const toolSearchFieldValuesOutputSchema = z.object({
+    result: z.string(),
+    metadata: baseOutputMetadataSchema,
+});
+//# sourceMappingURL=toolSearchFieldValuesArgs.js.map

package/dist/types/ee/AiAgent/schemas/tools/toolSearchFieldValuesArgs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolSearchFieldValuesArgs.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolSearchFieldValuesArgs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,wBAAwB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD,MAAM,CAAC,MAAM,oCAAoC,GAAG;;;;;;;;;;CAUnD,CAAC;AAEF,MAAM,CAAC,MAAM,+BAA+B,GAAG,gBAAgB,CAAC;IAC5D,IAAI,EAAE,qBAAqB;IAC3B,WAAW,EAAE,oCAAoC;CACpD,CAAC;KACG,MAAM,CAAC;IACJ,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACrD,OAAO,EAAE,gBAAgB,CAAC;QACtB,qBAAqB,EAAE,0CAA0C;KACpE,CAAC;IACF,KAAK,EAAE,CAAC;SACH,MAAM,EAAE;SACR,QAAQ,CAAC,qCAAqC,CAAC;SAC/C,QAAQ,EAAE;IACf,OAAO,EAAE,eAAe;SACnB,QAAQ,EAAE;SACV,QAAQ,CACL,oIAAoI,CACvI;CACR,CAAC;KACD,KAAK,EAAE,CAAC;AAMb,MAAM,CAAC,MAAM,0CAA0C,GACnD,+BAA+B,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACjD,GAAG,IAAI;IACP,OAAO,EAAE,IAAI,CAAC,OAAO;QACjB,CAAC,CAAC,wBAAwB,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;QAC9C,CAAC,CAAC,SAAS;IACf,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;CAC1B,CAAC,CAAC,CAAC;AAMR,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,QAAQ,EAAE,wBAAwB;CACrC,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/tools/toolTableVizArgs.js

@@ -0,0 +1,53 @@
+import { z } from 'zod';
+import { LegacyFollowUpTools, legacyFollowUpToolsTransform, } from '../../followUpTools';
+import { AiResultType } from '../../types';
+import { customMetricsSchema } from '../customMetrics';
+import { filtersSchemaTransformed, filtersSchemaV2 } from '../filters';
+import { baseOutputMetadataSchema } from '../outputMetadata';
+import { tableCalcsSchema } from '../tableCalcs/tableCalcs';
+import { createToolSchema } from '../toolSchemaBuilder';
+import visualizationMetadataSchema from '../visualizationMetadata';
+import { tableVizConfigSchema } from '../visualizations';
+export const TOOL_TABLE_VIZ_DESCRIPTION = `Use this tool to query data to display in a table or summarized if limit is set to 1.`;
+export const toolTableVizArgsSchema = createToolSchema({
+    type: AiResultType.TABLE_RESULT,
+    description: TOOL_TABLE_VIZ_DESCRIPTION,
+})
+    .extend({
+    ...visualizationMetadataSchema.shape,
+    customMetrics: customMetricsSchema,
+    tableCalculations: tableCalcsSchema,
+    vizConfig: tableVizConfigSchema,
+    filters: filtersSchemaV2
+        .nullable()
+        .describe('Filters to apply to the query. Filtered fields must exist in the selected explore or should be referenced from the custom metrics.'),
+    followUpTools: z
+        .array(z.union([
+        z.literal(AiResultType.VERTICAL_BAR_RESULT),
+        z.literal(AiResultType.TIME_SERIES_RESULT),
+    ]))
+        .describe('The actions the User can ask for after the AI has generated the table.'),
+})
+    .build();
+export const toolTableVizArgsSchemaTransformed = toolTableVizArgsSchema
+    .extend({
+    // backwards compatibility for old viz configs without customMetrics
+    customMetrics: customMetricsSchema.default(null),
+    tableCalculations: tableCalcsSchema.default(null),
+    followUpTools: z.array(z.union([
+        z.literal(AiResultType.VERTICAL_BAR_RESULT),
+        z.literal(AiResultType.TIME_SERIES_RESULT),
+        z.literal(LegacyFollowUpTools.GENERATE_BAR_VIZ),
+        z.literal(LegacyFollowUpTools.GENERATE_TIME_SERIES_VIZ),
+    ])),
+    filters: filtersSchemaTransformed,
+})
+    .transform((data) => ({
+    ...data,
+    followUpTools: legacyFollowUpToolsTransform(data.followUpTools),
+}));
+export const toolTableVizOutputSchema = z.object({
+    result: z.string(),
+    metadata: baseOutputMetadataSchema,
+});
+//# sourceMappingURL=toolTableVizArgs.js.map

package/dist/types/ee/AiAgent/schemas/tools/toolTableVizArgs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolTableVizArgs.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolTableVizArgs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EACH,mBAAmB,EACnB,4BAA4B,GAC/B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,2BAA2B,MAAM,0BAA0B,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,CAAC,MAAM,0BAA0B,GAAG,uFAAuF,CAAC;AAElI,MAAM,CAAC,MAAM,sBAAsB,GAAG,gBAAgB,CAAC;IACnD,IAAI,EAAE,YAAY,CAAC,YAAY;IAC/B,WAAW,EAAE,0BAA0B;CAC1C,CAAC;KACG,MAAM,CAAC;IACJ,GAAG,2BAA2B,CAAC,KAAK;IACpC,aAAa,EAAE,mBAAmB;IAClC,iBAAiB,EAAE,gBAAgB;IACnC,SAAS,EAAE,oBAAoB;IAC/B,OAAO,EAAE,eAAe;SACnB,QAAQ,EAAE;SACV,QAAQ,CACL,oIAAoI,CACvI;IACL,aAAa,EAAE,CAAC;SACX,KAAK,CACF,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC;QAC3C,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,kBAAkB,CAAC;KAC7C,CAAC,CACL;SACA,QAAQ,CACL,wEAAwE,CAC3E;CACR,CAAC;KACD,KAAK,EAAE,CAAC;AAIb,MAAM,CAAC,MAAM,iCAAiC,GAAG,sBAAsB;KAClE,MAAM,CAAC;IACJ,oEAAoE;IACpE,aAAa,EAAE,mBAAmB,CAAC,OAAO,CAAC,IAAI,CAAC;IAChD,iBAAiB,EAAE,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC;IACjD,aAAa,EAAE,CAAC,CAAC,KAAK,CAClB,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC;QAC3C,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,kBAAkB,CAAC;QAC1C,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,gBAAgB,CAAC;QAC/C,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,wBAAwB,CAAC;KAC1D,CAAC,CACL;IACD,OAAO,EAAE,wBAAwB;CACpC,CAAC;KACD,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClB,GAAG,IAAI;IACP,aAAa,EAAE,4BAA4B,CAAC,IAAI,CAAC,aAAa,CAAC;CAClE,CAAC,CAAC,CAAC;AAMR,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,QAAQ,EAAE,wBAAwB;CACrC,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/tools/toolTimeSeriesArgs.js

@@ -0,0 +1,58 @@
+import { z } from 'zod';
+import { LegacyFollowUpTools, legacyFollowUpToolsTransform, } from '../../followUpTools';
+import { AiResultType } from '../../types';
+import { customMetricsSchema } from '../customMetrics';
+import { filtersSchemaTransformed, filtersSchemaV2 } from '../filters';
+import { baseOutputMetadataSchema } from '../outputMetadata';
+import { tableCalcsSchema } from '../tableCalcs/tableCalcs';
+import { createToolSchema } from '../toolSchemaBuilder';
+import visualizationMetadataSchema from '../visualizationMetadata';
+import { timeSeriesMetricVizConfigSchema } from '../visualizations/timeSeriesViz';
+export const TOOL_TIME_SERIES_VIZ_DESCRIPTION = `Use this tool to generate a Time Series Chart.`;
+export const toolTimeSeriesArgsSchema = createToolSchema({
+    type: AiResultType.TIME_SERIES_RESULT,
+    description: TOOL_TIME_SERIES_VIZ_DESCRIPTION,
+})
+    .extend({
+    ...visualizationMetadataSchema.shape,
+    customMetrics: customMetricsSchema,
+    tableCalculations: tableCalcsSchema,
+    vizConfig: timeSeriesMetricVizConfigSchema,
+    filters: filtersSchemaV2
+        .nullable()
+        .describe('Filters to apply to the query. Filtered fields must exist in the selected explore or should be referenced from the custom metrics.'),
+    followUpTools: z
+        .array(z.union([
+        z.literal(AiResultType.TABLE_RESULT),
+        z.literal(AiResultType.VERTICAL_BAR_RESULT),
+    ]))
+        .describe(`The actions the User can ask for after the AI has generated the chart`),
+})
+    .build();
+export const toolTimeSeriesArgsSchemaTransformed = toolTimeSeriesArgsSchema
+    .extend({
+    // backwards compatibility for old viz configs
+    vizConfig: timeSeriesMetricVizConfigSchema.extend({
+        xAxisLabel: z.string().default(''),
+        yAxisLabel: z.string().default(''),
+    }),
+    // backwards compatibility for old viz configs without customMetrics
+    customMetrics: customMetricsSchema.default(null),
+    tableCalculations: tableCalcsSchema.default(null),
+    followUpTools: z.array(z.union([
+        z.literal(AiResultType.TABLE_RESULT),
+        z.literal(AiResultType.VERTICAL_BAR_RESULT),
+        z.literal(LegacyFollowUpTools.GENERATE_TABLE),
+        z.literal(LegacyFollowUpTools.GENERATE_BAR_VIZ),
+    ])),
+    filters: filtersSchemaTransformed,
+})
+    .transform((data) => ({
+    ...data,
+    followUpTools: legacyFollowUpToolsTransform(data.followUpTools),
+}));
+export const toolTimeSeriesOutputSchema = z.object({
+    result: z.string(),
+    metadata: baseOutputMetadataSchema,
+});
+//# sourceMappingURL=toolTimeSeriesArgs.js.map

package/dist/types/ee/AiAgent/schemas/tools/toolTimeSeriesArgs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolTimeSeriesArgs.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolTimeSeriesArgs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EACH,mBAAmB,EACnB,4BAA4B,GAC/B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,2BAA2B,MAAM,0BAA0B,CAAC;AACnE,OAAO,EAAE,+BAA+B,EAAE,MAAM,iCAAiC,CAAC;AAElF,MAAM,CAAC,MAAM,gCAAgC,GAAG,gDAAgD,CAAC;AAEjG,MAAM,CAAC,MAAM,wBAAwB,GAAG,gBAAgB,CAAC;IACrD,IAAI,EAAE,YAAY,CAAC,kBAAkB;IACrC,WAAW,EAAE,gCAAgC;CAChD,CAAC;KACG,MAAM,CAAC;IACJ,GAAG,2BAA2B,CAAC,KAAK;IACpC,aAAa,EAAE,mBAAmB;IAClC,iBAAiB,EAAE,gBAAgB;IACnC,SAAS,EAAE,+BAA+B;IAC1C,OAAO,EAAE,eAAe;SACnB,QAAQ,EAAE;SACV,QAAQ,CACL,oIAAoI,CACvI;IACL,aAAa,EAAE,CAAC;SACX,KAAK,CACF,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,CAAC;QACpC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC;KAC9C,CAAC,CACL;SACA,QAAQ,CACL,uEAAuE,CAC1E;CACR,CAAC;KACD,KAAK,EAAE,CAAC;AAIb,MAAM,CAAC,MAAM,mCAAmC,GAAG,wBAAwB;KACtE,MAAM,CAAC;IACJ,8CAA8C;IAC9C,SAAS,EAAE,+BAA+B,CAAC,MAAM,CAAC;QAC9C,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;QAClC,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;KACrC,CAAC;IACF,oEAAoE;IACpE,aAAa,EAAE,mBAAmB,CAAC,OAAO,CAAC,IAAI,CAAC;IAChD,iBAAiB,EAAE,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC;IACjD,aAAa,EAAE,CAAC,CAAC,KAAK,CAClB,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,CAAC;QACpC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC;QAC3C,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,cAAc,CAAC;QAC7C,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,gBAAgB,CAAC;KAClD,CAAC,CACL;IACD,OAAO,EAAE,wBAAwB;CACpC,CAAC;KACD,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClB,GAAG,IAAI;IACP,aAAa,EAAE,4BAA4B,CAAC,IAAI,CAAC,aAAa,CAAC;CAClE,CAAC,CAAC,CAAC;AAMR,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,QAAQ,EAAE,wBAAwB;CACrC,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/tools/toolVerticalBarArgs.js

@@ -0,0 +1,53 @@
+import { z } from 'zod';
+import { LegacyFollowUpTools, legacyFollowUpToolsTransform, } from '../../followUpTools';
+import { AiResultType } from '../../types';
+import { customMetricsSchema } from '../customMetrics';
+import { filtersSchemaTransformed, filtersSchemaV2 } from '../filters';
+import { baseOutputMetadataSchema } from '../outputMetadata';
+import { tableCalcsSchema } from '../tableCalcs/tableCalcs';
+import { createToolSchema } from '../toolSchemaBuilder';
+import visualizationMetadataSchema from '../visualizationMetadata';
+import { verticalBarMetricVizConfigSchema } from '../visualizations';
+export const TOOL_VERTICAL_BAR_VIZ_DESCRIPTION = `Use this tool to generate a Bar Chart Visualization.`;
+export const toolVerticalBarArgsSchema = createToolSchema({
+    type: AiResultType.VERTICAL_BAR_RESULT,
+    description: TOOL_VERTICAL_BAR_VIZ_DESCRIPTION,
+})
+    .extend({
+    ...visualizationMetadataSchema.shape,
+    customMetrics: customMetricsSchema,
+    tableCalculations: tableCalcsSchema,
+    vizConfig: verticalBarMetricVizConfigSchema,
+    filters: filtersSchemaV2
+        .nullable()
+        .describe('Filters to apply to the query. Filtered fields must exist in the selected explore or should be referenced from the custom metrics.'),
+    followUpTools: z
+        .array(z.union([
+        z.literal(AiResultType.TABLE_RESULT),
+        z.literal(AiResultType.TIME_SERIES_RESULT),
+    ]))
+        .describe(`The actions the User can ask for after the AI has generated the chart`),
+})
+    .build();
+export const toolVerticalBarArgsSchemaTransformed = toolVerticalBarArgsSchema
+    .extend({
+    // backwards compatibility for old viz configs without customMetrics
+    customMetrics: customMetricsSchema.default(null),
+    tableCalculations: tableCalcsSchema.default(null),
+    followUpTools: z.array(z.union([
+        z.literal(AiResultType.TABLE_RESULT),
+        z.literal(AiResultType.TIME_SERIES_RESULT),
+        z.literal(LegacyFollowUpTools.GENERATE_TABLE),
+        z.literal(LegacyFollowUpTools.GENERATE_TIME_SERIES_VIZ),
+    ])),
+    filters: filtersSchemaTransformed,
+})
+    .transform((data) => ({
+    ...data,
+    followUpTools: legacyFollowUpToolsTransform(data.followUpTools),
+}));
+export const toolVerticalBarOutputSchema = z.object({
+    result: z.string(),
+    metadata: baseOutputMetadataSchema,
+});
+//# sourceMappingURL=toolVerticalBarArgs.js.map

package/dist/types/ee/AiAgent/schemas/tools/toolVerticalBarArgs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolVerticalBarArgs.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolVerticalBarArgs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EACH,mBAAmB,EACnB,4BAA4B,GAC/B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,2BAA2B,MAAM,0BAA0B,CAAC;AACnE,OAAO,EAAE,gCAAgC,EAAE,MAAM,mBAAmB,CAAC;AAErE,MAAM,CAAC,MAAM,iCAAiC,GAAG,sDAAsD,CAAC;AAExG,MAAM,CAAC,MAAM,yBAAyB,GAAG,gBAAgB,CAAC;IACtD,IAAI,EAAE,YAAY,CAAC,mBAAmB;IACtC,WAAW,EAAE,iCAAiC;CACjD,CAAC;KACG,MAAM,CAAC;IACJ,GAAG,2BAA2B,CAAC,KAAK;IACpC,aAAa,EAAE,mBAAmB;IAClC,iBAAiB,EAAE,gBAAgB;IACnC,SAAS,EAAE,gCAAgC;IAC3C,OAAO,EAAE,eAAe;SACnB,QAAQ,EAAE;SACV,QAAQ,CACL,oIAAoI,CACvI;IACL,aAAa,EAAE,CAAC;SACX,KAAK,CACF,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,CAAC;QACpC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,kBAAkB,CAAC;KAC7C,CAAC,CACL;SACA,QAAQ,CACL,uEAAuE,CAC1E;CACR,CAAC;KACD,KAAK,EAAE,CAAC;AAIb,MAAM,CAAC,MAAM,oCAAoC,GAAG,yBAAyB;KACxE,MAAM,CAAC;IACJ,oEAAoE;IACpE,aAAa,EAAE,mBAAmB,CAAC,OAAO,CAAC,IAAI,CAAC;IAChD,iBAAiB,EAAE,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC;IACjD,aAAa,EAAE,CAAC,CAAC,KAAK,CAClB,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,YAAY,CAAC;QACpC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,kBAAkB,CAAC;QAC1C,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,cAAc,CAAC;QAC7C,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,wBAAwB,CAAC;KAC1D,CAAC,CACL;IACD,OAAO,EAAE,wBAAwB;CACpC,CAAC;KACD,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClB,GAAG,IAAI;IACP,aAAa,EAAE,4BAA4B,CAAC,IAAI,CAAC,aAAa,CAAC;CAClE,CAAC,CAAC,CAAC;AAMR,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,QAAQ,EAAE,wBAAwB;CACrC,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/visualizationMetadata.js

@@ -0,0 +1,9 @@
+import { z } from 'zod';
+const visualizationMetadataSchema = z.object({
+    title: z.string().describe('A descriptive title for the chart'),
+    description: z
+        .string()
+        .describe('A descriptive summary or explanation for the chart.'),
+});
+export default visualizationMetadataSchema;
+//# sourceMappingURL=visualizationMetadata.js.map

package/dist/types/ee/AiAgent/schemas/visualizationMetadata.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"visualizationMetadata.js","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/schemas/visualizationMetadata.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IAC/D,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CAAC,qDAAqD,CAAC;CACvE,CAAC,CAAC;AAEH,eAAe,2BAA2B,CAAC"}

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

@@ -0,0 +1,58 @@
+import { z } from 'zod';
+export * from './tableViz';
+export * from './timeSeriesViz';
+export * from './verticalBarViz';
+const VisualizationTools = [
+    'generateBarVizConfig',
+    'generateTableVizConfig',
+    'generateTimeSeriesVizConfig',
+];
+// define tool names
+export const ToolNameSchema = z.enum([
+    ...VisualizationTools,
+    'generateDashboard',
+    'generateDashboardV2',
+    'findContent',
+    'findExplores',
+    'findFields',
+    'searchFieldValues',
+    'findDashboards',
+    'findCharts',
+    'improveContext',
+    'proposeChange',
+    'runQuery',
+]);
+export const isToolName = (toolName) => ToolNameSchema.safeParse(toolName).success;
+// display messages schema
+export const ToolDisplayMessagesSchema = z.record(ToolNameSchema, z.string());
+export const TOOL_DISPLAY_MESSAGES = ToolDisplayMessagesSchema.parse({
+    findExplores: 'Finding relevant explores',
+    findDashboards: 'Finding relevant dashboards',
+    findFields: 'Finding relevant fields',
+    searchFieldValues: 'Searching field values',
+    generateBarVizConfig: 'Generating a bar chart',
+    generateTableVizConfig: 'Generating a table',
+    generateTimeSeriesVizConfig: 'Generating a line chart',
+    generateDashboard: 'Generating a dashboard',
+    generateDashboardV2: 'Generating a dashboard',
+    findCharts: 'Finding relevant charts',
+    improveContext: 'Improving context',
+    runQuery: 'Running query',
+});
+// after-tool-call messages
+export const TOOL_DISPLAY_MESSAGES_AFTER_TOOL_CALL = ToolDisplayMessagesSchema.parse({
+    findExplores: 'Found relevant explores',
+    findDashboards: 'Found relevant dashboards',
+    findFields: 'Found relevant fields',
+    searchFieldValues: 'Found field values',
+    generateBarVizConfig: 'Generated a bar chart',
+    generateTableVizConfig: 'Generated a table',
+    generateTimeSeriesVizConfig: 'Generated a line chart',
+    generateDashboard: 'Generated a dashboard',
+    generateDashboardV2: 'Generated a dashboard',
+    findCharts: 'Found relevant charts',
+    improveContext: 'Improved context',
+    runQuery: 'Ran query',
+});
+export const AVAILABLE_VISUALIZATION_TYPES = VisualizationTools;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","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;AAEjC,MAAM,kBAAkB,GAAG;IACvB,sBAAsB;IACtB,wBAAwB;IACxB,6BAA6B;CACvB,CAAC;AAEX,oBAAoB;AACpB,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,IAAI,CAAC;IACjC,GAAG,kBAAkB;IACrB,mBAAmB;IACnB,qBAAqB;IACrB,aAAa;IACb,cAAc;IACd,YAAY;IACZ,mBAAmB;IACnB,gBAAgB;IAChB,YAAY;IACZ,gBAAgB;IAChB,eAAe;IACf,UAAU;CACb,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAwB,EAAE,CACjE,cAAc,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;AAE/C,0BAA0B;AAC1B,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;AAE9E,MAAM,CAAC,MAAM,qBAAqB,GAAG,yBAAyB,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,mBAAmB,EAAE,wBAAwB;IAC7C,UAAU,EAAE,yBAAyB;IACrC,cAAc,EAAE,mBAAmB;IACnC,QAAQ,EAAE,eAAe;CAC5B,CAAC,CAAC;AAEH,2BAA2B;AAC3B,MAAM,CAAC,MAAM,qCAAqC,GAC9C,yBAAyB,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,mBAAmB,EAAE,uBAAuB;IAC5C,UAAU,EAAE,uBAAuB;IACnC,cAAc,EAAE,kBAAkB;IAClC,QAAQ,EAAE,WAAW;CACxB,CAAC,CAAC;AAEP,MAAM,CAAC,MAAM,6BAA6B,GAAG,kBAAkB,CAAC"}

package/dist/types/ee/AiAgent/schemas/visualizations/tableViz.js

@@ -0,0 +1,38 @@
+import { z } from 'zod';
+import { getValidAiQueryLimit } from '../../validators';
+import { getFieldIdSchema } from '../fieldId';
+import sortFieldSchema from '../sortField';
+export const tableVizConfigSchema = z
+    .object({
+    exploreName: z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for table query'),
+    metrics: z
+        .array(getFieldIdSchema({ additionalDescription: null }))
+        .describe('The field ids of the metrics to be calculated for the table. They will be grouped by the dimensions.'),
+    dimensions: z
+        .array(getFieldIdSchema({ additionalDescription: null }))
+        .describe('The field id for the dimensions to group the metrics by'),
+    sorts: z
+        .array(sortFieldSchema)
+        .describe('Sort configuration for the query, it can use a combination of metrics and dimensions.'),
+    limit: z
+        .number()
+        .nullable()
+        .describe('The maximum number of rows in the table.'),
+})
+    .describe('Configuration file for generating a table from a query with metrics and dimensions');
+export const metricQueryTableViz = ({ vizConfig, filters, maxLimit, customMetrics, tableCalculations, }) => ({
+    exploreName: vizConfig.exploreName,
+    metrics: vizConfig.metrics,
+    dimensions: vizConfig.dimensions || [],
+    sorts: vizConfig.sorts.map((sort) => ({
+        ...sort,
+        nullsFirst: sort.nullsFirst ?? undefined,
+    })),
+    limit: getValidAiQueryLimit(vizConfig.limit, maxLimit),
+    filters,
+    additionalMetrics: customMetrics ?? [],
+    tableCalculations,
+});
+//# sourceMappingURL=tableViz.js.map

package/dist/types/ee/AiAgent/schemas/visualizations/tableViz.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tableViz.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/tableViz.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,eAAe,MAAM,cAAc,CAAC;AAG3C,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC;KAChC,MAAM,CAAC;IACJ,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,oFAAoF,CACvF;IACL,OAAO,EAAE,CAAC;SACL,KAAK,CAAC,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,QAAQ,CACL,sGAAsG,CACzG;IACL,UAAU,EAAE,CAAC;SACR,KAAK,CAAC,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,QAAQ,CACL,yDAAyD,CAC5D;IACL,KAAK,EAAE,CAAC;SACH,KAAK,CAAC,eAAe,CAAC;SACtB,QAAQ,CACL,uFAAuF,CAC1F;IAEL,KAAK,EAAE,CAAC;SACH,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,0CAA0C,CAAC;CAC5D,CAAC;KACD,QAAQ,CACL,oFAAoF,CACvF,CAAC;AAIN,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,EAChC,SAAS,EACT,OAAO,EACP,QAAQ,EACR,aAAa,EACb,iBAAiB,GAOpB,EAA4B,EAAE,CAAC,CAAC;IAC7B,WAAW,EAAE,SAAS,CAAC,WAAW;IAClC,OAAO,EAAE,SAAS,CAAC,OAAO;IAC1B,UAAU,EAAE,SAAS,CAAC,UAAU,IAAI,EAAE;IACtC,KAAK,EAAE,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAClC,GAAG,IAAI;QACP,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,SAAS;KAC3C,CAAC,CAAC;IACH,KAAK,EAAE,oBAAoB,CAAC,SAAS,CAAC,KAAK,EAAE,QAAQ,CAAC;IACtD,OAAO;IACP,iBAAiB,EAAE,aAAa,IAAI,EAAE;IACtC,iBAAiB;CACpB,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/visualizations/timeSeriesViz.js

@@ -0,0 +1,63 @@
+import { z } from 'zod';
+import { AI_DEFAULT_MAX_QUERY_LIMIT } from '../../constants';
+import { getValidAiQueryLimit } from '../../validators';
+import { getFieldIdSchema } from '../fieldId';
+import sortFieldSchema from '../sortField';
+export const timeSeriesMetricVizConfigSchema = z.object({
+    exploreName: z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for the chart.'),
+    xDimension: getFieldIdSchema({
+        additionalDescription: 'The field id of the time dimension to be displayed on the x-axis.',
+    }),
+    yMetrics: z
+        .array(getFieldIdSchema({ additionalDescription: null }))
+        .min(1)
+        .describe('At least one metric is required. The field ids of the metrics to be displayed on the y-axis. If there are multiple metrics there will be one line per metric'),
+    sorts: z
+        .array(sortFieldSchema)
+        .describe('Sort configuration for the query, it can use a combination of metrics and dimensions.'),
+    breakdownByDimension: getFieldIdSchema({
+        additionalDescription: 'The field id of the dimension used to split the metrics into series for each dimension value. For example if you wanted to split a metric into multiple series based on City you would use the City dimension field id here. If this is not provided then the metric will be displayed as a single series.',
+    }).nullable(),
+    lineType: z
+        .union([z.literal('line'), z.literal('area')])
+        .describe('default line. The type of line to display. If area then the area under the line will be filled in.'),
+    limit: z
+        .number()
+        .max(AI_DEFAULT_MAX_QUERY_LIMIT)
+        .nullable()
+        .describe(`The total number of data points allowed on the chart.`),
+    xAxisLabel: z
+        .string()
+        .nullable()
+        .describe('A helpful label to explain the x-axis'),
+    yAxisLabel: z
+        .string()
+        .nullable()
+        .describe('A helpful label to explain the y-axis'),
+});
+export const metricQueryTimeSeriesViz = ({ vizConfig, filters, maxLimit, customMetrics, tableCalculations, }) => {
+    const metrics = vizConfig.yMetrics;
+    const dimensions = [
+        vizConfig.xDimension,
+        ...(vizConfig.breakdownByDimension
+            ? [vizConfig.breakdownByDimension]
+            : []),
+    ];
+    const { limit, sorts } = vizConfig;
+    return {
+        metrics,
+        dimensions,
+        limit: getValidAiQueryLimit(limit, maxLimit),
+        sorts: sorts.map((sort) => ({
+            ...sort,
+            nullsFirst: sort.nullsFirst ?? undefined,
+        })),
+        exploreName: vizConfig.exploreName,
+        filters,
+        additionalMetrics: customMetrics ?? [],
+        tableCalculations,
+    };
+};
+//# sourceMappingURL=timeSeriesViz.js.map

package/dist/types/ee/AiAgent/schemas/visualizations/timeSeriesViz.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeSeriesViz.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/timeSeriesViz.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAE7D,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,eAAe,MAAM,cAAc,CAAC;AAG3C,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACpD,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,mFAAmF,CACtF;IACL,UAAU,EAAE,gBAAgB,CAAC;QACzB,qBAAqB,EACjB,mEAAmE;KAC1E,CAAC;IACF,QAAQ,EAAE,CAAC;SACN,KAAK,CAAC,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACL,8JAA8J,CACjK;IACL,KAAK,EAAE,CAAC;SACH,KAAK,CAAC,eAAe,CAAC;SACtB,QAAQ,CACL,uFAAuF,CAC1F;IACL,oBAAoB,EAAE,gBAAgB,CAAC;QACnC,qBAAqB,EACjB,4SAA4S;KACnT,CAAC,CAAC,QAAQ,EAAE;IACb,QAAQ,EAAE,CAAC;SACN,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;SAC7C,QAAQ,CACL,oGAAoG,CACvG;IACL,KAAK,EAAE,CAAC;SACH,MAAM,EAAE;SACR,GAAG,CAAC,0BAA0B,CAAC;SAC/B,QAAQ,EAAE;SACV,QAAQ,CAAC,uDAAuD,CAAC;IACtE,UAAU,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;IACtD,UAAU,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;CACzD,CAAC,CAAC;AAMH,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,EACrC,SAAS,EACT,OAAO,EACP,QAAQ,EACR,aAAa,EACb,iBAAiB,GAOpB,EAA4B,EAAE;IAC3B,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,CAAC;IACnC,MAAM,UAAU,GAAG;QACf,SAAS,CAAC,UAAU;QACpB,GAAG,CAAC,SAAS,CAAC,oBAAoB;YAC9B,CAAC,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC;YAClC,CAAC,CAAC,EAAE,CAAC;KACZ,CAAC;IACF,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;IAEnC,OAAO;QACH,OAAO;QACP,UAAU;QACV,KAAK,EAAE,oBAAoB,CAAC,KAAK,EAAE,QAAQ,CAAC;QAC5C,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACxB,GAAG,IAAI;YACP,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,SAAS;SAC3C,CAAC,CAAC;QACH,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,OAAO;QACP,iBAAiB,EAAE,aAAa,IAAI,EAAE;QACtC,iBAAiB;KACpB,CAAC;AACN,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/visualizations/verticalBarViz.js

@@ -0,0 +1,68 @@
+import { z } from 'zod';
+import { AI_DEFAULT_MAX_QUERY_LIMIT } from '../../constants';
+import { getValidAiQueryLimit } from '../../validators';
+import { getFieldIdSchema } from '../fieldId';
+import sortFieldSchema from '../sortField';
+export const verticalBarMetricVizConfigSchema = z.object({
+    exploreName: z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for the chart.'),
+    xDimension: getFieldIdSchema({
+        additionalDescription: 'The field id of the dimension to be displayed on the x-axis.',
+    }),
+    yMetrics: z
+        .array(getFieldIdSchema({ additionalDescription: null }))
+        .min(1)
+        .describe('The field ids of the metrics to be displayed on the y-axis. The height of the bars'),
+    sorts: z
+        .array(sortFieldSchema)
+        .describe('Sort configuration for the query, it can use a combination of metrics and dimensions.'),
+    limit: z
+        .number()
+        .max(AI_DEFAULT_MAX_QUERY_LIMIT)
+        .nullable()
+        .describe(`The total number of data points / bars allowed on the chart.`),
+    breakdownByDimension: getFieldIdSchema({
+        additionalDescription: 'The field id of the dimension used to split the metrics into groups along the x-axis. If stacking is false then this will create multiple bars around each x value, if stacking is true then this will create multiple bars for each metric stacked on top of each other',
+    }).nullable(),
+    stackBars: z
+        .boolean()
+        .nullable()
+        .describe('If using breakdownByDimension then this will stack the bars on top of each other instead of side by side.'),
+    xAxisType: z
+        .union([z.literal('category'), z.literal('time')])
+        .describe('The x-axis type can be categorical for string value or time if the dimension is a date or timestamp.'),
+    xAxisLabel: z
+        .string()
+        .nullable()
+        .describe('A helpful label to explain the x-axis'),
+    yAxisLabel: z
+        .string()
+        .nullable()
+        .describe('A helpful label to explain the y-axis'),
+});
+export const metricQueryVerticalBarViz = ({ vizConfig, filters, maxLimit, customMetrics, tableCalculations, }) => {
+    const metrics = vizConfig.yMetrics;
+    const dimensions = [
+        vizConfig.xDimension,
+        ...(vizConfig.breakdownByDimension
+            ? [vizConfig.breakdownByDimension]
+            : []),
+    ];
+    const { limit, sorts } = vizConfig;
+    return {
+        metrics,
+        dimensions,
+        limit: getValidAiQueryLimit(limit, maxLimit),
+        sorts: sorts.map((sort) => ({
+            ...sort,
+            nullsFirst: sort.nullsFirst ?? undefined,
+        })),
+        exploreName: vizConfig.exploreName,
+        filters,
+        additionalMetrics: customMetrics ?? [],
+        // TODO: add tableCalculations
+        tableCalculations,
+    };
+};
+//# sourceMappingURL=verticalBarViz.js.map

package/dist/types/ee/AiAgent/schemas/visualizations/verticalBarViz.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"verticalBarViz.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/verticalBarViz.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAE7D,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,eAAe,MAAM,cAAc,CAAC;AAG3C,MAAM,CAAC,MAAM,gCAAgC,GAAG,CAAC,CAAC,MAAM,CAAC;IACrD,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,mFAAmF,CACtF;IACL,UAAU,EAAE,gBAAgB,CAAC;QACzB,qBAAqB,EACjB,8DAA8D;KACrE,CAAC;IACF,QAAQ,EAAE,CAAC;SACN,KAAK,CAAC,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;SACxD,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACL,oFAAoF,CACvF;IACL,KAAK,EAAE,CAAC;SACH,KAAK,CAAC,eAAe,CAAC;SACtB,QAAQ,CACL,uFAAuF,CAC1F;IACL,KAAK,EAAE,CAAC;SACH,MAAM,EAAE;SACR,GAAG,CAAC,0BAA0B,CAAC;SAC/B,QAAQ,EAAE;SACV,QAAQ,CACL,8DAA8D,CACjE;IACL,oBAAoB,EAAE,gBAAgB,CAAC;QACnC,qBAAqB,EACjB,0QAA0Q;KACjR,CAAC,CAAC,QAAQ,EAAE;IACb,SAAS,EAAE,CAAC;SACP,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,QAAQ,CACL,2GAA2G,CAC9G;IACL,SAAS,EAAE,CAAC;SACP,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;SACjD,QAAQ,CACL,sGAAsG,CACzG;IACL,UAAU,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;IACtD,UAAU,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;CACzD,CAAC,CAAC;AAMH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,EACtC,SAAS,EACT,OAAO,EACP,QAAQ,EACR,aAAa,EACb,iBAAiB,GAOpB,EAA4B,EAAE;IAC3B,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,CAAC;IACnC,MAAM,UAAU,GAAG;QACf,SAAS,CAAC,UAAU;QACpB,GAAG,CAAC,SAAS,CAAC,oBAAoB;YAC9B,CAAC,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC;YAClC,CAAC,CAAC,EAAE,CAAC;KACZ,CAAC;IACF,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;IACnC,OAAO;QACH,OAAO;QACP,UAAU;QACV,KAAK,EAAE,oBAAoB,CAAC,KAAK,EAAE,QAAQ,CAAC;QAC5C,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACxB,GAAG,IAAI;YACP,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,SAAS;SAC3C,CAAC,CAAC;QACH,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,OAAO;QACP,iBAAiB,EAAE,aAAa,IAAI,EAAE;QACtC,8BAA8B;QAC9B,iBAAiB;KACpB,CAAC;AACN,CAAC,CAAC"}

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

@@ -0,0 +1,12 @@
+export var AiResultType;
+(function (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["DASHBOARD_V2_RESULT"] = "dashboard_v2";
+    AiResultType["IMPROVE_CONTEXT"] = "improve_context";
+    AiResultType["PROPOSE_CHANGE"] = "propose_change";
+})(AiResultType || (AiResultType = {}));
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/types.ts"],"names":[],"mappings":"AASA,MAAM,CAAN,IAAY,YASX;AATD,WAAY,YAAY;IACpB,wDAAwC,CAAA;IACxC,0DAA0C,CAAA;IAC1C,sCAAsB,CAAA;IACtB,6CAA6B,CAAA;IAC7B,8CAA8B,CAAA;IAC9B,oDAAoC,CAAA;IACpC,mDAAmC,CAAA;IACnC,iDAAiC,CAAA;AACrC,CAAC,EATW,YAAY,KAAZ,YAAY,QASvB"}

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

@@ -329,14 +329,6 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
         })[] | null;
         type: AiResultType.QUERY_RESULT;
         description: string;
-        customMetrics: {
-            name: string;
-            label: string;
-            table: string;
-            type: import("../..").MetricType.PERCENTILE | import("../..").MetricType.AVERAGE | import("../..").MetricType.COUNT | import("../..").MetricType.COUNT_DISTINCT | import("../..").MetricType.SUM | import("../..").MetricType.MIN | import("../..").MetricType.MAX | import("../..").MetricType.MEDIAN;
-            description: string;
-            baseDimensionName: string;
-        }[] | null;
         chartConfig: {
             groupBy: string[] | null;
             stackBars: boolean | null;
@@ -351,6 +343,14 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
             secondaryYAxisMetric: string | null;
             secondaryYAxisLabel: string | null;
         } | null;
+        customMetrics: {
+            name: string;
+            label: string;
+            table: string;
+            type: import("../..").MetricType.PERCENTILE | import("../..").MetricType.AVERAGE | import("../..").MetricType.COUNT | import("../..").MetricType.COUNT_DISTINCT | import("../..").MetricType.SUM | import("../..").MetricType.MIN | import("../..").MetricType.MAX | import("../..").MetricType.MEDIAN;
+            description: string;
+            baseDimensionName: string;
+        }[] | null;
         title: string;
         queryConfig: {
             dimensions: string[];

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

@@ -0,0 +1,82 @@
+import { AI_DEFAULT_MAX_QUERY_LIMIT } from './constants';
+import { convertAiTableCalcsSchemaToTableCalcs, metricQueryTableViz, metricQueryTimeSeriesViz, metricQueryVerticalBarViz, toolRunQueryArgsSchemaTransformed, toolTableVizArgsSchemaTransformed, toolTimeSeriesArgsSchemaTransformed, toolVerticalBarArgsSchemaTransformed, } from './schemas';
+import { AiResultType } from './types';
+import { getValidAiQueryLimit } from './validators';
+export const parseVizConfig = (vizConfigUnknown, maxLimit) => {
+    if (!vizConfigUnknown) {
+        return null;
+    }
+    const toolVerticalBarArgsParsed = toolVerticalBarArgsSchemaTransformed.safeParse(vizConfigUnknown);
+    if (toolVerticalBarArgsParsed.success) {
+        const vizTool = toolVerticalBarArgsParsed.data;
+        const metricQuery = metricQueryVerticalBarViz({
+            vizConfig: vizTool.vizConfig,
+            filters: vizTool.filters,
+            maxLimit: maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT,
+            customMetrics: vizTool.customMetrics ?? null,
+            tableCalculations: convertAiTableCalcsSchemaToTableCalcs(vizTool.tableCalculations),
+        });
+        return {
+            type: AiResultType.VERTICAL_BAR_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    const toolTimeSeriesArgsParsed = toolTimeSeriesArgsSchemaTransformed.safeParse(vizConfigUnknown);
+    if (toolTimeSeriesArgsParsed.success) {
+        const vizTool = toolTimeSeriesArgsParsed.data;
+        const metricQuery = metricQueryTimeSeriesViz({
+            vizConfig: vizTool.vizConfig,
+            filters: vizTool.filters,
+            maxLimit: maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT,
+            customMetrics: vizTool.customMetrics ?? null,
+            tableCalculations: convertAiTableCalcsSchemaToTableCalcs(vizTool.tableCalculations),
+        });
+        return {
+            type: AiResultType.TIME_SERIES_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    const toolTableVizArgsParsed = toolTableVizArgsSchemaTransformed.safeParse(vizConfigUnknown);
+    if (toolTableVizArgsParsed.success) {
+        const vizTool = toolTableVizArgsParsed.data;
+        const metricQuery = metricQueryTableViz({
+            vizConfig: vizTool.vizConfig,
+            filters: vizTool.filters,
+            maxLimit: maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT,
+            customMetrics: vizTool.customMetrics ?? null,
+            tableCalculations: convertAiTableCalcsSchemaToTableCalcs(vizTool.tableCalculations),
+        });
+        return {
+            type: AiResultType.TABLE_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    // Parse runQuery tool
+    const toolRunQueryArgsParsed = toolRunQueryArgsSchemaTransformed.safeParse(vizConfigUnknown);
+    if (toolRunQueryArgsParsed.success) {
+        const vizTool = toolRunQueryArgsParsed.data;
+        const metricQuery = {
+            exploreName: vizTool.queryConfig.exploreName,
+            dimensions: vizTool.queryConfig.dimensions,
+            metrics: vizTool.queryConfig.metrics,
+            sorts: vizTool.queryConfig.sorts.map((sort) => ({
+                ...sort,
+                nullsFirst: sort.nullsFirst ?? undefined,
+            })),
+            limit: getValidAiQueryLimit(vizTool.queryConfig.limit, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT),
+            filters: vizTool.filters,
+            additionalMetrics: vizTool.customMetrics ?? [],
+            tableCalculations: convertAiTableCalcsSchemaToTableCalcs(vizTool.tableCalculations),
+        };
+        return {
+            type: AiResultType.QUERY_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    return null;
+};
+//# sourceMappingURL=utils.js.map