@burtson-labs/bandit-engine 2.0.125 → 2.0.126

package/dist/index.mjs

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-2WFTEJPS.mjs";
+} from "./chunk-JXFPCLEY.mjs";
 import {
   chat_provider_default
 } from "./chunk-SAQUEXYS.mjs";
@@ -10,7 +10,7 @@ import {
   useGatewayHealth,
   useGatewayMemory,
   useGatewayModels
-} from "./chunk-ID6CTZ3G.mjs";
+} from "./chunk-RKA3X3EY.mjs";
 import "./chunk-574ILQTT.mjs";
 import "./chunk-F7JXIW6T.mjs";
 import "./chunk-K2JGFKEB.mjs";

package/dist/management/management.js

@@ -20570,7 +20570,7 @@ var init_source_chips = __esm({
 });
 
 // src/chat/chat-messages.tsx
-var import_material36, import_jsx_runtime36, ChatMessages, chat_messages_default;
+var import_material36, import_jsx_runtime36, WORKING_TOKEN_RE, ToolWorkingChip, ChatMessages, chat_messages_default;
 var init_chat_messages = __esm({
   "src/chat/chat-messages.tsx"() {
     "use strict";
@@ -20580,6 +20580,38 @@ var init_chat_messages = __esm({
     init_source_chips();
     init_conversationStore();
     import_jsx_runtime36 = require("react/jsx-runtime");
+    WORKING_TOKEN_RE = /\s*<<working:([^>]*)>>\s*$/;
+    ToolWorkingChip = ({ label }) => /* @__PURE__ */ (0, import_jsx_runtime36.jsxs)(
+      import_material36.Box,
+      {
+        sx: (theme) => ({
+          display: "inline-flex",
+          alignItems: "center",
+          gap: 1.25,
+          mt: 1.5,
+          px: 1.75,
+          py: 0.9,
+          borderRadius: 999,
+          bgcolor: theme.palette.action.hover,
+          border: `1px solid ${theme.palette.divider}`
+        }),
+        children: [
+          /* @__PURE__ */ (0, import_jsx_runtime36.jsxs)("div", { className: "typing-only", children: [
+            /* @__PURE__ */ (0, import_jsx_runtime36.jsx)("span", { className: "dot" }),
+            /* @__PURE__ */ (0, import_jsx_runtime36.jsx)("span", { className: "dot" }),
+            /* @__PURE__ */ (0, import_jsx_runtime36.jsx)("span", { className: "dot" })
+          ] }),
+          /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(
+            import_material36.Box,
+            {
+              component: "span",
+              sx: { fontSize: "0.85rem", fontWeight: 500, color: "text.secondary", userSelect: "none" },
+              children: label || "Working on it"
+            }
+          )
+        ]
+      }
+    );
     ChatMessages = ({
       pendingMessage,
       streamBuffer,
@@ -20605,7 +20637,9 @@ var init_chat_messages = __esm({
           const isPlaceholder = entry.answer === "...";
           const showLoader = isLast && isStreaming && streamBuffer.trim() === "";
           const showThinking = showLoader && isThinking;
-          const content = isLast ? isStreaming ? streamBuffer || "" : isPlaceholder ? "" : entry.answer : entry.answer;
+          const rawContent = isLast ? isStreaming ? streamBuffer || "" : isPlaceholder ? "" : entry.answer : entry.answer;
+          const workingMatch = isLast && isStreaming ? rawContent.match(WORKING_TOKEN_RE) : null;
+          const content = workingMatch ? rawContent.replace(WORKING_TOKEN_RE, "") : rawContent;
           const rawSources = entry.sourceFiles;
           const sourceSummaries = rawSources ? rawSources.filter((doc) => doc && typeof doc.name === "string" && doc.name.trim()).map((doc) => ({ id: doc.id || doc.name, name: doc.name.trim() })) : void 0;
           const responseNode = /* @__PURE__ */ (0, import_jsx_runtime36.jsxs)(
@@ -20678,6 +20712,7 @@ var init_chat_messages = __esm({
                           sources: sourceSummaries
                         }
                       ),
+                      workingMatch && /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(ToolWorkingChip, { label: workingMatch[1] }),
                       /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(source_chips_default, { content })
                     ]
                   }
@@ -20690,7 +20725,7 @@ var init_chat_messages = __esm({
             {
               question: entry.question,
               response: responseNode,
-              responseText: isLast ? isStreaming ? streamBuffer || "" : isPlaceholder ? "" : entry.answer : entry.answer,
+              responseText: content,
               backgroundColor: "#444",
               images: entry.images,
               memoryUpdated: entry.memoryUpdated && !isPlaceholder,
@@ -24239,8 +24274,20 @@ ${protocol}`;
             return result.trimStart();
           };
           const stripToolBlocks = (text) => text.replace(/```(?:tool_code|TOOL_CODE)\s*\n[\s\S]*?\n```/gi, "").trim();
+          const truncateAtToolApparatus = (text) => {
+            let t = text;
+            const fenceIdx = t.search(/```(?:tool_code|TOOL_CODE)/);
+            if (fenceIdx !== -1) t = t.slice(0, fenceIdx);
+            const callIdx = t.search(
+              /\b(?:web_search|web_fetch|image_generation|create_file|ask_user)\s*\(/
+            );
+            if (callIdx !== -1) t = t.slice(0, callIdx);
+            t = t.replace(/```[\w-]*\s*$/, "");
+            const fenceCount = (t.match(/```/g) || []).length;
+            if (fenceCount % 2 === 1) t = t.slice(0, t.lastIndexOf("```"));
+            return t.trim();
+          };
           let workingLabel = "Working on it";
-          let workingPreamble = "";
           const stopWorking = () => {
             if (workingTimer) {
               clearInterval(workingTimer);
@@ -24249,17 +24296,10 @@ ${protocol}`;
           };
           const startWorking = (label = "Working on it") => {
             workingLabel = label.replace(/[.…]+\s*$/, "").trim() || "Working on it";
-            workingPreamble = stripToolBlocks(stripThinking(fullMessage)).trim();
-            if (workingTimer) return;
-            let dots = 0;
-            const render = () => {
-              dots = (dots + 1) % 4;
-              setStreamBuffer(
-                `${workingPreamble}${workingPreamble ? "\n\n" : ""}_${workingLabel}${".".repeat(dots)}_`
-              );
-            };
-            render();
-            workingTimer = setInterval(render, 450);
+            const preamble = truncateAtToolApparatus(stripToolBlocks(stripThinking(fullMessage)));
+            setStreamBuffer(
+              `${preamble}${preamble ? "\n\n" : ""}<<working:${workingLabel}>>`
+            );
           };
           const flushNow = () => {
             clearFlushTimer();
@@ -24305,10 +24345,16 @@ ${protocol}`;
               const inThinkBlock = /<think>/.test(fullMessage) && !/<think>[\s\S]*<\/think>/.test(fullMessage);
               setIsThinking?.(inThinkBlock);
               const visibleMessage = stripThinking(fullMessage);
-              if (/```(?:tool_code|TOOL_CODE)/.test(visibleMessage) || /\b(?:web_search|web_fetch|image_generation|create_file|ask_user)\s*\(/.test(visibleMessage)) {
+              const bareToolMatch = visibleMessage.match(
+                /\b(web_search|web_fetch|image_generation|create_file|ask_user)\s*\(/
+              );
+              if (/```(?:tool_code|TOOL_CODE)/.test(visibleMessage) || bareToolMatch) {
                 sawToolBlock = true;
                 clearFlushTimer();
-                startWorking();
+                const fn = bareToolMatch?.[1];
+                startWorking(
+                  fn === "create_file" ? "Creating your file" : fn === "web_search" ? "Searching the web" : fn === "web_fetch" ? "Reading the page" : fn === "image_generation" ? "Generating the image" : "Working on it"
+                );
               }
               latestDisplayMessage = visibleMessage;
               if (!sawToolBlock) {