@retrivora-ai/rag-engine 1.4.5 → 1.4.7

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

@@ -153,7 +153,7 @@ function getEnvConfig(env = process.env, base) {
       apiKey: (_$ = llmApiKeyByProvider[llmProvider]) != null ? _$ : "",
       baseUrl: readString(env, "LLM_BASE_URL"),
       systemPrompt: readString(env, "LLM_SYSTEM_PROMPT"),
-      maxTokens: readNumber(env, "LLM_MAX_TOKENS", 1024),
+      maxTokens: readNumber(env, "LLM_MAX_TOKENS", 4096),
       temperature: readNumber(env, "LLM_TEMPERATURE", 0.7),
       options: {
         profile: readString(env, "LLM_UNIVERSAL_PROFILE")
@@ -2257,29 +2257,22 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a, _b;
+    var _a;
     if (this.initialised) return;
     const chartInstruction = `
 
-When presenting structured data, statistics, or comparisons, decide if it is best presented as a table or a chart.
-- For tables: use standard Markdown tables.
-- For charts (bar, line, pie): You MUST output a fenced code block with the language 'chart'.
-- CRITICAL RULES:
-  1. Use ONLY valid JSON (double quotes, no trailing commas).
-  2. NO math expressions (e.g., use 100, NOT 50+50).
-  3. dataKeys MUST exactly match the property names in the data objects.
-  4. data objects MUST be FLAT (no nested objects or arrays).
-  5. DO NOT output naked JSON or markdown tables for visualizations.
-  6. ALWAYS wrap your visualization data in a \`\`\`chart ... \`\`\` code block.
-\`\`\`chart
-{
-  "type": "bar" | "line" | "pie",
-  "xAxisKey": "name",
-  "dataKeys": ["value"],
-  "data": [{"name": "A", "value": 10}]
-}
-\`\`\``;
-    if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes("chart"))) {
+### 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")) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
@@ -2296,7 +2289,7 @@ When presenting structured data, statistics, or comparisons, decide if it is bes
       this.entityExtractor = new EntityExtractor(this.llmProvider);
     }
     await this.vectorDB.initialize();
-    if (((_b = this.config.rag) == null ? void 0 : _b.architecture) === "agentic") {
+    if (((_a = this.config.rag) == null ? void 0 : _a.architecture) === "agentic") {
       this.agent = new LangChainAgent(this, this.config);
       await this.agent.initialize(this.llmProvider);
     }

package/dist/handlers/index.js

@@ -1760,7 +1760,7 @@ function getEnvConfig(env = process.env, base) {
       apiKey: (_$ = llmApiKeyByProvider[llmProvider]) != null ? _$ : "",
       baseUrl: readString(env, "LLM_BASE_URL"),
       systemPrompt: readString(env, "LLM_SYSTEM_PROMPT"),
-      maxTokens: readNumber(env, "LLM_MAX_TOKENS", 1024),
+      maxTokens: readNumber(env, "LLM_MAX_TOKENS", 4096),
       temperature: readNumber(env, "LLM_TEMPERATURE", 0.7),
       options: {
         profile: readString(env, "LLM_UNIVERSAL_PROFILE")
@@ -3820,29 +3820,22 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a, _b;
+    var _a;
     if (this.initialised) return;
     const chartInstruction = `
 
-When presenting structured data, statistics, or comparisons, decide if it is best presented as a table or a chart.
-- For tables: use standard Markdown tables.
-- For charts (bar, line, pie): You MUST output a fenced code block with the language 'chart'.
-- CRITICAL RULES:
-  1. Use ONLY valid JSON (double quotes, no trailing commas).
-  2. NO math expressions (e.g., use 100, NOT 50+50).
-  3. dataKeys MUST exactly match the property names in the data objects.
-  4. data objects MUST be FLAT (no nested objects or arrays).
-  5. DO NOT output naked JSON or markdown tables for visualizations.
-  6. ALWAYS wrap your visualization data in a \`\`\`chart ... \`\`\` code block.
-\`\`\`chart
-{
-  "type": "bar" | "line" | "pie",
-  "xAxisKey": "name",
-  "dataKeys": ["value"],
-  "data": [{"name": "A", "value": 10}]
-}
-\`\`\``;
-    if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes("chart"))) {
+### 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")) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
@@ -3859,7 +3852,7 @@ When presenting structured data, statistics, or comparisons, decide if it is bes
       this.entityExtractor = new EntityExtractor(this.llmProvider);
     }
     await this.vectorDB.initialize();
-    if (((_b = this.config.rag) == null ? void 0 : _b.architecture) === "agentic") {
+    if (((_a = this.config.rag) == null ? void 0 : _a.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-GYT3LWD6.mjs";
+} from "../chunk-YCL23GST.mjs";
 import "../chunk-YLTMFW4M.mjs";
 import "../chunk-X4TOT24V.mjs";
 export {

package/dist/server.js

@@ -1806,7 +1806,7 @@ function getEnvConfig(env = process.env, base) {
       apiKey: (_$ = llmApiKeyByProvider[llmProvider]) != null ? _$ : "",
       baseUrl: readString(env, "LLM_BASE_URL"),
       systemPrompt: readString(env, "LLM_SYSTEM_PROMPT"),
-      maxTokens: readNumber(env, "LLM_MAX_TOKENS", 1024),
+      maxTokens: readNumber(env, "LLM_MAX_TOKENS", 4096),
       temperature: readNumber(env, "LLM_TEMPERATURE", 0.7),
       options: {
         profile: readString(env, "LLM_UNIVERSAL_PROFILE")
@@ -3911,29 +3911,22 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a, _b;
+    var _a;
     if (this.initialised) return;
     const chartInstruction = `
 
-When presenting structured data, statistics, or comparisons, decide if it is best presented as a table or a chart.
-- For tables: use standard Markdown tables.
-- For charts (bar, line, pie): You MUST output a fenced code block with the language 'chart'.
-- CRITICAL RULES:
-  1. Use ONLY valid JSON (double quotes, no trailing commas).
-  2. NO math expressions (e.g., use 100, NOT 50+50).
-  3. dataKeys MUST exactly match the property names in the data objects.
-  4. data objects MUST be FLAT (no nested objects or arrays).
-  5. DO NOT output naked JSON or markdown tables for visualizations.
-  6. ALWAYS wrap your visualization data in a \`\`\`chart ... \`\`\` code block.
-\`\`\`chart
-{
-  "type": "bar" | "line" | "pie",
-  "xAxisKey": "name",
-  "dataKeys": ["value"],
-  "data": [{"name": "A", "value": 10}]
-}
-\`\`\``;
-    if (!((_a = this.config.llm.systemPrompt) == null ? void 0 : _a.includes("chart"))) {
+### 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")) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
     }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
@@ -3950,7 +3943,7 @@ When presenting structured data, statistics, or comparisons, decide if it is bes
       this.entityExtractor = new EntityExtractor(this.llmProvider);
     }
     await this.vectorDB.initialize();
-    if (((_b = this.config.rag) == null ? void 0 : _b.architecture) === "agentic") {
+    if (((_a = this.config.rag) == null ? void 0 : _a.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-GYT3LWD6.mjs";
+} from "./chunk-YCL23GST.mjs";
 import "./chunk-YLTMFW4M.mjs";
 import {
   PineconeProvider

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "1.4.5",
+  "version": "1.4.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/config/serverConfig.ts

@@ -146,7 +146,7 @@ export function getEnvConfig(env: Record<string, string | undefined> = process.e
       apiKey: llmApiKeyByProvider[llmProvider] ?? '',
       baseUrl: readString(env, 'LLM_BASE_URL'),
       systemPrompt: readString(env, 'LLM_SYSTEM_PROMPT'),
-      maxTokens: readNumber(env, 'LLM_MAX_TOKENS', 1024),
+      maxTokens: readNumber(env, 'LLM_MAX_TOKENS', 4096),
       temperature: readNumber(env, 'LLM_TEMPERATURE', 0.7),
       options: {
         profile: readString(env, 'LLM_UNIVERSAL_PROFILE'),

package/src/core/Pipeline.ts

@@ -99,27 +99,21 @@ export class Pipeline {
   async initialize(): Promise<void> {
     if (this.initialised) return;
 
-    // Augment system prompt with Chart/Table instructions if not already present
-    const chartInstruction = `\n\nWhen presenting structured data, statistics, or comparisons, decide if it is best presented as a table or a chart.
-- For tables: use standard Markdown tables.
-- For charts (bar, line, pie): You MUST output a fenced code block with the language 'chart'.
-- CRITICAL RULES:
-  1. Use ONLY valid JSON (double quotes, no trailing commas).
-  2. NO math expressions (e.g., use 100, NOT 50+50).
-  3. dataKeys MUST exactly match the property names in the data objects.
-  4. data objects MUST be FLAT (no nested objects or arrays).
-  5. DO NOT output naked JSON or markdown tables for visualizations.
-  6. ALWAYS wrap your visualization data in a \`\`\`chart ... \`\`\` code block.
-\`\`\`chart
-{
-  "type": "bar" | "line" | "pie",
-  "xAxisKey": "name",
-  "dataKeys": ["value"],
-  "data": [{"name": "A", "value": 10}]
-}
-\`\`\``;
-
-    if (!this.config.llm.systemPrompt?.includes('chart')) {
+    // 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')) {
       this.config.llm.systemPrompt = (this.config.llm.systemPrompt || '') + chartInstruction;
     }