@retrivora-ai/rag-engine 1.1.6 → 1.1.7

package/dist/index.js

@@ -250,7 +250,19 @@ function MessageBubble({
     }
     return { productsFromContent: products, cleanContent: content.trim() };
   }, [message.content, isUser]);
-  const allProducts = [...productsFromSources, ...productsFromContent];
+  const allProducts = import_react4.default.useMemo(() => {
+    if (productsFromContent.length > 0) return productsFromContent;
+    const uniqueProducts = [];
+    const seenIds = /* @__PURE__ */ new Set();
+    for (const p of productsFromSources) {
+      const identifier = p.id || p.name;
+      if (!seenIds.has(identifier)) {
+        seenIds.add(identifier);
+        uniqueProducts.push(p);
+      }
+    }
+    return uniqueProducts;
+  }, [productsFromSources, productsFromContent]);
   const hasProducts = allProducts.length > 0;
   return /* @__PURE__ */ import_react4.default.createElement("div", { className: `flex gap-3 ${isUser ? "flex-row-reverse" : "flex-row"} items-start` }, /* @__PURE__ */ import_react4.default.createElement(
     "div",

package/dist/index.mjs

@@ -208,7 +208,19 @@ function MessageBubble({
     }
     return { productsFromContent: products, cleanContent: content.trim() };
   }, [message.content, isUser]);
-  const allProducts = [...productsFromSources, ...productsFromContent];
+  const allProducts = React4.useMemo(() => {
+    if (productsFromContent.length > 0) return productsFromContent;
+    const uniqueProducts = [];
+    const seenIds = /* @__PURE__ */ new Set();
+    for (const p of productsFromSources) {
+      const identifier = p.id || p.name;
+      if (!seenIds.has(identifier)) {
+        seenIds.add(identifier);
+        uniqueProducts.push(p);
+      }
+    }
+    return uniqueProducts;
+  }, [productsFromSources, productsFromContent]);
   const hasProducts = allProducts.length > 0;
   return /* @__PURE__ */ React4.createElement("div", { className: `flex gap-3 ${isUser ? "flex-row-reverse" : "flex-row"} items-start` }, /* @__PURE__ */ React4.createElement(
     "div",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "1.1.6",
+  "version": "1.1.7",
   "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/components/MessageBubble.tsx

@@ -100,7 +100,26 @@ export function MessageBubble({
     return { productsFromContent: products, cleanContent: content.trim() };
   }, [message.content, isUser]);
 
-  const allProducts = [...productsFromSources, ...productsFromContent];
+  const allProducts = React.useMemo(() => {
+    // If the LLM explicitly provided products in a structured format, only show those.
+    // This ensures we only show what the assistant actually "displayed" as a result.
+    if (productsFromContent.length > 0) return productsFromContent;
+
+    // Fallback: If no structured products, show unique products from sources.
+    const uniqueProducts: Product[] = [];
+    const seenIds = new Set();
+
+    for (const p of productsFromSources) {
+      const identifier = p.id || p.name;
+      if (!seenIds.has(identifier)) {
+        seenIds.add(identifier);
+        uniqueProducts.push(p);
+      }
+    }
+
+    return uniqueProducts;
+  }, [productsFromSources, productsFromContent]);
+
   const hasProducts = allProducts.length > 0;
 
   return (