@retrivora-ai/rag-engine 1.7.3 → 1.7.5

package/dist/{chunk-M32ZXLTT.mjs → chunk-QOO37S2Z.mjs}

@@ -829,7 +829,8 @@ ${context}`;
         stopSequences: options == null ? void 0 : options.stop
       }
     });
-    return (_f = response.text) != null ? _f : "";
+    const text = typeof response.text === "function" ? response.text() : (_f = response.text) != null ? _f : "";
+    return text;
   }
   chatStream(messages, context, options) {
     return __asyncGenerator(this, null, function* () {
@@ -863,7 +864,7 @@ ${context}`;
       try {
         for (var iter = __forAwait(stream), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
           const chunk = temp.value;
-          const text = chunk.text;
+          const text = typeof chunk.text === "function" ? chunk.text() : chunk.text;
           if (text) yield text;
         }
       } catch (temp) {
@@ -2259,73 +2260,196 @@ var Pipeline = class {
   async initialize() {
     var _a, _b;
     if (this.initialised) return;
-    const CHART_MARKER = "<!-- UI_PROTOCOL_V6 -->";
+    const CHART_MARKER = "<!-- UI_PROTOCOL_V7 -->";
     const chartInstruction = `
 
 ${CHART_MARKER}
-### UNIVERSAL UI PROTOCOL
-When the user asks for a visual representation, comparison, distribution, breakdown, table, list, catalog, products, carousel, stock view, or category summary, you MUST include exactly one \`\`\`ui\`\`\` block.
-
-1. VIEW SELECTION
-   - Use "carousel" for browsing specific products, items, or a catalog. This is the DEFAULT for "show me", "what are", or "find me" requests for a specific category (e.g. "beauty").
-   - Use "chart" ONLY for numerical aggregations: distributions, percentages, counts, comparisons over time, or trends (e.g. "how many beauty products?").
-   - Use "table" for detailed technical specifications, row-heavy comparisons, or when the user explicitly asks for a list/table format.
-
-2. REQUIRED JSON SHAPE
-   {
-     "view": "chart" | "carousel" | "table",
-     "title": "Short heading",
-     "description": "One sentence describing the view",
-     "chartType": "pie" | "bar" | "line", // Required if view is "chart"
-     "xAxisKey": "label",
-     "dataKeys": ["value"],
-     "columns": ["dynamicFieldKey1", "dynamicFieldKey2"],
-     "insights": ["Short takeaway 1", "Short takeaway 2"],
-     "data": []
-   }
-
-3. DATA RULES
-   - Build the UI payload from retrieved context only. Do not invent products or values.
-   - For CAROUSEL: "data" must be a list of product objects (id, name, brand, price, image, link).
-   - For CHART: "data" must be aggregated counts or sums. Never put raw product lists in a chart view.
-   - If the user asks for "products in beauty", do NOT return a pie chart of categories; return a carousel of the beauty products themselves.
-
-4. FORMAT RULES
-   - Return valid JSON inside the \`\`\`ui\`\`\` block.
-   - Put natural language explanation before or after the block, not inside.
-
-5. EXAMPLES
+### UNIVERSAL UI PROTOCOL V7 (STRICT MODE)
+
+You are responsible for returning a SINGLE structured UI block when visualization is required.
+
+---
+
+## 1. INTENT CLASSIFICATION (MANDATORY)
+
+Classify the query into ONE of the following intents:
+
+- "explore" \u2192 user wants to browse items (e.g., "show me products")
+- "analyze" \u2192 user wants aggregation/distribution (e.g., "distribution of products")
+- "compare" \u2192 user wants side-by-side comparison
+- "detail" \u2192 user wants structured/tabular data
+
+---
+
+## 2. VIEW MAPPING (STRICT)
+
+| Intent   | View       |
+|----------|-----------|
+| explore  | carousel  |
+| analyze  | chart     |
+| compare  | table     |
+| detail   | table     |
+
+\u26A0\uFE0F Override Rules:
+- If query contains "distribution", "percentage", "ratio" \u2192 ALWAYS chart
+- If query contains "show products", "list items" \u2192 ALWAYS carousel
+- NEVER mix views
+
+---
+
+## 3. RESPONSE FORMAT (STRICT JSON ONLY INSIDE BLOCK)
+
+\`\`\`ui
+{
+  "view": "carousel" | "chart" | "table",
+  "title": "string",
+  "description": "string",
+  "metadata": {
+    "intent": "explore" | "analyze" | "compare" | "detail",
+    "confidence": 0.0-1.0
+  },
+
+  // CHART ONLY
+  "chart": {
+    "type": "pie" | "bar" | "line",
+    "xKey": "label",
+    "yKeys": ["value"],
+    "data": []
+  },
+
+  // TABLE ONLY
+  "table": {
+    "columns": [],
+    "rows": []
+  },
+
+  // CAROUSEL ONLY
+  "carousel": {
+    "items": []
+  },
+
+  "insights": []
+}
+\`\`\`
+
+---
+
+## 4. DATA CONTRACT (CRITICAL)
+
+### \u{1F539} CAROUSEL ITEM FORMAT
+{
+  "id": "string",
+  "name": "string",
+  "brand": "string",
+  "price": number,
+  "image": "url",
+  "link": "url",
+  "inStock": boolean
+}
+
+### \u{1F539} CHART DATA FORMAT
+{
+  "label": "string",
+  "value": number,
+  "inStock": boolean (optional)
+}
+
+### \u{1F539} TABLE FORMAT
+- columns MUST match row keys
+- rows MUST be flat objects
+
+---
+
+## 5. DATA RULES
+
+- NEVER hallucinate fields
+- ONLY use retrieved vector DB data
+- NO placeholders ("...", "N/A")
+- If image/link missing \u2192 REMOVE FIELD
+- Aggregate BEFORE rendering chart
+
+---
+
+## 6. SMART RULES (IMPORTANT)
+
+- If user asks BOTH:
+  "show products AND distribution"
+  \u2192 PRIORITIZE "explore" \u2192 carousel
+
+- If dataset size > 20:
+  \u2192 aggregate \u2192 chart or table
+
+---
+
+## 7. OUTPUT RULES
+
+- ALWAYS return EXACTLY ONE \`\`\`ui\`\`\` block
+- JSON must be VALID
+- No explanation inside block
+
+---
+
+## 8. EXAMPLES
+
 User: "show me beauty products"
-Assistant:
+
 \`\`\`ui
 {
   "view": "carousel",
   "title": "Beauty Products",
-  "description": "Browse our latest skincare and makeup collection.",
-  "data": [
-    { "id": "1", "name": "Mascara", "brand": "Essence", "price": "$9.99", "image": "...", "link": "..." },
-    { "id": "2", "name": "Lipstick", "brand": "Chic", "price": "$12.99", "image": "...", "link": "..." }
-  ]
+  "description": "Browse available beauty items",
+  "metadata": {
+    "intent": "explore",
+    "confidence": 0.95
+  },
+  "carousel": {
+    "items": [
+      {
+        "id": "1",
+        "name": "Mascara",
+        "brand": "Essence",
+        "price": 9.99,
+        "image": "https://example.com/mascara.jpg",
+        "link": "https://example.com/p1",
+        "inStock": true
+      }
+    ]
+  },
+  "insights": ["Most products are affordable"]
 }
 \`\`\`
 
-User: "breakdown of products by category"
-Assistant:
+---
+
+User: "distribution of products across categories"
+
 \`\`\`ui
 {
   "view": "chart",
-  "title": "Category Breakdown",
-  "chartType": "pie",
-  "xAxisKey": "label",
-  "dataKeys": ["value"],
-  "data": [
-    { "label": "Beauty", "value": 15 },
-    { "label": "Fragrances", "value": 8 }
-  ]
+  "title": "Product Distribution",
+  "description": "Category-wise product distribution",
+  "metadata": {
+    "intent": "analyze",
+    "confidence": 0.98
+  },
+  "chart": {
+    "type": "pie",
+    "xKey": "label",
+    "yKeys": ["value"],
+    "data": [
+      { "label": "Beauty", "value": 15 },
+      { "label": "Electronics", "value": 10 }
+    ]
+  },
+  "insights": ["Beauty dominates inventory"]
 }
-\`\`\``;
+\`\`\`
+`;
     if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes(CHART_MARKER))) {
-      this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
+      let cleanPrompt = this.config.llm.systemPrompt || "";
+      cleanPrompt = cleanPrompt.replace(/\n\n### UNIVERSAL UI PROTOCOL[\s\S]*?(?=\n\n|$)/g, "");
+      cleanPrompt = cleanPrompt.replace(/<!-- UI_PROTOCOL_V\d+ -->[\s\S]*?(?=\n\n|$)/g, "");
+      this.config.llm.systemPrompt = cleanPrompt.trim() + chartInstruction;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
     this.vectorDB = await ProviderRegistry.createVectorProvider(this.config.vectorDb);

package/dist/handlers/index.js

@@ -2437,7 +2437,8 @@ ${context}`;
         stopSequences: options == null ? void 0 : options.stop
       }
     });
-    return (_f = response.text) != null ? _f : "";
+    const text = typeof response.text === "function" ? response.text() : (_f = response.text) != null ? _f : "";
+    return text;
   }
   chatStream(messages, context, options) {
     return __asyncGenerator(this, null, function* () {
@@ -2471,7 +2472,7 @@ ${context}`;
       try {
         for (var iter = __forAwait(stream), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
           const chunk = temp.value;
-          const text = chunk.text;
+          const text = typeof chunk.text === "function" ? chunk.text() : chunk.text;
           if (text) yield text;
         }
       } catch (temp) {
@@ -3823,73 +3824,196 @@ var Pipeline = class {
   async initialize() {
     var _a, _b;
     if (this.initialised) return;
-    const CHART_MARKER = "<!-- UI_PROTOCOL_V6 -->";
+    const CHART_MARKER = "<!-- UI_PROTOCOL_V7 -->";
     const chartInstruction = `
 
 ${CHART_MARKER}
-### UNIVERSAL UI PROTOCOL
-When the user asks for a visual representation, comparison, distribution, breakdown, table, list, catalog, products, carousel, stock view, or category summary, you MUST include exactly one \`\`\`ui\`\`\` block.
+### UNIVERSAL UI PROTOCOL V7 (STRICT MODE)
 
-1. VIEW SELECTION
-   - Use "carousel" for browsing specific products, items, or a catalog. This is the DEFAULT for "show me", "what are", or "find me" requests for a specific category (e.g. "beauty").
-   - Use "chart" ONLY for numerical aggregations: distributions, percentages, counts, comparisons over time, or trends (e.g. "how many beauty products?").
-   - Use "table" for detailed technical specifications, row-heavy comparisons, or when the user explicitly asks for a list/table format.
+You are responsible for returning a SINGLE structured UI block when visualization is required.
 
-2. REQUIRED JSON SHAPE
-   {
-     "view": "chart" | "carousel" | "table",
-     "title": "Short heading",
-     "description": "One sentence describing the view",
-     "chartType": "pie" | "bar" | "line", // Required if view is "chart"
-     "xAxisKey": "label",
-     "dataKeys": ["value"],
-     "columns": ["dynamicFieldKey1", "dynamicFieldKey2"],
-     "insights": ["Short takeaway 1", "Short takeaway 2"],
-     "data": []
-   }
+---
 
-3. DATA RULES
-   - Build the UI payload from retrieved context only. Do not invent products or values.
-   - For CAROUSEL: "data" must be a list of product objects (id, name, brand, price, image, link).
-   - For CHART: "data" must be aggregated counts or sums. Never put raw product lists in a chart view.
-   - If the user asks for "products in beauty", do NOT return a pie chart of categories; return a carousel of the beauty products themselves.
+## 1. INTENT CLASSIFICATION (MANDATORY)
 
-4. FORMAT RULES
-   - Return valid JSON inside the \`\`\`ui\`\`\` block.
-   - Put natural language explanation before or after the block, not inside.
+Classify the query into ONE of the following intents:
+
+- "explore" \u2192 user wants to browse items (e.g., "show me products")
+- "analyze" \u2192 user wants aggregation/distribution (e.g., "distribution of products")
+- "compare" \u2192 user wants side-by-side comparison
+- "detail" \u2192 user wants structured/tabular data
+
+---
+
+## 2. VIEW MAPPING (STRICT)
+
+| Intent   | View       |
+|----------|-----------|
+| explore  | carousel  |
+| analyze  | chart     |
+| compare  | table     |
+| detail   | table     |
+
+\u26A0\uFE0F Override Rules:
+- If query contains "distribution", "percentage", "ratio" \u2192 ALWAYS chart
+- If query contains "show products", "list items" \u2192 ALWAYS carousel
+- NEVER mix views
+
+---
+
+## 3. RESPONSE FORMAT (STRICT JSON ONLY INSIDE BLOCK)
+
+\`\`\`ui
+{
+  "view": "carousel" | "chart" | "table",
+  "title": "string",
+  "description": "string",
+  "metadata": {
+    "intent": "explore" | "analyze" | "compare" | "detail",
+    "confidence": 0.0-1.0
+  },
+
+  // CHART ONLY
+  "chart": {
+    "type": "pie" | "bar" | "line",
+    "xKey": "label",
+    "yKeys": ["value"],
+    "data": []
+  },
+
+  // TABLE ONLY
+  "table": {
+    "columns": [],
+    "rows": []
+  },
+
+  // CAROUSEL ONLY
+  "carousel": {
+    "items": []
+  },
+
+  "insights": []
+}
+\`\`\`
+
+---
+
+## 4. DATA CONTRACT (CRITICAL)
+
+### \u{1F539} CAROUSEL ITEM FORMAT
+{
+  "id": "string",
+  "name": "string",
+  "brand": "string",
+  "price": number,
+  "image": "url",
+  "link": "url",
+  "inStock": boolean
+}
+
+### \u{1F539} CHART DATA FORMAT
+{
+  "label": "string",
+  "value": number,
+  "inStock": boolean (optional)
+}
+
+### \u{1F539} TABLE FORMAT
+- columns MUST match row keys
+- rows MUST be flat objects
+
+---
+
+## 5. DATA RULES
+
+- NEVER hallucinate fields
+- ONLY use retrieved vector DB data
+- NO placeholders ("...", "N/A")
+- If image/link missing \u2192 REMOVE FIELD
+- Aggregate BEFORE rendering chart
+
+---
+
+## 6. SMART RULES (IMPORTANT)
+
+- If user asks BOTH:
+  "show products AND distribution"
+  \u2192 PRIORITIZE "explore" \u2192 carousel
+
+- If dataset size > 20:
+  \u2192 aggregate \u2192 chart or table
+
+---
+
+## 7. OUTPUT RULES
+
+- ALWAYS return EXACTLY ONE \`\`\`ui\`\`\` block
+- JSON must be VALID
+- No explanation inside block
+
+---
+
+## 8. EXAMPLES
 
-5. EXAMPLES
 User: "show me beauty products"
-Assistant:
+
 \`\`\`ui
 {
   "view": "carousel",
   "title": "Beauty Products",
-  "description": "Browse our latest skincare and makeup collection.",
-  "data": [
-    { "id": "1", "name": "Mascara", "brand": "Essence", "price": "$9.99", "image": "...", "link": "..." },
-    { "id": "2", "name": "Lipstick", "brand": "Chic", "price": "$12.99", "image": "...", "link": "..." }
-  ]
+  "description": "Browse available beauty items",
+  "metadata": {
+    "intent": "explore",
+    "confidence": 0.95
+  },
+  "carousel": {
+    "items": [
+      {
+        "id": "1",
+        "name": "Mascara",
+        "brand": "Essence",
+        "price": 9.99,
+        "image": "https://example.com/mascara.jpg",
+        "link": "https://example.com/p1",
+        "inStock": true
+      }
+    ]
+  },
+  "insights": ["Most products are affordable"]
 }
 \`\`\`
 
-User: "breakdown of products by category"
-Assistant:
+---
+
+User: "distribution of products across categories"
+
 \`\`\`ui
 {
   "view": "chart",
-  "title": "Category Breakdown",
-  "chartType": "pie",
-  "xAxisKey": "label",
-  "dataKeys": ["value"],
-  "data": [
-    { "label": "Beauty", "value": 15 },
-    { "label": "Fragrances", "value": 8 }
-  ]
+  "title": "Product Distribution",
+  "description": "Category-wise product distribution",
+  "metadata": {
+    "intent": "analyze",
+    "confidence": 0.98
+  },
+  "chart": {
+    "type": "pie",
+    "xKey": "label",
+    "yKeys": ["value"],
+    "data": [
+      { "label": "Beauty", "value": 15 },
+      { "label": "Electronics", "value": 10 }
+    ]
+  },
+  "insights": ["Beauty dominates inventory"]
 }
-\`\`\``;
+\`\`\`
+`;
     if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes(CHART_MARKER))) {
-      this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
+      let cleanPrompt = this.config.llm.systemPrompt || "";
+      cleanPrompt = cleanPrompt.replace(/\n\n### UNIVERSAL UI PROTOCOL[\s\S]*?(?=\n\n|$)/g, "");
+      cleanPrompt = cleanPrompt.replace(/<!-- UI_PROTOCOL_V\d+ -->[\s\S]*?(?=\n\n|$)/g, "");
+      this.config.llm.systemPrompt = cleanPrompt.trim() + chartInstruction;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
     this.vectorDB = await ProviderRegistry.createVectorProvider(this.config.vectorDb);

package/dist/handlers/index.mjs

@@ -10,7 +10,7 @@ import {
   sseMetaFrame,
   sseTextFrame,
   sseUIFrame
-} from "../chunk-M32ZXLTT.mjs";
+} from "../chunk-QOO37S2Z.mjs";
 import "../chunk-YLTMFW4M.mjs";
 import "../chunk-X4TOT24V.mjs";
 export {

package/dist/index.js

@@ -405,7 +405,7 @@ function sanitizeJson(raw) {
       return `: ${result}`;
     }
   );
-  s = s.replace(/\/\/[^\n]*/g, "");
+  s = s.replace(/("(?:\\.|[^\\"])*")|\/\/[^\n]*|\/\*[\s\S]*?\*\//g, (m, g1) => g1 || "");
   s = s.replace(/([{,]\s*)([A-Za-z_$][A-Za-z0-9_$]*)\s*:/g, '$1"$2":');
   s = s.replace(/'([^'\\]*(?:\\.[^'\\]*)*)'/g, '"$1"');
   s = s.replace(/,\s*([}\]])/g, "$1");
@@ -535,11 +535,12 @@ function isLikelyUiOnlyMessage(text) {
   return trimmed.length === 0 || /^(chart data|table data|visualization|visual representation|product catalog|product recommendations|catalog summary)\s*:?\s*$/i.test(trimmed);
 }
 function resolveImage(data) {
+  const isPlaceholder = (val) => typeof val === "string" && (val === "..." || val === "" || val.toLowerCase() === "null" || val.toLowerCase() === "undefined");
   for (const key of ["image", "img", "thumbnail"]) {
     const v = data[key];
-    if (typeof v === "string" && v) return v;
+    if (typeof v === "string" && v && !isPlaceholder(v)) return v;
   }
-  if (Array.isArray(data.images) && typeof data.images[0] === "string") {
+  if (Array.isArray(data.images) && typeof data.images[0] === "string" && !isPlaceholder(data.images[0])) {
     return data.images[0];
   }
   return void 0;
@@ -636,6 +637,11 @@ function normalizeTabularRows(rows, columns) {
   }).filter((row) => Object.keys(row).length > 0);
 }
 function resolveStructuredRows(config) {
+  var _a, _b, _c, _d;
+  if (((_a = config.carousel) == null ? void 0 : _a.items) && Array.isArray(config.carousel.items)) return config.carousel.items;
+  if (((_b = config.chart) == null ? void 0 : _b.data) && Array.isArray(config.chart.data)) return config.chart.data;
+  if (((_c = config.table) == null ? void 0 : _c.rows) && Array.isArray(config.table.rows)) return config.table.rows;
+  if (((_d = config.table) == null ? void 0 : _d.data) && Array.isArray(config.table.data)) return config.table.data;
   if (Array.isArray(config.data) && config.data.length > 0) return config.data;
   if (Array.isArray(config.items) && config.items.length > 0) return config.items;
   if (Array.isArray(config.products) && config.products.length > 0) return config.products;
@@ -651,6 +657,14 @@ function UIDispatcher({ rawContent, primaryColor, accentColor, isStreaming, onAd
       const clean = rawContent.replace(/^```[a-z]*\s*/i, "").replace(/```\s*$/, "").trim();
       const parsed = JSON.parse(sanitizeJson(extractLikelyJsonBlock(clean)));
       const config2 = __spreadValues({}, parsed);
+      if (config2.chart) {
+        if (!config2.chartType) config2.chartType = config2.chart.type;
+        if (!config2.xAxisKey) config2.xAxisKey = config2.chart.xKey;
+        if (!config2.dataKeys) config2.dataKeys = config2.chart.yKeys;
+      }
+      if (config2.table) {
+        if (!config2.columns) config2.columns = config2.table.columns;
+      }
       const resolvedRows = resolveStructuredRows(config2);
       const hasProductLikeData = resolvedRows.length > 0 && typeof resolvedRows[0] === "object" && resolvedRows[0] !== null && !Array.isArray(resolvedRows[0]) && (["price", "brand", "image", "link", "thumbnail"].some((key) => key in resolvedRows[0]) || Object.keys(resolvedRows[0]).length >= 4);
       const hasAggregationLikeData = resolvedRows.length > 0 && typeof resolvedRows[0] === "object" && resolvedRows[0] !== null && !Array.isArray(resolvedRows[0]) && (["value", "count", "total", "percentage"].some((key) => key in resolvedRows[0]) || Object.keys(resolvedRows[0]).length <= 3 && Object.values(resolvedRows[0]).some((v) => typeof v === "number"));

package/dist/index.mjs

@@ -368,7 +368,7 @@ function sanitizeJson(raw) {
       return `: ${result}`;
     }
   );
-  s = s.replace(/\/\/[^\n]*/g, "");
+  s = s.replace(/("(?:\\.|[^\\"])*")|\/\/[^\n]*|\/\*[\s\S]*?\*\//g, (m, g1) => g1 || "");
   s = s.replace(/([{,]\s*)([A-Za-z_$][A-Za-z0-9_$]*)\s*:/g, '$1"$2":');
   s = s.replace(/'([^'\\]*(?:\\.[^'\\]*)*)'/g, '"$1"');
   s = s.replace(/,\s*([}\]])/g, "$1");
@@ -498,11 +498,12 @@ function isLikelyUiOnlyMessage(text) {
   return trimmed.length === 0 || /^(chart data|table data|visualization|visual representation|product catalog|product recommendations|catalog summary)\s*:?\s*$/i.test(trimmed);
 }
 function resolveImage(data) {
+  const isPlaceholder = (val) => typeof val === "string" && (val === "..." || val === "" || val.toLowerCase() === "null" || val.toLowerCase() === "undefined");
   for (const key of ["image", "img", "thumbnail"]) {
     const v = data[key];
-    if (typeof v === "string" && v) return v;
+    if (typeof v === "string" && v && !isPlaceholder(v)) return v;
   }
-  if (Array.isArray(data.images) && typeof data.images[0] === "string") {
+  if (Array.isArray(data.images) && typeof data.images[0] === "string" && !isPlaceholder(data.images[0])) {
     return data.images[0];
   }
   return void 0;
@@ -599,6 +600,11 @@ function normalizeTabularRows(rows, columns) {
   }).filter((row) => Object.keys(row).length > 0);
 }
 function resolveStructuredRows(config) {
+  var _a, _b, _c, _d;
+  if (((_a = config.carousel) == null ? void 0 : _a.items) && Array.isArray(config.carousel.items)) return config.carousel.items;
+  if (((_b = config.chart) == null ? void 0 : _b.data) && Array.isArray(config.chart.data)) return config.chart.data;
+  if (((_c = config.table) == null ? void 0 : _c.rows) && Array.isArray(config.table.rows)) return config.table.rows;
+  if (((_d = config.table) == null ? void 0 : _d.data) && Array.isArray(config.table.data)) return config.table.data;
   if (Array.isArray(config.data) && config.data.length > 0) return config.data;
   if (Array.isArray(config.items) && config.items.length > 0) return config.items;
   if (Array.isArray(config.products) && config.products.length > 0) return config.products;
@@ -614,6 +620,14 @@ function UIDispatcher({ rawContent, primaryColor, accentColor, isStreaming, onAd
       const clean = rawContent.replace(/^```[a-z]*\s*/i, "").replace(/```\s*$/, "").trim();
       const parsed = JSON.parse(sanitizeJson(extractLikelyJsonBlock(clean)));
       const config2 = __spreadValues({}, parsed);
+      if (config2.chart) {
+        if (!config2.chartType) config2.chartType = config2.chart.type;
+        if (!config2.xAxisKey) config2.xAxisKey = config2.chart.xKey;
+        if (!config2.dataKeys) config2.dataKeys = config2.chart.yKeys;
+      }
+      if (config2.table) {
+        if (!config2.columns) config2.columns = config2.table.columns;
+      }
       const resolvedRows = resolveStructuredRows(config2);
       const hasProductLikeData = resolvedRows.length > 0 && typeof resolvedRows[0] === "object" && resolvedRows[0] !== null && !Array.isArray(resolvedRows[0]) && (["price", "brand", "image", "link", "thumbnail"].some((key) => key in resolvedRows[0]) || Object.keys(resolvedRows[0]).length >= 4);
       const hasAggregationLikeData = resolvedRows.length > 0 && typeof resolvedRows[0] === "object" && resolvedRows[0] !== null && !Array.isArray(resolvedRows[0]) && (["value", "count", "total", "percentage"].some((key) => key in resolvedRows[0]) || Object.keys(resolvedRows[0]).length <= 3 && Object.values(resolvedRows[0]).some((v) => typeof v === "number"));