@retrivora-ai/rag-engine 1.8.9 → 1.9.1

package/src/types/index.ts

@@ -125,7 +125,12 @@ export type VisualizationType =
   | 'pie_chart'
   | 'bar_chart'
   | 'line_chart'
+  | 'histogram'
+  | 'horizontal_bar'
+  | 'scatter_plot'
   | 'radar_chart'
+  | 'metric_card'
+  | 'geo_map'
   | 'table'
   | 'product_carousel'
   | 'carousel'
@@ -173,6 +178,27 @@ export interface LineChartDataPoint {
   [key: string]: unknown;
 }
 
+/**
+ * Scatter plot data point
+ */
+export interface ScatterPlotDataPoint {
+  x: number;
+  y: number;
+  label?: string;
+  [key: string]: unknown;
+}
+
+/**
+ * KPI / metric card representation
+ */
+export interface MetricCardData {
+  label: string;
+  value: number;
+  operation?: 'sum' | 'average' | 'count' | 'min' | 'max' | 'median';
+  unit?: string;
+  details?: Record<string, string | number | boolean>;
+}
+
 /**
  * Table representation
  */

package/src/utils/ProductExtractor.ts

@@ -199,11 +199,13 @@ export function extractProductsFromSources(sources: VectorMatch[] | undefined, i
     .filter(s => {
       const m = s.metadata ?? {};
       const keys = Object.keys(m).map(k => k.toLowerCase());
-      const hasProductKey = keys.some(k =>
-        ['price', 'image', 'img', 'thumbnail', 'images', 'brand', 'product', 'sku', 'category', 'model', 'cost'].includes(k)
+      const hasStrongProductKey = keys.some(k =>
+        ['price', 'image', 'img', 'thumbnail', 'images', 'product', 'sku', 'cost'].includes(k)
       );
+      const hasProductIdentity = keys.some(k => ['brand', 'model'].includes(k)) &&
+        keys.some(k => ['name', 'title', 'product_name', 'product'].includes(k));
       const hasPricePattern = /\$\s*\d+/.test(s.content);
-      return hasProductKey || hasPricePattern || m.type === 'product';
+      return hasStrongProductKey || hasProductIdentity || hasPricePattern || m.type === 'product';
     })
     .map(s => {
       const m = (s.metadata ?? {}) as Record<string, unknown>;