@retrivora-ai/rag-engine 1.4.8 → 1.5.0

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

@@ -2257,22 +2257,25 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a;
+    var _a, _b;
     if (this.initialised) return;
+    const CHART_MARKER = "<!-- VIZ_PROTOCOL_V1 -->";
     const chartInstruction = `
 
-### VISUALIZATION PROTOCOL:
-When requested to show data visually (charts, graphs, or comparisons), you MUST use this format:
-1. Wrap the data in a fenced code block with the language 'chart'.
-2. Use ONLY valid JSON matching this schema:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "property_for_label",
-     "dataKeys": ["property_for_value"],
-     "data": [{ "property_for_label": "A", "property_for_value": 10 }]
-   }
-3. RULES: Data objects MUST be FLAT (no nesting). No math expressions in JSON. NEVER draw charts with text or markdown tables.`;
-    if (!this.config.llm.systemPrompt || !this.config.llm.systemPrompt.includes("chart")) {
+${CHART_MARKER}
+### 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;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
@@ -2289,7 +2292,7 @@ When requested to show data visually (charts, graphs, or comparisons), you MUST
       this.entityExtractor = new EntityExtractor(this.llmProvider);
     }
     await this.vectorDB.initialize();
-    if (((_a = this.config.rag) == null ? void 0 : _a.architecture) === "agentic") {
+    if (((_b = this.config.rag) == null ? void 0 : _b.architecture) === "agentic") {
       this.agent = new LangChainAgent(this, this.config);
       await this.agent.initialize(this.llmProvider);
     }

package/dist/handlers/index.js

@@ -3820,22 +3820,25 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a;
+    var _a, _b;
     if (this.initialised) return;
+    const CHART_MARKER = "<!-- VIZ_PROTOCOL_V1 -->";
     const chartInstruction = `
 
-### VISUALIZATION PROTOCOL:
-When requested to show data visually (charts, graphs, or comparisons), you MUST use this format:
-1. Wrap the data in a fenced code block with the language 'chart'.
-2. Use ONLY valid JSON matching this schema:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "property_for_label",
-     "dataKeys": ["property_for_value"],
-     "data": [{ "property_for_label": "A", "property_for_value": 10 }]
-   }
-3. RULES: Data objects MUST be FLAT (no nesting). No math expressions in JSON. NEVER draw charts with text or markdown tables.`;
-    if (!this.config.llm.systemPrompt || !this.config.llm.systemPrompt.includes("chart")) {
+${CHART_MARKER}
+### 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;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
@@ -3852,7 +3855,7 @@ When requested to show data visually (charts, graphs, or comparisons), you MUST
       this.entityExtractor = new EntityExtractor(this.llmProvider);
     }
     await this.vectorDB.initialize();
-    if (((_a = this.config.rag) == null ? void 0 : _a.architecture) === "agentic") {
+    if (((_b = this.config.rag) == null ? void 0 : _b.architecture) === "agentic") {
       this.agent = new LangChainAgent(this, this.config);
       await this.agent.initialize(this.llmProvider);
     }

package/dist/handlers/index.mjs

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

package/dist/index.js

@@ -503,9 +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 };
-    if (message.content.includes("```chart")) {
+    if (isUser || isVisualizationRequested) {
       return { productsFromContent: [], cleanContent: message.content };
     }
     const jsonRegex = /```json\s*([\s\S]*?)\s*```/g;
@@ -531,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,9 +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 };
-    if (message.content.includes("```chart")) {
+    if (isUser || isVisualizationRequested) {
       return { productsFromContent: [], cleanContent: message.content };
     }
     const jsonRegex = /```json\s*([\s\S]*?)\s*```/g;
@@ -494,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

@@ -3911,22 +3911,25 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a;
+    var _a, _b;
     if (this.initialised) return;
+    const CHART_MARKER = "<!-- VIZ_PROTOCOL_V1 -->";
     const chartInstruction = `
 
-### VISUALIZATION PROTOCOL:
-When requested to show data visually (charts, graphs, or comparisons), you MUST use this format:
-1. Wrap the data in a fenced code block with the language 'chart'.
-2. Use ONLY valid JSON matching this schema:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "property_for_label",
-     "dataKeys": ["property_for_value"],
-     "data": [{ "property_for_label": "A", "property_for_value": 10 }]
-   }
-3. RULES: Data objects MUST be FLAT (no nesting). No math expressions in JSON. NEVER draw charts with text or markdown tables.`;
-    if (!this.config.llm.systemPrompt || !this.config.llm.systemPrompt.includes("chart")) {
+${CHART_MARKER}
+### 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;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
@@ -3943,7 +3946,7 @@ When requested to show data visually (charts, graphs, or comparisons), you MUST
       this.entityExtractor = new EntityExtractor(this.llmProvider);
     }
     await this.vectorDB.initialize();
-    if (((_a = this.config.rag) == null ? void 0 : _a.architecture) === "agentic") {
+    if (((_b = this.config.rag) == null ? void 0 : _b.architecture) === "agentic") {
       this.agent = new LangChainAgent(this, this.config);
       await this.agent.initialize(this.llmProvider);
     }

package/dist/server.mjs

@@ -39,7 +39,7 @@ import {
   sseFrame,
   sseMetaFrame,
   sseTextFrame
-} from "./chunk-YCL23GST.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.8",
+  "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,12 +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, suppress the product carousel to keep focus on visualization
-    if (message.content.includes('```chart')) {
+    if (isUser || isVisualizationRequested) {
       return { productsFromContent: [] as Product[], cleanContent: message.content };
     }
 
@@ -342,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

@@ -100,20 +100,23 @@ export class Pipeline {
     if (this.initialised) return;
 
     // Augment system prompt with a Universal Visualization Protocol
-    const chartInstruction = `\n\n### VISUALIZATION PROTOCOL:
-When requested to show data visually (charts, graphs, or comparisons), you MUST use this format:
-1. Wrap the data in a fenced code block with the language 'chart'.
-2. Use ONLY valid JSON matching this schema:
-   {
-     "type": "bar" | "line" | "pie",
-     "xAxisKey": "property_for_label",
-     "dataKeys": ["property_for_value"],
-     "data": [{ "property_for_label": "A", "property_for_value": 10 }]
-   }
-3. RULES: Data objects MUST be FLAT (no nesting). No math expressions in JSON. NEVER draw charts with text or markdown tables.`;
-
-    // Inject chart instructions if not already present (including when systemPrompt is undefined)
-    if (!this.config.llm.systemPrompt || !this.config.llm.systemPrompt.includes('chart')) {
+    // 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}
+### 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;
     }