@burtson-labs/bandit-engine 2.0.70 → 2.0.72

package/dist/{chunk-HHMGNCBS.mjs → chunk-SXLI47FV.mjs}

@@ -6,7 +6,7 @@ import {
 } from "./chunk-6ITUH375.mjs";
 import {
   useMCPToolsStore
-} from "./chunk-IDZEEONG.mjs";
+} from "./chunk-DHYP4K5O.mjs";
 import {
   AUTH_TOKEN_CHANGED_EVENT,
   FeatureFlagProvider,
@@ -188,4 +188,4 @@ export {
   ChatProvider,
   chat_provider_default
 };
-//# sourceMappingURL=chunk-HHMGNCBS.mjs.map
+//# sourceMappingURL=chunk-SXLI47FV.mjs.map

package/dist/index.js

@@ -3264,7 +3264,7 @@ var init_knowledgeStore = __esm({
 });
 
 // src/store/mcpToolsStore.ts
-var import_zustand11, healthCheckTool, webSearchTool, webFetchTool, imageGenerationTool, defaultTools, useMCPToolsStore;
+var import_zustand11, healthCheckTool, webSearchTool, webFetchTool, imageGenerationTool, createFileTool, defaultTools, useMCPToolsStore;
 var init_mcpToolsStore = __esm({
   "src/store/mcpToolsStore.ts"() {
     "use strict";
@@ -3358,7 +3358,34 @@ var init_mcpToolsStore = __esm({
       method: "POST",
       isBuiltIn: true
     };
-    defaultTools = [healthCheckTool, webSearchTool, webFetchTool, imageGenerationTool];
+    createFileTool = {
+      id: "create-file",
+      name: "create_file",
+      description: "Create a downloadable file for the user (md, txt, csv, json, html, docx, pptx). Returns a temporary (~1 hour) download link.",
+      enabled: true,
+      type: "function",
+      function: {
+        name: "create_file",
+        description: "Generate a file the user can download. For docx/pptx write well-structured Markdown (headings, lists, tables; use '## ' headings to start each slide). Returns a short-lived (~1 hour) download URL \u2014 tell the user it expires.",
+        parameters: {
+          type: "object",
+          properties: {
+            content: { type: "string", description: "The file content (Markdown for docx/pptx; raw text for others)." },
+            filename: { type: "string", description: "Desired filename, e.g. 'report.docx' or 'notes.md'." },
+            format: {
+              type: "string",
+              enum: ["md", "txt", "csv", "json", "html", "xml", "yaml", "docx", "pptx"],
+              description: "File format. Defaults to md."
+            }
+          },
+          required: ["content", "format"]
+        }
+      },
+      endpoint: "/mcp/create-file",
+      method: "POST",
+      isBuiltIn: true
+    };
+    defaultTools = [healthCheckTool, webSearchTool, webFetchTool, imageGenerationTool, createFileTool];
     useMCPToolsStore = (0, import_zustand11.create)((set, get) => ({
       tools: defaultTools,
       isLoaded: false,
@@ -20939,6 +20966,7 @@ TOOL USAGE PROTOCOL (conservative approach)
   * web_search() - when asked about recent/current events, breaking news, live information (weather, prices, sports scores), or when you need to look up documentation, libraries, APIs, error messages, or verify a specific fact
   * web_fetch() - to read the FULL contents of a specific URL you already have. Reach for this when the user wants to "tell me more", "go deeper", "read/open that article", or asks for details about a specific source, link, or article from an EARLIER answer: take that item's URL from the previous Sources list in this conversation and fetch it, then answer from the page's actual content (not just the prior summary)
   * image_generation() - ONLY when explicitly asked to create or generate an image
+  * create_file({"content": "...", "filename": "report.docx", "format": "docx"}) - when the user asks for a downloadable FILE (a document, spreadsheet, slides, markdown, code, etc.) or to "export"/"download"/"save" something. Formats: md, txt, csv, json, html, xml, yaml, docx, pptx. For docx/pptx write well-structured Markdown (use "## " headings to start each slide for pptx). It returns a temporary download link \u2014 ALWAYS tell the user the file expires (~1 hour). If it is unclear whether they want it shown inline vs. as a downloadable file, use ask_user first.
   * ask_user({"questions": [{"question": "...", "header": "Format", "options": [{"label": "Inline (Recommended)"}, {"label": "Download a file"}]}]}) - when you are genuinely BLOCKED on a decision that is the USER's to make and cannot resolve from the request, context, or sensible defaults (e.g. show content inline vs. let them download it, which format/option they want). Renders clickable options the user answers in one step \u2014 better than asking in prose and ending your turn. Give 1-4 questions, each with 2-4 options; if one is clearly best, list it first and append " (Recommended)". The user may also type their own answer; act on it directly.
 - For general questions about concepts, definitions, explanations, or how-to topics, use your built-in knowledge WITHOUT calling tools.
 - Examples of what NOT to use tools for: "who are you?", "what is React?", "explain machine learning", "how does X work?", general programming questions.
@@ -21221,6 +21249,17 @@ _${toolStatus}_` : `_${toolStatus}_`);
 
 ${revisedPrompt ? `Prompt refinement: \u201C${revisedPrompt}\u201D
 ` : ""}Note: the image link may expire in ~2 hours.` : "Image generated successfully.";
+                        } else if (functionName === "create_file" || functionName === "create-file") {
+                          const fileData = result.data ?? {};
+                          if (fileData.url) {
+                            const mins = fileData.expiresInMinutes ?? 60;
+                            const name = fileData.filename || "your file";
+                            resultText = `\u{1F4C4} **[${name}](${fileData.url})** \u2014 ready to download.
+
+_This link is temporary and expires in about ${mins} minutes._`;
+                          } else {
+                            resultText = "The file was created.";
+                          }
                         } else if (typeof result.data === "string") {
                           resultText = result.data;
                         } else if (result.data) {
@@ -21235,7 +21274,7 @@ ${revisedPrompt ? `Prompt refinement: \u201C${revisedPrompt}\u201D
                       }
                       enhancedMessage = enhancedMessage.replace(placeholderToken, resultText);
                       if (result.success) {
-                        if (functionName === "image_generation" || functionName === "image-generation") {
+                        if (functionName === "image_generation" || functionName === "image-generation" || functionName === "create_file" || functionName === "create-file") {
                           inlineImageBlocks.push(resultText);
                         } else if (functionName !== "check_gateway_health") {
                           summarizableResults.push({ name: functionName, output: resultText });
@@ -21267,41 +21306,38 @@ ${revisedPrompt ? `Prompt refinement: \u201C${revisedPrompt}\u201D
                     try {
                       const toolResultsText = summarizableResults.map((r) => `## ${r.name}
 ${r.output}`).join("\n\n");
-                      const summaryMessages = [
-                        { role: "system", content: systemPromptForSummary },
+                      const MAX_CHAIN_ROUNDS = 4;
+                      const enabledToolsForChain = getEnabledMCPToolsForAI();
+                      const convo = [
+                        { role: "system", content: enhancedSystemPrompt },
                         ...contextMessages,
                         { role: "user", content: question },
-                        { role: "assistant", content: stripToolBlocks(fullMessage) || "Let me look that up." },
+                        { role: "assistant", content: stripToolBlocks(fullMessage) || "Let me work on that." },
                         {
                           role: "user",
-                          content: `I ran the tool(s) you requested. Here are the raw results:
+                          content: `Here are the results of the tool(s) so far:
 
 ${toolResultsText}
 
-Using these results together with your own knowledge, answer my original question concisely and in a natural, well-formatted way. Do NOT add a "Sources", "References", or "Citations" list of any kind \u2014 not names and not URLs. A clean, clickable Sources section is appended automatically below your answer, so any list you add just duplicates it. (Mentioning a source naturally inside a sentence is fine; a trailing list is not.) Do NOT output tool_code or call any tools again.`
+Use them to fully complete my original request. If you still need to take an action I asked for (for example, actually create a file I want to download), call the appropriate tool now with a \`\`\`tool_code\`\`\` block. Otherwise give your final answer. Do NOT add a "Sources"/"References"/"Citations" list \u2014 one is appended automatically.`
                         }
                       ];
-                      const summaryRequest = {
-                        model: modelName,
-                        messages: summaryMessages,
-                        stream: true,
-                        options: { num_predict: tokenLimit + 250 }
-                      };
-                      clearFlushTimer();
-                      setStreamBuffer("");
-                      setIsThinking?.(true);
-                      const summaryText = await new Promise((resolve) => {
+                      const streamTurn = (req) => new Promise((resolve) => {
                         let acc = "";
+                        const native = [];
                         let settled = false;
                         let timer;
-                        const done = (value) => {
+                        const finish = (value) => {
                           if (settled) return;
                           settled = true;
                           if (timer) clearTimeout(timer);
                           resolve(value);
                         };
-                        const summarySub = provider.chat(summaryRequest).subscribe({
+                        const sub2 = provider.chat(req).subscribe({
                           next: (data) => {
+                            if (Array.isArray(data?.message?.tool_calls) && data.message.tool_calls.length) {
+                              native.push(...data.message.tool_calls);
+                            }
                             if (data?.message?.content) {
                               acc += data.message.content;
                               const visible = stripThinking(acc);
@@ -21311,27 +21347,125 @@ Using these results together with your own knowledge, answer my original questio
                               setStreamBuffer(visible);
                             }
                           },
-                          error: (summaryErr) => {
-                            debugLogger.error("Summarization pass failed", {
-                              error: summaryErr instanceof Error ? summaryErr.message : String(summaryErr)
-                            });
-                            done("");
-                          },
-                          complete: () => done(stripThinking(acc).trim())
+                          error: () => finish({ text: stripThinking(acc).trim(), native }),
+                          complete: () => finish({ text: stripThinking(acc).trim(), native })
                         });
-                        currentSubRef.current = summarySub;
+                        currentSubRef.current = sub2;
                         timer = setTimeout(() => {
-                          debugLogger.warn("Summarization pass timed out; using inline tool output");
                           try {
-                            summarySub.unsubscribe();
+                            sub2.unsubscribe();
                           } catch {
                           }
-                          done("");
+                          finish({ text: stripThinking(acc).trim(), native });
                         }, 3e4);
                       });
+                      const runChainedTool = async (fn, params) => {
+                        if (fn === "ask_user" || fn === "ask-user") {
+                          const qs = parseAskUserQuestions(params.questions ?? params);
+                          if (!qs.length) return "ask_user failed: it needs a questions array.";
+                          const ans = await useAskUserStore.getState().ask(qs);
+                          return ans ? "The user answered:\n\n" + qs.map((q) => `Q: ${q.question}
+A: ${(ans[q.id] || "").trim() || "(no answer)"}`).join("\n\n") : "The user dismissed the question(s). Proceed with your best judgment.";
+                        }
+                        const status = fn === "create_file" || fn === "create-file" ? "Creating the file\u2026" : fn === "web_search" || fn === "web-search" ? "Searching the web\u2026" : fn === "web_fetch" || fn === "web-fetch" ? "Reading the page\u2026" : fn === "image_generation" || fn === "image-generation" ? "Generating the image\u2026" : "Working on it\u2026";
+                        setStreamBuffer(`_${status}_`);
+                        const result = await executeMCPTool({ toolName: fn, parameters: params });
+                        if (!result.success) return `That step failed: ${result.error || "unknown error"}.`;
+                        if (fn === "create_file" || fn === "create-file") {
+                          const f = result.data ?? {};
+                          if (f.url) {
+                            const mins = f.expiresInMinutes ?? 60;
+                            const name = f.filename || "your file";
+                            inlineImageBlocks.push(`\u{1F4C4} **[${name}](${f.url})** \u2014 ready to download.
+
+_This link is temporary and expires in about ${mins} minutes._`);
+                            return `File created and its download link is now shown to the user. Briefly confirm it's ready and that it expires in ~${mins} minutes.`;
+                          }
+                          return "The file was created.";
+                        }
+                        if (fn === "image_generation" || fn === "image-generation") {
+                          const img = result.data ?? {};
+                          if (img.imageUrl) {
+                            inlineImageBlocks.push(`![Generated image](${img.imageUrl})`);
+                            return "Image generated and shown to the user.";
+                          }
+                        }
+                        if (typeof result.data === "string") return result.data.slice(0, 2e3);
+                        if (result.data) return JSON.stringify(result.data).slice(0, 1500);
+                        return "Done.";
+                      };
+                      clearFlushTimer();
+                      let finalText = "";
+                      let lastTurnText = "";
+                      for (let round = 0; round < MAX_CHAIN_ROUNDS; round++) {
+                        setStreamBuffer("");
+                        setIsThinking?.(true);
+                        const turnRequest = {
+                          model: modelName,
+                          messages: convo,
+                          stream: true,
+                          tools: enabledToolsForChain.length ? enabledToolsForChain : void 0,
+                          options: { num_predict: tokenLimit + 250 }
+                        };
+                        const { text: turnText, native: turnNative } = await streamTurn(turnRequest);
+                        setIsThinking?.(false);
+                        if (turnText.trim()) lastTurnText = turnText;
+                        let toolText = turnText;
+                        if (turnNative.length && !/```(?:tool_code|TOOL_CODE)/.test(toolText)) {
+                          for (const raw of turnNative) {
+                            const tc = raw;
+                            const fnName = tc.function?.name ?? tc.name;
+                            if (!fnName) continue;
+                            const a = tc.function?.arguments ?? tc.arguments ?? {};
+                            toolText += `
+
+\`\`\`tool_code
+${fnName}(${typeof a === "string" ? a : JSON.stringify(a ?? {})})
+\`\`\``;
+                          }
+                        }
+                        const chainMatches = toolText.match(/```(?:tool_code|TOOL_CODE)\s*\n([^`]+)\n```/gi);
+                        if (!chainMatches || !chainMatches.length) {
+                          finalText = turnText;
+                          break;
+                        }
+                        const roundOut = [];
+                        for (const m of chainMatches) {
+                          const code = m.replace(/```(?:tool_code|TOOL_CODE)\s*\n|\n```/gi, "").trim();
+                          const fm = code.match(/^(\w+)\(\s*(.*?)\s*\)$/);
+                          if (!fm) continue;
+                          const [, fnName, rawParams] = fm;
+                          let parsed = {};
+                          const rp = rawParams.trim();
+                          if (rp) {
+                            try {
+                              parsed = JSON.parse(rp.startsWith("{") ? rp : `{${rp}}`);
+                            } catch {
+                              parsed = {};
+                            }
+                          }
+                          try {
+                            roundOut.push(`## ${fnName}
+${await runChainedTool(fnName, parsed)}`);
+                          } catch (e) {
+                            roundOut.push(`## ${fnName}
+That step failed: ${e instanceof Error ? e.message : String(e)}`);
+                          }
+                        }
+                        convo.push({ role: "assistant", content: stripToolBlocks(turnText) || "(using a tool)" });
+                        convo.push({
+                          role: "user",
+                          content: `Tool results:
+
+${roundOut.join("\n\n")}
+
+Now give your final answer to my original request, or call another tool if you still genuinely need to. Do NOT add a "Sources" list.`
+                        });
+                      }
                       setIsThinking?.(false);
-                      if (summaryText.trim()) {
-                        const cleanedSummary = summaryText.replace(
+                      const answerText = finalText.trim() ? finalText : lastTurnText;
+                      if (answerText.trim() || inlineImageBlocks.length) {
+                        const cleanedSummary = answerText.replace(
                           /\n{1,}\s*(?:[*_#>\s]*)(?:sources?|references?|citations?|further reading)(?:\s*:)?\s*(?:[*_]*)\s*\n[\s\S]*$/i,
                           ""
                         ).trimEnd();