@finos/legend-lego 2.0.188 → 2.0.189

package/lib/legend-ai/LegendAITypes.js

@@ -0,0 +1,281 @@
+/**
+ * Copyright (c) 2026-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { LegendAI_LegendApplicationPlugin_Extension, } from './LegendAI_LegendApplicationPlugin_Extension.js';
+export class TDSColumnSchema {
+    name;
+    type;
+    documentation;
+    sampleValues;
+}
+export var TDSServiceSourceType;
+(function (TDSServiceSourceType) {
+    TDSServiceSourceType["SERVICE"] = "service";
+    TDSServiceSourceType["ACCESS_POINT"] = "accessPoint";
+})(TDSServiceSourceType || (TDSServiceSourceType = {}));
+export class TDSServiceSchema {
+    title;
+    description;
+    pattern;
+    columns;
+    parameters;
+    /**
+     * Indicates the source of this service schema.
+     * - SERVICE: traditional DataSpace service executable (uses `FROM service(...)` SQL syntax)
+     * - ACCESS_POINT: data product access point (uses `FROM p(...)` SQL syntax with lakehouse runtime)
+     */
+    sourceType;
+    /** Full data product path (e.g. 'my::package::DataProduct'), used with `p()` syntax for access points. */
+    dataProductPath;
+}
+export class LegendAIConfig {
+    enabled;
+    llmServiceUrl;
+    llmModelName;
+    sqlExecutionUrl;
+    orchestratorUrl;
+    marketplaceSearchUrl;
+    engineUrl;
+    orchestratorAuthToken;
+    maxJudgeAttempts;
+    /** Lakehouse runtime environment name (e.g. 'prod', 'uat'). Defaults to 'prod' when not set. */
+    lakehouseEnvironment;
+}
+export const DEFAULT_LEGEND_AI_CONFIG = Object.freeze({
+    enabled: false,
+    llmServiceUrl: undefined,
+    llmModelName: undefined,
+    sqlExecutionUrl: undefined,
+    orchestratorUrl: undefined,
+    marketplaceSearchUrl: undefined,
+    engineUrl: undefined,
+});
+export function findLegendAIPlugin(plugins) {
+    return plugins.find((p) => p instanceof LegendAI_LegendApplicationPlugin_Extension);
+}
+export class LegendAIGridData {
+    columnDefs;
+    rowData;
+}
+export function buildColumnDefsFromNames(columns) {
+    return columns.map((col) => ({
+        colId: col,
+        headerName: col,
+        field: col,
+    }));
+}
+export var LegendAIThinkingStepStatus;
+(function (LegendAIThinkingStepStatus) {
+    LegendAIThinkingStepStatus["ACTIVE"] = "active";
+    LegendAIThinkingStepStatus["DONE"] = "done";
+    LegendAIThinkingStepStatus["ERROR"] = "error";
+})(LegendAIThinkingStepStatus || (LegendAIThinkingStepStatus = {}));
+export class LegendAIThinkingStep {
+    label;
+    status;
+}
+export var LegendAIMessageRole;
+(function (LegendAIMessageRole) {
+    LegendAIMessageRole["USER"] = "user";
+    LegendAIMessageRole["ASSISTANT"] = "assistant";
+})(LegendAIMessageRole || (LegendAIMessageRole = {}));
+export class LegendAIUserMessage {
+    id;
+    role;
+    text;
+}
+export class LegendAIAssistantMessage {
+    id;
+    role;
+    thinkingSteps;
+    sql;
+    textAnswer;
+    gridData;
+    error;
+    sqlGenTime;
+    execTime;
+    thinkingDuration;
+    isProcessing;
+    isExecuting;
+    suggestedQueries;
+}
+export class LegendAIConversationTurn {
+    question;
+    sql;
+    /**
+     * The intent classification of this turn, allowing downstream prompts
+     * to differentiate prior metadata answers from SQL query results.
+     */
+    intent;
+}
+export class LegendAIAccessPointGroupInfo {
+    title;
+    description;
+    accessPoints;
+}
+export class LegendAIProductMetadata {
+    name;
+    description;
+    coordinates;
+    serviceSummaries;
+    accessPointGroups;
+    tags;
+    supportInfo;
+}
+export var LegendAIQuestionIntent;
+(function (LegendAIQuestionIntent) {
+    LegendAIQuestionIntent["DATA_QUERY"] = "data_query";
+    LegendAIQuestionIntent["METADATA"] = "metadata";
+    LegendAIQuestionIntent["ORCHESTRATOR"] = "orchestrator";
+})(LegendAIQuestionIntent || (LegendAIQuestionIntent = {}));
+export const METADATA_SIGNAL_PATTERNS = Object.freeze([
+    /\b(?:what\s+does\s+this\s+(?:data\s*product|dataspace|data\s*space|product)\s+(?:do|provide|contain|offer|have))\b/,
+    /\b(?:describe|explain|summarize|summary|overview)\s+(?:this\s+)?(?:data\s*product|dataspace|data\s*space|product)\b/,
+    /\b(?:tell\s+me\s+about)\s+(?:this\s+)?(?:data\s*product|dataspace|data\s*space|product)\b/,
+    /\b(?:what\s+(?:services?|endpoints?|access\s*points?|columns?|fields?)\s+(?:are|is|does))\b/,
+    /\b(?:list|what\s+are)\s+(?:the\s+)?(?:available\s+)?(?:services?|endpoints?|capabilities)\b/,
+    /\b(?:list|what\s+are)\s+(?:the\s+)?(?:available\s+)?access\s*points?\b/,
+    /\b(?:who\s+(?:owns?|maintains?|created?|manages?|supports?))\b/,
+    /\b(?:owner(?:ship)?|maintainer|contact\s+(?:info|email|team|details)|support\s+(?:info|email|team))\b/,
+    /\b(?:classifications?|tags?|stereotypes?)\s+(?:of|for|on)\s+(?:this|the)\b/,
+    /\bsummarize\s+what\s+(?:you|this|it)\s+(?:have|has|provide|offers?|contains?)\b/,
+    /\bwhat\s+(?:do\s+you|does\s+it|does\s+this)\s+(?:have|provide|offer|contain)\b/,
+    /\b(?:how\s+many)\s+(?:services?|endpoints?|access\s*points?|columns?|fields?)\b/,
+    /\b(?:what\s+(?:type|kind|format))\s+(?:of|is)\b/,
+    /^(?:summarize|describe|explain|overview)\b/,
+    /\b(?:summary|overview|description)\s+(?:of|about)\b/,
+    /\bwhat\s+(?:is|are)\s+(?:this|the|it)\b/,
+    /\btell\s+me\s+(?:about|more)\b/,
+    /\bwhat\s+(?:\S+\s+)*(?:provides?|offers?|contains?|includes?)\s*$/,
+    /\bhow\s+does\s+(?:\S+\s+)*(?:work|function|operate)\b/,
+    /\bwhat\s+can\s+(?:i|we|you)\b/,
+    /\b(?:used\s+for|meant\s+for|designed\s+for|purpose\s+of)\b/,
+    /\b(?:help\s+me\s+(?:understand|with))\b/,
+    /\bwhat\s+(?:information|data|content|datasets?)\s+(?:is|are|does)\b/,
+    /\b(?:what\s+does)\s+\S+(?:\s+\S+)*\s+(?:do|provide|offer|contain|include|cover)\b/,
+]);
+export const DATA_QUERY_SIGNAL_PATTERNS = Object.freeze([
+    /\b(?:select|query|sql|rows?|records?|count|sum|avg|average|min|max|total)\b/,
+    /\b(?:top\s+\d+|first\s+\d+|last\s+\d+|limit\s+\d+)\b/,
+    /\b(?:filter|where|group\s+by|order\s+by|sort|join|aggregate)\b/,
+    /\b(?:distinct|unique)\s+(?:values?|entries?|items?)/,
+    /\bfrom\s+(?:\S+\s+)*service\b/,
+    /\b(?:show|give|get|fetch|retrieve|pull|find|provide|display|return)\s+(?:me\s+)?/,
+    /\b(?:compare|comparison|versus|vs\.?)\b/,
+    /\b(?:what\s+percentage|what\s+%|what\s+proportion|what\s+fraction)\b/,
+    /\b(?:which\s+\w+s?\s+(?:does|do|did|has|have|is|are|generate|earn|produce))\b/,
+    /\b(?:how\s+(?:much|many|often))\b/,
+    /\b\d{4}[-/]\d{2}[-/]\d{2}\b/,
+    /\b(?:as\s+of|since|before|after|between|from|until|latest|recent|yesterday|today)\b/,
+    /\blast\s+(?:year|month|week|quarter|fiscal)\b/,
+    /\b(?:fiscal\s+(?:year|quarter)|fy\s*\d|q[1-4]\s*\d{4})\b/,
+    /\b(?:sedol|isin|cusip|ticker|symbol)[\s:=]*\w+/,
+    /\b(?:revenue|sales|income|profit|earnings|margin|growth|volume|price|rate|exposure)\b/,
+    /\b(?:breakdown|distribution|composition|split|allocation|attribution|ranking|trend)\b/,
+    /\b(?:per\s+(?:country|region|year|month|quarter|segment|category))\b/,
+    /\b(?:grouped?\s+by|broken?\s+down\s+by|split\s+by|segmented?\s+by)\b/,
+]);
+const PRODUCT_REFERENCE_PATTERN = /\b(?:this|the)\s+(?:data\s*product|dataspace|data\s*space|product)\b/;
+const STRUCTURAL_KEYWORD_PATTERN = /\b(?:services?|endpoints?|access\s*points?|capabilities|owner|maintainer|support)\b/;
+function countPatternMatches(question, patterns) {
+    let count = 0;
+    for (const pattern of patterns) {
+        if (pattern.test(question)) {
+            count++;
+        }
+    }
+    return count;
+}
+/**
+ * Result of the fast deterministic regex-based classifier.
+ * Includes the scores and resolved intent so callers can decide
+ * whether the classification is confident or needs LLM arbitration.
+ */
+export class QuestionIntentClassification {
+    intent;
+    metaScore;
+    dataScore;
+    ambiguous;
+}
+/**
+ * Fast deterministic regex classifier (sync, < 1ms).
+ * Returns both the resolved intent AND whether the result is ambiguous,
+ * so callers can decide to escalate to an LLM for ambiguous cases.
+ */
+export function classifyQuestionIntentFast(question, hasServices) {
+    const q = question.toLowerCase().trim();
+    const metaScore = countPatternMatches(q, METADATA_SIGNAL_PATTERNS);
+    const dataScore = countPatternMatches(q, DATA_QUERY_SIGNAL_PATTERNS);
+    if (metaScore > 0 && dataScore === 0) {
+        return {
+            intent: LegendAIQuestionIntent.METADATA,
+            metaScore,
+            dataScore,
+            ambiguous: false,
+        };
+    }
+    if (dataScore > 0 && metaScore === 0) {
+        return {
+            intent: LegendAIQuestionIntent.DATA_QUERY,
+            metaScore,
+            dataScore,
+            ambiguous: false,
+        };
+    }
+    if (metaScore > 0 && dataScore > 0) {
+        if (dataScore >= metaScore * 2) {
+            return {
+                intent: LegendAIQuestionIntent.DATA_QUERY,
+                metaScore,
+                dataScore,
+                ambiguous: false,
+            };
+        }
+        if (PRODUCT_REFERENCE_PATTERN.test(q) ||
+            STRUCTURAL_KEYWORD_PATTERN.test(q)) {
+            return {
+                intent: LegendAIQuestionIntent.METADATA,
+                metaScore,
+                dataScore,
+                ambiguous: false,
+            };
+        }
+        const tentative = metaScore > dataScore
+            ? LegendAIQuestionIntent.METADATA
+            : LegendAIQuestionIntent.DATA_QUERY;
+        return {
+            intent: tentative,
+            metaScore,
+            dataScore,
+            ambiguous: true,
+        };
+    }
+    return {
+        intent: hasServices
+            ? LegendAIQuestionIntent.DATA_QUERY
+            : LegendAIQuestionIntent.METADATA,
+        metaScore,
+        dataScore,
+        ambiguous: true,
+    };
+}
+/**
+ * Legacy synchronous classifier for backward compatibility.
+ * Returns just the intent, discarding ambiguity info.
+ */
+export function classifyQuestionIntent(question, hasServices) {
+    return classifyQuestionIntentFast(question, hasServices).intent;
+}
+//# sourceMappingURL=LegendAITypes.js.map

package/lib/legend-ai/LegendAITypes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LegendAITypes.js","sourceRoot":"","sources":["../../src/legend-ai/LegendAITypes.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAQH,OAAO,EACL,0CAA0C,GAE3C,MAAM,iDAAiD,CAAC;AAEzD,MAAM,OAAO,eAAe;IAC1B,IAAI,CAAU;IACd,IAAI,CAAU;IACd,aAAa,CAAU;IACvB,YAAY,CAAU;CACvB;AAED,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,2CAAmB,CAAA;IACnB,oDAA4B,CAAA;AAC9B,CAAC,EAHW,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,gBAAgB;IAC3B,KAAK,CAAU;IACf,WAAW,CAAU;IACrB,OAAO,CAAU;IACjB,OAAO,CAAqB;IAC5B,UAAU,CAAY;IACtB;;;;OAIG;IACH,UAAU,CAAwB;IAClC,0GAA0G;IAC1G,eAAe,CAAU;CAC1B;AAED,MAAM,OAAO,cAAc;IACzB,OAAO,CAAW;IAClB,aAAa,CAAsB;IACnC,YAAY,CAAsB;IAClC,eAAe,CAAsB;IACrC,eAAe,CAAsB;IACrC,oBAAoB,CAAsB;IAC1C,SAAS,CAAsB;IAC/B,qBAAqB,CAAU;IAC/B,gBAAgB,CAAU;IAC1B,gGAAgG;IAChG,oBAAoB,CAAU;CAC/B;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAmB,MAAM,CAAC,MAAM,CAAC;IACpE,OAAO,EAAE,KAAK;IACd,aAAa,EAAE,SAAS;IACxB,YAAY,EAAE,SAAS;IACvB,eAAe,EAAE,SAAS;IAC1B,eAAe,EAAE,SAAS;IAC1B,oBAAoB,EAAE,SAAS;IAC/B,SAAS,EAAE,SAAS;CACrB,CAAC,CAAC;AAEH,MAAM,UAAU,kBAAkB,CAChC,OAAkC;IAElC,OAAO,OAAO,CAAC,IAAI,CACjB,CAAC,CAAC,EAAmD,EAAE,CACrD,CAAC,YAAY,0CAA0C,CAC1D,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,gBAAgB;IAC3B,UAAU,CAA8B;IACxC,OAAO,CAAoB;CAC5B;AAED,MAAM,UAAU,wBAAwB,CACtC,OAAiB;IAEjB,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC3B,KAAK,EAAE,GAAG;QACV,UAAU,EAAE,GAAG;QACf,KAAK,EAAE,GAAG;KACX,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,CAAN,IAAY,0BAIX;AAJD,WAAY,0BAA0B;IACpC,+CAAiB,CAAA;IACjB,2CAAa,CAAA;IACb,6CAAe,CAAA;AACjB,CAAC,EAJW,0BAA0B,KAA1B,0BAA0B,QAIrC;AAED,MAAM,OAAO,oBAAoB;IAC/B,KAAK,CAAU;IACf,MAAM,CAA8B;CACrC;AAED,MAAM,CAAN,IAAY,mBAGX;AAHD,WAAY,mBAAmB;IAC7B,oCAAa,CAAA;IACb,8CAAuB,CAAA;AACzB,CAAC,EAHW,mBAAmB,KAAnB,mBAAmB,QAG9B;AAED,MAAM,OAAO,mBAAmB;IAC9B,EAAE,CAAU;IACZ,IAAI,CAA4B;IAChC,IAAI,CAAU;CACf;AAED,MAAM,OAAO,wBAAwB;IACnC,EAAE,CAAU;IACZ,IAAI,CAAiC;IACrC,aAAa,CAA0B;IACvC,GAAG,CAAiB;IACpB,UAAU,CAAiB;IAC3B,QAAQ,CAA2B;IACnC,KAAK,CAAiB;IACtB,UAAU,CAAiB;IAC3B,QAAQ,CAAiB;IACzB,gBAAgB,CAAiB;IACjC,YAAY,CAAW;IACvB,WAAW,CAAW;IACtB,gBAAgB,CAAY;CAC7B;AAID,MAAM,OAAO,wBAAwB;IACnC,QAAQ,CAAU;IAClB,GAAG,CAAU;IACb;;;OAGG;IACH,MAAM,CAA0B;CACjC;AA8BD,MAAM,OAAO,4BAA4B;IACvC,KAAK,CAAU;IACf,WAAW,CAAU;IACrB,YAAY,CAA6B;CAC1C;AAED,MAAM,OAAO,uBAAuB;IAClC,IAAI,CAAU;IACd,WAAW,CAAU;IACrB,WAAW,CAAU;IACrB,gBAAgB,CAA4B;IAC5C,iBAAiB,CAAkC;IACnD,IAAI,CAAqB;IACzB,WAAW,CAAU;CACtB;AAED,MAAM,CAAN,IAAY,sBAIX;AAJD,WAAY,sBAAsB;IAChC,mDAAyB,CAAA;IACzB,+CAAqB,CAAA;IACrB,uDAA6B,CAAA;AAC/B,CAAC,EAJW,sBAAsB,KAAtB,sBAAsB,QAIjC;AAED,MAAM,CAAC,MAAM,wBAAwB,GAAsB,MAAM,CAAC,MAAM,CAAC;IACvE,oHAAoH;IACpH,qHAAqH;IACrH,2FAA2F;IAC3F,6FAA6F;IAC7F,6FAA6F;IAC7F,wEAAwE;IACxE,gEAAgE;IAChE,uGAAuG;IACvG,4EAA4E;IAC5E,iFAAiF;IACjF,gFAAgF;IAChF,iFAAiF;IACjF,iDAAiD;IACjD,4CAA4C;IAC5C,qDAAqD;IACrD,yCAAyC;IACzC,gCAAgC;IAChC,mEAAmE;IACnE,uDAAuD;IACvD,+BAA+B;IAC/B,4DAA4D;IAC5D,yCAAyC;IACzC,qEAAqE;IACrE,mFAAmF;CACpF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,0BAA0B,GAAsB,MAAM,CAAC,MAAM,CAAC;IACzE,6EAA6E;IAC7E,sDAAsD;IACtD,gEAAgE;IAChE,qDAAqD;IACrD,+BAA+B;IAC/B,kFAAkF;IAClF,yCAAyC;IACzC,sEAAsE;IACtE,+EAA+E;IAC/E,mCAAmC;IACnC,6BAA6B;IAC7B,qFAAqF;IACrF,+CAA+C;IAC/C,0DAA0D;IAC1D,gDAAgD;IAChD,uFAAuF;IACvF,uFAAuF;IACvF,sEAAsE;IACtE,sEAAsE;CACvE,CAAC,CAAC;AAEH,MAAM,yBAAyB,GAC7B,sEAAsE,CAAC;AAEzE,MAAM,0BAA0B,GAC9B,qFAAqF,CAAC;AAExF,SAAS,mBAAmB,CAC1B,QAAgB,EAChB,QAA2B;IAE3B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,KAAK,EAAE,CAAC;QACV,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,4BAA4B;IACvC,MAAM,CAA0B;IAChC,SAAS,CAAU;IACnB,SAAS,CAAU;IACnB,SAAS,CAAW;CACrB;AAED;;;;GAIG;AACH,MAAM,UAAU,0BAA0B,CACxC,QAAgB,EAChB,WAAoB;IAEpB,MAAM,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAExC,MAAM,SAAS,GAAG,mBAAmB,CAAC,CAAC,EAAE,wBAAwB,CAAC,CAAC;IACnE,MAAM,SAAS,GAAG,mBAAmB,CAAC,CAAC,EAAE,0BAA0B,CAAC,CAAC;IAErE,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;YACvC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IACD,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,sBAAsB,CAAC,UAAU;YACzC,SAAS;YACT,SAAS;YACT,SAAS,EAAE,KAAK;SACjB,CAAC;IACJ,CAAC;IAED,IAAI,SAAS,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;QACnC,IAAI,SAAS,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,MAAM,EAAE,sBAAsB,CAAC,UAAU;gBACzC,SAAS;gBACT,SAAS;gBACT,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,CAAC;QACD,IACE,yBAAyB,CAAC,IAAI,CAAC,CAAC,CAAC;YACjC,0BAA0B,CAAC,IAAI,CAAC,CAAC,CAAC,EAClC,CAAC;YACD,OAAO;gBACL,MAAM,EAAE,sBAAsB,CAAC,QAAQ;gBACvC,SAAS;gBACT,SAAS;gBACT,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GACb,SAAS,GAAG,SAAS;YACnB,CAAC,CAAC,sBAAsB,CAAC,QAAQ;YACjC,CAAC,CAAC,sBAAsB,CAAC,UAAU,CAAC;QACxC,OAAO;YACL,MAAM,EAAE,SAAS;YACjB,SAAS;YACT,SAAS;YACT,SAAS,EAAE,IAAI;SAChB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,MAAM,EAAE,WAAW;YACjB,CAAC,CAAC,sBAAsB,CAAC,UAAU;YACnC,CAAC,CAAC,sBAAsB,CAAC,QAAQ;QACnC,SAAS;QACT,SAAS;QACT,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CACpC,QAAgB,EAChB,WAAoB;IAEpB,OAAO,0BAA0B,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,MAAM,CAAC;AAClE,CAAC"}

package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.d.ts

@@ -0,0 +1,127 @@
+/**
+ * Copyright (c) 2026-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { LegendApplicationPlugin, type LegendApplicationPluginManager } from '@finos/legend-application';
+import type { QueryExplicitExecutionContextInfo, TDSRowDataType } from '@finos/legend-graph';
+import type { TDSServiceSchema, LegendAIConfig, LegendAIConversationTurn, LegendAIProductMetadata, LegendAIQuestionIntent } from './LegendAITypes.js';
+export declare class LegendAISqlExtractionResult {
+    sql: string | null;
+    failure: string | null;
+    suggestion?: string;
+}
+export declare enum LegendAIJudgeVerdict {
+    PASS = "PASS",
+    FAIL = "FAIL"
+}
+export declare class LegendAIJudgeResult {
+    verdict: LegendAIJudgeVerdict;
+    issues?: string;
+    correctedSql?: string;
+}
+export declare class LegendAISqlExecutionResultData {
+    columns: string[];
+    rows: TDSRowDataType[];
+}
+export declare class LegendAIOrchestratorDataProductCoordinates {
+    data_product: string;
+    group_id: string;
+    artifact_id: string;
+    version: string;
+}
+export interface LegendAISemanticSearchResolutionDetails {
+    data_product_coordinates: LegendAIOrchestratorDataProductCoordinates;
+    root_entity: string;
+    related_entities: string[];
+}
+export declare class LegendAIOrchestratorRequest {
+    user_question: string;
+    semantic_search_resolution_details: LegendAISemanticSearchResolutionDetails;
+}
+export declare class LegendAIOrchestratorResponse {
+    legend_query: string;
+}
+export declare class LegendAIResolvedEntities {
+    rootEntity: string;
+    relatedEntities: string[];
+}
+/**
+ * @deprecated Use {@link QueryExplicitExecutionContextInfo} from `@finos/legend-graph` directly.
+ */
+export type LegendAIPureExecutionContext = QueryExplicitExecutionContextInfo;
+export declare abstract class LegendAI_LegendApplicationPlugin_Extension extends LegendApplicationPlugin {
+    /**
+     * This helps to better type-check for this empty abstract type
+     * See https://github.com/finos/legend-studio/blob/master/docs/technical/typescript-usage.md#understand-typescript-structual-type-system
+     */
+    private readonly _$nominalTypeBrand;
+    install(pluginManager: LegendApplicationPluginManager<LegendApplicationPlugin>): void;
+    /**
+     * Classify the user question as either a data query or a metadata question.
+     *
+     * This is async to allow implementations to use an LLM micro-classifier
+     * for ambiguous questions while falling back to fast regex for clear-cut cases.
+     */
+    abstract classifyQuestionIntent(question: string, hasServices: boolean, config: LegendAIConfig, serviceNames?: string[]): Promise<LegendAIQuestionIntent>;
+    /**
+     * Build the LLM prompt for answering a metadata question about the product.
+     */
+    abstract buildMetadataPrompt(question: string, metadata: LegendAIProductMetadata, history?: LegendAIConversationTurn[]): string;
+    /**
+     * Build the LLM prompt for generating a SQL query from the user question.
+     */
+    abstract buildGeneratorPrompt(question: string, services: TDSServiceSchema[], coordinates: string, history?: LegendAIConversationTurn[]): string;
+    /**
+     * Build the LLM prompt for verifying and correcting a generated SQL query.
+     */
+    abstract buildJudgePrompt(sql: string, question: string, services: TDSServiceSchema[], coordinates: string, history?: LegendAIConversationTurn[]): string;
+    /**
+     * Send a prompt to the LLM service and return the raw response text.
+     */
+    abstract callLLM(prompt: string, config: LegendAIConfig): Promise<string>;
+    /**
+     * Execute a SQL query against the data platform and return tabular results.
+     */
+    abstract executeSql(sql: string, config: LegendAIConfig): Promise<LegendAISqlExecutionResultData>;
+    /**
+     * Parse the LLM generator response to extract the SQL query or failure reason.
+     */
+    abstract extractSqlFromResponse(answerText: string): LegendAISqlExtractionResult;
+    /**
+     * Parse the LLM judge response to extract the verdict and optional corrections.
+     */
+    abstract extractJudgeResult(answerText: string): LegendAIJudgeResult;
+    /**
+     * Call the Legend AI Orchestrator to generate a Legend/PURE query from
+     * natural language using resolved entity information.
+     */
+    abstract generateQueryViaOrchestrator(request: LegendAIOrchestratorRequest, config: LegendAIConfig): Promise<LegendAIOrchestratorResponse>;
+    /**
+     * Resolve root and related entities for a given user query within a data product.
+     * Uses the dataset search API to determine which entities are most relevant.
+     */
+    abstract resolveEntitiesForQuery(query: string, dataProductCoordinates: LegendAIOrchestratorDataProductCoordinates, config: LegendAIConfig): Promise<LegendAIResolvedEntities>;
+    /**
+     * Execute a Pure/Legend query against the engine and return tabular results.
+     * Converts the Pure expression text to a lambda, then executes it using
+     * the engine's execution endpoint with the provided execution context.
+     */
+    abstract executePureQuery(pureQuery: string, executionContext: QueryExplicitExecutionContextInfo, dataProductCoordinates: LegendAIOrchestratorDataProductCoordinates, config: LegendAIConfig): Promise<LegendAISqlExecutionResultData>;
+    /**
+     * Execute a SQL query that uses `p()` syntax against data product access points.
+     * Wraps the SQL in `#SQL{...}#`, builds a lakehouse runtime, and executes via the engine.
+     */
+    abstract executeLakehouseSql(sql: string, dataProductCoordinates: LegendAIOrchestratorDataProductCoordinates, config: LegendAIConfig): Promise<LegendAISqlExecutionResultData>;
+}
+//# sourceMappingURL=LegendAI_LegendApplicationPlugin_Extension.d.ts.map

package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LegendAI_LegendApplicationPlugin_Extension.d.ts","sourceRoot":"","sources":["../../src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,EACvB,KAAK,8BAA8B,EACpC,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EACV,iCAAiC,EACjC,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EACV,gBAAgB,EAChB,cAAc,EACd,wBAAwB,EACxB,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,oBAAoB,CAAC;AAE5B,qBAAa,2BAA2B;IACtC,GAAG,EAAG,MAAM,GAAG,IAAI,CAAC;IACpB,OAAO,EAAG,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,oBAAY,oBAAoB;IAC9B,IAAI,SAAS;IACb,IAAI,SAAS;CACd;AAED,qBAAa,mBAAmB;IAC9B,OAAO,EAAG,oBAAoB,CAAC;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,8BAA8B;IACzC,OAAO,EAAG,MAAM,EAAE,CAAC;IACnB,IAAI,EAAG,cAAc,EAAE,CAAC;CACzB;AAED,qBAAa,0CAA0C;IACrD,YAAY,EAAG,MAAM,CAAC;IACtB,QAAQ,EAAG,MAAM,CAAC;IAClB,WAAW,EAAG,MAAM,CAAC;IACrB,OAAO,EAAG,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,uCAAuC;IACtD,wBAAwB,EAAE,0CAA0C,CAAC;IACrE,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,2BAA2B;IACtC,aAAa,EAAG,MAAM,CAAC;IACvB,kCAAkC,EAAG,uCAAuC,CAAC;CAC9E;AAED,qBAAa,4BAA4B;IACvC,YAAY,EAAG,MAAM,CAAC;CACvB;AAED,qBAAa,wBAAwB;IACnC,UAAU,EAAG,MAAM,CAAC;IACpB,eAAe,EAAG,MAAM,EAAE,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,MAAM,4BAA4B,GAAG,iCAAiC,CAAC;AAE7E,8BAAsB,0CAA2C,SAAQ,uBAAuB;IAC9F;;;OAGG;IACH,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAgD;IAEnF,OAAO,CACL,aAAa,EAAE,8BAA8B,CAAC,uBAAuB,CAAC,GACrE,IAAI;IAIP;;;;;OAKG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,OAAO,EACpB,MAAM,EAAE,cAAc,EACtB,YAAY,CAAC,EAAE,MAAM,EAAE,GACtB,OAAO,CAAC,sBAAsB,CAAC;IAElC;;OAEG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,uBAAuB,EACjC,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,oBAAoB,CAC3B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,gBAAgB,CACvB,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,wBAAwB,EAAE,GACnC,MAAM;IAET;;OAEG;IACH,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;IAEzE;;OAEG;IACH,QAAQ,CAAC,UAAU,CACjB,GAAG,EAAE,MAAM,EACX,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;OAEG;IACH,QAAQ,CAAC,sBAAsB,CAC7B,UAAU,EAAE,MAAM,GACjB,2BAA2B;IAE9B;;OAEG;IACH,QAAQ,CAAC,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,mBAAmB;IAEpE;;;OAGG;IACH,QAAQ,CAAC,4BAA4B,CACnC,OAAO,EAAE,2BAA2B,EACpC,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,4BAA4B,CAAC;IAExC;;;OAGG;IACH,QAAQ,CAAC,uBAAuB,CAC9B,KAAK,EAAE,MAAM,EACb,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,wBAAwB,CAAC;IAEpC;;;;OAIG;IACH,QAAQ,CAAC,gBAAgB,CACvB,SAAS,EAAE,MAAM,EACjB,gBAAgB,EAAE,iCAAiC,EACnD,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;IAE1C;;;OAGG;IACH,QAAQ,CAAC,mBAAmB,CAC1B,GAAG,EAAE,MAAM,EACX,sBAAsB,EAAE,0CAA0C,EAClE,MAAM,EAAE,cAAc,GACrB,OAAO,CAAC,8BAA8B,CAAC;CAC3C"}

package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.js

@@ -0,0 +1,63 @@
+/**
+ * Copyright (c) 2026-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { LegendApplicationPlugin, } from '@finos/legend-application';
+export class LegendAISqlExtractionResult {
+    sql;
+    failure;
+    suggestion;
+}
+export var LegendAIJudgeVerdict;
+(function (LegendAIJudgeVerdict) {
+    LegendAIJudgeVerdict["PASS"] = "PASS";
+    LegendAIJudgeVerdict["FAIL"] = "FAIL";
+})(LegendAIJudgeVerdict || (LegendAIJudgeVerdict = {}));
+export class LegendAIJudgeResult {
+    verdict;
+    issues;
+    correctedSql;
+}
+export class LegendAISqlExecutionResultData {
+    columns;
+    rows;
+}
+export class LegendAIOrchestratorDataProductCoordinates {
+    data_product;
+    group_id;
+    artifact_id;
+    version;
+}
+export class LegendAIOrchestratorRequest {
+    user_question;
+    semantic_search_resolution_details;
+}
+export class LegendAIOrchestratorResponse {
+    legend_query;
+}
+export class LegendAIResolvedEntities {
+    rootEntity;
+    relatedEntities;
+}
+export class LegendAI_LegendApplicationPlugin_Extension extends LegendApplicationPlugin {
+    /**
+     * This helps to better type-check for this empty abstract type
+     * See https://github.com/finos/legend-studio/blob/master/docs/technical/typescript-usage.md#understand-typescript-structual-type-system
+     */
+    _$nominalTypeBrand;
+    install(pluginManager) {
+        pluginManager.registerApplicationPlugin(this);
+    }
+}
+//# sourceMappingURL=LegendAI_LegendApplicationPlugin_Extension.js.map

package/lib/legend-ai/LegendAI_LegendApplicationPlugin_Extension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LegendAI_LegendApplicationPlugin_Extension.js","sourceRoot":"","sources":["../../src/legend-ai/LegendAI_LegendApplicationPlugin_Extension.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,uBAAuB,GAExB,MAAM,2BAA2B,CAAC;AAanC,MAAM,OAAO,2BAA2B;IACtC,GAAG,CAAiB;IACpB,OAAO,CAAiB;IACxB,UAAU,CAAU;CACrB;AAED,MAAM,CAAN,IAAY,oBAGX;AAHD,WAAY,oBAAoB;IAC9B,qCAAa,CAAA;IACb,qCAAa,CAAA;AACf,CAAC,EAHW,oBAAoB,KAApB,oBAAoB,QAG/B;AAED,MAAM,OAAO,mBAAmB;IAC9B,OAAO,CAAwB;IAC/B,MAAM,CAAU;IAChB,YAAY,CAAU;CACvB;AAED,MAAM,OAAO,8BAA8B;IACzC,OAAO,CAAY;IACnB,IAAI,CAAoB;CACzB;AAED,MAAM,OAAO,0CAA0C;IACrD,YAAY,CAAU;IACtB,QAAQ,CAAU;IAClB,WAAW,CAAU;IACrB,OAAO,CAAU;CAClB;AAQD,MAAM,OAAO,2BAA2B;IACtC,aAAa,CAAU;IACvB,kCAAkC,CAA2C;CAC9E;AAED,MAAM,OAAO,4BAA4B;IACvC,YAAY,CAAU;CACvB;AAED,MAAM,OAAO,wBAAwB;IACnC,UAAU,CAAU;IACpB,eAAe,CAAY;CAC5B;AAOD,MAAM,OAAgB,0CAA2C,SAAQ,uBAAuB;IAC9F;;;OAGG;IACc,kBAAkB,CAAgD;IAEnF,OAAO,CACL,aAAsE;QAEtE,aAAa,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;IAChD,CAAC;CA8GF"}

package/lib/legend-ai/__test-utils__/LegendAITestUtils.d.ts

@@ -0,0 +1,29 @@
+/**
+ * Copyright (c) 2026-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import type { MessageSetter } from '../stores/LegendAIChatState.js';
+import { type LegendAIMessage, type LegendAIAssistantMessage, type LegendAIConfig, type LegendAIProductMetadata, type TDSServiceSchema } from '../LegendAITypes.js';
+import { type LegendAI_LegendApplicationPlugin_Extension } from '../LegendAI_LegendApplicationPlugin_Extension.js';
+export declare const TEST__createMockSetter: () => {
+    setter: MessageSetter;
+    getMessages: () => LegendAIMessage[];
+};
+export declare const TEST__makeAssistantMessage: (overrides?: Partial<LegendAIAssistantMessage>) => LegendAIAssistantMessage;
+export declare const TEST__seedAssistant: (setter: MessageSetter) => void;
+export declare const TEST_DATA__legendAIConfig: LegendAIConfig;
+export declare const TEST_DATA__legendAIMetadata: LegendAIProductMetadata;
+export declare const TEST_DATA__legendAIServices: TDSServiceSchema[];
+export declare const TEST__createMockLegendAIPlugin: (overrides?: Partial<LegendAI_LegendApplicationPlugin_Extension>) => LegendAI_LegendApplicationPlugin_Extension;
+//# sourceMappingURL=LegendAITestUtils.d.ts.map

package/lib/legend-ai/__test-utils__/LegendAITestUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LegendAITestUtils.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/__test-utils__/LegendAITestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,EACL,KAAK,eAAe,EACpB,KAAK,wBAAwB,EAC7B,KAAK,cAAc,EACnB,KAAK,uBAAuB,EAC5B,KAAK,gBAAgB,EAItB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,0CAA0C,EAIhD,MAAM,kDAAkD,CAAC;AAE1D,eAAO,MAAM,sBAAsB,QAAO;IACxC,MAAM,EAAE,aAAa,CAAC;IACtB,WAAW,EAAE,MAAM,eAAe,EAAE,CAAC;CAWtC,CAAC;AACF,eAAO,MAAM,0BAA0B,GACrC,YAAY,OAAO,CAAC,wBAAwB,CAAC,KAC5C,wBAeD,CAAC;AACH,eAAO,MAAM,mBAAmB,GAAI,QAAQ,aAAa,KAAG,IAK3D,CAAC;AACF,eAAO,MAAM,yBAAyB,EAAE,cASvC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,uBAIzC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,gBAAgB,EAUzD,CAAC;AACF,eAAO,MAAM,8BAA8B,GACzC,YAAY,OAAO,CAAC,0CAA0C,CAAC,KAC9D,0CAmC+C,CAAC"}

package/lib/legend-ai/__test-utils__/LegendAITestUtils.js

@@ -0,0 +1,98 @@
+/**
+ * Copyright (c) 2026-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { createMock } from '@finos/legend-shared/test';
+import { LegendAIMessageRole, LegendAIQuestionIntent, } from '../LegendAITypes.js';
+import { LegendAIJudgeVerdict, } from '../LegendAI_LegendApplicationPlugin_Extension.js';
+export const TEST__createMockSetter = () => {
+    let messages = [];
+    const setter = (action) => {
+        if (typeof action === 'function') {
+            messages = action(messages);
+        }
+        else {
+            messages = action;
+        }
+    };
+    return { setter, getMessages: () => messages };
+};
+export const TEST__makeAssistantMessage = (overrides) => ({
+    id: 'test-assistant-msg',
+    role: LegendAIMessageRole.ASSISTANT,
+    thinkingSteps: [],
+    sql: null,
+    textAnswer: null,
+    gridData: null,
+    error: null,
+    sqlGenTime: null,
+    execTime: null,
+    thinkingDuration: null,
+    isProcessing: true,
+    isExecuting: false,
+    suggestedQueries: [],
+    ...overrides,
+});
+export const TEST__seedAssistant = (setter) => {
+    setter([
+        { id: 'test-user-msg', role: LegendAIMessageRole.USER, text: 'test' },
+        TEST__makeAssistantMessage(),
+    ]);
+};
+export const TEST_DATA__legendAIConfig = {
+    enabled: true,
+    llmServiceUrl: 'http://localhost/llm',
+    llmModelName: 'test-model',
+    sqlExecutionUrl: 'http://localhost/sql',
+    orchestratorUrl: undefined,
+    marketplaceSearchUrl: undefined,
+    engineUrl: undefined,
+    maxJudgeAttempts: 2,
+};
+export const TEST_DATA__legendAIMetadata = {
+    name: 'TestProduct',
+    coordinates: 'com.test:prod:1.0.0',
+    serviceSummaries: [{ title: 'Svc', description: 'desc' }],
+};
+export const TEST_DATA__legendAIServices = [
+    {
+        title: 'TradeService',
+        pattern: '/trades',
+        columns: [
+            { name: 'tradeId', type: 'String' },
+            { name: 'amount', type: 'Number' },
+        ],
+        parameters: [],
+    },
+];
+export const TEST__createMockLegendAIPlugin = (overrides) => ({
+    getName: () => 'test-plugin',
+    getVersion: () => '1.0.0',
+    install: () => { },
+    classifyQuestionIntent: (_q, _h) => LegendAIQuestionIntent.DATA_QUERY,
+    buildMetadataPrompt: (_q, _m, _h) => 'metadata prompt',
+    buildGeneratorPrompt: (_q, _s, _c, _h) => 'generator prompt',
+    buildJudgePrompt: (_sql, _q, _s, _c, _h) => 'judge prompt',
+    callLLM: createMock(),
+    executeSql: createMock(),
+    extractSqlFromResponse: (_a) => ({
+        sql: 'SELECT * FROM t',
+        failure: null,
+    }),
+    extractJudgeResult: (_a) => ({
+        verdict: LegendAIJudgeVerdict.PASS,
+    }),
+    ...overrides,
+});
+//# sourceMappingURL=LegendAITestUtils.js.map

package/lib/legend-ai/__test-utils__/LegendAITestUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LegendAITestUtils.js","sourceRoot":"","sources":["../../../src/legend-ai/__test-utils__/LegendAITestUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAEvD,OAAO,EAOL,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAIL,oBAAoB,GACrB,MAAM,kDAAkD,CAAC;AAE1D,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAGpC,EAAE;IACF,IAAI,QAAQ,GAAsB,EAAE,CAAC;IACrC,MAAM,MAAM,GAAkB,CAAC,MAAM,EAAE,EAAE;QACvC,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;YACjC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,QAAQ,GAAG,MAAM,CAAC;QACpB,CAAC;IACH,CAAC,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;AACjD,CAAC,CAAC;AACF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CACxC,SAA6C,EACnB,EAAE,CAAC,CAAC;IAC9B,EAAE,EAAE,oBAAoB;IACxB,IAAI,EAAE,mBAAmB,CAAC,SAAS;IACnC,aAAa,EAAE,EAAE;IACjB,GAAG,EAAE,IAAI;IACT,UAAU,EAAE,IAAI;IAChB,QAAQ,EAAE,IAAI;IACd,KAAK,EAAE,IAAI;IACX,UAAU,EAAE,IAAI;IAChB,QAAQ,EAAE,IAAI;IACd,gBAAgB,EAAE,IAAI;IACtB,YAAY,EAAE,IAAI;IAClB,WAAW,EAAE,KAAK;IAClB,gBAAgB,EAAE,EAAE;IACpB,GAAG,SAAS;CACb,CAAC,CAAC;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,MAAqB,EAAQ,EAAE;IACjE,MAAM,CAAC;QACL,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE;QACrE,0BAA0B,EAAE;KAC7B,CAAC,CAAC;AACL,CAAC,CAAC;AACF,MAAM,CAAC,MAAM,yBAAyB,GAAmB;IACvD,OAAO,EAAE,IAAI;IACb,aAAa,EAAE,sBAAsB;IACrC,YAAY,EAAE,YAAY;IAC1B,eAAe,EAAE,sBAAsB;IACvC,eAAe,EAAE,SAAS;IAC1B,oBAAoB,EAAE,SAAS;IAC/B,SAAS,EAAE,SAAS;IACpB,gBAAgB,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAA4B;IAClE,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,qBAAqB;IAClC,gBAAgB,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;CAC1D,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAuB;IAC7D;QACE,KAAK,EAAE,cAAc;QACrB,OAAO,EAAE,SAAS;QAClB,OAAO,EAAE;YACP,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE;YACnC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE;SACnC;QACD,UAAU,EAAE,EAAE;KACf;CACF,CAAC;AACF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAC5C,SAA+D,EACnB,EAAE,CAC9C,CAAC;IACC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa;IAC5B,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO;IACzB,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;IACjB,sBAAsB,EAAE,CAAC,EAAU,EAAE,EAAW,EAA0B,EAAE,CAC1E,sBAAsB,CAAC,UAAU;IACnC,mBAAmB,EAAE,CACnB,EAAU,EACV,EAA2B,EAC3B,EAA+B,EAC/B,EAAE,CAAC,iBAAiB;IACtB,oBAAoB,EAAE,CACpB,EAAU,EACV,EAAsB,EACtB,EAAU,EACV,EAA+B,EAC/B,EAAE,CAAC,kBAAkB;IACvB,gBAAgB,EAAE,CAChB,IAAY,EACZ,EAAU,EACV,EAAsB,EACtB,EAAU,EACV,EAA+B,EAC/B,EAAE,CAAC,cAAc;IACnB,OAAO,EAAE,UAAU,EAAE;IACrB,UAAU,EAAE,UAAU,EAAE;IACxB,sBAAsB,EAAE,CAAC,EAAU,EAA+B,EAAE,CAAC,CAAC;QACpE,GAAG,EAAE,iBAAiB;QACtB,OAAO,EAAE,IAAI;KACd,CAAC;IACF,kBAAkB,EAAE,CAAC,EAAU,EAAuB,EAAE,CAAC,CAAC;QACxD,OAAO,EAAE,oBAAoB,CAAC,IAAI;KACnC,CAAC;IACF,GAAG,SAAS;CACb,CAA+C,CAAC"}

package/lib/legend-ai/components/LegendAIChat.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Copyright (c) 2026-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { type LegendAIChatProps, type LegendAIProductMetadata, type TDSServiceSchema, type TDSColumnSchema } from '../LegendAITypes.js';
+export declare const LEGEND_AI_ANCHOR_ID = "legend-ai-anchor";
+export declare function isStringColumn(c: TDSColumnSchema): boolean;
+export declare function isNumericColumn(c: TDSColumnSchema): boolean;
+export declare function isDateColumn(c: TDSColumnSchema): boolean;
+export declare function buildSuggestedQueries(services: TDSServiceSchema[], metadata: LegendAIProductMetadata): string[];
+export declare const LegendAIChat: (props: LegendAIChatProps) => React.ReactNode;
+//# sourceMappingURL=LegendAIChat.d.ts.map

package/lib/legend-ai/components/LegendAIChat.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LegendAIChat.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIChat.tsx"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAeH,OAAO,EACL,KAAK,iBAAiB,EAEtB,KAAK,uBAAuB,EAC5B,KAAK,gBAAgB,EACrB,KAAK,eAAe,EAIrB,MAAM,qBAAqB,CAAC;AAI7B,eAAO,MAAM,mBAAmB,qBAAqB,CAAC;AAoBtD,wBAAgB,cAAc,CAAC,CAAC,EAAE,eAAe,GAAG,OAAO,CAE1D;AAED,wBAAgB,eAAe,CAAC,CAAC,EAAE,eAAe,GAAG,OAAO,CAE3D;AAED,wBAAgB,YAAY,CAAC,CAAC,EAAE,eAAe,GAAG,OAAO,CAMxD;AAwDD,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,gBAAgB,EAAE,EAC5B,QAAQ,EAAE,uBAAuB,GAChC,MAAM,EAAE,CAqCV;AAmLD,eAAO,MAAM,YAAY,GAAI,OAAO,iBAAiB,KAAG,KAAK,CAAC,SAsJ7D,CAAC"}