@burtson-labs/bandit-engine 2.0.72 → 2.0.73

package/dist/index.js

@@ -21021,6 +21021,7 @@ ${protocol}`;
           let latestDisplayMessage = "";
           let sawToolBlock = false;
           const nativeToolCalls = [];
+          let workingTimer = null;
           const stripThinking = (text) => {
             let result = text.replace(/<think>[\s\S]*?<\/think>/g, "");
             const openIdx = result.indexOf("<think>");
@@ -21028,6 +21029,28 @@ ${protocol}`;
             return result.trimStart();
           };
           const stripToolBlocks = (text) => text.replace(/```(?:tool_code|TOOL_CODE)\s*\n[\s\S]*?\n```/gi, "").trim();
+          let workingLabel = "Working on it";
+          let workingPreamble = "";
+          const stopWorking = () => {
+            if (workingTimer) {
+              clearInterval(workingTimer);
+              workingTimer = null;
+            }
+          };
+          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 flushNow = () => {
             clearFlushTimer();
             if (!sawToolBlock) {
@@ -21063,6 +21086,7 @@ ${protocol}`;
                 nativeToolCalls.push(...data.message.tool_calls);
                 sawToolBlock = true;
                 clearFlushTimer();
+                startWorking();
               }
               if (data.message.content) {
                 fullMessage += data.message.content;
@@ -21074,6 +21098,7 @@ ${protocol}`;
               if (/```(?:tool_code|TOOL_CODE)/.test(visibleMessage)) {
                 sawToolBlock = true;
                 clearFlushTimer();
+                startWorking();
               }
               latestDisplayMessage = visibleMessage;
               if (!sawToolBlock) {
@@ -21082,6 +21107,7 @@ ${protocol}`;
             },
             error: (err) => {
               debugLogger.error("Stream error:", err);
+              stopWorking();
               overrideComponentStatus("Idle");
               setIsSubmitting(false);
               setIsStreaming(false);
@@ -21166,6 +21192,7 @@ ${fn}(${argStr})
                       if (functionName === "ask_user" || functionName === "ask-user") {
                         enhancedMessage = enhancedMessage.replace(match, "");
                         clearFlushTimer();
+                        stopWorking();
                         const askPreamble = stripToolBlocks(fullMessage).trim();
                         setStreamBuffer(askPreamble || "_Waiting for your answer\u2026_");
                         const questions = parseAskUserQuestions(
@@ -21188,11 +21215,8 @@ A: ${(answers[q.id] || "").trim() || "(no answer)"}`).join("\n\n") : "The user d
                       const placeholderToken = `<<TOOL_LOADING_${functionName}_${Math.random().toString(36).slice(2)}>>`;
                       enhancedMessage = enhancedMessage.replace(match, placeholderToken);
                       clearFlushTimer();
-                      const toolStatus = functionName === "web_search" || functionName === "web-search" ? "Searching the web\u2026" : functionName === "web_fetch" || functionName === "web-fetch" ? "Reading the page\u2026" : functionName === "image_generation" || functionName === "image-generation" ? "Generating the image\u2026" : "Working on it\u2026";
-                      const toolPreamble = stripToolBlocks(fullMessage).trim();
-                      setStreamBuffer(toolPreamble ? `${toolPreamble}
-
-_${toolStatus}_` : `_${toolStatus}_`);
+                      const toolStatus = functionName === "web_search" || functionName === "web-search" ? "Searching the web" : functionName === "web_fetch" || functionName === "web-fetch" ? "Reading the page" : functionName === "image_generation" || functionName === "image-generation" ? "Generating the image" : functionName === "create_file" || functionName === "create-file" ? "Creating your file" : "Working on it";
+                      startWorking(toolStatus);
                       telemetryEvent("tool_loop:tool_execute", { name: functionName, params: parsedParams });
                       const result = await executeMCPTool({
                         toolName: functionName,
@@ -21343,7 +21367,10 @@ Use them to fully complete my original request. If you still need to take an act
                               const visible = stripThinking(acc);
                               latestDisplayMessage = visible;
                               lastPartialRef.current.text = visible;
-                              if (visible) setIsThinking?.(false);
+                              if (visible) {
+                                stopWorking();
+                                setIsThinking?.(false);
+                              }
                               setStreamBuffer(visible);
                             }
                           },
@@ -21367,8 +21394,8 @@ Use them to fully complete my original request. If you still need to take an act
                           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 status = fn === "create_file" || fn === "create-file" ? "Creating your file" : fn === "web_search" || fn === "web-search" ? "Searching the web" : fn === "web_fetch" || fn === "web-fetch" ? "Reading the page" : fn === "image_generation" || fn === "image-generation" ? "Generating the image" : "Working on it";
+                        startWorking(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") {
@@ -21398,6 +21425,7 @@ _This link is temporary and expires in about ${mins} minutes._`);
                       let finalText = "";
                       let lastTurnText = "";
                       for (let round = 0; round < MAX_CHAIN_ROUNDS; round++) {
+                        stopWorking();
                         setStreamBuffer("");
                         setIsThinking?.(true);
                         const turnRequest = {
@@ -21491,6 +21519,7 @@ ${inlineImageBlocks.join("\n\n")}` : "");
                     }
                   }
                 }
+                stopWorking();
                 overrideComponentStatus("Idle");
                 setIsSubmitting(false);
                 setPreviousQuestion(question);