@retrivora-ai/rag-engine 1.2.2 → 1.2.4

package/dist/{chunk-GCPPRD2G.mjs → chunk-X4TMJV23.mjs}

@@ -1692,11 +1692,30 @@ Sources Used: ${JSON.stringify(
           )}`;
         }
       });
+      const defaultAgentPrompt = "You are a helpful AI assistant with access to a document search tool.";
+      let finalSystemPrompt = this.config.llm.systemPrompt || defaultAgentPrompt;
+      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): Output a JSON code block with language 'chart'.
+- CRITICAL: Use ONLY valid JSON (double quotes, no trailing commas).
+\`\`\`chart
+{
+  "type": "bar" | "line" | "pie",
+  "xAxisKey": "name",
+  "dataKeys": ["value"],
+  "data": [{"name": "A", "value": 10}]
+}
+\`\`\``;
+      if (!finalSystemPrompt.includes("chart")) {
+        finalSystemPrompt += chartInstruction;
+      }
       this.agent = createAgent({
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         model: chatModel,
         tools: [searchTool],
-        systemPrompt: this.config.llm.systemPrompt || "You are a helpful AI assistant with access to a document search tool."
+        systemPrompt: finalSystemPrompt
       });
       void HumanMessage;
     } catch (error) {
@@ -2231,8 +2250,25 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a;
+    var _a, _b;
     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): Output a JSON code block with language 'chart'.
+- CRITICAL: Use ONLY valid JSON (double quotes, no trailing commas).
+\`\`\`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"))) {
+      this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
+    }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
     this.vectorDB = await ProviderRegistry.createVectorProvider(this.config.vectorDb);
     const { llmProvider, embeddingProvider } = await EmbeddingStrategyResolver.resolve(
@@ -2247,7 +2283,7 @@ var Pipeline = class {
       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

@@ -3261,11 +3261,30 @@ Sources Used: ${JSON.stringify(
           )}`;
         }
       });
+      const defaultAgentPrompt = "You are a helpful AI assistant with access to a document search tool.";
+      let finalSystemPrompt = this.config.llm.systemPrompt || defaultAgentPrompt;
+      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): Output a JSON code block with language 'chart'.
+- CRITICAL: Use ONLY valid JSON (double quotes, no trailing commas).
+\`\`\`chart
+{
+  "type": "bar" | "line" | "pie",
+  "xAxisKey": "name",
+  "dataKeys": ["value"],
+  "data": [{"name": "A", "value": 10}]
+}
+\`\`\``;
+      if (!finalSystemPrompt.includes("chart")) {
+        finalSystemPrompt += chartInstruction;
+      }
       this.agent = createAgent({
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         model: chatModel,
         tools: [searchTool],
-        systemPrompt: this.config.llm.systemPrompt || "You are a helpful AI assistant with access to a document search tool."
+        systemPrompt: finalSystemPrompt
       });
       void HumanMessage;
     } catch (error) {
@@ -3794,8 +3813,25 @@ var Pipeline = class {
     this.reranker = new Reranker();
   }
   async initialize() {
-    var _a;
+    var _a, _b;
     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): Output a JSON code block with language 'chart'.
+- CRITICAL: Use ONLY valid JSON (double quotes, no trailing commas).
+\`\`\`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"))) {
+      this.config.llm.systemPrompt = (this.config.llm.systemPrompt || "") + chartInstruction;
+    }
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
     this.vectorDB = await ProviderRegistry.createVectorProvider(this.config.vectorDb);
     const { llmProvider, embeddingProvider } = await EmbeddingStrategyResolver.resolve(
@@ -3810,7 +3846,7 @@ var Pipeline = class {
       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-GCPPRD2G.mjs";
+} from "../chunk-X4TMJV23.mjs";
 import "../chunk-YLTMFW4M.mjs";
 import "../chunk-X4TOT24V.mjs";
 export {