@lightdash/common 0.1776.0 → 0.1778.0

package/dist/esm/ee/AiAgent/schemas/tools/toolVerticalBarArgs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolVerticalBarArgs.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/tools/toolVerticalBarArgs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,KAAK,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,EAEH,KAAK,oCAAoC,EAC5C,MAAM,mBAAmB,CAAC;AAE3B,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAkBpC,CAAC;AAEH,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,yBAAyB,CAAC,CAAC;AAE5E,eAAO,MAAM,qBAAqB,WACtB,OAAO,KAChB,MAAM,IAAI,mBAC0C,CAAC;AAExD,eAAO,MAAM,oCAAoC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;UAK/B,YAAY,CAAC,mBAAmB;eAC3B,oCAAoC;aACtC,OAAO;mBACD,aAAa,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAKrC,CAAC;AAEN,MAAM,MAAM,8BAA8B,GAAG,CAAC,CAAC,KAAK,CAChD,OAAO,oCAAoC,CAC9C,CAAC"}

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

@@ -0,0 +1,24 @@
+import { z } from 'zod';
+import { FollowUpTools } from '../../followUpTools';
+import { AiResultType } from '../../types';
+import { filtersSchema, filtersSchemaTransformed } from '../filters';
+import { verticalBarMetricVizConfigSchema, } from '../visualizations';
+export const toolVerticalBarArgsSchema = z.object({
+    type: z.literal(AiResultType.VERTICAL_BAR_RESULT),
+    vizConfig: verticalBarMetricVizConfigSchema,
+    filters: filtersSchema
+        .nullable()
+        .describe('Filters to apply to the query. Filtered fields must exist in the selected explore.'),
+    followUpTools: z
+        .array(z.union([
+        z.literal(FollowUpTools.GENERATE_BAR_VIZ),
+        z.literal(FollowUpTools.GENERATE_TIME_SERIES_VIZ),
+    ]))
+        .describe(`The actions the User can ask for after the AI has generated the chart. NEVER include ${FollowUpTools.GENERATE_BAR_VIZ} in this list.`),
+});
+export const isToolVerticalBarArgs = (config) => toolVerticalBarArgsSchema.safeParse(config).success;
+export const toolVerticalBarArgsSchemaTransformed = toolVerticalBarArgsSchema.transform((data) => ({
+    ...data,
+    filters: filtersSchemaTransformed.parse(data.filters),
+}));
+//# sourceMappingURL=toolVerticalBarArgs.js.map

package/dist/esm/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;AAExB,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,YAAY,CAAC;AACrE,OAAO,EACH,gCAAgC,GAEnC,MAAM,mBAAmB,CAAC;AAE3B,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC;IACjD,SAAS,EAAE,gCAAgC;IAC3C,OAAO,EAAE,aAAa;SACjB,QAAQ,EAAE;SACV,QAAQ,CACL,oFAAoF,CACvF;IACL,aAAa,EAAE,CAAC;SACX,KAAK,CACF,CAAC,CAAC,KAAK,CAAC;QACJ,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,gBAAgB,CAAC;QACzC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,wBAAwB,CAAC;KACpD,CAAC,CACL;SACA,QAAQ,CACL,wFAAwF,aAAa,CAAC,gBAAgB,gBAAgB,CACzI;CACR,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACjC,MAAe,EACc,EAAE,CAC/B,yBAAyB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;AAExD,MAAM,CAAC,MAAM,oCAAoC,GAC7C,yBAAyB,CAAC,SAAS,CAC/B,CACI,IAAI,EAMN,EAAE,CAAC,CAAC;IACF,GAAG,IAAI;IACP,OAAO,EAAE,wBAAwB,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;CACxD,CAAC,CACL,CAAC"}

package/dist/esm/ee/AiAgent/schemas/visualizationMetadata.d.ts

@@ -0,0 +1,13 @@
+import { z } from 'zod';
+declare const visualizationMetadataSchema: z.ZodObject<{
+    title: z.ZodString;
+    description: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    description: string;
+    title: string;
+}, {
+    description: string;
+    title: string;
+}>;
+export default visualizationMetadataSchema;
+//# sourceMappingURL=visualizationMetadata.d.ts.map

package/dist/esm/ee/AiAgent/schemas/visualizationMetadata.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"visualizationMetadata.d.ts","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/schemas/visualizationMetadata.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,QAAA,MAAM,2BAA2B;;;;;;;;;EAK/B,CAAC;AAEH,eAAe,2BAA2B,CAAC"}

package/dist/esm/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/esm/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/esm/ee/AiAgent/schemas/visualizations/index.d.ts

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

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

@@ -0,0 +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;AAGjC,eAAO,MAAM,cAAc,4HAMzB,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,sJAAuC,CAAC;AAE9E,eAAO,MAAM,qBAAqB,4IAMhC,CAAC;AAGH,eAAO,MAAM,qCAAqC,4IAO5C,CAAC"}

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

@@ -0,0 +1,31 @@
+import { z } from 'zod';
+export * from './tableViz';
+export * from './timeSeriesViz';
+export * from './verticalBarViz';
+// define tool names
+export const ToolNameSchema = z.enum([
+    'findExplores',
+    'findFields',
+    'generateBarVizConfig',
+    'generateTableVizConfig',
+    'generateTimeSeriesVizConfig',
+]);
+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',
+    findFields: 'Finding relevant fields',
+    generateBarVizConfig: 'Generating a bar chart',
+    generateTableVizConfig: 'Generating a table',
+    generateTimeSeriesVizConfig: 'Generating a line chart',
+});
+// after-tool-call messages
+export const TOOL_DISPLAY_MESSAGES_AFTER_TOOL_CALL = ToolDisplayMessagesSchema.parse({
+    findExplores: 'Found relevant explores',
+    findFields: 'Found relevant fields',
+    generateBarVizConfig: 'Generated a bar chart',
+    generateTableVizConfig: 'Generated a table',
+    generateTimeSeriesVizConfig: 'Generated a line chart',
+});
+//# sourceMappingURL=index.js.map

package/dist/esm/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,oBAAoB;AACpB,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,IAAI,CAAC;IACjC,cAAc;IACd,YAAY;IACZ,sBAAsB;IACtB,wBAAwB;IACxB,6BAA6B;CAChC,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,UAAU,EAAE,yBAAyB;IACrC,oBAAoB,EAAE,wBAAwB;IAC9C,sBAAsB,EAAE,oBAAoB;IAC5C,2BAA2B,EAAE,yBAAyB;CACzD,CAAC,CAAC;AAEH,2BAA2B;AAC3B,MAAM,CAAC,MAAM,qCAAqC,GAC9C,yBAAyB,CAAC,KAAK,CAAC;IAC5B,YAAY,EAAE,yBAAyB;IACvC,UAAU,EAAE,uBAAuB;IACnC,oBAAoB,EAAE,uBAAuB;IAC7C,sBAAsB,EAAE,mBAAmB;IAC3C,2BAA2B,EAAE,wBAAwB;CACxD,CAAC,CAAC"}

package/dist/esm/ee/AiAgent/schemas/visualizations/tableViz.d.ts

@@ -0,0 +1,47 @@
+import { z } from 'zod';
+import type { Filters } from '../../../../types/filter';
+import type { AiMetricQueryWithFilters } from '../../types';
+export declare const tableVizConfigSchema: z.ZodObject<{
+    title: z.ZodString;
+    description: z.ZodString;
+} & {
+    exploreName: z.ZodString;
+    metrics: z.ZodArray<z.ZodString, "many">;
+    dimensions: z.ZodNullable<z.ZodArray<z.ZodString, "many">>;
+    sorts: z.ZodArray<z.ZodObject<{
+        fieldId: z.ZodString;
+        descending: z.ZodBoolean;
+    }, "strip", z.ZodTypeAny, {
+        descending: boolean;
+        fieldId: string;
+    }, {
+        descending: boolean;
+        fieldId: string;
+    }>, "many">;
+    limit: z.ZodNullable<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    description: string;
+    exploreName: string;
+    dimensions: string[] | null;
+    metrics: string[];
+    sorts: {
+        descending: boolean;
+        fieldId: string;
+    }[];
+    limit: number | null;
+    title: string;
+}, {
+    description: string;
+    exploreName: string;
+    dimensions: string[] | null;
+    metrics: string[];
+    sorts: {
+        descending: boolean;
+        fieldId: string;
+    }[];
+    limit: number | null;
+    title: string;
+}>;
+export type TableVizConfigSchemaType = z.infer<typeof tableVizConfigSchema>;
+export declare const metricQueryTableViz: (vizConfig: TableVizConfigSchemaType, filters: Filters, maxLimit: number) => AiMetricQueryWithFilters;
+//# sourceMappingURL=tableViz.d.ts.map

package/dist/esm/ee/AiAgent/schemas/visualizations/tableViz.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tableViz.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/tableViz.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAK5D,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgC5B,CAAC;AAEN,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAE5E,eAAO,MAAM,mBAAmB,cACjB,wBAAwB,WAC1B,OAAO,YACN,MAAM,KACjB,wBAOD,CAAC"}

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

@@ -0,0 +1,35 @@
+import { z } from 'zod';
+import { getValidAiQueryLimit } from '../../validators';
+import sortFieldSchema from '../sortField';
+import visualizationMetadataSchema from '../visualizationMetadata';
+export const tableVizConfigSchema = visualizationMetadataSchema
+    .extend({
+    exploreName: z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for table query'),
+    metrics: z
+        .array(z.string())
+        .min(1)
+        .describe('At least one metric is required. The field ids of the metrics to be calculated for the CSV. They will be grouped by the dimensions.'),
+    dimensions: z
+        .array(z.string())
+        .nullable()
+        .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 CSV file from a query with metrics and dimensions');
+export const metricQueryTableViz = (vizConfig, filters, maxLimit) => ({
+    exploreName: vizConfig.exploreName,
+    metrics: vizConfig.metrics,
+    dimensions: vizConfig.dimensions || [],
+    sorts: vizConfig.sorts,
+    limit: getValidAiQueryLimit(vizConfig.limit, maxLimit),
+    filters,
+});
+//# sourceMappingURL=tableViz.js.map

package/dist/esm/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;AAGxB,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,eAAe,MAAM,cAAc,CAAC;AAC3C,OAAO,2BAA2B,MAAM,0BAA0B,CAAC;AAEnE,MAAM,CAAC,MAAM,oBAAoB,GAAG,2BAA2B;KAC1D,MAAM,CAAC;IACJ,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,oFAAoF,CACvF;IACL,OAAO,EAAE,CAAC;SACL,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACL,qIAAqI,CACxI;IACL,UAAU,EAAE,CAAC;SACR,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,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,uFAAuF,CAC1F,CAAC;AAIN,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAC/B,SAAmC,EACnC,OAAgB,EAChB,QAAgB,EACQ,EAAE,CAAC,CAAC;IAC5B,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;IACtB,KAAK,EAAE,oBAAoB,CAAC,SAAS,CAAC,KAAK,EAAE,QAAQ,CAAC;IACtD,OAAO;CACV,CAAC,CAAC"}

package/dist/esm/ee/AiAgent/schemas/visualizations/timeSeriesViz.d.ts

@@ -0,0 +1,53 @@
+import { z } from 'zod';
+import type { Filters } from '../../../../types/filter';
+import type { AiMetricQueryWithFilters } from '../../types';
+export declare const timeSeriesMetricVizConfigSchema: z.ZodObject<{
+    title: z.ZodString;
+    description: z.ZodString;
+} & {
+    exploreName: z.ZodString;
+    xDimension: z.ZodString;
+    yMetrics: z.ZodArray<z.ZodString, "many">;
+    sorts: z.ZodArray<z.ZodObject<{
+        fieldId: z.ZodString;
+        descending: z.ZodBoolean;
+    }, "strip", z.ZodTypeAny, {
+        descending: boolean;
+        fieldId: string;
+    }, {
+        descending: boolean;
+        fieldId: string;
+    }>, "many">;
+    breakdownByDimension: z.ZodNullable<z.ZodString>;
+    lineType: z.ZodUnion<[z.ZodLiteral<"line">, z.ZodLiteral<"area">]>;
+    limit: z.ZodNullable<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    description: string;
+    exploreName: string;
+    sorts: {
+        descending: boolean;
+        fieldId: string;
+    }[];
+    limit: number | null;
+    title: string;
+    xDimension: string;
+    yMetrics: string[];
+    breakdownByDimension: string | null;
+    lineType: "line" | "area";
+}, {
+    description: string;
+    exploreName: string;
+    sorts: {
+        descending: boolean;
+        fieldId: string;
+    }[];
+    limit: number | null;
+    title: string;
+    xDimension: string;
+    yMetrics: string[];
+    breakdownByDimension: string | null;
+    lineType: "line" | "area";
+}>;
+export type TimeSeriesMetricVizConfigSchemaType = z.infer<typeof timeSeriesMetricVizConfigSchema>;
+export declare const metricQueryTimeSeriesViz: (vizConfig: TimeSeriesMetricVizConfigSchemaType, filters: Filters, maxLimit: number) => AiMetricQueryWithFilters;
+//# sourceMappingURL=timeSeriesViz.d.ts.map

package/dist/esm/ee/AiAgent/schemas/visualizations/timeSeriesViz.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeSeriesViz.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/timeSeriesViz.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAK5D,eAAO,MAAM,+BAA+B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAuCtC,CAAC;AAEP,MAAM,MAAM,mCAAmC,GAAG,CAAC,CAAC,KAAK,CACrD,OAAO,+BAA+B,CACzC,CAAC;AAEF,eAAO,MAAM,wBAAwB,cACtB,mCAAmC,WACrC,OAAO,YACN,MAAM,KACjB,wBAkBF,CAAC"}

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

@@ -0,0 +1,51 @@
+import { z } from 'zod';
+import { AI_DEFAULT_MAX_QUERY_LIMIT } from '../../constants';
+import { getValidAiQueryLimit } from '../../validators';
+import sortFieldSchema from '../sortField';
+import visualizationMetadataSchema from '../visualizationMetadata';
+export const timeSeriesMetricVizConfigSchema = visualizationMetadataSchema.extend({
+    exploreName: z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for the chart.'),
+    xDimension: z
+        .string()
+        .describe('The field id of the time dimension to be displayed on the x-axis.'),
+    yMetrics: z
+        .array(z.string())
+        .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: z
+        .string()
+        .nullable()
+        .describe('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.'),
+    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.`),
+});
+export const metricQueryTimeSeriesViz = (vizConfig, filters, maxLimit) => {
+    const metrics = vizConfig.yMetrics;
+    const dimensions = [
+        vizConfig.xDimension,
+        ...(vizConfig.breakdownByDimension
+            ? [vizConfig.breakdownByDimension]
+            : []),
+    ];
+    const { limit, sorts } = vizConfig;
+    return {
+        metrics,
+        dimensions,
+        limit: getValidAiQueryLimit(limit, maxLimit),
+        sorts,
+        exploreName: vizConfig.exploreName,
+        filters,
+    };
+};
+//# sourceMappingURL=timeSeriesViz.js.map

package/dist/esm/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;AAExB,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAE7D,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,eAAe,MAAM,cAAc,CAAC;AAC3C,OAAO,2BAA2B,MAAM,0BAA0B,CAAC;AAEnE,MAAM,CAAC,MAAM,+BAA+B,GACxC,2BAA2B,CAAC,MAAM,CAAC;IAC/B,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,mFAAmF,CACtF;IACL,UAAU,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,CACL,mEAAmE,CACtE;IACL,QAAQ,EAAE,CAAC;SACN,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,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,CAAC;SAClB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACL,4SAA4S,CAC/S;IACL,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;CACzE,CAAC,CAAC;AAMP,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACpC,SAA8C,EAC9C,OAAgB,EAChB,QAAgB,EACQ,EAAE;IAC1B,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;QACL,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,OAAO;KACV,CAAC;AACN,CAAC,CAAC"}

package/dist/esm/ee/AiAgent/schemas/visualizations/verticalBarViz.d.ts

@@ -0,0 +1,62 @@
+import { z } from 'zod';
+import type { Filters } from '../../../../types/filter';
+import type { AiMetricQueryWithFilters } from '../../types';
+export declare const verticalBarMetricVizConfigSchema: z.ZodObject<{
+    title: z.ZodString;
+    description: z.ZodString;
+} & {
+    exploreName: z.ZodString;
+    xDimension: z.ZodString;
+    yMetrics: z.ZodArray<z.ZodString, "many">;
+    sorts: z.ZodArray<z.ZodObject<{
+        fieldId: z.ZodString;
+        descending: z.ZodBoolean;
+    }, "strip", z.ZodTypeAny, {
+        descending: boolean;
+        fieldId: string;
+    }, {
+        descending: boolean;
+        fieldId: string;
+    }>, "many">;
+    limit: z.ZodNullable<z.ZodNumber>;
+    breakdownByDimension: z.ZodNullable<z.ZodString>;
+    stackBars: z.ZodNullable<z.ZodBoolean>;
+    xAxisType: z.ZodUnion<[z.ZodLiteral<"category">, z.ZodLiteral<"time">]>;
+    xAxisLabel: z.ZodNullable<z.ZodString>;
+    yAxisLabel: z.ZodNullable<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    description: string;
+    exploreName: string;
+    sorts: {
+        descending: boolean;
+        fieldId: string;
+    }[];
+    limit: number | null;
+    title: string;
+    xDimension: string;
+    yMetrics: string[];
+    breakdownByDimension: string | null;
+    stackBars: boolean | null;
+    xAxisType: "time" | "category";
+    xAxisLabel: string | null;
+    yAxisLabel: string | null;
+}, {
+    description: string;
+    exploreName: string;
+    sorts: {
+        descending: boolean;
+        fieldId: string;
+    }[];
+    limit: number | null;
+    title: string;
+    xDimension: string;
+    yMetrics: string[];
+    breakdownByDimension: string | null;
+    stackBars: boolean | null;
+    xAxisType: "time" | "category";
+    xAxisLabel: string | null;
+    yAxisLabel: string | null;
+}>;
+export type VerticalBarMetricVizConfigSchemaType = z.infer<typeof verticalBarMetricVizConfigSchema>;
+export declare const metricQueryVerticalBarViz: (vizConfig: VerticalBarMetricVizConfigSchemaType, filters: Filters, maxLimit: number) => AiMetricQueryWithFilters;
+//# sourceMappingURL=verticalBarViz.d.ts.map

package/dist/esm/ee/AiAgent/schemas/visualizations/verticalBarViz.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"verticalBarViz.d.ts","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/visualizations/verticalBarViz.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAK5D,eAAO,MAAM,gCAAgC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAuDvC,CAAC;AAEP,MAAM,MAAM,oCAAoC,GAAG,CAAC,CAAC,KAAK,CACtD,OAAO,gCAAgC,CAC1C,CAAC;AAEF,eAAO,MAAM,yBAAyB,cACvB,oCAAoC,WACtC,OAAO,YACN,MAAM,KACjB,wBAiBF,CAAC"}

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

@@ -0,0 +1,63 @@
+import { z } from 'zod';
+import { AI_DEFAULT_MAX_QUERY_LIMIT } from '../../constants';
+import { getValidAiQueryLimit } from '../../validators';
+import sortFieldSchema from '../sortField';
+import visualizationMetadataSchema from '../visualizationMetadata';
+export const verticalBarMetricVizConfigSchema = visualizationMetadataSchema.extend({
+    exploreName: z
+        .string()
+        .describe('The name of the explore containing the metrics and dimensions used for the chart.'),
+    xDimension: z
+        .string()
+        .describe('The field id of the dimension to be displayed on the x-axis.'),
+    yMetrics: z
+        .array(z.string())
+        .min(1)
+        .describe('At least one metric is required. 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: z
+        .string()
+        .nullable()
+        .describe('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'),
+    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) => {
+    const metrics = vizConfig.yMetrics;
+    const dimensions = [
+        vizConfig.xDimension,
+        ...(vizConfig.breakdownByDimension
+            ? [vizConfig.breakdownByDimension]
+            : []),
+    ];
+    const { limit, sorts } = vizConfig;
+    return {
+        metrics,
+        dimensions,
+        limit: getValidAiQueryLimit(limit, maxLimit),
+        sorts,
+        exploreName: vizConfig.exploreName,
+        filters,
+    };
+};
+//# sourceMappingURL=verticalBarViz.js.map

package/dist/esm/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;AAExB,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAE7D,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,eAAe,MAAM,cAAc,CAAC;AAC3C,OAAO,2BAA2B,MAAM,0BAA0B,CAAC;AAEnE,MAAM,CAAC,MAAM,gCAAgC,GACzC,2BAA2B,CAAC,MAAM,CAAC;IAC/B,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,mFAAmF,CACtF;IACL,UAAU,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,CACL,8DAA8D,CACjE;IACL,QAAQ,EAAE,CAAC;SACN,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACL,qHAAqH,CACxH;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,CAAC;SAClB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACL,0QAA0Q,CAC7Q;IACL,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;AAMP,MAAM,CAAC,MAAM,yBAAyB,GAAG,CACrC,SAA+C,EAC/C,OAAgB,EAChB,QAAgB,EACQ,EAAE;IAC1B,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;QACL,WAAW,EAAE,SAAS,CAAC,WAAW;QAClC,OAAO;KACV,CAAC;AACN,CAAC,CAAC"}

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

@@ -0,0 +1,24 @@
+import type { Filters } from '../../types/filter';
+import type { MetricQuery } from '../../types/metricQuery';
+import type { ToolTableVizArgs, ToolTimeSeriesArgs, ToolVerticalBarArgs } from './schemas/tools';
+export declare enum AiResultType {
+    TIME_SERIES_RESULT = "time_series_chart",
+    VERTICAL_BAR_RESULT = "vertical_bar_chart",
+    ONE_LINE_RESULT = "one_line_result",
+    TABLE_RESULT = "table"
+}
+export type AiMetricQuery = Pick<MetricQuery, 'metrics' | 'dimensions' | 'sorts' | 'limit' | 'exploreName'>;
+export type AiMetricQueryWithFilters = AiMetricQuery & {
+    filters: Filters;
+};
+export type AiAgentVizConfig = {
+    type: 'vertical_bar_chart';
+    config: ToolVerticalBarArgs;
+} | {
+    type: 'time_series_chart';
+    config: ToolTimeSeriesArgs;
+} | {
+    type: 'table';
+    config: ToolTableVizArgs;
+};
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +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,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EACR,gBAAgB,EAChB,kBAAkB,EAClB,mBAAmB,EACtB,MAAM,iBAAiB,CAAC;AAEzB,oBAAY,YAAY;IACpB,kBAAkB,sBAAsB;IACxC,mBAAmB,uBAAuB;IAC1C,eAAe,oBAAoB;IACnC,YAAY,UAAU;CACzB;AAED,MAAM,MAAM,aAAa,GAAG,IAAI,CAC5B,WAAW,EACX,SAAS,GAAG,YAAY,GAAG,OAAO,GAAG,OAAO,GAAG,aAAa,CAC/D,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"}

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

@@ -0,0 +1,8 @@
+export var AiResultType;
+(function (AiResultType) {
+    AiResultType["TIME_SERIES_RESULT"] = "time_series_chart";
+    AiResultType["VERTICAL_BAR_RESULT"] = "vertical_bar_chart";
+    AiResultType["ONE_LINE_RESULT"] = "one_line_result";
+    AiResultType["TABLE_RESULT"] = "table";
+})(AiResultType || (AiResultType = {}));
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/types.ts"],"names":[],"mappings":"AAQA,MAAM,CAAN,IAAY,YAKX;AALD,WAAY,YAAY;IACpB,wDAAwC,CAAA;IACxC,0DAA0C,CAAA;IAC1C,mDAAmC,CAAA;IACnC,sCAAsB,CAAA;AAC1B,CAAC,EALW,YAAY,KAAZ,YAAY,QAKvB"}

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

@@ -0,0 +1,43 @@
+import { AiResultType } from './types';
+export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?: number | undefined) => {
+    readonly type: AiResultType.VERTICAL_BAR_RESULT;
+    readonly vizTool: {
+        type: AiResultType.VERTICAL_BAR_RESULT;
+        vizConfig: import("./schemas").VerticalBarMetricVizConfigSchemaType;
+        filters: import("../..").Filters;
+        followUpTools: import("./followUpTools").FollowUpTools[];
+    };
+    readonly metricQuery: import("./types").AiMetricQueryWithFilters;
+} | {
+    readonly type: AiResultType.TIME_SERIES_RESULT;
+    readonly vizTool: {
+        type: AiResultType.TIME_SERIES_RESULT;
+        vizConfig: import("./schemas").TimeSeriesMetricVizConfigSchemaType;
+        filters: import("../..").Filters;
+        followUpTools: import("./followUpTools").FollowUpTools[];
+    };
+    readonly metricQuery: import("./types").AiMetricQueryWithFilters;
+} | {
+    readonly type: AiResultType.TABLE_RESULT;
+    readonly vizTool: {
+        type: AiResultType.TABLE_RESULT;
+        vizConfig: import("./schemas").TableVizConfigSchemaType;
+        filters: import("../..").Filters;
+    };
+    readonly metricQuery: import("./types").AiMetricQueryWithFilters;
+} | {
+    readonly type: AiResultType.ONE_LINE_RESULT;
+    readonly metricQuery: {
+        filters: import("../..").Filters;
+        exploreName: string;
+        dimensions: string[];
+        metrics: string[];
+        sorts: {
+            descending: boolean;
+            fieldId: string;
+        }[];
+        limit: number;
+    };
+    readonly vizTool?: undefined;
+} | null;
+//# sourceMappingURL=utils.d.ts.map

package/dist/esm/ee/AiAgent/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../src/ee/AiAgent/utils.ts"],"names":[],"mappings":"AAcA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEvC,eAAO,MAAM,cAAc,qBACL,MAAM,GAAG,IAAI,aACpB,MAAM,GAAG,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;QA+DhC,CAAC"}

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

@@ -0,0 +1,48 @@
+import { AI_DEFAULT_MAX_QUERY_LIMIT } from './constants';
+import { isToolOneLineArgs, isToolTableVizArgs, isToolTimeSeriesArgs, isToolVerticalBarArgs, metricQueryTableViz, metricQueryTimeSeriesViz, metricQueryVerticalBarViz, toolOneLineArgsSchemaTransformed, toolTableVizArgsSchemaTransformed, toolTimeSeriesArgsSchemaTransformed, toolVerticalBarArgsSchemaTransformed, } from './schemas';
+import { AiResultType } from './types';
+export const parseVizConfig = (vizConfigUnknown, maxLimit) => {
+    if (!vizConfigUnknown) {
+        return null;
+    }
+    if (isToolVerticalBarArgs(vizConfigUnknown)) {
+        const vizTool = toolVerticalBarArgsSchemaTransformed.parse(vizConfigUnknown);
+        const metricQuery = metricQueryVerticalBarViz(vizTool.vizConfig, vizTool.filters, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT);
+        return {
+            type: AiResultType.VERTICAL_BAR_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    if (isToolTimeSeriesArgs(vizConfigUnknown)) {
+        const vizTool = toolTimeSeriesArgsSchemaTransformed.parse(vizConfigUnknown);
+        const metricQuery = metricQueryTimeSeriesViz(vizTool.vizConfig, vizTool.filters, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT);
+        return {
+            type: AiResultType.TIME_SERIES_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    if (isToolTableVizArgs(vizConfigUnknown)) {
+        const vizTool = toolTableVizArgsSchemaTransformed.parse(vizConfigUnknown);
+        const metricQuery = metricQueryTableViz(vizTool.vizConfig, vizTool.filters, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT);
+        return {
+            type: AiResultType.TABLE_RESULT,
+            vizTool,
+            metricQuery,
+        };
+    }
+    if (isToolOneLineArgs(vizConfigUnknown)) {
+        const vizTool = toolOneLineArgsSchemaTransformed.parse(vizConfigUnknown);
+        const metricQuery = {
+            ...vizTool.metricQuery,
+            filters: vizTool.filters,
+        };
+        return {
+            type: AiResultType.ONE_LINE_RESULT,
+            metricQuery,
+        };
+    }
+    return null;
+};
+//# sourceMappingURL=utils.js.map

package/dist/esm/ee/AiAgent/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EACH,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,qBAAqB,EACrB,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,gCAAgC,EAChC,iCAAiC,EACjC,mCAAmC,EACnC,oCAAoC,GACvC,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEvC,MAAM,CAAC,MAAM,cAAc,GAAG,CAC1B,gBAA+B,EAC/B,QAA6B,EAC/B,EAAE;IACA,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,IAAI,qBAAqB,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC1C,MAAM,OAAO,GACT,oCAAoC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACjE,MAAM,WAAW,GAAG,yBAAyB,CACzC,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,OAAO,EACf,QAAQ,IAAI,0BAA0B,CACzC,CAAC;QACF,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,mBAAmB;YACtC,OAAO;YACP,WAAW;SACL,CAAC;IACf,CAAC;IACD,IAAI,oBAAoB,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACzC,MAAM,OAAO,GACT,mCAAmC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAChE,MAAM,WAAW,GAAG,wBAAwB,CACxC,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,OAAO,EACf,QAAQ,IAAI,0BAA0B,CACzC,CAAC;QACF,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,kBAAkB;YACrC,OAAO;YACP,WAAW;SACL,CAAC;IACf,CAAC;IACD,IAAI,kBAAkB,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACvC,MAAM,OAAO,GACT,iCAAiC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC9D,MAAM,WAAW,GAAG,mBAAmB,CACnC,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,OAAO,EACf,QAAQ,IAAI,0BAA0B,CACzC,CAAC;QACF,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,YAAY;YAC/B,OAAO;YACP,WAAW;SACL,CAAC;IACf,CAAC;IAED,IAAI,iBAAiB,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACtC,MAAM,OAAO,GACT,gCAAgC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAG;YAChB,GAAG,OAAO,CAAC,WAAW;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;SAC3B,CAAC;QACF,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,eAAe;YAClC,WAAW;SACL,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AAChB,CAAC,CAAC"}

package/dist/esm/ee/AiAgent/validators.d.ts

@@ -0,0 +1,2 @@
+export declare function getValidAiQueryLimit(limit: number | null, maxLimit: number): number;
+//# sourceMappingURL=validators.d.ts.map

package/dist/esm/ee/AiAgent/validators.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"validators.d.ts","sourceRoot":"","sources":["../../../../src/ee/AiAgent/validators.ts"],"names":[],"mappings":"AAAA,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,EAAE,QAAQ,EAAE,MAAM,UAY1E"}

package/dist/esm/ee/AiAgent/validators.js

@@ -0,0 +1,10 @@
+export function getValidAiQueryLimit(limit, maxLimit) {
+    if (!limit) {
+        return maxLimit;
+    }
+    if (limit > maxLimit) {
+        throw new Error(`The limit provided is greater than the maximum allowed limit of ${maxLimit}`);
+    }
+    return limit;
+}
+//# sourceMappingURL=validators.js.map