@lightdash/common 0.2104.7 → 0.2104.8

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

@@ -48,9 +48,9 @@ export declare const baseAgentSchema: z.ZodObject<{
     name: string;
     description: string;
     tags: string[] | null;
+    model: string;
     uuid: string;
     organizationUuid: string;
-    model: string;
     userAccess: string[];
     version: number;
     imageUrl: string | null;
@@ -70,9 +70,9 @@ export declare const baseAgentSchema: z.ZodObject<{
     name: string;
     description: string;
     tags: string[] | null;
+    model: string;
     uuid: string;
     organizationUuid: string;
-    model: string;
     userAccess: string[];
     version: number;
     imageUrl: string | null;

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

@@ -0,0 +1,43 @@
+import { z } from 'zod';
+export * from './adminTypes';
+export * from './chartConfig/slack';
+export * from './chartConfig/web';
+export * from './constants';
+export * from './filterExploreByTags';
+export * from './followUpTools';
+export * from './requestTypes';
+export * from './schemas';
+export * from './types';
+export * from './utils';
+export * from './validators';
+export const baseAgentSchema = z.object({
+    uuid: z.string(),
+    projectUuid: z.string(),
+    organizationUuid: z.string(),
+    name: z.string(),
+    description: z.string(),
+    imageUrl: z.string().url().nullable(),
+    tags: z.array(z.string()).nullable(),
+    integrations: z.array(
+    // z.union([
+    // TODO: once we add more integrations, we should use union
+    z.object({
+        type: z.literal('slack'),
+        channelId: z.string(),
+    })),
+    createdAt: z.coerce.date(),
+    updatedAt: z.coerce.date(),
+    instruction: z
+        .string()
+        .max(8192, // 8kb
+    'Custom instruction is too long. Maximum allowed is 8,100 characters.')
+        .nullable(),
+    provider: z.string(),
+    model: z.string(),
+    groupAccess: z.array(z.string()),
+    userAccess: z.array(z.string()),
+    enableDataAccess: z.boolean(),
+    enableSelfImprovement: z.boolean(),
+    version: z.number(),
+});
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAoBxB,cAAc,cAAc,CAAC;AAC7B,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,aAAa,CAAC;AAC5B,cAAc,uBAAuB,CAAC;AACtC,cAAc,iBAAiB,CAAC;AAChC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,WAAW,CAAC;AAC1B,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,cAAc,CAAC;AAE7B,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE;IAE5B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;IAErC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAEpC,YAAY,EAAE,CAAC,CAAC,KAAK;IACjB,YAAY;IACZ,2DAA2D;IAC3D,CAAC,CAAC,MAAM,CAAC;QACL,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QACxB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;KACxB,CAAC,CAEL;IAED,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IAC1B,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IAE1B,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,GAAG,CACA,IAAI,EAAE,MAAM;IACZ,sEAAsE,CACzE;SACA,QAAQ,EAAE;IACf,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAChC,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC/B,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE;IAC7B,qBAAqB,EAAE,CAAC,CAAC,OAAO,EAAE;IAClC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC"}

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

@@ -0,0 +1,2 @@
+export const isSlackPrompt = (prompt) => 'slackUserId' in prompt;
+//# sourceMappingURL=requestTypes.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"requestTypes.js","sourceRoot":"","sources":["../../../../src/ee/AiAgent/requestTypes.ts"],"names":[],"mappings":"AAsDA,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,MAAgB,EAAyB,EAAE,CACrE,aAAa,IAAI,MAAM,CAAC"}

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

@@ -0,0 +1,50 @@
+import { z } from 'zod';
+import { MetricType } from '../../../types/field';
+export const customMetricBaseSchema = z.object({
+    name: z
+        .string()
+        .describe('Unique metric name using snake_case (e.g., "avg_customer_age", "total_revenue")'),
+    label: z
+        .string()
+        .describe('Human-readable label for the metric (e.g., "Average Customer Age", "Total Revenue")'),
+    description: z
+        .string()
+        .describe('Brief explanation of what the metric represents, how it is calculated, and why it matters. Example: "Calculates the total revenue by summing all transaction amounts in the sales table."'),
+    baseDimensionName: z
+        .string()
+        .describe('Name of the base dimension/column this metric calculates from'),
+    table: z
+        .string()
+        .describe('Table name where the base column exists. Match with available dimensions in the explore.'),
+    type: z
+        .enum([
+        MetricType.AVERAGE,
+        MetricType.COUNT,
+        MetricType.COUNT_DISTINCT,
+        MetricType.MAX,
+        MetricType.MIN,
+        MetricType.SUM,
+        MetricType.PERCENTILE,
+        MetricType.MEDIAN,
+    ])
+        .describe(`Choose based on the user's request. If the base dimension type is STRING, TIMESTAMP, DATE, BOOLEAN, use COUNT_DISTINCT, COUNT, MIN, MAX. If the base dimension type is NUMBER, use MIN, MAX, SUM, PERCENTILE, MEDIAN, AVERAGE, COUNT_DISTINCT, COUNT. If the base dimension type is BOOLEAN, use COUNT_DISTINCT, COUNT.`),
+});
+export const customMetricsSchema = z
+    .array(customMetricBaseSchema)
+    .nullable()
+    .describe(`Create custom metrics when requested metrics don't exist. Only create if no existing metric matches the user's request. Use null if no custom metrics needed.
+
+IMPORTANT: If the user requests metrics that don't exist (like "average customer age"), create them using the customMetrics field. Analyze available dimensions from findFields results and create appropriate SQL aggregations.
+
+When using custom metrics:
+1. Create the metric in customMetrics array with just the metric name (e.g., "avg_customer_age")
+2. Reference it in metrics array using the format "table_metricname" (e.g., "customers_avg_customer_age")
+3. Reference it in sorts array using the format "table_metricname" (e.g., "customers_avg_customer_age")
+4. DO NOT use the raw metric name in metrics or sorts arrays
+
+For example:
+- "Show me average customer age sorted descending" →
+customMetrics: [{name: "avg_customer_age", label: "Average Customer Age", type: "AVERAGE", baseDimensionName: "age", table: "customers"}]
+metrics: ["customers_avg_customer_age"]
+sorts: [{fieldId: "customers_avg_customer_age", descending: true}]`);
+//# sourceMappingURL=customMetrics.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"customMetrics.js","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/schemas/customMetrics.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAElD,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,IAAI,EAAE,CAAC;SACF,MAAM,EAAE;SACR,QAAQ,CACL,iFAAiF,CACpF;IACL,KAAK,EAAE,CAAC;SACH,MAAM,EAAE;SACR,QAAQ,CACL,qFAAqF,CACxF;IACL,WAAW,EAAE,CAAC;SACT,MAAM,EAAE;SACR,QAAQ,CACL,2LAA2L,CAC9L;IACL,iBAAiB,EAAE,CAAC;SACf,MAAM,EAAE;SACR,QAAQ,CACL,+DAA+D,CAClE;IACL,KAAK,EAAE,CAAC;SACH,MAAM,EAAE;SACR,QAAQ,CACL,0FAA0F,CAC7F;IACL,IAAI,EAAE,CAAC;SACF,IAAI,CAAC;QACF,UAAU,CAAC,OAAO;QAClB,UAAU,CAAC,KAAK;QAChB,UAAU,CAAC,cAAc;QACzB,UAAU,CAAC,GAAG;QACd,UAAU,CAAC,GAAG;QACd,UAAU,CAAC,GAAG;QACd,UAAU,CAAC,UAAU;QACrB,UAAU,CAAC,MAAM;KACpB,CAAC;SACD,QAAQ,CACL,yTAAyT,CAC5T;CACR,CAAC,CAAC;AAIH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC;KAC/B,KAAK,CAAC,sBAAsB,CAAC;KAC7B,QAAQ,EAAE;KACV,QAAQ,CACL;;;;;;;;;;;;;;mEAc2D,CAC9D,CAAC"}

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

@@ -0,0 +1,11 @@
+import { z } from 'zod';
+const getFieldIdSchema = (args) => z
+    .string()
+    .describe([
+    args.additionalDescription,
+    '"fieldId" must come from the previously searched Fields; otherwise, it will throw an error',
+]
+    .filter(Boolean)
+    .join(' '));
+export { getFieldIdSchema };
+//# sourceMappingURL=fieldId.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"fieldId.js","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/schemas/fieldId.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,gBAAgB,GAAG,CAAC,IAA8C,EAAE,EAAE,CACxE,CAAC;KACI,MAAM,EAAE;KACR,QAAQ,CACL;IACI,IAAI,CAAC,qBAAqB;IAC1B,4FAA4F;CAC/F;KACI,MAAM,CAAC,OAAO,CAAC;KACf,IAAI,CAAC,GAAG,CAAC,CACjB,CAAC;AAEV,OAAO,EAAE,gBAAgB,EAAE,CAAC"}

package/dist/types/ee/AiAgent/schemas/filters/booleanFilters.js

@@ -0,0 +1,29 @@
+import { z } from 'zod';
+import { DimensionType, MetricType } from '../../../../types/field';
+import { FilterOperator, FilterType } from '../../../../types/filter';
+import { getFieldIdSchema } from '../fieldId';
+const commonBooleanFilterRuleSchema = z.object({
+    fieldId: getFieldIdSchema({ additionalDescription: null }),
+    fieldType: z.union([
+        z.literal(DimensionType.BOOLEAN),
+        z.literal(MetricType.BOOLEAN),
+    ]),
+    fieldFilterType: z.literal(FilterType.BOOLEAN),
+});
+const booleanFilterSchema = z.union([
+    commonBooleanFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.NULL),
+            z.literal(FilterOperator.NOT_NULL),
+        ]),
+    }),
+    commonBooleanFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.EQUALS),
+            z.literal(FilterOperator.NOT_EQUALS),
+        ]),
+        values: z.array(z.boolean()).length(1),
+    }),
+]);
+export default booleanFilterSchema;
+//# sourceMappingURL=booleanFilters.js.map

package/dist/types/ee/AiAgent/schemas/filters/booleanFilters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"booleanFilters.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/filters/booleanFilters.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,OAAO,EAAE,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;IAC1D,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC;QACf,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC;QAChC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;KAChC,CAAC;IACF,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;CACjD,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC;IAChC,6BAA6B,CAAC,MAAM,CAAC;QACjC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC;YAC9B,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC;SACrC,CAAC;KACL,CAAC;IACF,6BAA6B,CAAC,MAAM,CAAC;QACjC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC;YAChC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC;SACvC,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;KACzC,CAAC;CACL,CAAC,CAAC;AAEH,eAAe,mBAAmB,CAAC"}

package/dist/types/ee/AiAgent/schemas/filters/dateFilters.js

@@ -0,0 +1,84 @@
+import { z } from 'zod';
+import { DimensionType, MetricType } from '../../../../types/field';
+import { FilterOperator, FilterType, UnitOfTime, } from '../../../../types/filter';
+import { getFieldIdSchema } from '../fieldId';
+const dateOrDateTimeSchema = z.union([
+    z.string().date(),
+    z.string().datetime(),
+]);
+const commonDateFilterRuleSchema = z.object({
+    fieldId: getFieldIdSchema({ additionalDescription: null }),
+    fieldType: z.union([
+        z.literal(DimensionType.DATE),
+        z.literal(DimensionType.TIMESTAMP),
+        z.literal(MetricType.DATE),
+        z.literal(MetricType.TIMESTAMP),
+    ]),
+    fieldFilterType: z.literal(FilterType.DATE),
+});
+const dateFilterSchema = z.union([
+    commonDateFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.NULL),
+            z.literal(FilterOperator.NOT_NULL),
+        ]),
+    }),
+    commonDateFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.EQUALS),
+            z.literal(FilterOperator.NOT_EQUALS),
+        ]),
+        values: z.array(dateOrDateTimeSchema),
+    }),
+    commonDateFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.IN_THE_PAST),
+            z.literal(FilterOperator.NOT_IN_THE_PAST),
+            z.literal(FilterOperator.IN_THE_NEXT),
+            // NOTE: NOT_IN_THE_NEXT is not supported...
+        ]),
+        values: z.array(z.number()).length(1),
+        settings: z.object({
+            completed: z.boolean(),
+            unitOfTime: z.union([
+                z.literal(UnitOfTime.days),
+                z.literal(UnitOfTime.weeks),
+                z.literal(UnitOfTime.months),
+                z.literal(UnitOfTime.quarters),
+                z.literal(UnitOfTime.years),
+            ]),
+        }),
+    }),
+    commonDateFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.IN_THE_CURRENT),
+            z.literal(FilterOperator.NOT_IN_THE_CURRENT),
+        ]),
+        values: z.array(z.literal(1)).length(1),
+        settings: z.object({
+            completed: z.literal(false),
+            unitOfTime: z.union([
+                z.literal(UnitOfTime.days),
+                z.literal(UnitOfTime.weeks),
+                z.literal(UnitOfTime.months),
+                z.literal(UnitOfTime.quarters),
+                z.literal(UnitOfTime.years),
+            ]),
+        }),
+    }),
+    commonDateFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.LESS_THAN),
+            z.literal(FilterOperator.LESS_THAN_OR_EQUAL),
+            z.literal(FilterOperator.GREATER_THAN),
+            z.literal(FilterOperator.GREATER_THAN_OR_EQUAL),
+        ]),
+        values: z.array(dateOrDateTimeSchema).length(1),
+    }),
+    commonDateFilterRuleSchema.extend({
+        operator: z.literal(FilterOperator.IN_BETWEEN),
+        values: z.array(dateOrDateTimeSchema).length(2),
+    }),
+]);
+export default dateFilterSchema;
+//# sourceMappingURL=dateFilters.js.map

package/dist/types/ee/AiAgent/schemas/filters/dateFilters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dateFilters.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/filters/dateFilters.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EACH,cAAc,EACd,UAAU,EACV,UAAU,GACb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC;IACjC,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;IACjB,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACxB,CAAC,CAAC;AAEH,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,OAAO,EAAE,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;IAC1D,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC;QACf,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC;QAC7B,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC;QAClC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;QAC1B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC;KAClC,CAAC;IACF,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;CAC9C,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7B,0BAA0B,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC;YAC9B,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC;SACrC,CAAC;KACL,CAAC;IACF,0BAA0B,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC;YAChC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC;SACvC,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC;KACxC,CAAC;IACF,0BAA0B,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC;YACrC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,eAAe,CAAC;YACzC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC;YACrC,4CAA4C;SAC/C,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;QACrC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC;YACf,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE;YACtB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC;gBAChB,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;gBAC1B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC3B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;gBAC5B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAC9B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;aAC9B,CAAC;SACL,CAAC;KACL,CAAC;IACF,0BAA0B,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC;YACxC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,kBAAkB,CAAC;SAC/C,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;QACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC;YACf,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;YAC3B,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC;gBAChB,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;gBAC1B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC3B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;gBAC5B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAC9B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;aAC9B,CAAC;SACL,CAAC;KACL,CAAC;IACF,0BAA0B,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC;YACnC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,kBAAkB,CAAC;YAC5C,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,YAAY,CAAC;YACtC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,qBAAqB,CAAC;SAClD,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;KAClD,CAAC;IACF,0BAA0B,CAAC,MAAM,CAAC;QAC9B,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC;QAC9C,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;KAClD,CAAC;CACL,CAAC,CAAC;AAEH,eAAe,gBAAgB,CAAC"}

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

@@ -0,0 +1,98 @@
+import { v4 as uuid } from 'uuid';
+import { z } from 'zod';
+import assertUnreachable from '../../../../utils/assertUnreachable';
+import booleanFilterSchema from './booleanFilters';
+import dateFilterSchema from './dateFilters';
+import numberFilterSchema from './numberFilters';
+import stringFilterSchema from './stringFilters';
+export { booleanFilterSchema, dateFilterSchema, numberFilterSchema, stringFilterSchema, };
+const filterAndOrSchema = z
+    .union([z.literal('and'), z.literal('or')])
+    .describe('Type of filter group operation');
+const filterRuleSchema = z.union([
+    booleanFilterSchema,
+    stringFilterSchema,
+    numberFilterSchema,
+    dateFilterSchema,
+]);
+const filterRuleSchemaTransformed = filterRuleSchema.transform((data) => ({
+    id: uuid(),
+    target: {
+        fieldId: data.fieldId,
+        fieldFilterType: data.fieldFilterType,
+    },
+    operator: data.operator,
+    values: 'values' in data ? data.values : [],
+    ...('settings' in data ? { settings: data.settings } : {}),
+}));
+// TODO: deprecate this in 2 weeks
+const filtersSchemaV1 = z.object({
+    type: filterAndOrSchema,
+    dimensions: z.array(filterRuleSchema).nullable(),
+    metrics: z.array(filterRuleSchema).nullable(),
+});
+export const filtersSchemaV2 = filtersSchemaV1.extend({
+    tableCalculations: z.array(numberFilterSchema).nullable(),
+});
+const filtersSchemaAndFilterRulesTransformedV1 = z.object({
+    type: filterAndOrSchema,
+    dimensions: z.array(filterRuleSchemaTransformed).nullable(),
+    metrics: z.array(filterRuleSchemaTransformed).nullable(),
+});
+const filtersSchemaAndFilterRulesTransformedV2 = filtersSchemaAndFilterRulesTransformedV1.extend({
+    tableCalculations: z.array(filterRuleSchemaTransformed).nullable(),
+});
+const filtersSchemaAndFilterRulesTransformed = z
+    .union([
+    filtersSchemaAndFilterRulesTransformedV2,
+    filtersSchemaAndFilterRulesTransformedV1,
+])
+    .nullable();
+export const filtersSchemaTransformed = filtersSchemaAndFilterRulesTransformed.transform((data) => {
+    if (!data) {
+        return {
+            dimensions: { id: uuid(), and: [] },
+            metrics: { id: uuid(), and: [] },
+            tableCalculations: { id: uuid(), and: [] },
+        };
+    }
+    switch (data.type) {
+        case 'and':
+            return {
+                dimensions: {
+                    id: uuid(),
+                    and: data.dimensions ?? [],
+                },
+                metrics: {
+                    id: uuid(),
+                    and: data.metrics ?? [],
+                },
+                tableCalculations: {
+                    id: uuid(),
+                    and: 'tableCalculations' in data
+                        ? data.tableCalculations ?? []
+                        : [],
+                },
+            };
+        case 'or':
+            return {
+                dimensions: {
+                    id: uuid(),
+                    or: data.dimensions ?? [],
+                },
+                metrics: {
+                    id: uuid(),
+                    or: data.metrics ?? [],
+                },
+                tableCalculations: {
+                    id: uuid(),
+                    or: 'tableCalculations' in data
+                        ? data.tableCalculations ?? []
+                        : [],
+                },
+            };
+        default:
+            return assertUnreachable(data.type, 'Invalid filter type');
+    }
+});
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/filters/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,IAAI,IAAI,EAAE,MAAM,MAAM,CAAC;AAClC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAOxB,OAAO,iBAAiB,MAAM,qCAAqC,CAAC;AACpE,OAAO,mBAAmB,MAAM,kBAAkB,CAAC;AACnD,OAAO,gBAAgB,MAAM,eAAe,CAAC;AAC7C,OAAO,kBAAkB,MAAM,iBAAiB,CAAC;AACjD,OAAO,kBAAkB,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EACH,mBAAmB,EACnB,gBAAgB,EAChB,kBAAkB,EAClB,kBAAkB,GACrB,CAAC;AAEF,MAAM,iBAAiB,GAAG,CAAC;KACtB,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;KAC1C,QAAQ,CAAC,gCAAgC,CAAC,CAAC;AAEhD,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7B,mBAAmB;IACnB,kBAAkB;IAClB,kBAAkB;IAClB,gBAAgB;CACnB,CAAC,CAAC;AAOH,MAAM,2BAA2B,GAAG,gBAAgB,CAAC,SAAS,CAC1D,CAAC,IAAI,EAAgB,EAAE,CAAC,CAAC;IACrB,EAAE,EAAE,IAAI,EAAE;IACV,MAAM,EAAE;QACJ,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,eAAe,EAAE,IAAI,CAAC,eAAe;KACxC;IACD,QAAQ,EAAE,IAAI,CAAC,QAAQ;IACvB,MAAM,EAAE,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;IAC3C,GAAG,CAAC,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;CAC7D,CAAC,CACL,CAAC;AAEF,kCAAkC;AAClC,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7B,IAAI,EAAE,iBAAiB;IACvB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,QAAQ,EAAE;IAChD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,QAAQ,EAAE;CAChD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC;IAClD,iBAAiB,EAAE,CAAC,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,QAAQ,EAAE;CAC5D,CAAC,CAAC;AAEH,MAAM,wCAAwC,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,IAAI,EAAE,iBAAiB;IACvB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,QAAQ,EAAE;IAC3D,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,QAAQ,EAAE;CAC3D,CAAC,CAAC;AAEH,MAAM,wCAAwC,GAC1C,wCAAwC,CAAC,MAAM,CAAC;IAC5C,iBAAiB,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,QAAQ,EAAE;CACrE,CAAC,CAAC;AAEP,MAAM,sCAAsC,GAAG,CAAC;KAC3C,KAAK,CAAC;IACH,wCAAwC;IACxC,wCAAwC;CAC3C,CAAC;KACD,QAAQ,EAAE,CAAC;AAEhB,MAAM,CAAC,MAAM,wBAAwB,GACjC,sCAAsC,CAAC,SAAS,CAAC,CAAC,IAAI,EAAW,EAAE;IAC/D,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,OAAO;YACH,UAAU,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;YACnC,OAAO,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;YAChC,iBAAiB,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;SAC7C,CAAC;IACN,CAAC;IACD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,KAAK;YACN,OAAO;gBACH,UAAU,EAAE;oBACR,EAAE,EAAE,IAAI,EAAE;oBACV,GAAG,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;iBAC7B;gBACD,OAAO,EAAE;oBACL,EAAE,EAAE,IAAI,EAAE;oBACV,GAAG,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;iBAC1B;gBACD,iBAAiB,EAAE;oBACf,EAAE,EAAE,IAAI,EAAE;oBACV,GAAG,EACC,mBAAmB,IAAI,IAAI;wBACvB,CAAC,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE;wBAC9B,CAAC,CAAC,EAAE;iBACf;aACJ,CAAC;QACN,KAAK,IAAI;YACL,OAAO;gBACH,UAAU,EAAE;oBACR,EAAE,EAAE,IAAI,EAAE;oBACV,EAAE,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;iBAC5B;gBACD,OAAO,EAAE;oBACL,EAAE,EAAE,IAAI,EAAE;oBACV,EAAE,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;iBACzB;gBACD,iBAAiB,EAAE;oBACf,EAAE,EAAE,IAAI,EAAE;oBACV,EAAE,EACE,mBAAmB,IAAI,IAAI;wBACvB,CAAC,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE;wBAC9B,CAAC,CAAC,EAAE;iBACf;aACJ,CAAC;QACN;YACI,OAAO,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;IACnE,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/filters/numberFilters.js

@@ -0,0 +1,53 @@
+import { z } from 'zod';
+import { DimensionType, MetricType } from '../../../../types/field';
+import { FilterOperator, FilterType } from '../../../../types/filter';
+import { getFieldIdSchema } from '../fieldId';
+const commonNumberFilterRuleSchema = z.object({
+    fieldId: getFieldIdSchema({ additionalDescription: null }),
+    fieldType: z.union([
+        z.literal(DimensionType.NUMBER),
+        z.literal(MetricType.NUMBER),
+        z.literal(MetricType.PERCENTILE),
+        z.literal(MetricType.MEDIAN),
+        z.literal(MetricType.AVERAGE),
+        z.literal(MetricType.COUNT),
+        z.literal(MetricType.COUNT_DISTINCT),
+        z.literal(MetricType.SUM),
+        z.literal(MetricType.MIN),
+        z.literal(MetricType.MAX),
+    ]),
+    fieldFilterType: z.literal(FilterType.NUMBER),
+});
+const numberFilterSchema = z.union([
+    commonNumberFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.NULL),
+            z.literal(FilterOperator.NOT_NULL),
+        ]),
+    }),
+    commonNumberFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.EQUALS),
+            z.literal(FilterOperator.NOT_EQUALS),
+        ]),
+        values: z.array(z.number()),
+    }),
+    commonNumberFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.LESS_THAN),
+            z.literal(FilterOperator.LESS_THAN_OR_EQUAL),
+            z.literal(FilterOperator.GREATER_THAN),
+            z.literal(FilterOperator.GREATER_THAN_OR_EQUAL),
+        ]),
+        values: z.array(z.number()).length(1),
+    }),
+    commonNumberFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.IN_BETWEEN),
+            z.literal(FilterOperator.NOT_IN_BETWEEN),
+        ]),
+        values: z.array(z.number()).length(2),
+    }),
+]);
+export default numberFilterSchema;
+//# sourceMappingURL=numberFilters.js.map

package/dist/types/ee/AiAgent/schemas/filters/numberFilters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"numberFilters.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/filters/numberFilters.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,4BAA4B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,OAAO,EAAE,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;IAC1D,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC;QACf,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC;QAC/B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;QAC5B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC;QAChC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;QAC5B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;QAC7B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC;QAC3B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC;QACpC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QACzB,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QACzB,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;KAC5B,CAAC;IACF,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;CAChD,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC/B,4BAA4B,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC;YAC9B,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC;SACrC,CAAC;KACL,CAAC;IACF,4BAA4B,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC;YAChC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC;SACvC,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC9B,CAAC;IACF,4BAA4B,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC;YACnC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,kBAAkB,CAAC;YAC5C,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,YAAY,CAAC;YACtC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,qBAAqB,CAAC;SAClD,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;KACxC,CAAC;IACF,4BAA4B,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC;YACpC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC;SAC3C,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;KACxC,CAAC;CACL,CAAC,CAAC;AAEH,eAAe,kBAAkB,CAAC"}

package/dist/types/ee/AiAgent/schemas/filters/stringFilters.js

@@ -0,0 +1,33 @@
+import { z } from 'zod';
+import { DimensionType, MetricType } from '../../../../types/field';
+import { FilterOperator, FilterType } from '../../../../types/filter';
+import { getFieldIdSchema } from '../fieldId';
+const commonStringFilterRuleSchema = z.object({
+    fieldId: getFieldIdSchema({ additionalDescription: null }),
+    fieldType: z.union([
+        z.literal(DimensionType.STRING),
+        z.literal(MetricType.STRING),
+    ]),
+    fieldFilterType: z.literal(FilterType.STRING),
+});
+const stringFilterSchema = z.union([
+    commonStringFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.NULL),
+            z.literal(FilterOperator.NOT_NULL),
+        ]),
+    }),
+    commonStringFilterRuleSchema.extend({
+        operator: z.union([
+            z.literal(FilterOperator.EQUALS),
+            z.literal(FilterOperator.NOT_EQUALS),
+            z.literal(FilterOperator.STARTS_WITH),
+            z.literal(FilterOperator.ENDS_WITH),
+            z.literal(FilterOperator.INCLUDE),
+            z.literal(FilterOperator.NOT_INCLUDE),
+        ]),
+        values: z.array(z.string()),
+    }),
+]);
+export default stringFilterSchema;
+//# sourceMappingURL=stringFilters.js.map

package/dist/types/ee/AiAgent/schemas/filters/stringFilters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stringFilters.js","sourceRoot":"","sources":["../../../../../../src/ee/AiAgent/schemas/filters/stringFilters.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,4BAA4B,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,OAAO,EAAE,gBAAgB,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC;IAC1D,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC;QACf,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC;QAC/B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;KAC/B,CAAC;IACF,eAAe,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC;CAChD,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC/B,4BAA4B,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC;YAC9B,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC;SACrC,CAAC;KACL,CAAC;IACF,4BAA4B,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC;YACd,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC;YAChC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC;YACpC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC;YACrC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC;YACnC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC;YACjC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,WAAW,CAAC;SACxC,CAAC;QACF,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC9B,CAAC;CACL,CAAC,CAAC;AAEH,eAAe,kBAAkB,CAAC"}

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

@@ -0,0 +1,9 @@
+export * from './customMetrics';
+export * from './filters';
+export * from './outputMetadata';
+export * from './parser';
+export * from './sortField';
+export * from './tableCalcs/tableCalcs';
+export * from './tools';
+export * from './visualizations';
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/schemas/index.ts"],"names":[],"mappings":"AAgBA,cAAc,iBAAiB,CAAC;AAChC,cAAc,WAAW,CAAC;AAC1B,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AACzB,cAAc,aAAa,CAAC;AAC5B,cAAc,yBAAyB,CAAC;AACxC,cAAc,SAAS,CAAC;AACxB,cAAc,kBAAkB,CAAC"}

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

@@ -0,0 +1,5 @@
+import { z } from 'zod';
+export const baseOutputMetadataSchema = z.object({
+    status: z.enum(['success', 'error']),
+});
+//# sourceMappingURL=outputMetadata.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"outputMetadata.js","sourceRoot":"","sources":["../../../../../src/ee/AiAgent/schemas/outputMetadata.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;CACvC,CAAC,CAAC"}

package/dist/types/ee/AiAgent/schemas/parser.d.ts

@@ -3052,14 +3052,6 @@ export declare const parseToolArgs: (toolName: ToolName, toolArgs: unknown) => i
     })[] | null;
     type: import("..").AiResultType.QUERY_RESULT;
     description: string;
-    customMetrics: {
-        name: string;
-        label: string;
-        table: string;
-        type: import("../../..").MetricType.PERCENTILE | import("../../..").MetricType.AVERAGE | import("../../..").MetricType.COUNT | import("../../..").MetricType.COUNT_DISTINCT | import("../../..").MetricType.SUM | import("../../..").MetricType.MIN | import("../../..").MetricType.MAX | import("../../..").MetricType.MEDIAN;
-        description: string;
-        baseDimensionName: string;
-    }[] | null;
     chartConfig: {
         groupBy: string[] | null;
         stackBars: boolean | null;
@@ -3074,6 +3066,14 @@ export declare const parseToolArgs: (toolName: ToolName, toolArgs: unknown) => i
         secondaryYAxisMetric: string | null;
         secondaryYAxisLabel: string | null;
     } | null;
+    customMetrics: {
+        name: string;
+        label: string;
+        table: string;
+        type: import("../../..").MetricType.PERCENTILE | import("../../..").MetricType.AVERAGE | import("../../..").MetricType.COUNT | import("../../..").MetricType.COUNT_DISTINCT | import("../../..").MetricType.SUM | import("../../..").MetricType.MIN | import("../../..").MetricType.MAX | import("../../..").MetricType.MEDIAN;
+        description: string;
+        baseDimensionName: string;
+    }[] | null;
     title: string;
     queryConfig: {
         dimensions: string[];
@@ -3363,14 +3363,6 @@ export declare const parseToolArgs: (toolName: ToolName, toolArgs: unknown) => i
             frameType: "rows" | "range";
         } | null;
     })[] | null | undefined;
-    customMetrics?: {
-        name: string;
-        label: string;
-        table: string;
-        type: import("../../..").MetricType.PERCENTILE | import("../../..").MetricType.AVERAGE | import("../../..").MetricType.COUNT | import("../../..").MetricType.COUNT_DISTINCT | import("../../..").MetricType.SUM | import("../../..").MetricType.MIN | import("../../..").MetricType.MAX | import("../../..").MetricType.MEDIAN;
-        description: string;
-        baseDimensionName: string;
-    }[] | null | undefined;
     chartConfig?: {
         groupBy: string[] | null;
         stackBars: boolean | null;
@@ -3385,6 +3377,14 @@ export declare const parseToolArgs: (toolName: ToolName, toolArgs: unknown) => i
         secondaryYAxisMetric: string | null;
         secondaryYAxisLabel: string | null;
     } | null | undefined;
+    customMetrics?: {
+        name: string;
+        label: string;
+        table: string;
+        type: import("../../..").MetricType.PERCENTILE | import("../../..").MetricType.AVERAGE | import("../../..").MetricType.COUNT | import("../../..").MetricType.COUNT_DISTINCT | import("../../..").MetricType.SUM | import("../../..").MetricType.MIN | import("../../..").MetricType.MAX | import("../../..").MetricType.MEDIAN;
+        description: string;
+        baseDimensionName: string;
+    }[] | null | undefined;
 }> | import("zod").SafeParseSuccess<{
     visualizations: {
         filters: import("../../..").Filters;
@@ -3447,14 +3447,6 @@ export declare const parseToolArgs: (toolName: ToolName, toolArgs: unknown) => i
         })[] | null;
         type: import("..").AiResultType.QUERY_RESULT;
         description: string;
-        customMetrics: {
-            name: string;
-            label: string;
-            table: string;
-            type: import("../../..").MetricType.PERCENTILE | import("../../..").MetricType.AVERAGE | import("../../..").MetricType.COUNT | import("../../..").MetricType.COUNT_DISTINCT | import("../../..").MetricType.SUM | import("../../..").MetricType.MIN | import("../../..").MetricType.MAX | import("../../..").MetricType.MEDIAN;
-            description: string;
-            baseDimensionName: string;
-        }[] | null;
         chartConfig: {
             groupBy: string[] | null;
             stackBars: boolean | null;
@@ -3469,6 +3461,14 @@ export declare const parseToolArgs: (toolName: ToolName, toolArgs: unknown) => i
             secondaryYAxisMetric: string | null;
             secondaryYAxisLabel: string | null;
         } | null;
+        customMetrics: {
+            name: string;
+            label: string;
+            table: string;
+            type: import("../../..").MetricType.PERCENTILE | import("../../..").MetricType.AVERAGE | import("../../..").MetricType.COUNT | import("../../..").MetricType.COUNT_DISTINCT | import("../../..").MetricType.SUM | import("../../..").MetricType.MIN | import("../../..").MetricType.MAX | import("../../..").MetricType.MEDIAN;
+            description: string;
+            baseDimensionName: string;
+        }[] | null;
         title: string;
         queryConfig: {
             dimensions: string[];

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

@@ -0,0 +1,43 @@
+import assertUnreachable from '../../../utils/assertUnreachable';
+import { toolDashboardArgsSchemaTransformed, toolDashboardV2ArgsSchemaTransformed, toolFindChartsArgsSchemaTransformed, toolFindContentArgsSchema, toolFindDashboardsArgsSchemaTransformed, toolFindExploresArgsSchemaTransformed, toolFindFieldsArgsSchemaTransformed, toolImproveContextArgsSchema, toolProposeChangeArgsSchema, toolRunQueryArgsSchemaTransformed, toolSearchFieldValuesArgsSchemaTransformed, toolTableVizArgsSchemaTransformed, toolTimeSeriesArgsSchemaTransformed, toolVerticalBarArgsSchemaTransformed, } from './tools';
+/**
+ * Parse the tool args using the specific schema for each tool name
+ * @param toolName - The name of the tool to parse
+ * @param toolArgs - The tool args to parse
+ * @returns The parsed tool args
+ */
+export const parseToolArgs = (toolName, toolArgs) => {
+    switch (toolName) {
+        case 'findExplores':
+            return toolFindExploresArgsSchemaTransformed.safeParse(toolArgs);
+        case 'findFields':
+            return toolFindFieldsArgsSchemaTransformed.safeParse(toolArgs);
+        case 'findContent':
+            return toolFindContentArgsSchema.safeParse(toolArgs);
+        case 'searchFieldValues':
+            return toolSearchFieldValuesArgsSchemaTransformed.safeParse(toolArgs);
+        case 'generateBarVizConfig':
+            return toolVerticalBarArgsSchemaTransformed.safeParse(toolArgs);
+        case 'generateTableVizConfig':
+            return toolTableVizArgsSchemaTransformed.safeParse(toolArgs);
+        case 'generateTimeSeriesVizConfig':
+            return toolTimeSeriesArgsSchemaTransformed.safeParse(toolArgs);
+        case 'generateDashboard':
+            return toolDashboardArgsSchemaTransformed.safeParse(toolArgs);
+        case 'generateDashboardV2':
+            return toolDashboardV2ArgsSchemaTransformed.safeParse(toolArgs);
+        case 'findDashboards':
+            return toolFindDashboardsArgsSchemaTransformed.safeParse(toolArgs);
+        case 'findCharts':
+            return toolFindChartsArgsSchemaTransformed.safeParse(toolArgs);
+        case 'improveContext':
+            return toolImproveContextArgsSchema.safeParse(toolArgs);
+        case 'proposeChange':
+            return toolProposeChangeArgsSchema.safeParse(toolArgs);
+        case 'runQuery':
+            return toolRunQueryArgsSchemaTransformed.safeParse(toolArgs);
+        default:
+            return assertUnreachable(toolName, `Unknown tool name: ${toolName}`);
+    }
+};
+//# sourceMappingURL=parser.js.map