@retrivora-ai/rag-engine 2.0.7 → 2.1.1

package/src/utils/UITransformer.ts

@@ -560,8 +560,7 @@ RULES:
   ): UITransformationResponse {
     const products: CarouselProduct[] = data
       .map(item => this.extractProductInfo(item, config, trainedSchema))
-      .filter((p): p is CarouselProduct => p !== null)
-      .slice(0, 15);
+      .filter((p): p is CarouselProduct => p !== null);
 
     return {
       type: 'product_carousel',
@@ -1035,7 +1034,7 @@ RULES:
   }
 
   private static profileData(data: VectorMatch[]): DataProfile {
-    const records = data.map((item): DataRecord => {
+    const records = (data || []).filter((item): item is VectorMatch => Boolean(item && typeof item === 'object')).map((item): DataRecord => {
       const fields: Record<string, PrimitiveValue> = {};
       Object.entries(item.metadata || {}).forEach(([key, value]) => {
         const primitive = this.toPrimitive(value);
@@ -1044,8 +1043,8 @@ RULES:
       if (!fields.content && item.content) fields.content = item.content.substring(0, 500);
       return {
         id: item.id,
-        content: item.content,
-        score: item.score,
+        content: item.content ?? '',
+        score: item.score ?? 0,
         fields,
         source: item,
       };
@@ -1122,6 +1121,15 @@ RULES:
     query: string,
   ): UITransformationResponse | null {
     if (profile.records.length === 0) return null;
+
+    const q = query.toLowerCase().trim();
+    // Do not automatically force charts unless query contains visual or analytical intent
+    const isExplicitVisualOrAnalytic = /\b(chart|graph|plot|visualize|trend|monthly|revenue|top\s*\d+|breakdown|analytics|compare|versus|vs\.?|histogram|pie|bar|line|distribution|percentage|share|summary of metrics)\b/i.test(q);
+
+    if (!isExplicitVisualOrAnalytic) {
+      return null;
+    }
+
     if (profile.dateFields.length > 0 && profile.numericFields.length > 0) {
       return this.transformToLineChart(profile);
     }
@@ -1266,6 +1274,7 @@ RULES:
   }
 
   private static getProductCategory(item: VectorMatch): string | null {
+    if (!item) return null;
     const meta = item.metadata || {};
     const metadataCategory = resolveMetadataValue(meta, 'category');
     if (this.isUsableCategory(metadataCategory)) return String(metadataCategory).trim();
@@ -1402,6 +1411,7 @@ RULES:
     };
 
     data.forEach(item => {
+      if (!item) return;
       Object.keys(item.metadata || {}).forEach(addField);
       for (const match of (item.content || '').matchAll(/(?:^|\n)\s*([A-Za-z][A-Za-z0-9_ /-]{1,50})\s*[:\-–]\s*([^\n]+)/g)) {
         addField(match[1]);
@@ -1552,6 +1562,7 @@ RULES:
   }
 
   private static extractStockQuantity(item: VectorMatch): number | null {
+    if (!item) return null;
     const meta = item.metadata || {};
     const stockValue = resolveMetadataValue(meta, 'stock');
     const numericStock = this.toFiniteNumber(stockValue);