@sisense/sdk-ai-core 0.5.1 → 0.6.0

package/dist/constants.d.ts

@@ -5,7 +5,14 @@ export declare const TOOL_NAME_RECOMMEND_QUERIES = "recommendQueries";
 export declare const TOOL_NAME_GET_DASHBOARDS = "getDashboards";
 export declare const TOOL_NAME_GET_DATA_SOURCES = "getDataSources";
 export declare const TOOL_NAME_GET_DATA_SOURCE_FIELDS = "getDataSourceFields";
+export declare const TOOL_NAME_BUILD_QUERY = "buildQuery";
 export declare const TOOL_CONTEXT_MAX_ASSETS = 10;
 export declare const MAX_DATA_SOURCE_FIELDS_SUMMARY = 200;
 export declare const DEFAULT_LLM = "LLM_PLACEHOLDER";
+export declare const HEADER_USER_ACTION_ID = "x-user-action-id";
+export declare const HEADER_TRACEPARENT = "traceparent";
+export declare const HEADER_INTERFACE_NAME = "x-interface-name";
+export declare const HEADER_FEATURE_NAME = "x-feature-name";
+export declare const HEADER_QUOTA_ID = "x-quota-id";
+export declare const ALLOWED_HEADERS: string[];
 //# sourceMappingURL=constants.d.ts.map

package/dist/constants.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,uBAAuB,eAAe,CAAC;AACpD,eAAO,MAAM,2BAA2B,mBAAmB,CAAC;AAC5D,eAAO,MAAM,4BAA4B,mBAAmB,CAAC;AAC7D,eAAO,MAAM,2BAA2B,qBAAqB,CAAC;AAC9D,eAAO,MAAM,wBAAwB,kBAAkB,CAAC;AACxD,eAAO,MAAM,0BAA0B,mBAAmB,CAAC;AAC3D,eAAO,MAAM,gCAAgC,wBAAwB,CAAC;AAGtE,eAAO,MAAM,uBAAuB,KAAK,CAAC;AAG1C,eAAO,MAAM,8BAA8B,MAAM,CAAC;AAGlD,eAAO,MAAM,WAAW,oBAAoB,CAAC"}
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,uBAAuB,eAAe,CAAC;AACpD,eAAO,MAAM,2BAA2B,mBAAmB,CAAC;AAC5D,eAAO,MAAM,4BAA4B,mBAAmB,CAAC;AAC7D,eAAO,MAAM,2BAA2B,qBAAqB,CAAC;AAC9D,eAAO,MAAM,wBAAwB,kBAAkB,CAAC;AACxD,eAAO,MAAM,0BAA0B,mBAAmB,CAAC;AAC3D,eAAO,MAAM,gCAAgC,wBAAwB,CAAC;AACtE,eAAO,MAAM,qBAAqB,eAAe,CAAC;AAGlD,eAAO,MAAM,uBAAuB,KAAK,CAAC;AAG1C,eAAO,MAAM,8BAA8B,MAAM,CAAC;AAGlD,eAAO,MAAM,WAAW,oBAAoB,CAAC;AAG7C,eAAO,MAAM,qBAAqB,qBAAqB,CAAC;AACxD,eAAO,MAAM,kBAAkB,gBAAgB,CAAC;AAChD,eAAO,MAAM,qBAAqB,qBAAqB,CAAC;AACxD,eAAO,MAAM,mBAAmB,mBAAmB,CAAC;AACpD,eAAO,MAAM,eAAe,eAAe,CAAC;AAE5C,eAAO,MAAM,eAAe,UAM3B,CAAC"}

package/dist/engines/build-query/build-query-engine.d.ts

@@ -0,0 +1,11 @@
+import { z } from 'zod';
+import { buildQuerySchema, BuildQueryContext, QueryResult } from './types';
+/**
+ * Standalone build query engine - completely decoupled from any UI or store
+ * Converts a natural language query into an NLQ result and dataset
+ * @param params - The parameters containing data source title and query prompt
+ * @param context - Context with tool call ID and optional httpClient
+ * @returns The query result with queryId, nlqResult, and dataset
+ */
+export declare function buildQueryEngine(params: z.infer<typeof buildQuerySchema>, context: BuildQueryContext): Promise<QueryResult>;
+//# sourceMappingURL=build-query-engine.d.ts.map

package/dist/engines/build-query/build-query-engine.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"build-query-engine.d.ts","sourceRoot":"","sources":["../../../src/engines/build-query/build-query-engine.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,OAAO,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAE3E;;;;;;GAMG;AACH,wBAAsB,gBAAgB,CACrC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,EACxC,OAAO,EAAE,iBAAiB,GACxB,OAAO,CAAC,WAAW,CAAC,CAuCtB"}

package/dist/engines/build-query/index.d.ts

@@ -0,0 +1,3 @@
+export { buildQueryEngine } from './build-query-engine';
+export * from './types';
+//# sourceMappingURL=index.d.ts.map

package/dist/engines/build-query/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/engines/build-query/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,cAAc,SAAS,CAAC"}

package/dist/engines/build-query/types.d.ts

@@ -0,0 +1,34 @@
+import { z } from 'zod';
+import { HttpClient } from '@sisense/sdk-rest-client';
+import { NlqV3Result } from '../../services/nlq/nlq-service';
+/**
+ * Schema for buildQuery engine parameters
+ */
+export declare const buildQuerySchema: z.ZodObject<{
+    dataSourceTitle: z.ZodString;
+    queryPrompt: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    dataSourceTitle: string;
+    queryPrompt: string;
+}, {
+    dataSourceTitle: string;
+    queryPrompt: string;
+}>;
+/**
+ * Context interface for buildQueryEngine
+ */
+export interface BuildQueryContext {
+    /** Tool call ID for tracking */
+    toolCallId: string;
+    /** HTTP client for API calls (optional, uses singleton if not provided) */
+    httpClient?: HttpClient;
+    /** Optional ID generator for testability (defaults to nanoidHex(12)) */
+    generateQueryId?: () => string;
+}
+export interface QueryResult {
+    queryId: string;
+    title: string;
+    nlqResult: NlqV3Result;
+    dataset: {} | string | null;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/engines/build-query/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/engines/build-query/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAE7D;;GAEG;AACH,eAAO,MAAM,gBAAgB;;;;;;;;;EAS3B,CAAC;AAEH;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC,gCAAgC;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,2EAA2E;IAC3E,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,wEAAwE;IACxE,eAAe,CAAC,EAAE,MAAM,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,WAAW;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,WAAW,CAAC;IACvB,OAAO,EAAE,EAAE,GAAG,MAAM,GAAG,IAAI,CAAC;CAC5B"}

package/dist/engines/chart-builder/chart-builder-engine.d.ts

@@ -1,14 +1,15 @@
-import { ChartSummary, BuildChartContext, buildChartSchema } from './types';
+import { ChartSummary, BuildChartContext, buildChartSchema, buildChartSchemaNaturalConversation } from './types';
+import { QueryResult } from '../build-query/types';
 import { default as z } from 'zod';
 /**
  * Build contextual system prompt for chart builder
  */
-export declare function buildContextualSystemPrompt(chartSummaries: ChartSummary[]): string;
+export declare function buildContextualSystemPrompt(chartSummaries: ChartSummary[], querySummaries?: QueryResult[]): string;
 /**
  * Standalone chart builder engine - completely decoupled from any UI or store
  * @param params - The parameters for the chart builder
  * @param context - All external dependencies injected
  * @returns The chart summary
  */
-export declare function buildChartEngine(params: z.infer<typeof buildChartSchema>, context: BuildChartContext): Promise<ChartSummary>;
+export declare function buildChartEngine(params: z.infer<typeof buildChartSchema> | z.infer<typeof buildChartSchemaNaturalConversation>, context: BuildChartContext): Promise<ChartSummary>;
 //# sourceMappingURL=chart-builder-engine.d.ts.map

package/dist/engines/chart-builder/chart-builder-engine.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chart-builder-engine.d.ts","sourceRoot":"","sources":["../../../src/engines/chart-builder/chart-builder-engine.ts"],"names":[],"mappings":"AACA,OAAO,EACN,YAAY,EAUZ,iBAAiB,EACjB,gBAAgB,EAChB,MAAM,SAAS,CAAC;AAajB,OAAO,CAAC,MAAM,KAAK,CAAC;AAqHpB;;GAEG;AACH,wBAAgB,2BAA2B,CAAC,cAAc,EAAE,YAAY,EAAE,UA4BzE;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CACrC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,EACxC,OAAO,EAAE,iBAAiB,GACxB,OAAO,CAAC,YAAY,CAAC,CAoEvB"}
+{"version":3,"file":"chart-builder-engine.d.ts","sourceRoot":"","sources":["../../../src/engines/chart-builder/chart-builder-engine.ts"],"names":[],"mappings":"AACA,OAAO,EACN,YAAY,EAUZ,iBAAiB,EACjB,gBAAgB,EAChB,mCAAmC,EACnC,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAanD,OAAO,CAAC,MAAM,KAAK,CAAC;AAqHpB;;GAEG;AACH,wBAAgB,2BAA2B,CAC1C,cAAc,EAAE,YAAY,EAAE,EAC9B,cAAc,CAAC,EAAE,WAAW,EAAE,UAkC9B;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CACrC,MAAM,EACH,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,GAChC,CAAC,CAAC,KAAK,CAAC,OAAO,mCAAmC,CAAC,EACtD,OAAO,EAAE,iBAAiB,GACxB,OAAO,CAAC,YAAY,CAAC,CAsEvB"}

package/dist/engines/chart-builder/create-chart/create-chart.d.ts

@@ -6,9 +6,10 @@ type GenerateChartParams = {
     chartPrompt: string;
     chartType: NullableChartType;
     chartTitle: string;
+    queryId?: string | null;
 };
-export declare const generateChart: ({ dataSourceTitle, queryPrompt, chartPrompt, chartType, chartTitle, }: GenerateChartParams, context: BuildChartContext) => Promise<ExtendedChartWidgetProps>;
-export declare const createChart: ({ queryPrompt, chartPrompt, chartType, chartTitle, }: z.infer<typeof createChartSchema>, context: BuildChartContext) => Promise<{
+export declare const generateChart: ({ dataSourceTitle, queryPrompt, chartPrompt, chartType, chartTitle, queryId, }: GenerateChartParams, context: BuildChartContext) => Promise<ExtendedChartWidgetProps>;
+export declare const createChart: ({ queryPrompt, chartPrompt, chartType, chartTitle, queryId: paramsQueryId, }: z.infer<typeof createChartSchema>, context: BuildChartContext) => Promise<{
     message: string;
 }>;
 export {};

package/dist/engines/chart-builder/create-chart/create-chart.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"create-chart.d.ts","sourceRoot":"","sources":["../../../../src/engines/chart-builder/create-chart/create-chart.ts"],"names":[],"mappings":"AAKA,OAAO,EACN,iBAAiB,EACjB,iBAAiB,EACjB,wBAAwB,EACxB,iBAAiB,EACjB,MAAM,UAAU,CAAC;AAKlB,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,KAAK,mBAAmB,GAAG;IAC1B,eAAe,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,iBAAiB,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,eAAO,MAAM,aAAa,GACzB,uEAMG,mBAAmB,EACtB,SAAS,iBAAiB,KACxB,OAAO,CAAC,wBAAwB,CA8ClC,CAAC;AAEF,eAAO,MAAM,WAAW,GACvB,sDAKG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,EACpC,SAAS,iBAAiB;;EA4B1B,CAAC"}
+{"version":3,"file":"create-chart.d.ts","sourceRoot":"","sources":["../../../../src/engines/chart-builder/create-chart/create-chart.ts"],"names":[],"mappings":"AAKA,OAAO,EACN,iBAAiB,EACjB,iBAAiB,EACjB,wBAAwB,EACxB,iBAAiB,EACjB,MAAM,UAAU,CAAC;AAKlB,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,KAAK,mBAAmB,GAAG;IAC1B,eAAe,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,iBAAiB,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB,CAAC;AAEF,eAAO,MAAM,aAAa,GACzB,gFAOG,mBAAmB,EACtB,SAAS,iBAAiB,KACxB,OAAO,CAAC,wBAAwB,CAgDlC,CAAC;AAEF,eAAO,MAAM,WAAW,GACvB,8EAMG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,EACpC,SAAS,iBAAiB;;EAiC1B,CAAC"}

package/dist/engines/chart-builder/create-chart/generate-chart-table.d.ts

@@ -1,4 +1,5 @@
 import { ExtendedChartWidgetProps } from '../types';
 import { HttpClient } from '@sisense/sdk-rest-client';
-export declare const generateChartTable: (dataSourceTitle: string, queryPrompt: string, isNlqV3Enabled?: boolean, httpClient?: HttpClient) => Promise<ExtendedChartWidgetProps>;
+import { QueryResult } from '../../build-query/types';
+export declare const generateChartTable: (dataSourceTitle: string, queryPrompt: string, isNlqV3Enabled?: boolean, httpClient?: HttpClient, queryId?: string | null, retrieveQuery?: (queryId: string) => QueryResult | null) => Promise<ExtendedChartWidgetProps>;
 //# sourceMappingURL=generate-chart-table.d.ts.map

package/dist/engines/chart-builder/create-chart/generate-chart-table.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generate-chart-table.d.ts","sourceRoot":"","sources":["../../../../src/engines/chart-builder/create-chart/generate-chart-table.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AAOpD,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AA4HtD,eAAO,MAAM,kBAAkB,GAC9B,iBAAiB,MAAM,EACvB,aAAa,MAAM,EACnB,iBAAiB,OAAO,EACxB,aAAa,UAAU,sCAyBvB,CAAC"}
+{"version":3,"file":"generate-chart-table.d.ts","sourceRoot":"","sources":["../../../../src/engines/chart-builder/create-chart/generate-chart-table.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AAOpD,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAQtD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAwItD,eAAO,MAAM,kBAAkB,GAC9B,iBAAiB,MAAM,EACvB,aAAa,MAAM,EACnB,iBAAiB,OAAO,EACxB,aAAa,UAAU,EACvB,UAAU,MAAM,GAAG,IAAI,EACvB,gBAAgB,CAAC,OAAO,EAAE,MAAM,KAAK,WAAW,GAAG,IAAI,sCA2BvD,CAAC"}

package/dist/engines/chart-builder/types.d.ts

@@ -4,6 +4,7 @@ import { HttpClient } from '@sisense/sdk-rest-client';
 import { default as OpenAI } from 'openai';
 import { z } from 'zod';
 import { NlqResponseData } from '@sisense/sdk-ui/ai';
+import { QueryResult } from '../build-query/types';
 export type ArtifactReference = {
     artifactId?: string;
 };
@@ -64,12 +65,12 @@ export declare const axesMapping: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     breakBy: z.ZodArray<z.ZodNumber, "many">;
 }, "strict", z.ZodTypeAny, {
     value: number[];
-    type: "line" | "area" | "bar" | "column" | "polar";
+    type: "area" | "column" | "line" | "bar" | "polar";
     category: number[];
     breakBy: number[];
 }, {
     value: number[];
-    type: "line" | "area" | "bar" | "column" | "polar";
+    type: "area" | "column" | "line" | "bar" | "polar";
     category: number[];
     breakBy: number[];
 }>, z.ZodObject<{
@@ -192,21 +193,22 @@ export declare const axesMapping: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     locationLevel: z.ZodEnum<["auto", "country", "state", "city"]>;
 }, "strict", z.ZodTypeAny, {
     type: "scattermap";
+    details: number[];
     size: number[];
     geo: number[];
     colorBy: number[];
-    details: number[];
     locationLevel: "auto" | "country" | "state" | "city";
 }, {
     type: "scattermap";
+    details: number[];
     size: number[];
     geo: number[];
     colorBy: number[];
-    details: number[];
     locationLevel: "auto" | "country" | "state" | "city";
 }>]>;
 export declare const nullableChartType: z.ZodNullable<z.ZodEnum<["bar", "column", "area", "line", "polar", "pie", "funnel", "treemap", "sunburst", "scatter", "table", "indicator", "boxplot", "areamap", "scattermap", "arearange"]>>;
 export declare const chartIdSchema: z.ZodDefault<z.ZodString>;
+export declare const queryIdSchema: z.ZodDefault<z.ZodNullable<z.ZodString>>;
 export declare const chartRecommendationsSchema: z.ZodObject<{
     chartType: z.ZodEnum<["bar", "column", "area", "line", "polar", "pie", "funnel", "treemap", "sunburst", "scatter", "table", "indicator", "boxplot", "areamap", "scattermap", "arearange"]>;
     axesMapping: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
@@ -216,12 +218,12 @@ export declare const chartRecommendationsSchema: z.ZodObject<{
         breakBy: z.ZodArray<z.ZodNumber, "many">;
     }, "strict", z.ZodTypeAny, {
         value: number[];
-        type: "line" | "area" | "bar" | "column" | "polar";
+        type: "area" | "column" | "line" | "bar" | "polar";
         category: number[];
         breakBy: number[];
     }, {
         value: number[];
-        type: "line" | "area" | "bar" | "column" | "polar";
+        type: "area" | "column" | "line" | "bar" | "polar";
         category: number[];
         breakBy: number[];
     }>, z.ZodObject<{
@@ -344,24 +346,24 @@ export declare const chartRecommendationsSchema: z.ZodObject<{
         locationLevel: z.ZodEnum<["auto", "country", "state", "city"]>;
     }, "strict", z.ZodTypeAny, {
         type: "scattermap";
+        details: number[];
         size: number[];
         geo: number[];
         colorBy: number[];
-        details: number[];
         locationLevel: "auto" | "country" | "state" | "city";
     }, {
         type: "scattermap";
+        details: number[];
         size: number[];
         geo: number[];
         colorBy: number[];
-        details: number[];
         locationLevel: "auto" | "country" | "state" | "city";
     }>]>;
 }, "strip", z.ZodTypeAny, {
-    chartType: "line" | "area" | "bar" | "column" | "polar" | "pie" | "funnel" | "treemap" | "sunburst" | "scatter" | "indicator" | "areamap" | "boxplot" | "scattermap" | "arearange" | "table";
+    chartType: "area" | "table" | "indicator" | "column" | "line" | "pie" | "bar" | "polar" | "funnel" | "treemap" | "sunburst" | "scatter" | "areamap" | "boxplot" | "scattermap" | "arearange";
     axesMapping: {
         value: number[];
-        type: "line" | "area" | "bar" | "column" | "polar";
+        type: "area" | "column" | "line" | "bar" | "polar";
         category: number[];
         breakBy: number[];
     } | {
@@ -403,17 +405,17 @@ export declare const chartRecommendationsSchema: z.ZodObject<{
         color: number[];
     } | {
         type: "scattermap";
+        details: number[];
         size: number[];
         geo: number[];
         colorBy: number[];
-        details: number[];
         locationLevel: "auto" | "country" | "state" | "city";
     };
 }, {
-    chartType: "line" | "area" | "bar" | "column" | "polar" | "pie" | "funnel" | "treemap" | "sunburst" | "scatter" | "indicator" | "areamap" | "boxplot" | "scattermap" | "arearange" | "table";
+    chartType: "area" | "table" | "indicator" | "column" | "line" | "pie" | "bar" | "polar" | "funnel" | "treemap" | "sunburst" | "scatter" | "areamap" | "boxplot" | "scattermap" | "arearange";
     axesMapping: {
         value: number[];
-        type: "line" | "area" | "bar" | "column" | "polar";
+        type: "area" | "column" | "line" | "bar" | "polar";
         category: number[];
         breakBy: number[];
     } | {
@@ -455,10 +457,10 @@ export declare const chartRecommendationsSchema: z.ZodObject<{
         color: number[];
     } | {
         type: "scattermap";
+        details: number[];
         size: number[];
         geo: number[];
         colorBy: number[];
-        details: number[];
         locationLevel: "auto" | "country" | "state" | "city";
     };
 }>;
@@ -472,21 +474,37 @@ export declare const buildChartSchema: z.ZodObject<{
     dataSourceTitle: string;
     userPrompt: string;
 }>;
+export declare const buildChartSchemaNaturalConversation: z.ZodObject<{
+    dataSourceTitle: z.ZodString;
+    userPrompt: z.ZodString;
+    queryId: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+}, "strip", z.ZodTypeAny, {
+    dataSourceTitle: string;
+    userPrompt: string;
+    queryId?: string | null | undefined;
+}, {
+    dataSourceTitle: string;
+    userPrompt: string;
+    queryId?: string | null | undefined;
+}>;
 export declare const createChartSchema: z.ZodObject<{
     queryPrompt: z.ZodString;
     chartPrompt: z.ZodString;
     chartType: z.ZodNullable<z.ZodEnum<["bar", "column", "area", "line", "polar", "pie", "funnel", "treemap", "sunburst", "scatter", "table", "indicator", "boxplot", "areamap", "scattermap", "arearange"]>>;
     chartTitle: z.ZodString;
+    queryId: z.ZodDefault<z.ZodNullable<z.ZodString>>;
 }, "strip", z.ZodTypeAny, {
-    chartType: "line" | "area" | "bar" | "column" | "polar" | "pie" | "funnel" | "treemap" | "sunburst" | "scatter" | "indicator" | "areamap" | "boxplot" | "scattermap" | "arearange" | "table" | null;
     queryPrompt: string;
+    chartType: "area" | "table" | "indicator" | "column" | "line" | "pie" | "bar" | "polar" | "funnel" | "treemap" | "sunburst" | "scatter" | "areamap" | "boxplot" | "scattermap" | "arearange" | null;
+    queryId: string | null;
     chartPrompt: string;
     chartTitle: string;
 }, {
-    chartType: "line" | "area" | "bar" | "column" | "polar" | "pie" | "funnel" | "treemap" | "sunburst" | "scatter" | "indicator" | "areamap" | "boxplot" | "scattermap" | "arearange" | "table" | null;
     queryPrompt: string;
+    chartType: "area" | "table" | "indicator" | "column" | "line" | "pie" | "bar" | "polar" | "funnel" | "treemap" | "sunburst" | "scatter" | "areamap" | "boxplot" | "scattermap" | "arearange" | null;
     chartPrompt: string;
     chartTitle: string;
+    queryId?: string | null | undefined;
 }>;
 export declare const updateChartSchema: z.ZodObject<{
     userPrompt: z.ZodString;
@@ -521,11 +539,11 @@ export declare const updateChartTypeSchema: z.ZodObject<{
 } & {
     chartType: z.ZodEnum<["bar", "column", "area", "line", "polar", "pie", "funnel", "treemap", "sunburst", "scatter", "table", "indicator", "boxplot", "areamap", "scattermap", "arearange"]>;
 }, "strip", z.ZodTypeAny, {
-    chartType: "line" | "area" | "bar" | "column" | "polar" | "pie" | "funnel" | "treemap" | "sunburst" | "scatter" | "indicator" | "areamap" | "boxplot" | "scattermap" | "arearange" | "table";
+    chartType: "area" | "table" | "indicator" | "column" | "line" | "pie" | "bar" | "polar" | "funnel" | "treemap" | "sunburst" | "scatter" | "areamap" | "boxplot" | "scattermap" | "arearange";
     userPrompt: string;
     chartId: string;
 }, {
-    chartType: "line" | "area" | "bar" | "column" | "polar" | "pie" | "funnel" | "treemap" | "sunburst" | "scatter" | "indicator" | "areamap" | "boxplot" | "scattermap" | "arearange" | "table";
+    chartType: "area" | "table" | "indicator" | "column" | "line" | "pie" | "bar" | "polar" | "funnel" | "treemap" | "sunburst" | "scatter" | "areamap" | "boxplot" | "scattermap" | "arearange";
     userPrompt: string;
     chartId?: string | undefined;
 }>;
@@ -598,6 +616,8 @@ export interface BuildChartContext {
     dataSourceTitle: string;
     /** Chart summaries to be used as context */
     chartSummaries?: ChartSummary[];
+    /** query results to be used as context */
+    querySummaries?: QueryResult[];
     /**
      * Retrieve a chart by its ID
      * @param chartId - The ID of the chart to retrieve
@@ -610,9 +630,17 @@ export interface BuildChartContext {
      * @param chartWidgetProps - The chart widget props to save
      */
     saveChart?: (chartId: string, chartWidgetProps: ExtendedChartWidgetProps) => void;
+    /**
+     * Retrieve a query result by its ID
+     * @param queryId - The ID of the query to retrieve
+     * @returns The query result, or null if not found
+     */
+    retrieveQuery?: (queryId: string) => QueryResult | null;
     /** Whether NLQ V3 is enabled */
     isNlqV3Enabled?: boolean;
     httpClient?: HttpClient;
     openAIClient?: OpenAI;
+    /** ID of a prior query result when this chart uses that query; otherwise null/undefined */
+    queryId?: string | null;
 }
 //# sourceMappingURL=types.d.ts.map

package/dist/engines/chart-builder/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/engines/chart-builder/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EACN,SAAS,EACT,gBAAgB,EAChB,YAAY,EACZ,mBAAmB,EACnB,qBAAqB,EACrB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD,MAAM,MAAM,iBAAiB,GAAG;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAGF,MAAM,MAAM,wBAAwB,GAAG,qBAAqB,CAC3D,gBAAgB,EAChB,OAAO,CACP,GACA,iBAAiB,GAAG;IACnB;;OAEG;IACH,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC;IAC3B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;CAC9B,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG,SAAS,GAAG,IAAI,GAAG,SAAS,CAAC;AAE7D;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC9B,wCAAwC;IACxC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,iDAAiD;IACjD,eAAe,EAAE,MAAM,CAAC;IACxB,2CAA2C;IAC3C,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,iCAAiC;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,4CAA4C;IAC5C,WAAW,EAAE,MAAM,CAAC;IACpB,4CAA4C;IAC5C,YAAY,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,YAAY;IAC5B,mDAAmD;IACnD,OAAO,EAAE,MAAM,CAAC;IAChB,6DAA6D;IAC7D,SAAS,EAAE,MAAM,CAAC;IAClB,mEAAmE;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,wDAAwD;IACxD,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,uDAAuD;IACvD,OAAO,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,MAAM,UAAU,GACnB,YAAY,GACZ,mBAAmB,GACnB,MAAM,GACN,aAAa,GACb,uBAAuB,CAAC;AAE3B,MAAM,WAAW,MAAM;IACtB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;CACd;AAgBD,eAAO,MAAM,SAAS,iLAIpB,CAAC;AAwFH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAUtB,CAAC;AAEH,eAAO,MAAM,iBAAiB,gMAI5B,CAAC;AAEH,eAAO,MAAM,aAAa,2BAKO,CAAC;AAElC,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGrC,CAAC;AAEH,eAAO,MAAM,gBAAgB;;;;;;;;;EAS3B,CAAC;AAUH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;EAa5B,CAAC;AAEH,eAAO,MAAM,iBAAiB;;;;;;;;;;EAE5B,CAAC;AAEH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;EAEjC,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;EAEhC,CAAC;AAEH,eAAO,MAAM,+BAA+B;;;;;;;;;;EAAoB,CAAC;AAEjE,eAAO,MAAM,wBAAwB;;;;;;;;;;EAAoB,CAAC;AAE1D,eAAO,MAAM,4BAA4B;;;;;;;;;;EAAoB,CAAC;AAE9D,eAAO,MAAM,iCAAiC;;;;;;;;;;EAAoB,CAAC;AAEnE,eAAO,MAAM,6BAA6B;;;;;;;;;;EAAoB,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,eAAe,EAAE,MAAM,CAAC;IACxB,4CAA4C;IAC5C,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC;IAChC;;;;OAIG;IACH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,wBAAwB,GAAG,IAAI,CAAC;IACrE;;;;OAIG;IACH,SAAS,CAAC,EAAE,CACX,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,wBAAwB,KACtC,IAAI,CAAC;IACV,gCAAgC;IAChC,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/engines/chart-builder/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EACN,SAAS,EACT,gBAAgB,EAChB,YAAY,EACZ,mBAAmB,EACnB,qBAAqB,EACrB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAGnD,MAAM,MAAM,iBAAiB,GAAG;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAGF,MAAM,MAAM,wBAAwB,GAAG,qBAAqB,CAC3D,gBAAgB,EAChB,OAAO,CACP,GACA,iBAAiB,GAAG;IACnB;;OAEG;IACH,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC;IAC3B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;CAC9B,CAAC;AAEH,MAAM,MAAM,iBAAiB,GAAG,SAAS,GAAG,IAAI,GAAG,SAAS,CAAC;AAE7D;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC9B,wCAAwC;IACxC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,iDAAiD;IACjD,eAAe,EAAE,MAAM,CAAC;IACxB,2CAA2C;IAC3C,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB,iCAAiC;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,4CAA4C;IAC5C,WAAW,EAAE,MAAM,CAAC;IACpB,4CAA4C;IAC5C,YAAY,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,YAAY;IAC5B,mDAAmD;IACnD,OAAO,EAAE,MAAM,CAAC;IAChB,6DAA6D;IAC7D,SAAS,EAAE,MAAM,CAAC;IAClB,mEAAmE;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,wDAAwD;IACxD,KAAK,CAAC,EAAE,gBAAgB,CAAC;IACzB,uDAAuD;IACvD,OAAO,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,MAAM,UAAU,GACnB,YAAY,GACZ,mBAAmB,GACnB,MAAM,GACN,aAAa,GACb,uBAAuB,CAAC;AAE3B,MAAM,WAAW,MAAM;IACtB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;CACd;AAgBD,eAAO,MAAM,SAAS,iLAIpB,CAAC;AAwFH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAUtB,CAAC;AAEH,eAAO,MAAM,iBAAiB,gMAI5B,CAAC;AAEH,eAAO,MAAM,aAAa,2BAKO,CAAC;AAElC,eAAO,MAAM,aAAa,0CAMxB,CAAC;AAEH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGrC,CAAC;AAEH,eAAO,MAAM,gBAAgB;;;;;;;;;EAS3B,CAAC;AAEH,eAAO,MAAM,mCAAmC;;;;;;;;;;;;EAgB9C,CAAC;AAUH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;EAc5B,CAAC;AAEH,eAAO,MAAM,iBAAiB;;;;;;;;;;EAE5B,CAAC;AAEH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;EAEjC,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;EAEhC,CAAC;AAEH,eAAO,MAAM,+BAA+B;;;;;;;;;;EAAoB,CAAC;AAEjE,eAAO,MAAM,wBAAwB;;;;;;;;;;EAAoB,CAAC;AAE1D,eAAO,MAAM,4BAA4B;;;;;;;;;;EAAoB,CAAC;AAE9D,eAAO,MAAM,iCAAiC;;;;;;;;;;EAAoB,CAAC;AAEnE,eAAO,MAAM,6BAA6B;;;;;;;;;;EAAoB,CAAC;AAE/D;;GAEG;AACH,MAAM,WAAW,iBAAiB;IACjC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB;;OAEG;IACH,eAAe,EAAE,MAAM,CAAC;IACxB,4CAA4C;IAC5C,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC;IAChC,0CAA0C;IAC1C,cAAc,CAAC,EAAE,WAAW,EAAE,CAAC;IAC/B;;;;OAIG;IACH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,wBAAwB,GAAG,IAAI,CAAC;IACrE;;;;OAIG;IACH,SAAS,CAAC,EAAE,CACX,OAAO,EAAE,MAAM,EACf,gBAAgB,EAAE,wBAAwB,KACtC,IAAI,CAAC;IACV;;;;OAIG;IACH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,WAAW,GAAG,IAAI,CAAC;IACxD,gCAAgC;IAChC,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,2FAA2F;IAC3F,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB"}