@retrivora-ai/rag-engine 1.5.3 → 1.5.5

package/dist/index.mjs

@@ -233,8 +233,58 @@ function DynamicChart({ config, primaryColor = "#6366f1", accentColor = "#8b5cf6
   }).slice(0, 5);
   const finalXKey = String(resolvedXKey);
   const finalDataKeys = resolvedDataKeys.map(String).filter((k) => k !== "undefined");
+  const pieDataKey = finalDataKeys[0] || "value";
+  const stockAwareColor = (entry, index) => {
+    var _a, _b, _c;
+    const stockStatus = String(
+      (_c = (_b = (_a = entry.stockStatus) != null ? _a : entry.status) != null ? _b : entry.availability) != null ? _c : ""
+    ).toLowerCase();
+    const inStockCount = typeof entry.inStockCount === "number" ? entry.inStockCount : void 0;
+    const outOfStockCount = typeof entry.outOfStockCount === "number" ? entry.outOfStockCount : void 0;
+    const inStockFlag = entry.inStock;
+    if (stockStatus.includes("in stock")) return "#10b981";
+    if (stockStatus.includes("out of stock")) return "#f97316";
+    if (typeof inStockFlag === "string" && inStockFlag.toLowerCase() === "true") return "#10b981";
+    if (inStockFlag === 1) return "#10b981";
+    if (inStockFlag === 0) return "#f97316";
+    if (typeof inStockCount === "number" || typeof outOfStockCount === "number") {
+      const inCount = inStockCount != null ? inStockCount : 0;
+      const outCount = outOfStockCount != null ? outOfStockCount : 0;
+      if (inCount > outCount) return "#10b981";
+      if (outCount > inCount) return "#f97316";
+    }
+    return colors[index % colors.length];
+  };
+  const tooltipLabelMap = {
+    value: "Value",
+    count: "Count",
+    inStock: "In stock",
+    inStockCount: "In stock",
+    outOfStockCount: "Out of stock",
+    stockStatus: "Stock status",
+    category: "Category",
+    label: "Label",
+    name: "Name"
+  };
+  const renderTooltip = (row, label) => {
+    var _a, _b;
+    if (!row) return null;
+    return /* @__PURE__ */ React4.createElement("div", { className: "rounded-xl border border-slate-200 bg-white px-3 py-2 text-xs shadow-xl dark:border-white/10 dark:bg-slate-950" }, /* @__PURE__ */ React4.createElement("p", { className: "mb-2 font-semibold text-slate-800 dark:text-white/90" }, String((_b = (_a = row[finalXKey]) != null ? _a : label) != null ? _b : "Details")), Object.entries(row).map(([key, value]) => {
+      var _a2;
+      if (value === null || value === void 0 || key === finalXKey) return null;
+      if (typeof value === "object") return null;
+      return /* @__PURE__ */ React4.createElement("div", { key, className: "flex items-center justify-between gap-3 text-slate-600 dark:text-white/70" }, /* @__PURE__ */ React4.createElement("span", null, (_a2 = tooltipLabelMap[key]) != null ? _a2 : key), /* @__PURE__ */ React4.createElement("span", { className: "font-medium text-slate-900 dark:text-white/90" }, String(value)));
+    }));
+  };
+  const getTooltipRow = (payload) => {
+    if (!Array.isArray(payload) || payload.length === 0) return null;
+    const first = payload[0];
+    if (!first || typeof first !== "object" || !("payload" in first)) return null;
+    const row = first.payload;
+    if (!row || typeof row !== "object" || Array.isArray(row)) return null;
+    return row;
+  };
   if (type === "pie") {
-    const pieDataKey = finalDataKeys[0] || "value";
     return /* @__PURE__ */ React4.createElement("div", { className: "w-full h-72 min-h-[280px] mt-4 mb-2 select-none overflow-hidden" }, /* @__PURE__ */ React4.createElement(ResponsiveContainer, { width: "99%", height: "100%" }, /* @__PURE__ */ React4.createElement(PieChart, null, /* @__PURE__ */ React4.createElement(
       Pie,
       {
@@ -246,21 +296,10 @@ function DynamicChart({ config, primaryColor = "#6366f1", accentColor = "#8b5cf6
         outerRadius: 80,
         label: ({ name, percent }) => `${name} ${((percent != null ? percent : 0) * 100).toFixed(0)}%`
       },
-      data.map((entry, index) => /* @__PURE__ */ React4.createElement(Cell, { key: `cell-${index}`, fill: colors[index % colors.length] }))
-    ), /* @__PURE__ */ React4.createElement(
-      Tooltip,
-      {
-        contentStyle: { borderRadius: "8px", border: "none", boxShadow: "0 4px 6px -1px rgb(0 0 0 / 0.1)" }
-      }
-    ), /* @__PURE__ */ React4.createElement(Legend, null))));
+      sanitizedData.map((entry, index) => /* @__PURE__ */ React4.createElement(Cell, { key: `cell-${index}`, fill: stockAwareColor(entry, index) }))
+    ), /* @__PURE__ */ React4.createElement(Tooltip, { content: (props) => renderTooltip(getTooltipRow(props.payload), typeof props.label === "string" ? props.label : void 0) }), /* @__PURE__ */ React4.createElement(Legend, null))));
   }
-  return /* @__PURE__ */ React4.createElement("div", { className: "w-full h-72 min-h-[280px] mt-4 mb-2 select-none overflow-hidden" }, /* @__PURE__ */ React4.createElement(ResponsiveContainer, { width: "99%", height: "100%" }, type === "bar" ? /* @__PURE__ */ React4.createElement(BarChart, { data: sanitizedData, margin: { top: 10, right: 10, left: -20, bottom: 0 } }, /* @__PURE__ */ React4.createElement(CartesianGrid, { strokeDasharray: "3 3", vertical: false, stroke: "#e2e8f0" }), /* @__PURE__ */ React4.createElement(XAxis, { dataKey: finalXKey, tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: { stroke: "#cbd5e1" } }), /* @__PURE__ */ React4.createElement(YAxis, { tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: false }), /* @__PURE__ */ React4.createElement(
-    Tooltip,
-    {
-      contentStyle: { borderRadius: "8px", border: "none", boxShadow: "0 4px 6px -1px rgb(0 0 0 / 0.1)" },
-      cursor: { fill: "#f1f5f9" }
-    }
-  ), /* @__PURE__ */ React4.createElement(Legend, { wrapperStyle: { fontSize: 12 } }), finalDataKeys.map((key, index) => /* @__PURE__ */ React4.createElement(
+  return /* @__PURE__ */ React4.createElement("div", { className: "w-full h-72 min-h-[280px] mt-4 mb-2 select-none overflow-hidden" }, /* @__PURE__ */ React4.createElement(ResponsiveContainer, { width: "99%", height: "100%" }, type === "bar" ? /* @__PURE__ */ React4.createElement(BarChart, { data: sanitizedData, margin: { top: 10, right: 10, left: -20, bottom: 0 } }, /* @__PURE__ */ React4.createElement(CartesianGrid, { strokeDasharray: "3 3", vertical: false, stroke: "#e2e8f0" }), /* @__PURE__ */ React4.createElement(XAxis, { dataKey: finalXKey, tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: { stroke: "#cbd5e1" } }), /* @__PURE__ */ React4.createElement(YAxis, { tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: false }), /* @__PURE__ */ React4.createElement(Tooltip, { content: (props) => renderTooltip(getTooltipRow(props.payload), typeof props.label === "string" ? props.label : void 0), cursor: { fill: "#f1f5f9" } }), /* @__PURE__ */ React4.createElement(Legend, { wrapperStyle: { fontSize: 12 } }), finalDataKeys.map((key, index) => /* @__PURE__ */ React4.createElement(
     Bar,
     {
       key,
@@ -269,12 +308,7 @@ function DynamicChart({ config, primaryColor = "#6366f1", accentColor = "#8b5cf6
       radius: [4, 4, 0, 0],
       barSize: Math.max(20, 60 / data.length)
     }
-  ))) : /* @__PURE__ */ React4.createElement(LineChart, { data: sanitizedData, margin: { top: 10, right: 10, left: -20, bottom: 0 } }, /* @__PURE__ */ React4.createElement(CartesianGrid, { strokeDasharray: "3 3", vertical: false, stroke: "#e2e8f0" }), /* @__PURE__ */ React4.createElement(XAxis, { dataKey: finalXKey, tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: { stroke: "#cbd5e1" } }), /* @__PURE__ */ React4.createElement(YAxis, { tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: false }), /* @__PURE__ */ React4.createElement(
-    Tooltip,
-    {
-      contentStyle: { borderRadius: "8px", border: "none", boxShadow: "0 4px 6px -1px rgb(0 0 0 / 0.1)" }
-    }
-  ), /* @__PURE__ */ React4.createElement(Legend, { wrapperStyle: { fontSize: 12 } }), finalDataKeys.map((key, index) => /* @__PURE__ */ React4.createElement(
+  ))) : /* @__PURE__ */ React4.createElement(LineChart, { data: sanitizedData, margin: { top: 10, right: 10, left: -20, bottom: 0 } }, /* @__PURE__ */ React4.createElement(CartesianGrid, { strokeDasharray: "3 3", vertical: false, stroke: "#e2e8f0" }), /* @__PURE__ */ React4.createElement(XAxis, { dataKey: finalXKey, tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: { stroke: "#cbd5e1" } }), /* @__PURE__ */ React4.createElement(YAxis, { tick: { fontSize: 12, fill: "#64748b" }, tickLine: false, axisLine: false }), /* @__PURE__ */ React4.createElement(Tooltip, { content: (props) => renderTooltip(getTooltipRow(props.payload), typeof props.label === "string" ? props.label : void 0) }), /* @__PURE__ */ React4.createElement(Legend, { wrapperStyle: { fontSize: 12 } }), finalDataKeys.map((key, index) => /* @__PURE__ */ React4.createElement(
     Line,
     {
       key,
@@ -403,37 +437,84 @@ function DataTable({ config }) {
     ))
   ))))), /* @__PURE__ */ React5.createElement("div", { className: "px-4 py-2 bg-slate-50 dark:bg-white/5 border-t border-slate-100 dark:border-white/5" }, /* @__PURE__ */ React5.createElement("p", { className: "text-[11px] text-slate-400 dark:text-white/30" }, config.data.length, " row", config.data.length !== 1 ? "s" : "")));
 }
-function ChartBlock({ rawContent, primaryColor, accentColor, isStreaming }) {
+function UIDispatcher({ rawContent, primaryColor, accentColor, isStreaming, onAddToCart }) {
+  var _a;
   const result = React5.useMemo(() => {
     if (isStreaming) return { loading: true };
-    if (!rawContent || rawContent === "undefined") return { error: "Empty visualization config." };
     try {
       const clean = rawContent.replace(/^```[a-z]*\s*/i, "").replace(/```\s*$/, "").trim();
-      const config2 = JSON.parse(sanitizeJson(clean));
+      const parsed = JSON.parse(sanitizeJson(clean));
+      const config2 = __spreadValues({}, parsed);
+      if (!config2.view) {
+        if (config2.type === "products" || Array.isArray(config2.items)) {
+          config2.view = "carousel";
+          config2.data = config2.data || config2.items || [];
+        } else if (["pie", "bar", "line"].includes(config2.type) || ["pie", "bar", "line"].includes(config2.chartType)) {
+          config2.view = "chart";
+          config2.chartType = config2.chartType || config2.type || "bar";
+          config2.data = config2.data || [];
+        } else {
+          config2.view = "table";
+          config2.data = Array.isArray(config2.data) ? config2.data : Array.isArray(config2) ? config2 : [];
+        }
+      }
       return { config: config2 };
     } catch (err) {
-      console.error("[ChartBlock] Parsing failed.\nError:", err);
       return { error: String(err) };
     }
   }, [rawContent, isStreaming]);
   if ("loading" in result) {
-    return /* @__PURE__ */ React5.createElement("div", { className: "my-4 p-8 bg-slate-50 dark:bg-white/5 rounded-xl border border-dashed border-slate-300 dark:border-white/10 flex flex-col items-center justify-center gap-3 select-none" }, /* @__PURE__ */ React5.createElement("div", { className: "w-6 h-6 border-2 border-indigo-500 border-t-transparent rounded-full animate-spin" }), /* @__PURE__ */ React5.createElement("p", { className: "text-xs text-slate-500 font-medium animate-pulse" }, "Preparing data visualization\u2026"));
+    return /* @__PURE__ */ React5.createElement("div", { className: "my-4 p-8 bg-slate-50 dark:bg-white/5 rounded-xl border border-dashed border-slate-300 dark:border-white/10 flex flex-col items-center justify-center gap-3 animate-pulse" }, /* @__PURE__ */ React5.createElement("div", { className: "w-5 h-5 border-2 border-indigo-500 border-t-transparent rounded-full animate-spin" }), /* @__PURE__ */ React5.createElement("p", { className: "text-xs text-slate-500 font-medium italic" }, "Preparing view..."));
   }
   if ("error" in result) {
-    return /* @__PURE__ */ React5.createElement("div", { className: "p-4 bg-red-50 dark:bg-red-900/20 text-red-600 dark:text-red-400 rounded-lg text-sm border border-red-100 dark:border-red-900/30" }, /* @__PURE__ */ React5.createElement("p", { className: "font-medium mb-1" }, "Failed to render visualization"), /* @__PURE__ */ React5.createElement("p", { className: "opacity-70 text-xs" }, "The generated configuration is invalid. Check the console for details."));
+    return /* @__PURE__ */ React5.createElement("pre", { className: "p-4 my-2 bg-slate-900 text-slate-100 rounded-lg text-[10px] overflow-auto max-h-40" }, /* @__PURE__ */ React5.createElement("code", null, rawContent));
   }
   const { config } = result;
-  if (config.type === "table") {
-    return /* @__PURE__ */ React5.createElement(DataTable, { config });
+  const hasInsights = Array.isArray(config.insights) && config.insights.length > 0;
+  const hasDescription = typeof config.description === "string" && config.description.trim().length > 0;
+  switch (config.view) {
+    case "chart":
+      return /* @__PURE__ */ React5.createElement("div", { className: "my-6 p-4 bg-white dark:bg-slate-900 rounded-2xl border border-slate-200 dark:border-white/10 shadow-sm overflow-hidden" }, config.title && /* @__PURE__ */ React5.createElement("h4", { className: "text-xs font-semibold text-slate-500 mb-4 px-2" }, config.title), hasDescription && /* @__PURE__ */ React5.createElement("p", { className: "px-2 text-sm text-slate-500 dark:text-white/60" }, config.description), /* @__PURE__ */ React5.createElement(
+        DynamicChart,
+        {
+          config: {
+            type: config.chartType || "bar",
+            data: config.data,
+            xAxisKey: config.xAxisKey,
+            dataKeys: config.dataKeys,
+            colors: config.colors
+          },
+          primaryColor,
+          accentColor
+        }
+      ), hasInsights && /* @__PURE__ */ React5.createElement("div", { className: "mt-4 flex flex-wrap gap-2 px-2" }, (_a = config.insights) == null ? void 0 : _a.map((insight) => /* @__PURE__ */ React5.createElement(
+        "span",
+        {
+          key: insight,
+          className: "rounded-full border border-emerald-200 bg-emerald-50 px-3 py-1 text-[11px] font-medium text-emerald-700 dark:border-emerald-500/20 dark:bg-emerald-500/10 dark:text-emerald-300"
+        },
+        insight
+      ))));
+    case "carousel":
+      return /* @__PURE__ */ React5.createElement("div", { className: "my-4" }, config.title && /* @__PURE__ */ React5.createElement("h4", { className: "mb-2 text-xs font-semibold text-slate-500" }, config.title), hasDescription && /* @__PURE__ */ React5.createElement("p", { className: "mb-3 text-sm text-slate-500 dark:text-white/60" }, config.description), /* @__PURE__ */ React5.createElement(ProductCarousel, { products: config.data.map((item) => {
+        var _a2;
+        return __spreadProps(__spreadValues({}, item), {
+          image: (_a2 = item.image) != null ? _a2 : typeof resolveImage === "function" ? resolveImage(item) : void 0
+        });
+      }), primaryColor, onAddToCart }));
+    case "table":
+      return /* @__PURE__ */ React5.createElement("div", { className: "my-4 p-4 bg-white dark:bg-slate-900 rounded-xl border border-slate-200 dark:border-white/10 shadow-sm overflow-auto max-h-[400px]" }, config.title && /* @__PURE__ */ React5.createElement("h4", { className: "text-xs font-semibold text-slate-500 mb-2" }, config.title), hasDescription && /* @__PURE__ */ React5.createElement("p", { className: "mb-3 text-sm text-slate-500 dark:text-white/60" }, config.description), /* @__PURE__ */ React5.createElement(DataTable, { config: {
+        type: "table",
+        title: config.title,
+        description: config.description,
+        columns: config.columns,
+        dataKeys: config.dataKeys,
+        xAxisKey: config.xAxisKey,
+        data: config.data
+      } }));
+    default:
+      return null;
   }
-  return /* @__PURE__ */ React5.createElement("div", { className: "my-4 p-4 bg-white dark:bg-slate-900 rounded-xl border border-slate-200 dark:border-white/10 shadow-sm" }, /* @__PURE__ */ React5.createElement(
-    DynamicChart,
-    {
-      config,
-      primaryColor,
-      accentColor
-    }
-  ));
 }
 function MessageBubble({
   message,
@@ -445,6 +526,10 @@ function MessageBubble({
 }) {
   const isUser = message.role === "user";
   const [showSources, setShowSources] = React5.useState(false);
+  const hasStructuredProductBlock = React5.useMemo(
+    () => /```(?:ui|json|chart)\s*[\s\S]*?"(?:view|type)"\s*:\s*"(?:carousel|products)"/i.test(message.content),
+    [message.content]
+  );
   const productsFromSources = React5.useMemo(() => {
     if (isUser || !sources) return [];
     return sources.filter((s) => {
@@ -608,24 +693,40 @@ ${match.trim()}
         ]);
         var _a;
         const lang = (_a = /language-(\w+)/.exec(className != null ? className : "")) == null ? void 0 : _a[1];
-        if (!inline && lang === "chart") {
+        if (!inline && (lang === "ui" || lang === "chart")) {
           return /* @__PURE__ */ React5.createElement(
-            ChartBlock,
+            UIDispatcher,
             {
               rawContent: String(children != null ? children : "").trim(),
               primaryColor,
               accentColor,
-              isStreaming
+              isStreaming,
+              onAddToCart
             }
           );
         }
+        if (!inline && lang === "json") {
+          const content = String(children != null ? children : "").trim();
+          if (content.includes('"type": "products"') || content.includes('"type":"products"')) {
+            return /* @__PURE__ */ React5.createElement(
+              UIDispatcher,
+              {
+                rawContent: content,
+                primaryColor,
+                accentColor,
+                isStreaming,
+                onAddToCart
+              }
+            );
+          }
+        }
         if (!inline && lang === "table-loading") {
           return /* @__PURE__ */ React5.createElement("div", { className: "my-5 p-6 bg-slate-50 dark:bg-white/5 rounded-xl border border-dashed border-slate-300 dark:border-white/10 flex items-center justify-center gap-3 select-none" }, /* @__PURE__ */ React5.createElement("div", { className: "w-4 h-4 border-2 border-indigo-500 border-t-transparent rounded-full animate-spin" }), /* @__PURE__ */ React5.createElement("p", { className: "text-xs text-slate-500 font-medium italic animate-pulse" }, "Generating data table..."));
         }
         return /* @__PURE__ */ React5.createElement("code", __spreadValues({ className }, props), typeof children === "string" || typeof children === "number" ? children : JSON.stringify(children));
       }
     }),
-    [primaryColor, accentColor, isStreaming]
+    [primaryColor, accentColor, isStreaming, onAddToCart]
   );
   return /* @__PURE__ */ React5.createElement("div", { className: `flex gap-3 ${isUser ? "flex-row-reverse" : "flex-row"} items-start` }, /* @__PURE__ */ React5.createElement(
     "div",
@@ -641,7 +742,7 @@ ${match.trim()}
       style: isUser ? { background: `linear-gradient(135deg, ${primaryColor}, ${accentColor})` } : {}
     },
     isStreaming && !message.content ? /* @__PURE__ */ React5.createElement("span", { className: "flex items-center gap-1 py-1" }, /* @__PURE__ */ React5.createElement("span", { className: "w-1.5 h-1.5 rounded-full bg-slate-400 dark:bg-white/60 animate-bounce [animation-delay:0ms]" }), /* @__PURE__ */ React5.createElement("span", { className: "w-1.5 h-1.5 rounded-full bg-slate-400 dark:bg-white/60 animate-bounce [animation-delay:150ms]" }), /* @__PURE__ */ React5.createElement("span", { className: "w-1.5 h-1.5 rounded-full bg-slate-400 dark:bg-white/60 animate-bounce [animation-delay:300ms]" })) : /* @__PURE__ */ React5.createElement("div", { className: `prose prose-sm max-w-none ${isUser ? "prose-invert" : "dark:prose-invert"}` }, /* @__PURE__ */ React5.createElement(ReactMarkdown, { remarkPlugins: [remarkGfm], components: markdownComponents }, processedMarkdown), isStreaming && message.content && /* @__PURE__ */ React5.createElement("span", { className: "inline-block w-1.5 h-3.5 ml-1 bg-emerald-500 animate-pulse align-middle" }))
-  ), !isUser && allProducts.length > 0 && /* @__PURE__ */ React5.createElement("div", { className: "w-full mt-1" }, /* @__PURE__ */ React5.createElement(
+  ), !isUser && !hasStructuredProductBlock && allProducts.length > 0 && /* @__PURE__ */ React5.createElement("div", { className: "w-full mt-1" }, /* @__PURE__ */ React5.createElement(
     ProductCarousel,
     {
       products: allProducts,

package/dist/server.js

@@ -3913,17 +3913,68 @@ var Pipeline = class {
   async initialize() {
     var _a, _b;
     if (this.initialised) return;
-    const CHART_MARKER = "<!-- VIZ_PROTOCOL_V2 -->";
+    const CHART_MARKER = "<!-- UI_PROTOCOL_V5 -->";
     const chartInstruction = `
 
 ${CHART_MARKER}
-### MANDATORY VISUALIZATION RULES:
-- If asked for statistics, distribution, or comparisons: You MUST use the \`\`\`chart block.
-- FORMAT: \`\`\`chart { "type": "pie" | "bar" | "line", "data": [ { "label": "X", "value": 10 } ] } \`\`\`
-- RULES:
-  1. ALWAYS include the \`\`\`chart and \`\`\` markers.
-  2. NEVER use text, tables, or ASCII art for charts.
-  3. Output ONLY the chart block. No Python, no text descriptions.`;
+### 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 "chart" for distributions, percentages, counts, comparisons, trends, or "show me in a chart".
+   - Use "carousel" for browseable product recommendations or when the user asks to see products/items/cards.
+   - Use "table" for tabular/raw detail, explicit table/list requests, or when there are many attributes to compare.
+
+2. REQUIRED JSON SHAPE
+   {
+     "view": "chart" | "carousel" | "table",
+     "title": "Short heading",
+     "description": "One sentence describing the view",
+     "chartType": "pie" | "bar" | "line",
+     "xAxisKey": "label",
+     "dataKeys": ["value"],
+     "columns": ["name", "category", "price", "stockStatus"],
+     "insights": ["Short takeaway 1", "Short takeaway 2"],
+     "data": []
+   }
+
+3. DATA RULES
+   - Build the UI payload from retrieved context only. Do not invent products, categories, or stock values.
+   - Aggregate raw product rows when the user asks for a chart.
+   - For charts, every row in "data" must be flat JSON with primitive values only.
+   - Prefer { "label": "...", "value": number } for pie charts.
+   - If stock information exists, preserve it in each row with fields like "inStock", "inStockCount", "outOfStockCount", or "stockStatus".
+   - If the user asks to highlight what is in stock, include those stock-related fields in the chart data and mention the pattern in "insights".
+   - For carousel rows, include product-friendly fields such as id, name, brand, price, image, link, category, stockStatus.
+   - For tables, keep "data" row-oriented and include "columns" when possible.
+
+4. FORMAT RULES
+   - Never use ASCII art, markdown tables as the primary answer, or text-only fake charts when a UI block is requested.
+   - Put any short natural-language explanation before or after the \`\`\`ui\`\`\` block, but keep it concise.
+   - Return valid JSON inside the \`\`\`ui\`\`\` block.
+
+5. EXAMPLE
+User: "provide me the distribution of products across categories and highlight which ones are in stock in a pie chart"
+Assistant:
+\`\`\`ui
+{
+  "view": "chart",
+  "title": "Product Distribution Across Categories",
+  "description": "Pie chart showing product count by category with stock signals preserved per slice.",
+  "chartType": "pie",
+  "xAxisKey": "label",
+  "dataKeys": ["value"],
+  "insights": [
+    "Beauty has the highest product count.",
+    "Electronics has the largest number of in-stock products."
+  ],
+  "data": [
+    { "label": "Beauty", "value": 12, "inStockCount": 9, "outOfStockCount": 3 },
+    { "label": "Electronics", "value": 8, "inStockCount": 7, "outOfStockCount": 1 },
+    { "label": "Home", "value": 5, "inStockCount": 2, "outOfStockCount": 3 }
+  ]
+}
+\`\`\``;
     if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes(CHART_MARKER))) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
     }

package/dist/server.mjs

@@ -39,7 +39,7 @@ import {
   sseFrame,
   sseMetaFrame,
   sseTextFrame
-} from "./chunk-C4UPQX3E.mjs";
+} from "./chunk-Q4MDH6C4.mjs";
 import "./chunk-YLTMFW4M.mjs";
 import {
   PineconeProvider

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "1.5.3",
+  "version": "1.5.5",
   "description": "Retrivora AI is a plug-and-play AI engine for RAG chat experiences — generic vector DB + LLM provider, embeddable or standalone.",
   "author": "Abhinav Alkuchi",
   "license": "MIT",

package/src/app/api/chat/route.ts

@@ -1,4 +1,4 @@
-import { createChatHandler } from '@/handlers';
+import { createStreamHandler } from '@/handlers';
 import { getRagConfig } from '@/config/serverConfig';
 
-export const POST = createChatHandler(getRagConfig());
+export const POST = createStreamHandler(getRagConfig());

package/src/components/DynamicChart.tsx

@@ -96,11 +96,85 @@ export function DynamicChart({ config, primaryColor = '#6366f1', accentColor = '
   // Final sanity check: ensure all keys are strings
   const finalXKey = String(resolvedXKey);
   const finalDataKeys = resolvedDataKeys.map(String).filter(k => k !== 'undefined');
+  const pieDataKey = finalDataKeys[0] || 'value';
+
+  const stockAwareColor = (entry: Record<string, string | number | null>, index: number) => {
+    const stockStatus = String(
+      entry.stockStatus ??
+      entry.status ??
+      entry.availability ??
+      '',
+    ).toLowerCase();
+    const inStockCount = typeof entry.inStockCount === 'number' ? entry.inStockCount : undefined;
+    const outOfStockCount = typeof entry.outOfStockCount === 'number' ? entry.outOfStockCount : undefined;
+    const inStockFlag = entry.inStock;
+
+    if (stockStatus.includes('in stock')) return '#10b981';
+    if (stockStatus.includes('out of stock')) return '#f97316';
+    if (typeof inStockFlag === 'string' && inStockFlag.toLowerCase() === 'true') return '#10b981';
+    if (inStockFlag === 1) return '#10b981';
+    if (inStockFlag === 0) return '#f97316';
+    if (typeof inStockCount === 'number' || typeof outOfStockCount === 'number') {
+      const inCount = inStockCount ?? 0;
+      const outCount = outOfStockCount ?? 0;
+      if (inCount > outCount) return '#10b981';
+      if (outCount > inCount) return '#f97316';
+    }
+
+    return colors[index % colors.length];
+  };
+
+  const tooltipLabelMap: Record<string, string> = {
+    value: 'Value',
+    count: 'Count',
+    inStock: 'In stock',
+    inStockCount: 'In stock',
+    outOfStockCount: 'Out of stock',
+    stockStatus: 'Stock status',
+    category: 'Category',
+    label: 'Label',
+    name: 'Name',
+  };
+
+  const renderTooltip = (
+    row: Record<string, string | number | null> | null,
+    label?: string,
+  ) => {
+    if (!row) return null;
+
+    return (
+      <div className="rounded-xl border border-slate-200 bg-white px-3 py-2 text-xs shadow-xl dark:border-white/10 dark:bg-slate-950">
+        <p className="mb-2 font-semibold text-slate-800 dark:text-white/90">
+          {String(row[finalXKey] ?? label ?? 'Details')}
+        </p>
+        {Object.entries(row).map(([key, value]) => {
+          if (value === null || value === undefined || key === finalXKey) return null;
+          if (typeof value === 'object') return null;
+
+          return (
+            <div key={key} className="flex items-center justify-between gap-3 text-slate-600 dark:text-white/70">
+              <span>{tooltipLabelMap[key] ?? key}</span>
+              <span className="font-medium text-slate-900 dark:text-white/90">{String(value)}</span>
+            </div>
+          );
+        })}
+      </div>
+    );
+  };
+
+  const getTooltipRow = (payload: unknown): Record<string, string | number | null> | null => {
+    if (!Array.isArray(payload) || payload.length === 0) return null;
+    const first = payload[0];
+    if (!first || typeof first !== 'object' || !('payload' in first)) return null;
+
+    const row = (first as { payload?: unknown }).payload;
+    if (!row || typeof row !== 'object' || Array.isArray(row)) return null;
+
+    return row as Record<string, string | number | null>;
+  };
 
   // Handle Pie Chart separately as it has a different structure
   if (type === 'pie') {
-    const pieDataKey = finalDataKeys[0] || 'value';
-    
     return (
       <div className="w-full h-72 min-h-[280px] mt-4 mb-2 select-none overflow-hidden">
         <ResponsiveContainer width="99%" height="100%">
@@ -114,13 +188,11 @@ export function DynamicChart({ config, primaryColor = '#6366f1', accentColor = '
               outerRadius={80}
               label={({ name, percent }) => `${name} ${((percent ?? 0) * 100).toFixed(0)}%`}
             >
-              {data.map((entry, index) => (
-                <Cell key={`cell-${index}`} fill={colors[index % colors.length]} />
+              {sanitizedData.map((entry, index) => (
+                <Cell key={`cell-${index}`} fill={stockAwareColor(entry, index)} />
               ))}
             </Pie>
-            <Tooltip 
-              contentStyle={{ borderRadius: '8px', border: 'none', boxShadow: '0 4px 6px -1px rgb(0 0 0 / 0.1)' }}
-            />
+            <Tooltip content={(props) => renderTooltip(getTooltipRow(props.payload), typeof props.label === 'string' ? props.label : undefined)} />
             <Legend />
           </PieChart>
         </ResponsiveContainer>
@@ -137,10 +209,7 @@ export function DynamicChart({ config, primaryColor = '#6366f1', accentColor = '
             <CartesianGrid strokeDasharray="3 3" vertical={false} stroke="#e2e8f0" />
             <XAxis dataKey={finalXKey} tick={{ fontSize: 12, fill: '#64748b' }} tickLine={false} axisLine={{ stroke: '#cbd5e1' }} />
             <YAxis tick={{ fontSize: 12, fill: '#64748b' }} tickLine={false} axisLine={false} />
-            <Tooltip 
-              contentStyle={{ borderRadius: '8px', border: 'none', boxShadow: '0 4px 6px -1px rgb(0 0 0 / 0.1)' }}
-              cursor={{ fill: '#f1f5f9' }}
-            />
+            <Tooltip content={(props) => renderTooltip(getTooltipRow(props.payload), typeof props.label === 'string' ? props.label : undefined)} cursor={{ fill: '#f1f5f9' }} />
             <Legend wrapperStyle={{ fontSize: 12 }} />
             {finalDataKeys.map((key, index) => (
               <Bar 
@@ -157,9 +226,7 @@ export function DynamicChart({ config, primaryColor = '#6366f1', accentColor = '
             <CartesianGrid strokeDasharray="3 3" vertical={false} stroke="#e2e8f0" />
             <XAxis dataKey={finalXKey} tick={{ fontSize: 12, fill: '#64748b' }} tickLine={false} axisLine={{ stroke: '#cbd5e1' }} />
             <YAxis tick={{ fontSize: 12, fill: '#64748b' }} tickLine={false} axisLine={false} />
-            <Tooltip 
-              contentStyle={{ borderRadius: '8px', border: 'none', boxShadow: '0 4px 6px -1px rgb(0 0 0 / 0.1)' }}
-            />
+            <Tooltip content={(props) => renderTooltip(getTooltipRow(props.payload), typeof props.label === 'string' ? props.label : undefined)} />
             <Legend wrapperStyle={{ fontSize: 12 }} />
             {finalDataKeys.map((key, index) => (
               <Line