@finos/legend-lego 2.0.196 → 2.0.198

package/lib/legend-ai/components/LegendAIAnalysisUtils.js

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 import { LegendAIChartType, } from '../LegendAI_LegendApplicationPlugin_Extension.js';
-import { isNonNullable, isNumber, isString } from '@finos/legend-shared';
+import { isNumber, isString } from '@finos/legend-shared';
 const CHART_PALETTE_COUNT = 10;
 const MAX_CHART_ITEMS = 10;
 const TOP_N_ITEMS = 5;
@@ -29,19 +29,38 @@ function profileColumns(gridData) {
         : gridData.rowData;
     return gridData.columnDefs.map((col) => {
         const field = col.field ?? col.colId ?? '';
-        const values = rows.map((r) => r[field]).filter(isNonNullable);
-        const numericValues = values.filter(isNumber);
-        const unique = new Set(values.map(String));
+        let nonNullCount = 0;
+        let stringCount = 0;
+        const numericValues = [];
+        const unique = new Set();
+        for (const r of rows) {
+            const v = r[field];
+            if (v !== null && v !== undefined) {
+                nonNullCount++;
+                unique.add(String(v));
+                if (isNumber(v)) {
+                    numericValues.push(v);
+                }
+                if (isString(v)) {
+                    stringCount++;
+                }
+            }
+        }
         return {
             name: field,
-            isNumeric: numericValues.length === values.length && values.length > 0,
-            isString: values.length > 0 && values.every(isString),
+            isNumeric: numericValues.length === nonNullCount && nonNullCount > 0,
+            isString: nonNullCount > 0 && stringCount === nonNullCount,
             uniqueCount: unique.size,
-            values,
+            nonNullCount,
             numericValues,
         };
     });
 }
+function findBestCategoricalColumn(profiles, rowCount) {
+    return profiles
+        .filter((c) => c.isString && c.uniqueCount > 1 && c.uniqueCount < rowCount)
+        .sort((a, b) => a.uniqueCount - b.uniqueCount)[0];
+}
 function formatNumber(n) {
     if (Number.isInteger(n) && Math.abs(n) < 1_000_000) {
         return n.toLocaleString();
@@ -84,10 +103,8 @@ function computeNumericMetrics(col, metrics) {
         });
     }
 }
-export function computeKeyMetrics(gridData) {
-    const profiles = profileColumns(gridData);
+function computeKeyMetricsFromProfiles(profiles, rowCount) {
     const metrics = [];
-    const rowCount = gridData.rowData.length;
     metrics.push({
         label: 'Total Rows',
         value: rowCount.toLocaleString(),
@@ -105,59 +122,91 @@ export function computeKeyMetrics(gridData) {
     }
     return metrics.slice(0, MAX_KEY_METRICS);
 }
-export function inferChartType(gridData) {
-    const profiles = profileColumns(gridData);
+export function computeKeyMetrics(gridData) {
+    return computeKeyMetricsFromProfiles(profileColumns(gridData), gridData.rowData.length);
+}
+function inferChartTypeFromProfiles(profiles, rowCount) {
     const numericCols = profiles.filter((c) => c.isNumeric);
     const stringCols = profiles.filter((c) => c.isString && c.uniqueCount > 1);
     if (stringCols.length >= 1 &&
         numericCols.length >= 1 &&
-        gridData.rowData.length <= MAX_BAR_CHART_ROWS) {
-        if (gridData.rowData.length <= MAX_PIE_CHART_ROWS) {
+        rowCount <= MAX_BAR_CHART_ROWS) {
+        if (rowCount <= MAX_PIE_CHART_ROWS) {
             return LegendAIChartType.PIE;
         }
         return LegendAIChartType.BAR;
     }
-    if (numericCols.length >= 1 && gridData.rowData.length > 1) {
+    if (numericCols.length >= 1 && rowCount > 1) {
         return LegendAIChartType.BAR;
     }
+    if (rowCount > 1) {
+        const categoricalCol = findBestCategoricalColumn(profiles, rowCount);
+        if (categoricalCol) {
+            return categoricalCol.uniqueCount <= MAX_PIE_CHART_ROWS
+                ? LegendAIChartType.PIE
+                : LegendAIChartType.BAR;
+        }
+    }
     return LegendAIChartType.NONE;
 }
-export function computeChartData(gridData) {
-    const profiles = profileColumns(gridData);
+export function inferChartType(gridData) {
+    return inferChartTypeFromProfiles(profileColumns(gridData), gridData.rowData.length);
+}
+function computeChartDataFromProfiles(profiles, gridData) {
     const numericCol = profiles.find((c) => c.isNumeric);
     const labelCol = profiles.find((c) => c.isString && c.uniqueCount > 1);
-    if (!numericCol) {
+    if (numericCol) {
+        const field = numericCol.name;
+        const labelField = labelCol?.name;
+        const rows = gridData.rowData.length > MAX_PROFILE_SAMPLE
+            ? gridData.rowData.slice(0, MAX_PROFILE_SAMPLE)
+            : gridData.rowData;
+        const entries = rows
+            .map((row) => {
+            const rawValue = row[field];
+            return {
+                label: labelField
+                    ? String(row[labelField] ?? '')
+                    : String(row[gridData.columnDefs[0]?.field ?? ''] ?? ''),
+                value: typeof rawValue === 'number' ? rawValue : 0,
+            };
+        })
+            .filter((e) => e.label.length > 0)
+            .sort((a, b) => b.value - a.value)
+            .slice(0, MAX_CHART_ITEMS);
+        return entries.map((e, i) => ({
+            label: e.label,
+            value: e.value,
+            colorIndex: i % CHART_PALETTE_COUNT,
+        }));
+    }
+    const categoricalCol = findBestCategoricalColumn(profiles, gridData.rowData.length);
+    if (!categoricalCol) {
         return [];
     }
-    const field = numericCol.name;
-    const labelField = labelCol?.name;
-    const rows = gridData.rowData.length > MAX_PROFILE_SAMPLE
-        ? gridData.rowData.slice(0, MAX_PROFILE_SAMPLE)
-        : gridData.rowData;
-    const entries = rows
-        .map((row) => {
-        const rawValue = row[field];
-        return {
-            label: labelField
-                ? String(row[labelField] ?? '')
-                : String(row[gridData.columnDefs[0]?.field ?? ''] ?? ''),
-            value: typeof rawValue === 'number' ? rawValue : 0,
-        };
-    })
-        .filter((e) => e.label.length > 0)
-        .sort((a, b) => b.value - a.value)
-        .slice(0, MAX_CHART_ITEMS);
-    return entries.map((e, i) => ({
-        label: e.label,
-        value: e.value,
+    const freqMap = new Map();
+    for (const row of gridData.rowData) {
+        const val = String(row[categoricalCol.name] ?? '');
+        if (val.length > 0) {
+            freqMap.set(val, (freqMap.get(val) ?? 0) + 1);
+        }
+    }
+    return [...freqMap.entries()]
+        .sort((a, b) => b[1] - a[1])
+        .slice(0, MAX_CHART_ITEMS)
+        .map(([label, value], i) => ({
+        label,
+        value,
         colorIndex: i % CHART_PALETTE_COUNT,
     }));
 }
+export function computeChartData(gridData) {
+    return computeChartDataFromProfiles(profileColumns(gridData), gridData);
+}
 export function computeTopItems(gridData) {
     return computeChartData(gridData).slice(0, TOP_N_ITEMS);
 }
-export function findNumericColumnName(gridData) {
-    const profiles = profileColumns(gridData);
+function findNumericColumnNameFromProfiles(profiles, gridData) {
     const numericCol = profiles.find((c) => c.isNumeric);
     if (!numericCol) {
         return undefined;
@@ -165,4 +214,20 @@ export function findNumericColumnName(gridData) {
     const colDef = gridData.columnDefs.find((c) => (c.field ?? c.colId ?? '') === numericCol.name);
     return colDef?.headerName ?? colDef?.field;
 }
+export function findNumericColumnName(gridData) {
+    return findNumericColumnNameFromProfiles(profileColumns(gridData), gridData);
+}
+export function analyzeGridData(gridData) {
+    const profiles = profileColumns(gridData);
+    const rowCount = gridData.rowData.length;
+    const chartType = inferChartTypeFromProfiles(profiles, rowCount);
+    return {
+        metrics: computeKeyMetricsFromProfiles(profiles, rowCount),
+        chartType,
+        chartData: chartType === LegendAIChartType.NONE
+            ? []
+            : computeChartDataFromProfiles(profiles, gridData),
+        numericColumnName: findNumericColumnNameFromProfiles(profiles, gridData),
+    };
+}
 //# sourceMappingURL=LegendAIAnalysisUtils.js.map

package/lib/legend-ai/components/LegendAIAnalysisUtils.js.map

@@ -1 +1 @@
-{"version":3,"file":"LegendAIAnalysisUtils.js","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAGL,iBAAiB,GAClB,MAAM,kDAAkD,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEzE,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAC/B,MAAM,eAAe,GAAG,EAAE,CAAC;AAC3B,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAChC,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAC7B,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAC9B,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAW7B,SAAS,cAAc,CAAC,QAA0B;IAChD,MAAM,IAAI,GACR,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,kBAAkB;QAC1C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC;QAC/C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;IAEvB,OAAO,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACrC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC/D,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAC3C,OAAO;YACL,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,aAAa,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;YACtE,QAAQ,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC;YACrD,WAAW,EAAE,MAAM,CAAC,IAAI;YACxB,MAAM;YACN,aAAa;SACd,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,YAAY,CAAC,CAAS;IAC7B,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS,EAAE,CAAC;QACnD,OAAO,CAAC,CAAC,cAAc,EAAE,CAAC;IAC5B,CAAC;IACD,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,SAAS,EAAE,CAAC;QAC7B,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC1C,CAAC;IACD,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC;QACzB,OAAO,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACtC,CAAC;IACD,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,qBAAqB,CAC5B,GAAkB,EAClB,OAA4B;IAE5B,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,GAAG,GAAG,MAAM,CAAC,iBAAiB,CAAC;IACnC,IAAI,GAAG,GAAG,MAAM,CAAC,iBAAiB,CAAC;IACnC,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC;QACT,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;YACZ,GAAG,GAAG,CAAC,CAAC;QACV,CAAC;QACD,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;YACZ,GAAG,GAAG,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IACD,MAAM,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC;IAE3C,IAAI,GAAG,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC;YACX,KAAK,EAAE,OAAO,GAAG,CAAC,IAAI,EAAE;YACxB,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC;YACxB,GAAG,CAAC,GAAG,KAAK,GAAG;gBACb,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,YAAY,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,IAAI,GAAG,CAAC,WAAW,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QAC/D,OAAO,CAAC,IAAI,CAAC;YACX,KAAK,EAAE,SAAS,GAAG,CAAC,IAAI,EAAE;YAC1B,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC;SACzB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,QAA0B;IAE1B,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,OAAO,GAAwB,EAAE,CAAC;IACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;IAEzC,OAAO,CAAC,IAAI,CAAC;QACX,KAAK,EAAE,YAAY;QACnB,KAAK,EAAE,QAAQ,CAAC,cAAc,EAAE;KACjC,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAC9B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CACjD,CAAC;IACF,IAAI,UAAU,EAAE,CAAC;QACf,qBAAqB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,CAAC,WAAW,IAAI,QAAQ,CACpE,CAAC;IACF,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC;YACX,KAAK,EAAE,UAAU,SAAS,CAAC,IAAI,EAAE;YACjC,KAAK,EAAE,SAAS,CAAC,WAAW,CAAC,cAAc,EAAE;SAC9C,CAAC,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,QAA0B;IACvD,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACxD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;IAE3E,IACE,UAAU,CAAC,MAAM,IAAI,CAAC;QACtB,WAAW,CAAC,MAAM,IAAI,CAAC;QACvB,QAAQ,CAAC,OAAO,CAAC,MAAM,IAAI,kBAAkB,EAC7C,CAAC;QACD,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,IAAI,kBAAkB,EAAE,CAAC;YAClD,OAAO,iBAAiB,CAAC,GAAG,CAAC;QAC/B,CAAC;QACD,OAAO,iBAAiB,CAAC,GAAG,CAAC;IAC/B,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,IAAI,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3D,OAAO,iBAAiB,CAAC,GAAG,CAAC;IAC/B,CAAC;IAED,OAAO,iBAAiB,CAAC,IAAI,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,QAA0B;IAE1B,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;IAEvE,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC;IAC9B,MAAM,UAAU,GAAG,QAAQ,EAAE,IAAI,CAAC;IAClC,MAAM,IAAI,GACR,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,kBAAkB;QAC1C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC;QAC/C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;IAEvB,MAAM,OAAO,GAAG,IAAI;SACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5B,OAAO;YACL,KAAK,EAAE,UAAU;gBACf,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gBAC/B,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YAC1D,KAAK,EAAE,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;SACnD,CAAC;IACJ,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;SACjC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;SACjC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;IAE7B,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5B,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,UAAU,EAAE,CAAC,GAAG,mBAAmB;KACpC,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,QAA0B;IAE1B,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,QAA0B;IAE1B,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACrD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,CACrC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,UAAU,CAAC,IAAI,CACtD,CAAC;IACF,OAAO,MAAM,EAAE,UAAU,IAAI,MAAM,EAAE,KAAK,CAAC;AAC7C,CAAC"}
+{"version":3,"file":"LegendAIAnalysisUtils.js","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIAnalysisUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAGL,iBAAiB,GAClB,MAAM,kDAAkD,CAAC;AAE1D,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAE1D,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAC/B,MAAM,eAAe,GAAG,EAAE,CAAC;AAC3B,MAAM,WAAW,GAAG,CAAC,CAAC;AACtB,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAChC,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAC7B,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAC9B,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAkB7B,SAAS,cAAc,CAAC,QAA0B;IAChD,MAAM,IAAI,GACR,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,kBAAkB;QAC1C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC;QAC/C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;IAEvB,OAAO,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACrC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;QAC3C,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;QAEjC,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;YACnB,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;gBAClC,YAAY,EAAE,CAAC;gBACf,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBACtB,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChB,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACxB,CAAC;gBACD,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;oBAChB,WAAW,EAAE,CAAC;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,aAAa,CAAC,MAAM,KAAK,YAAY,IAAI,YAAY,GAAG,CAAC;YACpE,QAAQ,EAAE,YAAY,GAAG,CAAC,IAAI,WAAW,KAAK,YAAY;YAC1D,WAAW,EAAE,MAAM,CAAC,IAAI;YACxB,YAAY;YACZ,aAAa;SACd,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,yBAAyB,CAChC,QAAyB,EACzB,QAAgB;IAEhB,OAAO,QAAQ;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,CAAC,WAAW,GAAG,QAAQ,CAAC;SAC1E,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,YAAY,CAAC,CAAS;IAC7B,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS,EAAE,CAAC;QACnD,OAAO,CAAC,CAAC,cAAc,EAAE,CAAC;IAC5B,CAAC;IACD,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,SAAS,EAAE,CAAC;QAC7B,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC1C,CAAC;IACD,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC;QACzB,OAAO,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACtC,CAAC;IACD,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,qBAAqB,CAC5B,GAAkB,EAClB,OAA4B;IAE5B,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,GAAG,GAAG,MAAM,CAAC,iBAAiB,CAAC;IACnC,IAAI,GAAG,GAAG,MAAM,CAAC,iBAAiB,CAAC;IACnC,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC;QACT,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;YACZ,GAAG,GAAG,CAAC,CAAC;QACV,CAAC;QACD,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;YACZ,GAAG,GAAG,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IACD,MAAM,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC;IAE3C,IAAI,GAAG,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC;YACX,KAAK,EAAE,OAAO,GAAG,CAAC,IAAI,EAAE;YACxB,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC;YACxB,GAAG,CAAC,GAAG,KAAK,GAAG;gBACb,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,YAAY,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,IAAI,GAAG,CAAC,WAAW,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;QAC/D,OAAO,CAAC,IAAI,CAAC;YACX,KAAK,EAAE,SAAS,GAAG,CAAC,IAAI,EAAE;YAC1B,KAAK,EAAE,YAAY,CAAC,GAAG,CAAC;SACzB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CACpC,QAAyB,EACzB,QAAgB;IAEhB,MAAM,OAAO,GAAwB,EAAE,CAAC;IAExC,OAAO,CAAC,IAAI,CAAC;QACX,KAAK,EAAE,YAAY;QACnB,KAAK,EAAE,QAAQ,CAAC,cAAc,EAAE;KACjC,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAC9B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CACjD,CAAC;IACF,IAAI,UAAU,EAAE,CAAC;QACf,qBAAqB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,IAAI,CAAC,CAAC,WAAW,IAAI,QAAQ,CACpE,CAAC;IACF,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC;YACX,KAAK,EAAE,UAAU,SAAS,CAAC,IAAI,EAAE;YACjC,KAAK,EAAE,SAAS,CAAC,WAAW,CAAC,cAAc,EAAE;SAC9C,CAAC,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,QAA0B;IAE1B,OAAO,6BAA6B,CAClC,cAAc,CAAC,QAAQ,CAAC,EACxB,QAAQ,CAAC,OAAO,CAAC,MAAM,CACxB,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CACjC,QAAyB,EACzB,QAAgB;IAEhB,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACxD,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;IAE3E,IACE,UAAU,CAAC,MAAM,IAAI,CAAC;QACtB,WAAW,CAAC,MAAM,IAAI,CAAC;QACvB,QAAQ,IAAI,kBAAkB,EAC9B,CAAC;QACD,IAAI,QAAQ,IAAI,kBAAkB,EAAE,CAAC;YACnC,OAAO,iBAAiB,CAAC,GAAG,CAAC;QAC/B,CAAC;QACD,OAAO,iBAAiB,CAAC,GAAG,CAAC;IAC/B,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,IAAI,CAAC,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;QAC5C,OAAO,iBAAiB,CAAC,GAAG,CAAC;IAC/B,CAAC;IAED,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;QACjB,MAAM,cAAc,GAAG,yBAAyB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACrE,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,cAAc,CAAC,WAAW,IAAI,kBAAkB;gBACrD,CAAC,CAAC,iBAAiB,CAAC,GAAG;gBACvB,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,iBAAiB,CAAC,IAAI,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,QAA0B;IACvD,OAAO,0BAA0B,CAC/B,cAAc,CAAC,QAAQ,CAAC,EACxB,QAAQ,CAAC,OAAO,CAAC,MAAM,CACxB,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CACnC,QAAyB,EACzB,QAA0B;IAE1B,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC;IAEvE,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC;QAC9B,MAAM,UAAU,GAAG,QAAQ,EAAE,IAAI,CAAC;QAClC,MAAM,IAAI,GACR,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,kBAAkB;YAC1C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC;YAC/C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;QAEvB,MAAM,OAAO,GAAG,IAAI;aACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACX,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;YAC5B,OAAO;gBACL,KAAK,EAAE,UAAU;oBACf,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;oBAC/B,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC1D,KAAK,EAAE,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;aACnD,CAAC;QACJ,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;aACjC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;QAE7B,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5B,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,UAAU,EAAE,CAAC,GAAG,mBAAmB;SACpC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,cAAc,GAAG,yBAAyB,CAC9C,QAAQ,EACR,QAAQ,CAAC,OAAO,CAAC,MAAM,CACxB,CAAC;IACF,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC1C,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;QACnC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACnD,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;SAC1B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3B,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC;SACzB,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3B,KAAK;QACL,KAAK;QACL,UAAU,EAAE,CAAC,GAAG,mBAAmB;KACpC,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,QAA0B;IAE1B,OAAO,4BAA4B,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,QAA0B;IAE1B,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,iCAAiC,CACxC,QAAyB,EACzB,QAA0B;IAE1B,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACrD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,CACrC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,UAAU,CAAC,IAAI,CACtD,CAAC;IACF,OAAO,MAAM,EAAE,UAAU,IAAI,MAAM,EAAE,KAAK,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,QAA0B;IAE1B,OAAO,iCAAiC,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;AAC/E,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,QAA0B;IAE1B,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;IACzC,MAAM,SAAS,GAAG,0BAA0B,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACjE,OAAO;QACL,OAAO,EAAE,6BAA6B,CAAC,QAAQ,EAAE,QAAQ,CAAC;QAC1D,SAAS;QACT,SAAS,EACP,SAAS,KAAK,iBAAiB,CAAC,IAAI;YAClC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,4BAA4B,CAAC,QAAQ,EAAE,QAAQ,CAAC;QACtD,iBAAiB,EAAE,iCAAiC,CAAC,QAAQ,EAAE,QAAQ,CAAC;KACzE,CAAC;AACJ,CAAC"}

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

@@ -13,12 +13,8 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { type LegendAIChatProps, type LegendAIProductMetadata, type TDSServiceSchema, type TDSColumnSchema, LegendAIThinkingStepStatus } from '../LegendAITypes.js';
+import { type LegendAIChatProps, LegendAIThinkingStepStatus } 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 function renderStepStatusIcon(status: LegendAIThinkingStepStatus): React.ReactNode;
 export declare const LegendAIChat: (props: LegendAIChatProps) => React.ReactNode;
 //# sourceMappingURL=LegendAIChat.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"LegendAIChat.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIChat.tsx"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AA2BH,OAAO,EACL,KAAK,iBAAiB,EAEtB,KAAK,uBAAuB,EAC5B,KAAK,gBAAgB,EACrB,KAAK,eAAe,EACpB,0BAA0B,EAG3B,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;AAED,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,0BAA0B,GACjC,KAAK,CAAC,SAAS,CASjB;AAuMD,eAAO,MAAM,YAAY,GAAI,OAAO,iBAAiB,KAAG,KAAK,CAAC,SAyJ7D,CAAC"}
+{"version":3,"file":"LegendAIChat.d.ts","sourceRoot":"","sources":["../../../src/legend-ai/components/LegendAIChat.tsx"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAsBH,OAAO,EACL,KAAK,iBAAiB,EAOtB,0BAA0B,EAI3B,MAAM,qBAAqB,CAAC;AAO7B,eAAO,MAAM,mBAAmB,qBAAqB,CAAC;AAgHtD,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,0BAA0B,GACjC,KAAK,CAAC,SAAS,CASjB;AAmWD,eAAO,MAAM,YAAY,GAAI,OAAO,iBAAiB,KAAG,KAAK,CAAC,SAuM7D,CAAC"}