@burtson-labs/bandit-engine 2.0.125 → 2.0.126

package/dist/{chat-SC6CFUED.mjs → chat-Z3GRZ7GA.mjs}

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-2WFTEJPS.mjs";
+} from "./chunk-JXFPCLEY.mjs";
 import "./chunk-ONQMRE2G.mjs";
 import "./chunk-574ILQTT.mjs";
 import "./chunk-K2JGFKEB.mjs";
@@ -13,4 +13,4 @@ import "./chunk-BJTO5JO5.mjs";
 export {
   chat_default as default
 };
-//# sourceMappingURL=chat-SC6CFUED.mjs.map
+//# sourceMappingURL=chat-Z3GRZ7GA.mjs.map

package/dist/{chunk-2WFTEJPS.mjs → chunk-JXFPCLEY.mjs}

@@ -723,6 +723,38 @@ var source_chips_default = SourceChips;
 
 // src/chat/chat-messages.tsx
 import { jsx as jsx6, jsxs as jsxs3 } from "react/jsx-runtime";
+var WORKING_TOKEN_RE = /\s*<<working:([^>]*)>>\s*$/;
+var ToolWorkingChip = ({ label }) => /* @__PURE__ */ jsxs3(
+  Box4,
+  {
+    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__ */ jsxs3("div", { className: "typing-only", children: [
+        /* @__PURE__ */ jsx6("span", { className: "dot" }),
+        /* @__PURE__ */ jsx6("span", { className: "dot" }),
+        /* @__PURE__ */ jsx6("span", { className: "dot" })
+      ] }),
+      /* @__PURE__ */ jsx6(
+        Box4,
+        {
+          component: "span",
+          sx: { fontSize: "0.85rem", fontWeight: 500, color: "text.secondary", userSelect: "none" },
+          children: label || "Working on it"
+        }
+      )
+    ]
+  }
+);
 var ChatMessages = ({
   pendingMessage,
   streamBuffer,
@@ -748,7 +780,9 @@ var ChatMessages = ({
       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__ */ jsxs3(
@@ -821,6 +855,7 @@ var ChatMessages = ({
                       sources: sourceSummaries
                     }
                   ),
+                  workingMatch && /* @__PURE__ */ jsx6(ToolWorkingChip, { label: workingMatch[1] }),
                   /* @__PURE__ */ jsx6(source_chips_default, { content })
                 ]
               }
@@ -833,7 +868,7 @@ var ChatMessages = ({
         {
           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,
@@ -4214,8 +4249,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);
@@ -4224,17 +4271,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();
@@ -4280,10 +4320,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) {
@@ -11498,4 +11544,4 @@ var chat_default = Chat;
 export {
   chat_default
 };
-//# sourceMappingURL=chunk-2WFTEJPS.mjs.map
+//# sourceMappingURL=chunk-JXFPCLEY.mjs.map