@retrivora-ai/rag-engine 1.4.9 → 1.5.0

package/dist/{chunk-FHNGNXXS.mjs → chunk-YTL3SLQU.mjs}

@@ -2263,18 +2263,18 @@ var Pipeline = class {
     const chartInstruction = `
 
 ${CHART_MARKER}
-### MASTER DATA VISUALIZATION PROTOCOL:
-You are an expert data analyst. When asked to visualize or compare data retrieved from the context:
-1. DECIDE the best chart type: 'pie' (for parts-of-a-whole), 'bar' (for category comparisons), or 'line' (for trends).
-2. TRANSFORM arbitrary metadata from the context into a FLAT list of objects. Extract relevant keys dynamically.
-3. OUTPUT a \`\`\`chart\`\`\` code block using this JSON format:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "key_for_labels",
-     "dataKeys": ["key_for_values"],
-     "data": [ { "key_for_labels": "Category A", "key_for_values": 100 }, ... ]
-   }
-4. RULES: Valid JSON only. No math. No text-based charts. No markdown tables for graphs. If showing a chart, DO NOT output a "products" JSON carousel.`;
+### MANDATORY CHART PROTOCOL:
+If asked for a chart (pie, bar, line), you MUST output exactly ONE block using this structure:
+\`\`\`chart
+{
+  "type": "pie",
+  "data": [ { "label": "Item A", "value": 10 }, { "label": "Item B", "value": 20 } ]
+}
+\`\`\`
+- RULES: 
+1. You MUST include the \`\`\`chart and \`\`\` markers. Naked JSON will NOT render.
+2. Output ONLY the chart block. No Python snippets. No "here is the data" text.
+3. Data objects MUST be FLAT. No math (e.g. use 10, NOT 5+5).`;
     if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes(CHART_MARKER))) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
     }

package/dist/handlers/index.js

@@ -3826,18 +3826,18 @@ var Pipeline = class {
     const chartInstruction = `
 
 ${CHART_MARKER}
-### MASTER DATA VISUALIZATION PROTOCOL:
-You are an expert data analyst. When asked to visualize or compare data retrieved from the context:
-1. DECIDE the best chart type: 'pie' (for parts-of-a-whole), 'bar' (for category comparisons), or 'line' (for trends).
-2. TRANSFORM arbitrary metadata from the context into a FLAT list of objects. Extract relevant keys dynamically.
-3. OUTPUT a \`\`\`chart\`\`\` code block using this JSON format:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "key_for_labels",
-     "dataKeys": ["key_for_values"],
-     "data": [ { "key_for_labels": "Category A", "key_for_values": 100 }, ... ]
-   }
-4. RULES: Valid JSON only. No math. No text-based charts. No markdown tables for graphs. If showing a chart, DO NOT output a "products" JSON carousel.`;
+### MANDATORY CHART PROTOCOL:
+If asked for a chart (pie, bar, line), you MUST output exactly ONE block using this structure:
+\`\`\`chart
+{
+  "type": "pie",
+  "data": [ { "label": "Item A", "value": 10 }, { "label": "Item B", "value": 20 } ]
+}
+\`\`\`
+- RULES: 
+1. You MUST include the \`\`\`chart and \`\`\` markers. Naked JSON will NOT render.
+2. Output ONLY the chart block. No Python snippets. No "here is the data" text.
+3. Data objects MUST be FLAT. No math (e.g. use 10, NOT 5+5).`;
     if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes(CHART_MARKER))) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
     }

package/dist/handlers/index.mjs

@@ -9,7 +9,7 @@ import {
   sseFrame,
   sseMetaFrame,
   sseTextFrame
-} from "../chunk-FHNGNXXS.mjs";
+} from "../chunk-YTL3SLQU.mjs";
 import "../chunk-YLTMFW4M.mjs";
 import "../chunk-X4TOT24V.mjs";
 export {

package/dist/index.js

@@ -503,10 +503,13 @@ function MessageBubble({
       };
     });
   }, [sources, isUser]);
+  const isVisualizationRequested = import_react5.default.useMemo(() => {
+    if (isUser) return false;
+    const text = message.content.toLowerCase();
+    return message.content.includes("```chart") || text.includes("pie chart") || text.includes("bar chart") || text.includes("line chart") || text.includes("distribution across");
+  }, [message.content, isUser]);
   const { productsFromContent, cleanContent } = import_react5.default.useMemo(() => {
-    if (isUser) return { productsFromContent: [], cleanContent: message.content };
-    const isVisualization = message.content.includes("```chart") || message.content.toLowerCase().includes("pie chart") || message.content.toLowerCase().includes("bar chart") || message.content.toLowerCase().includes("distribution");
-    if (isVisualization) {
+    if (isUser || isVisualizationRequested) {
       return { productsFromContent: [], cleanContent: message.content };
     }
     const jsonRegex = /```json\s*([\s\S]*?)\s*```/g;
@@ -532,7 +535,7 @@ function MessageBubble({
       }
     }
     return { productsFromContent: products, cleanContent: content.trim() };
-  }, [message.content, isUser]);
+  }, [message.content, isUser, isVisualizationRequested]);
   const allProducts = import_react5.default.useMemo(() => {
     if (productsFromContent.length > 0) return productsFromContent;
     const seen = /* @__PURE__ */ new Set();

package/dist/index.mjs

@@ -466,10 +466,13 @@ function MessageBubble({
       };
     });
   }, [sources, isUser]);
+  const isVisualizationRequested = React5.useMemo(() => {
+    if (isUser) return false;
+    const text = message.content.toLowerCase();
+    return message.content.includes("```chart") || text.includes("pie chart") || text.includes("bar chart") || text.includes("line chart") || text.includes("distribution across");
+  }, [message.content, isUser]);
   const { productsFromContent, cleanContent } = React5.useMemo(() => {
-    if (isUser) return { productsFromContent: [], cleanContent: message.content };
-    const isVisualization = message.content.includes("```chart") || message.content.toLowerCase().includes("pie chart") || message.content.toLowerCase().includes("bar chart") || message.content.toLowerCase().includes("distribution");
-    if (isVisualization) {
+    if (isUser || isVisualizationRequested) {
       return { productsFromContent: [], cleanContent: message.content };
     }
     const jsonRegex = /```json\s*([\s\S]*?)\s*```/g;
@@ -495,7 +498,7 @@ function MessageBubble({
       }
     }
     return { productsFromContent: products, cleanContent: content.trim() };
-  }, [message.content, isUser]);
+  }, [message.content, isUser, isVisualizationRequested]);
   const allProducts = React5.useMemo(() => {
     if (productsFromContent.length > 0) return productsFromContent;
     const seen = /* @__PURE__ */ new Set();

package/dist/server.js

@@ -3917,18 +3917,18 @@ var Pipeline = class {
     const chartInstruction = `
 
 ${CHART_MARKER}
-### MASTER DATA VISUALIZATION PROTOCOL:
-You are an expert data analyst. When asked to visualize or compare data retrieved from the context:
-1. DECIDE the best chart type: 'pie' (for parts-of-a-whole), 'bar' (for category comparisons), or 'line' (for trends).
-2. TRANSFORM arbitrary metadata from the context into a FLAT list of objects. Extract relevant keys dynamically.
-3. OUTPUT a \`\`\`chart\`\`\` code block using this JSON format:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "key_for_labels",
-     "dataKeys": ["key_for_values"],
-     "data": [ { "key_for_labels": "Category A", "key_for_values": 100 }, ... ]
-   }
-4. RULES: Valid JSON only. No math. No text-based charts. No markdown tables for graphs. If showing a chart, DO NOT output a "products" JSON carousel.`;
+### MANDATORY CHART PROTOCOL:
+If asked for a chart (pie, bar, line), you MUST output exactly ONE block using this structure:
+\`\`\`chart
+{
+  "type": "pie",
+  "data": [ { "label": "Item A", "value": 10 }, { "label": "Item B", "value": 20 } ]
+}
+\`\`\`
+- RULES: 
+1. You MUST include the \`\`\`chart and \`\`\` markers. Naked JSON will NOT render.
+2. Output ONLY the chart block. No Python snippets. No "here is the data" text.
+3. Data objects MUST be FLAT. No math (e.g. use 10, NOT 5+5).`;
     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-FHNGNXXS.mjs";
+} from "./chunk-YTL3SLQU.mjs";
 import "./chunk-YLTMFW4M.mjs";
 import {
   PineconeProvider

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "1.4.9",
+  "version": "1.5.0",
   "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

@@ -309,18 +309,23 @@ export function MessageBubble({
       });
   }, [sources, isUser]);
 
+  // ── Visualization Intent Detection ────────────────────────────────────────
+  // Robust detection to prevent UI clutter and ensure atomic rendering.
+  const isVisualizationRequested = React.useMemo(() => {
+    if (isUser) return false;
+    const text = message.content.toLowerCase();
+    return (
+      message.content.includes('```chart') ||
+      text.includes('pie chart') ||
+      text.includes('bar chart') ||
+      text.includes('line chart') ||
+      text.includes('distribution across')
+    );
+  }, [message.content, isUser]);
+
   // ── Products + cleaned content from structured JSON blocks in the message ──
   const { productsFromContent, cleanContent } = React.useMemo(() => {
-    if (isUser) return { productsFromContent: [] as Product[], cleanContent: message.content };
-
-    // If the message contains a chart or refers to visualization, suppress the product carousel
-    const isVisualization = 
-      message.content.includes('```chart') || 
-      message.content.toLowerCase().includes('pie chart') ||
-      message.content.toLowerCase().includes('bar chart') ||
-      message.content.toLowerCase().includes('distribution');
-
-    if (isVisualization) {
+    if (isUser || isVisualizationRequested) {
       return { productsFromContent: [] as Product[], cleanContent: message.content };
     }
 
@@ -348,7 +353,7 @@ export function MessageBubble({
     }
 
     return { productsFromContent: products, cleanContent: content.trim() };
-  }, [message.content, isUser]);
+  }, [message.content, isUser, isVisualizationRequested]);
 
   // ── Merge & deduplicate products ───────────────────────────────────────────
   const allProducts = React.useMemo<Product[]>(() => {

package/src/core/Pipeline.ts

@@ -103,18 +103,18 @@ export class Pipeline {
     // We use a unique marker to ensure this is only injected once but is ALWAYS present.
     const CHART_MARKER = '<!-- VIZ_PROTOCOL_V1 -->';
     const chartInstruction = `\n\n${CHART_MARKER}
-### MASTER DATA VISUALIZATION PROTOCOL:
-You are an expert data analyst. When asked to visualize or compare data retrieved from the context:
-1. DECIDE the best chart type: 'pie' (for parts-of-a-whole), 'bar' (for category comparisons), or 'line' (for trends).
-2. TRANSFORM arbitrary metadata from the context into a FLAT list of objects. Extract relevant keys dynamically.
-3. OUTPUT a \`\`\`chart\`\`\` code block using this JSON format:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "key_for_labels",
-     "dataKeys": ["key_for_values"],
-     "data": [ { "key_for_labels": "Category A", "key_for_values": 100 }, ... ]
-   }
-4. RULES: Valid JSON only. No math. No text-based charts. No markdown tables for graphs. If showing a chart, DO NOT output a "products" JSON carousel.`;
+### MANDATORY CHART PROTOCOL:
+If asked for a chart (pie, bar, line), you MUST output exactly ONE block using this structure:
+\`\`\`chart
+{
+  "type": "pie",
+  "data": [ { "label": "Item A", "value": 10 }, { "label": "Item B", "value": 20 } ]
+}
+\`\`\`
+- RULES: 
+1. You MUST include the \`\`\`chart and \`\`\` markers. Naked JSON will NOT render.
+2. Output ONLY the chart block. No Python snippets. No "here is the data" text.
+3. Data objects MUST be FLAT. No math (e.g. use 10, NOT 5+5).`;
 
     if (!this.config.llm.systemPrompt?.includes(CHART_MARKER)) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || '') + chartInstruction;