@retrivora-ai/rag-engine 1.2.2 → 1.2.3

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

@@ -1692,11 +1692,29 @@ 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' matching this schema:
+\`\`\`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 +2249,24 @@ 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' matching this schema:
+\`\`\`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 +2281,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,29 @@ 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' matching this schema:
+\`\`\`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 +3812,24 @@ 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' matching this schema:
+\`\`\`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 +3844,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-WGPNEAK3.mjs";
 import "../chunk-YLTMFW4M.mjs";
 import "../chunk-X4TOT24V.mjs";
 export {