@lightdash/common 0.1969.1 → 0.1971.0

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

@@ -5,6 +5,25 @@ export declare const TOOL_VERTICAL_BAR_VIZ_DESCRIPTION = "Use this tool to gener
 export declare const toolVerticalBarArgsSchema: z.ZodObject<{
     type: z.ZodLiteral<AiResultType.VERTICAL_BAR_RESULT>;
 } & {
+    customMetrics: z.ZodNullable<z.ZodArray<z.ZodObject<{
+        name: z.ZodString;
+        label: z.ZodString;
+        baseDimensionName: z.ZodString;
+        table: z.ZodString;
+        type: z.ZodEnum<[import("../../../..").MetricType.AVERAGE, import("../../../..").MetricType.COUNT, import("../../../..").MetricType.COUNT_DISTINCT, import("../../../..").MetricType.MAX, import("../../../..").MetricType.MIN, import("../../../..").MetricType.SUM, import("../../../..").MetricType.PERCENTILE, import("../../../..").MetricType.MEDIAN]>;
+    }, "strip", z.ZodTypeAny, {
+        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;
+        baseDimensionName: string;
+    }, {
+        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;
+        baseDimensionName: string;
+    }>, "many">>;
     vizConfig: z.ZodObject<{
         exploreName: z.ZodString;
         xDimension: z.ZodString;
@@ -1191,6 +1210,14 @@ export declare const toolVerticalBarArgsSchema: z.ZodObject<{
         })[] | null;
         type: "and" | "or";
     } | 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;
+        baseDimensionName: string;
+    }[] | null;
+    title: string;
     vizConfig: {
         limit: number | null;
         exploreName: string;
@@ -1207,7 +1234,6 @@ export declare const toolVerticalBarArgsSchema: z.ZodObject<{
         stackBars: boolean | null;
         xAxisType: "time" | "category";
     };
-    title: string;
     followUpTools: (FollowUpTools.GENERATE_BAR_VIZ | FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
 }, {
     type: AiResultType.VERTICAL_BAR_RESULT;
@@ -1393,6 +1419,14 @@ export declare const toolVerticalBarArgsSchema: z.ZodObject<{
         })[] | null;
         type: "and" | "or";
     } | 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;
+        baseDimensionName: string;
+    }[] | null;
+    title: string;
     vizConfig: {
         limit: number | null;
         exploreName: string;
@@ -1409,13 +1443,11 @@ export declare const toolVerticalBarArgsSchema: z.ZodObject<{
         stackBars: boolean | null;
         xAxisType: "time" | "category";
     };
-    title: string;
     followUpTools: (FollowUpTools.GENERATE_BAR_VIZ | FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
 }>;
 export type ToolVerticalBarArgs = z.infer<typeof toolVerticalBarArgsSchema>;
 export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObject<{
     type: z.ZodLiteral<AiResultType.VERTICAL_BAR_RESULT>;
-} & {
     vizConfig: z.ZodObject<{
         exploreName: z.ZodString;
         xDimension: z.ZodString;
@@ -2418,6 +2450,26 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
     followUpTools: z.ZodArray<z.ZodUnion<[z.ZodLiteral<FollowUpTools.GENERATE_BAR_VIZ>, z.ZodLiteral<FollowUpTools.GENERATE_TIME_SERIES_VIZ>]>, "many">;
     title: z.ZodString;
     description: z.ZodString;
+} & {
+    customMetrics: z.ZodDefault<z.ZodNullable<z.ZodArray<z.ZodObject<{
+        name: z.ZodString;
+        label: z.ZodString;
+        baseDimensionName: z.ZodString;
+        table: z.ZodString;
+        type: z.ZodEnum<[import("../../../..").MetricType.AVERAGE, import("../../../..").MetricType.COUNT, import("../../../..").MetricType.COUNT_DISTINCT, import("../../../..").MetricType.MAX, import("../../../..").MetricType.MIN, import("../../../..").MetricType.SUM, import("../../../..").MetricType.PERCENTILE, import("../../../..").MetricType.MEDIAN]>;
+    }, "strip", z.ZodTypeAny, {
+        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;
+        baseDimensionName: string;
+    }, {
+        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;
+        baseDimensionName: string;
+    }>, "many">>>;
 }, z.UnknownKeysParam, z.ZodTypeAny, {
     type: AiResultType.VERTICAL_BAR_RESULT;
     description: string;
@@ -2602,6 +2654,14 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         })[] | null;
         type: "and" | "or";
     } | 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;
+        baseDimensionName: string;
+    }[] | null;
+    title: string;
     vizConfig: {
         limit: number | null;
         exploreName: string;
@@ -2618,7 +2678,6 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         stackBars: boolean | null;
         xAxisType: "time" | "category";
     };
-    title: string;
     followUpTools: (FollowUpTools.GENERATE_BAR_VIZ | FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
 }, {
     type: AiResultType.VERTICAL_BAR_RESULT;
@@ -2804,6 +2863,7 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         })[] | null;
         type: "and" | "or";
     } | null;
+    title: string;
     vizConfig: {
         limit: number | null;
         exploreName: string;
@@ -2820,12 +2880,26 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         stackBars: boolean | null;
         xAxisType: "time" | "category";
     };
-    title: string;
     followUpTools: (FollowUpTools.GENERATE_BAR_VIZ | FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
+    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;
+        baseDimensionName: string;
+    }[] | null | undefined;
 }>, {
     filters: import("../../../..").Filters;
     type: AiResultType.VERTICAL_BAR_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;
+        baseDimensionName: string;
+    }[] | null;
+    title: string;
     vizConfig: {
         limit: number | null;
         exploreName: string;
@@ -2842,7 +2916,6 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         stackBars: boolean | null;
         xAxisType: "time" | "category";
     };
-    title: string;
     followUpTools: (FollowUpTools.GENERATE_BAR_VIZ | FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
 }, {
     type: AiResultType.VERTICAL_BAR_RESULT;
@@ -3028,6 +3101,7 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         })[] | null;
         type: "and" | "or";
     } | null;
+    title: string;
     vizConfig: {
         limit: number | null;
         exploreName: string;
@@ -3044,8 +3118,14 @@ export declare const toolVerticalBarArgsSchemaTransformed: z.ZodEffects<z.ZodObj
         stackBars: boolean | null;
         xAxisType: "time" | "category";
     };
-    title: string;
     followUpTools: (FollowUpTools.GENERATE_BAR_VIZ | FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
+    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;
+        baseDimensionName: string;
+    }[] | null | undefined;
 }>;
 export type ToolVerticalBarArgsTransformed = z.infer<typeof toolVerticalBarArgsSchemaTransformed>;
 //# sourceMappingURL=toolVerticalBarArgs.d.ts.map

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

@@ -1 +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,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAM3C,eAAO,MAAM,iCAAiC,yDAAyD,CAAC;AAExG,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAuB1B,CAAC;AAEb,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,yBAAyB,CAAC,CAAC;AAE5E,eAAO,MAAM,oCAAoC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAI1C,CAAC;AAER,MAAM,MAAM,8BAA8B,GAAG,CAAC,CAAC,KAAK,CAChD,OAAO,oCAAoC,CAC9C,CAAC"}
+{"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,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO3C,eAAO,MAAM,iCAAiC,yDAAyD,CAAC;AAExG,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAwB1B,CAAC;AAEb,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,yBAAyB,CAAC,CAAC;AAE5E,eAAO,MAAM,oCAAoC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAQ1C,CAAC;AAER,MAAM,MAAM,8BAA8B,GAAG,CAAC,CAAC,KAAK,CAChD,OAAO,oCAAoC,CAC9C,CAAC"}

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

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import { FollowUpTools } from '../../followUpTools';
 import { AiResultType } from '../../types';
+import { customMetricsSchema } from '../customMetrics';
 import { filtersSchema, filtersSchemaTransformed } from '../filters';
 import { createToolSchema } from '../toolSchemaBuilder';
 import visualizationMetadataSchema from '../visualizationMetadata';
@@ -9,6 +10,7 @@ export const TOOL_VERTICAL_BAR_VIZ_DESCRIPTION = `Use this tool to generate a Ba
 export const toolVerticalBarArgsSchema = createToolSchema(AiResultType.VERTICAL_BAR_RESULT, TOOL_VERTICAL_BAR_VIZ_DESCRIPTION)
     .extend({
     ...visualizationMetadataSchema.shape,
+    customMetrics: customMetricsSchema,
     vizConfig: verticalBarMetricVizConfigSchema,
     filters: filtersSchema
         .nullable()
@@ -21,7 +23,12 @@ export const toolVerticalBarArgsSchema = createToolSchema(AiResultType.VERTICAL_
         .describe(`The actions the User can ask for after the AI has generated the chart. NEVER include ${FollowUpTools.GENERATE_BAR_VIZ} in this list.`),
 })
     .build();
-export const toolVerticalBarArgsSchemaTransformed = toolVerticalBarArgsSchema.transform((data) => ({
+export const toolVerticalBarArgsSchemaTransformed = toolVerticalBarArgsSchema
+    .extend({
+    // backwards compatibility for old viz configs without customMetrics
+    customMetrics: customMetricsSchema.default(null),
+})
+    .transform((data) => ({
     ...data,
     filters: filtersSchemaTransformed.parse(data.filters),
 }));

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

@@ -1 +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,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,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,CACrD,YAAY,CAAC,mBAAmB,EAChC,iCAAiC,CACpC;KACI,MAAM,CAAC;IACJ,GAAG,2BAA2B,CAAC,KAAK;IACpC,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;KACD,KAAK,EAAE,CAAC;AAIb,MAAM,CAAC,MAAM,oCAAoC,GAC7C,yBAAyB,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC3C,GAAG,IAAI;IACP,OAAO,EAAE,wBAAwB,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;CACxD,CAAC,CAAC,CAAC"}
+{"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,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,MAAM,YAAY,CAAC;AACrE,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,CACrD,YAAY,CAAC,mBAAmB,EAChC,iCAAiC,CACpC;KACI,MAAM,CAAC;IACJ,GAAG,2BAA2B,CAAC,KAAK;IACpC,aAAa,EAAE,mBAAmB;IAClC,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;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;CACnD,CAAC;KACD,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClB,GAAG,IAAI;IACP,OAAO,EAAE,wBAAwB,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;CACxD,CAAC,CAAC,CAAC"}

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

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import type { Filters } from '../../../../types/filter';
 import type { AiMetricQueryWithFilters } from '../../types';
+import type { ToolTableVizArgsTransformed } from '../tools';
 export declare const tableVizConfigSchema: z.ZodObject<{
     exploreName: z.ZodString;
     metrics: z.ZodArray<z.ZodString, "many">;
@@ -41,5 +42,10 @@ export declare const tableVizConfigSchema: z.ZodObject<{
     }[];
 }>;
 export type TableVizConfigSchemaType = z.infer<typeof tableVizConfigSchema>;
-export declare const metricQueryTableViz: (vizConfig: TableVizConfigSchemaType, filters: Filters, maxLimit: number) => AiMetricQueryWithFilters;
+export declare const metricQueryTableViz: ({ vizConfig, filters, maxLimit, customMetrics, }: {
+    vizConfig: TableVizConfigSchemaType;
+    filters: Filters;
+    maxLimit: number;
+    customMetrics: ToolTableVizArgsTransformed["customMetrics"] | null;
+}) => AiMetricQueryWithFilters;
 //# sourceMappingURL=tableViz.d.ts.map

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

@@ -1 +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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8B5B,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,wBAUD,CAAC"}
+{"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;AAI5D,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,UAAU,CAAC;AAE5D,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8B5B,CAAC;AAEN,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAE5E,eAAO,MAAM,mBAAmB,qDAK7B;IACC,SAAS,EAAE,wBAAwB,CAAC;IACpC,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,2BAA2B,CAAC,eAAe,CAAC,GAAG,IAAI,CAAC;CACtE,KAAG,wBAWF,CAAC"}

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

@@ -22,7 +22,7 @@ export const tableVizConfigSchema = z
         .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) => ({
+export const metricQueryTableViz = ({ vizConfig, filters, maxLimit, customMetrics, }) => ({
     exploreName: vizConfig.exploreName,
     metrics: vizConfig.metrics,
     dimensions: vizConfig.dimensions || [],
@@ -32,5 +32,6 @@ export const metricQueryTableViz = (vizConfig, filters, maxLimit) => ({
     })),
     limit: getValidAiQueryLimit(vizConfig.limit, maxLimit),
     filters,
+    additionalMetrics: customMetrics ?? [],
 });
 //# sourceMappingURL=tableViz.js.map

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

@@ -1 +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,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,eAAe,MAAM,cAAc,CAAC;AAE3C,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,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,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;CACV,CAAC,CAAC"}
+{"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,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,GAMhB,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;CACzC,CAAC,CAAC"}

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

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import type { Filters } from '../../../../types/filter';
 import type { AiMetricQueryWithFilters } from '../../types';
+import type { ToolTimeSeriesArgsTransformed } from '../tools';
 export declare const timeSeriesMetricVizConfigSchema: z.ZodObject<{
     exploreName: z.ZodString;
     xDimension: z.ZodString;
@@ -53,5 +54,10 @@ export declare const timeSeriesMetricVizConfigSchema: z.ZodObject<{
     yAxisLabel: string | null;
 }>;
 export type TimeSeriesMetricVizConfigSchemaType = z.infer<typeof timeSeriesMetricVizConfigSchema>;
-export declare const metricQueryTimeSeriesViz: (vizConfig: TimeSeriesMetricVizConfigSchemaType, filters: Filters, maxLimit: number) => AiMetricQueryWithFilters;
+export declare const metricQueryTimeSeriesViz: ({ vizConfig, filters, maxLimit, customMetrics, }: {
+    vizConfig: TimeSeriesMetricVizConfigSchemaType;
+    filters: Filters;
+    maxLimit: number;
+    customMetrics: ToolTimeSeriesArgsTransformed["customMetrics"] | null;
+}) => AiMetricQueryWithFilters;
 //# sourceMappingURL=timeSeriesViz.d.ts.map

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

@@ -1 +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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2C1C,CAAC;AAEH,MAAM,MAAM,mCAAmC,GAAG,CAAC,CAAC,KAAK,CACrD,OAAO,+BAA+B,CACzC,CAAC;AAEF,eAAO,MAAM,wBAAwB,cACtB,mCAAmC,WACrC,OAAO,YACN,MAAM,KACjB,wBAqBF,CAAC"}
+{"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;AAI5D,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,UAAU,CAAC;AAE9D,eAAO,MAAM,+BAA+B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2C1C,CAAC;AAEH,MAAM,MAAM,mCAAmC,GAAG,CAAC,CAAC,KAAK,CACrD,OAAO,+BAA+B,CACzC,CAAC;AAEF,eAAO,MAAM,wBAAwB,qDAKlC;IACC,SAAS,EAAE,mCAAmC,CAAC;IAC/C,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,6BAA6B,CAAC,eAAe,CAAC,GAAG,IAAI,CAAC;CACxE,KAAG,wBAsBH,CAAC"}

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

@@ -37,7 +37,7 @@ export const timeSeriesMetricVizConfigSchema = z.object({
         .nullable()
         .describe('A helpful label to explain the y-axis'),
 });
-export const metricQueryTimeSeriesViz = (vizConfig, filters, maxLimit) => {
+export const metricQueryTimeSeriesViz = ({ vizConfig, filters, maxLimit, customMetrics, }) => {
     const metrics = vizConfig.yMetrics;
     const dimensions = [
         vizConfig.xDimension,
@@ -56,6 +56,7 @@ export const metricQueryTimeSeriesViz = (vizConfig, filters, maxLimit) => {
         })),
         exploreName: vizConfig.exploreName,
         filters,
+        additionalMetrics: customMetrics ?? [],
     };
 };
 //# sourceMappingURL=timeSeriesViz.js.map

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

@@ -1 +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,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,eAAe,MAAM,cAAc,CAAC;AAE3C,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,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,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;KACV,CAAC;AACN,CAAC,CAAC"}
+{"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,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,GAMhB,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;KACzC,CAAC;AACN,CAAC,CAAC"}

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

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import type { Filters } from '../../../../types/filter';
 import type { AiMetricQueryWithFilters } from '../../types';
+import type { ToolVerticalBarArgsTransformed } from '../tools';
 export declare const verticalBarMetricVizConfigSchema: z.ZodObject<{
     exploreName: z.ZodString;
     xDimension: z.ZodString;
@@ -56,5 +57,10 @@ export declare const verticalBarMetricVizConfigSchema: z.ZodObject<{
     xAxisType: "time" | "category";
 }>;
 export type VerticalBarMetricVizConfigSchemaType = z.infer<typeof verticalBarMetricVizConfigSchema>;
-export declare const metricQueryVerticalBarViz: (vizConfig: VerticalBarMetricVizConfigSchemaType, filters: Filters, maxLimit: number) => AiMetricQueryWithFilters;
+export declare const metricQueryVerticalBarViz: ({ vizConfig, filters, maxLimit, customMetrics, }: {
+    vizConfig: VerticalBarMetricVizConfigSchemaType;
+    filters: Filters;
+    maxLimit: number;
+    customMetrics: ToolVerticalBarArgsTransformed["customMetrics"] | null;
+}) => AiMetricQueryWithFilters;
 //# sourceMappingURL=verticalBarViz.d.ts.map

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

@@ -1 +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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAmD3C,CAAC;AAEH,MAAM,MAAM,oCAAoC,GAAG,CAAC,CAAC,KAAK,CACtD,OAAO,gCAAgC,CAC1C,CAAC;AAEF,eAAO,MAAM,yBAAyB,cACvB,oCAAoC,WACtC,OAAO,YACN,MAAM,KACjB,wBAoBF,CAAC"}
+{"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;AAI5D,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,UAAU,CAAC;AAE/D,eAAO,MAAM,gCAAgC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAmD3C,CAAC;AAEH,MAAM,MAAM,oCAAoC,GAAG,CAAC,CAAC,KAAK,CACtD,OAAO,gCAAgC,CAC1C,CAAC;AAEF,eAAO,MAAM,yBAAyB,qDAKnC;IACC,SAAS,EAAE,oCAAoC,CAAC;IAChD,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,8BAA8B,CAAC,eAAe,CAAC,GAAG,IAAI,CAAC;CACzE,KAAG,wBAqBH,CAAC"}

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

@@ -41,7 +41,7 @@ export const verticalBarMetricVizConfigSchema = z.object({
         .nullable()
         .describe('A helpful label to explain the y-axis'),
 });
-export const metricQueryVerticalBarViz = (vizConfig, filters, maxLimit) => {
+export const metricQueryVerticalBarViz = ({ vizConfig, filters, maxLimit, customMetrics, }) => {
     const metrics = vizConfig.yMetrics;
     const dimensions = [
         vizConfig.xDimension,
@@ -60,6 +60,7 @@ export const metricQueryVerticalBarViz = (vizConfig, filters, maxLimit) => {
         })),
         exploreName: vizConfig.exploreName,
         filters,
+        additionalMetrics: customMetrics ?? [],
     };
 };
 //# sourceMappingURL=verticalBarViz.js.map

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

@@ -1 +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,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,eAAe,MAAM,cAAc,CAAC;AAE3C,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,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,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;KACV,CAAC;AACN,CAAC,CAAC"}
+{"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,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,GAMhB,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;KACzC,CAAC;AACN,CAAC,CAAC"}

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

@@ -1,5 +1,5 @@
 import type { Filters } from '../../types/filter';
-import type { MetricQuery } from '../../types/metricQuery';
+import type { AdditionalMetric, MetricQuery } from '../../types/metricQuery';
 import type { ToolTableVizArgs, ToolTimeSeriesArgs, ToolVerticalBarArgs } from './schemas/tools';
 export declare enum AiResultType {
     TIME_SERIES_RESULT = "time_series_chart",
@@ -7,7 +7,9 @@ export declare enum AiResultType {
     ONE_LINE_RESULT = "one_line_result",
     TABLE_RESULT = "table"
 }
-export type AiMetricQuery = Pick<MetricQuery, 'metrics' | 'dimensions' | 'sorts' | 'limit' | 'exploreName'>;
+export type AiMetricQuery = Pick<MetricQuery, 'metrics' | 'dimensions' | 'sorts' | 'limit' | 'exploreName'> & {
+    additionalMetrics: Omit<AdditionalMetric, 'sql'>[];
+};
 export type AiMetricQueryWithFilters = AiMetricQuery & {
     filters: Filters;
 };

package/dist/esm/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,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"}
+{"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,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,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"}

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

@@ -5,6 +5,14 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
         filters: import("../..").Filters;
         type: AiResultType.VERTICAL_BAR_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;
+            baseDimensionName: string;
+        }[] | null;
+        title: string;
         vizConfig: {
             limit: number | null;
             exploreName: string;
@@ -21,7 +29,6 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
             stackBars: boolean | null;
             xAxisType: "time" | "category";
         };
-        title: string;
         followUpTools: (import("./followUpTools").FollowUpTools.GENERATE_BAR_VIZ | import("./followUpTools").FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
     };
     readonly metricQuery: import("./types").AiMetricQueryWithFilters;
@@ -31,6 +38,14 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
         filters: import("../..").Filters;
         type: AiResultType.TIME_SERIES_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;
+            baseDimensionName: string;
+        }[] | null;
+        title: string;
         vizConfig: {
             limit: number | null;
             exploreName: string;
@@ -46,7 +61,6 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
             xAxisLabel: string;
             yAxisLabel: string;
         };
-        title: string;
         followUpTools: (import("./followUpTools").FollowUpTools.GENERATE_BAR_VIZ | import("./followUpTools").FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
     };
     readonly metricQuery: import("./types").AiMetricQueryWithFilters;
@@ -56,6 +70,14 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
         filters: import("../..").Filters;
         type: AiResultType.TABLE_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;
+            baseDimensionName: string;
+        }[] | null;
+        title: string;
         vizConfig: {
             dimensions: string[];
             metrics: string[];
@@ -67,7 +89,6 @@ export declare const parseVizConfig: (vizConfigUnknown: object | null, maxLimit?
                 nullsFirst: boolean | null;
             }[];
         };
-        title: string;
         followUpTools: (import("./followUpTools").FollowUpTools.GENERATE_BAR_VIZ | import("./followUpTools").FollowUpTools.GENERATE_TIME_SERIES_VIZ)[];
     };
     readonly metricQuery: import("./types").AiMetricQueryWithFilters;

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

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

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

@@ -8,7 +8,12 @@ export const parseVizConfig = (vizConfigUnknown, maxLimit) => {
     const toolVerticalBarArgsParsed = toolVerticalBarArgsSchemaTransformed.safeParse(vizConfigUnknown);
     if (toolVerticalBarArgsParsed.success) {
         const vizTool = toolVerticalBarArgsParsed.data;
-        const metricQuery = metricQueryVerticalBarViz(vizTool.vizConfig, vizTool.filters, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT);
+        const metricQuery = metricQueryVerticalBarViz({
+            vizConfig: vizTool.vizConfig,
+            filters: vizTool.filters,
+            maxLimit: maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT,
+            customMetrics: vizTool.customMetrics ?? null,
+        });
         return {
             type: AiResultType.VERTICAL_BAR_RESULT,
             vizTool,
@@ -18,7 +23,12 @@ export const parseVizConfig = (vizConfigUnknown, maxLimit) => {
     const toolTimeSeriesArgsParsed = toolTimeSeriesArgsSchemaTransformed.safeParse(vizConfigUnknown);
     if (toolTimeSeriesArgsParsed.success) {
         const vizTool = toolTimeSeriesArgsParsed.data;
-        const metricQuery = metricQueryTimeSeriesViz(vizTool.vizConfig, vizTool.filters, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT);
+        const metricQuery = metricQueryTimeSeriesViz({
+            vizConfig: vizTool.vizConfig,
+            filters: vizTool.filters,
+            maxLimit: maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT,
+            customMetrics: vizTool.customMetrics ?? null,
+        });
         return {
             type: AiResultType.TIME_SERIES_RESULT,
             vizTool,
@@ -28,7 +38,12 @@ export const parseVizConfig = (vizConfigUnknown, maxLimit) => {
     const toolTableVizArgsParsed = toolTableVizArgsSchemaTransformed.safeParse(vizConfigUnknown);
     if (toolTableVizArgsParsed.success) {
         const vizTool = toolTableVizArgsParsed.data;
-        const metricQuery = metricQueryTableViz(vizTool.vizConfig, vizTool.filters, maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT);
+        const metricQuery = metricQueryTableViz({
+            vizConfig: vizTool.vizConfig,
+            filters: vizTool.filters,
+            maxLimit: maxLimit ?? AI_DEFAULT_MAX_QUERY_LIMIT,
+            customMetrics: vizTool.customMetrics ?? null,
+        });
         return {
             type: AiResultType.TABLE_RESULT,
             vizTool,

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

@@ -1 +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,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,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,MAAM,yBAAyB,GAC3B,oCAAoC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAErE,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC;QAC/C,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;IAED,MAAM,wBAAwB,GAC1B,mCAAmC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACpE,IAAI,wBAAwB,CAAC,OAAO,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,wBAAwB,CAAC,IAAI,CAAC;QAC9C,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;IAED,MAAM,sBAAsB,GACxB,iCAAiC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAClE,IAAI,sBAAsB,CAAC,OAAO,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,sBAAsB,CAAC,IAAI,CAAC;QAC5C,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,OAAO,IAAI,CAAC;AAChB,CAAC,CAAC"}
+{"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,mBAAmB,EACnB,wBAAwB,EACxB,yBAAyB,EACzB,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,MAAM,yBAAyB,GAC3B,oCAAoC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAErE,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC;QAC/C,MAAM,WAAW,GAAG,yBAAyB,CAAC;YAC1C,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,QAAQ,IAAI,0BAA0B;YAChD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI;SAC/C,CAAC,CAAC;QACH,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,mBAAmB;YACtC,OAAO;YACP,WAAW;SACL,CAAC;IACf,CAAC;IAED,MAAM,wBAAwB,GAC1B,mCAAmC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACpE,IAAI,wBAAwB,CAAC,OAAO,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,wBAAwB,CAAC,IAAI,CAAC;QAC9C,MAAM,WAAW,GAAG,wBAAwB,CAAC;YACzC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,QAAQ,IAAI,0BAA0B;YAChD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI;SAC/C,CAAC,CAAC;QACH,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,kBAAkB;YACrC,OAAO;YACP,WAAW;SACL,CAAC;IACf,CAAC;IAED,MAAM,sBAAsB,GACxB,iCAAiC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAClE,IAAI,sBAAsB,CAAC,OAAO,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,sBAAsB,CAAC,IAAI,CAAC;QAC5C,MAAM,WAAW,GAAG,mBAAmB,CAAC;YACpC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,QAAQ,IAAI,0BAA0B;YAChD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI;SAC/C,CAAC,CAAC;QACH,OAAO;YACH,IAAI,EAAE,YAAY,CAAC,YAAY;YAC/B,OAAO;YACP,WAAW;SACL,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AAChB,CAAC,CAAC"}

package/dist/esm/utils/additionalMetrics.d.ts

@@ -1,10 +1,16 @@
 import { type CompiledTable } from '../types/explore';
-import { type Metric } from '../types/field';
+import { DimensionType, MetricType, type Metric } from '../types/field';
 import { type AdditionalMetric } from '../types/metricQuery';
 type ConvertAdditionalMetricArgs = {
     additionalMetric: AdditionalMetric;
     table: CompiledTable;
 };
 export declare const convertAdditionalMetric: ({ additionalMetric, table, }: ConvertAdditionalMetricArgs) => Metric;
+/**
+ * Get the custom metric types for a given dimension type
+ * @param type
+ * @returns
+ */
+export declare const getCustomMetricType: (type: DimensionType) => MetricType[];
 export {};
 //# sourceMappingURL=additionalMetrics.d.ts.map

package/dist/esm/utils/additionalMetrics.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"additionalMetrics.d.ts","sourceRoot":"","sources":["../../../src/utils/additionalMetrics.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,KAAK,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,KAAK,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAE7D,KAAK,2BAA2B,GAAG;IAC/B,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,KAAK,EAAE,aAAa,CAAC;CACxB,CAAC;AAEF,eAAO,MAAM,uBAAuB,iCAGjC,2BAA2B,KAAG,MAkBhC,CAAC"}
+{"version":3,"file":"additionalMetrics.d.ts","sourceRoot":"","sources":["../../../src/utils/additionalMetrics.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxE,OAAO,EAAE,KAAK,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAE7D,KAAK,2BAA2B,GAAG;IAC/B,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,KAAK,EAAE,aAAa,CAAC;CACxB,CAAC;AAEF,eAAO,MAAM,uBAAuB,iCAGjC,2BAA2B,KAAG,MAkBhC,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,mBAAmB,SAAU,aAAa,KAAG,UAAU,EA2BnE,CAAC"}