@burtson-labs/bandit-engine 2.0.125 → 2.0.127

package/dist/{chat-SC6CFUED.mjs → chat-2XYHFUT2.mjs}

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-2WFTEJPS.mjs";
+} from "./chunk-RRSACJI2.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-2XYHFUT2.mjs.map

package/dist/{chunk-ID6CTZ3G.mjs → chunk-7NBLIL76.mjs}

@@ -9754,7 +9754,7 @@ var MCPToolsTabV2_default = MCPToolsTabV2;
 
 // src/management/management.tsx
 import { jsx as jsx14, jsxs as jsxs14 } from "react/jsx-runtime";
-var preloadChatPage = () => import("./chat-SC6CFUED.mjs");
+var preloadChatPage = () => import("./chat-2XYHFUT2.mjs");
 var buildCapabilitiesUrl = (gatewayApiUrl) => {
   const trimmed = gatewayApiUrl.replace(/\/$/, "");
   if (trimmed.endsWith("/api")) {
@@ -11110,4 +11110,4 @@ export {
   useGatewayMemory,
   management_default
 };
-//# sourceMappingURL=chunk-ID6CTZ3G.mjs.map
+//# sourceMappingURL=chunk-7NBLIL76.mjs.map

package/dist/{chunk-2WFTEJPS.mjs → chunk-RRSACJI2.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,
@@ -1705,23 +1740,24 @@ ${sanitize(
                 bgcolor: inputBackground,
                 borderRadius: "24px",
                 boxShadow: isMobile ? `0 0 0 1px ${theme.palette.mode === "dark" ? "#444" : "#eee"}` : "0 4px 20px rgba(0, 0, 0, 0.4)",
-                px: isMobile ? compactMobile ? 1 : 1.5 : 3,
-                pt: isMobile ? compactMobile ? 0.6 : 1 : 1,
-                pb: isMobile ? compactMobile ? 0.9 : 1.5 : 1.5,
+                px: isMobile ? compactMobile ? 1 : 1.5 : 2,
+                pt: isMobile ? compactMobile ? 0.6 : 1 : 0.75,
+                pb: isMobile ? compactMobile ? 0.9 : 1.5 : 0.75,
                 display: "flex",
                 flexDirection: "column",
-                gap: isMobile ? compactMobile ? 0.75 : 1 : 1
+                gap: isMobile ? compactMobile ? 0.75 : 1 : 0.5
               },
               children: [
                 /* @__PURE__ */ jsxs6(
                   Box6,
                   {
                     sx: {
-                      display: "flex",
+                      display: fileInputs.length > 0 || pastedImages.length > 0 ? "flex" : "none",
                       flexWrap: "wrap",
                       gap: attachmentsGap,
                       alignItems: "center",
-                      flexDirection: "row"
+                      flexDirection: "row",
+                      pt: isMobile ? 0 : 0.5
                     },
                     children: [
                       fileInputs.map((file, idx) => /* @__PURE__ */ jsxs6(
@@ -1845,7 +1881,7 @@ ${sanitize(
                     ]
                   }
                 ),
-                /* @__PURE__ */ jsx9(
+                isMobile && /* @__PURE__ */ jsx9(
                   Box6,
                   {
                     sx: {
@@ -1858,7 +1894,7 @@ ${sanitize(
                       {
                         fullWidth: true,
                         multiline: true,
-                        maxRows: isMobile ? compactMobile ? 6 : 12 : 6,
+                        maxRows: compactMobile ? 6 : 12,
                         placeholder: isIncognito ? "Incognito chat \u2014 not saved" : "What's on your mind?",
                         value: inputValue,
                         onChange: (e) => setInputValue(e.target.value),
@@ -1869,18 +1905,15 @@ ${sanitize(
                           disableUnderline: true,
                           sx: {
                             flex: "1 1 auto",
-                            maxHeight: isMobile ? compactMobile ? "120px" : "150px" : "none",
+                            maxHeight: compactMobile ? "120px" : "150px",
                             overflowY: "auto",
                             color: fileText,
                             "& .MuiInputBase-inputMultiline": {
-                              fontSize: isMobile ? compactMobile ? "0.95rem" : "1rem" : "1rem"
+                              fontSize: compactMobile ? "0.95rem" : "1rem"
                             }
                           }
                         },
-                        sx: {
-                          flex: 1,
-                          "& .MuiInputBase-root": {}
-                        }
+                        sx: { flex: 1 }
                       }
                     )
                   }
@@ -1891,9 +1924,9 @@ ${sanitize(
                     sx: {
                       display: "flex",
                       justifyContent: "space-between",
-                      alignItems: "center",
+                      alignItems: "flex-end",
                       gap: isMobile ? 0.6 : 1,
-                      mt: isMobile ? 0.5 : 1
+                      mt: isMobile ? 0.5 : 0
                     },
                     children: [
                       /* @__PURE__ */ jsxs6(
@@ -1980,6 +2013,30 @@ ${sanitize(
                           ]
                         }
                       ),
+                      !isMobile && /* @__PURE__ */ jsx9(
+                        TextField2,
+                        {
+                          fullWidth: true,
+                          multiline: true,
+                          maxRows: 6,
+                          placeholder: isIncognito ? "Incognito chat \u2014 not saved" : "What's on your mind?",
+                          value: inputValue,
+                          onChange: (e) => setInputValue(e.target.value),
+                          onKeyDown: handleKeyPress,
+                          inputRef,
+                          variant: "standard",
+                          InputProps: {
+                            disableUnderline: true,
+                            sx: {
+                              color: fileText,
+                              maxHeight: "160px",
+                              overflowY: "auto",
+                              "& .MuiInputBase-inputMultiline": { fontSize: "1rem" }
+                            }
+                          },
+                          sx: { flex: 1, alignSelf: "center", mx: 0.5 }
+                        }
+                      ),
                       /* @__PURE__ */ jsx9(Box6, { sx: { display: "flex", alignItems: "center", gap: isMobile ? 0.6 : 1 }, children: /* @__PURE__ */ jsx9(Tooltip2, { title: isStreaming ? "Stop response" : "Send message", children: /* @__PURE__ */ jsx9("span", { children: /* @__PURE__ */ jsx9(
                         IconButton5,
                         {
@@ -2082,8 +2139,9 @@ var chat_input_default = ChatInput;
 
 // src/chat/ask-user-card.tsx
 import { useEffect as useEffect5, useState as useState7 } from "react";
-import { Box as Box7, Paper, Typography as Typography4, Button as Button2, TextField as TextField3, Chip, Stack } from "@mui/material";
+import { Box as Box7, Paper, Typography as Typography4, Button as Button2, TextField as TextField3, Chip, Grow } from "@mui/material";
 import { useTheme as useTheme7, alpha as alpha4 } from "@mui/material/styles";
+import { CircleHelp as QuestionIcon, Check as CheckIcon3 } from "lucide-react";
 
 // src/store/askUserStore.ts
 import { create as create3 } from "zustand";
@@ -2174,57 +2232,142 @@ var AskUserCard = () => {
     });
     submit(final);
   };
-  return /* @__PURE__ */ jsx10(Box7, { sx: { width: "100%", display: "flex", justifyContent: "center", px: { xs: 1, sm: 2 }, mb: 1.5 }, children: /* @__PURE__ */ jsxs7(
+  const isSingleQuestion = pending.questions.length === 1;
+  const handleOptionTap = (qid, label) => {
+    setFreeform((p) => ({ ...p, [qid]: "" }));
+    if (isSingleQuestion) {
+      submit({ [qid]: label });
+      return;
+    }
+    setSelected((p) => ({ ...p, [qid]: label }));
+  };
+  return /* @__PURE__ */ jsx10(Grow, { in: true, timeout: 300, children: /* @__PURE__ */ jsx10(Box7, { sx: { width: "100%", display: "flex", justifyContent: "center", px: { xs: 1, sm: 2 }, mb: 1.5 }, children: /* @__PURE__ */ jsxs7(
     Paper,
     {
       elevation: 0,
       sx: {
         width: "100%",
         maxWidth: 760,
-        p: { xs: 1.75, sm: 2.25 },
-        borderRadius: 2,
-        border: `1px solid ${alpha4(theme.palette.primary.main, 0.45)}`,
-        bgcolor: alpha4(theme.palette.primary.main, 0.06)
+        p: { xs: 2, sm: 2.5 },
+        borderRadius: 3,
+        border: `1px solid ${theme.palette.divider}`,
+        bgcolor: theme.palette.background.paper,
+        boxShadow: `0 8px 32px ${alpha4(theme.palette.common.black, theme.palette.mode === "dark" ? 0.35 : 0.1)}`
       },
       children: [
-        /* @__PURE__ */ jsx10(Typography4, { variant: "caption", sx: { color: theme.palette.primary.main, fontWeight: 700, letterSpacing: 0.4 }, children: "BANDIT NEEDS A QUICK DECISION" }),
-        pending.questions.map((q) => /* @__PURE__ */ jsxs7(Box7, { sx: { mt: 1.5 }, children: [
-          q.header && /* @__PURE__ */ jsx10(Chip, { label: q.header, size: "small", sx: { mb: 0.75, fontWeight: 600 }, color: "primary", variant: "outlined" }),
-          /* @__PURE__ */ jsx10(Typography4, { sx: { fontWeight: 600, color: theme.palette.text.primary, mb: 1 }, children: q.question }),
-          q.options && q.options.length > 0 && /* @__PURE__ */ jsx10(Stack, { spacing: 1, children: q.options.map((opt) => {
-            const isSel = selected[q.id] === opt.label && !freeform[q.id]?.trim();
-            return /* @__PURE__ */ jsx10(
-              Button2,
+        /* @__PURE__ */ jsxs7(Box7, { sx: { display: "flex", alignItems: "center", gap: 1, mb: 0.5 }, children: [
+          /* @__PURE__ */ jsx10(
+            Box7,
+            {
+              sx: {
+                width: 26,
+                height: 26,
+                borderRadius: "50%",
+                display: "flex",
+                alignItems: "center",
+                justifyContent: "center",
+                bgcolor: alpha4(theme.palette.primary.main, 0.12),
+                color: theme.palette.primary.main,
+                flexShrink: 0
+              },
+              children: /* @__PURE__ */ jsx10(QuestionIcon, { size: 16 })
+            }
+          ),
+          /* @__PURE__ */ jsx10(Typography4, { variant: "body2", sx: { color: theme.palette.text.secondary, fontWeight: 600 }, children: "Quick question before I continue" })
+        ] }),
+        pending.questions.map((q) => /* @__PURE__ */ jsxs7(Box7, { sx: { mt: 1.75 }, children: [
+          /* @__PURE__ */ jsxs7(Box7, { sx: { display: "flex", alignItems: "baseline", gap: 1, flexWrap: "wrap", mb: 1.25 }, children: [
+            /* @__PURE__ */ jsx10(Typography4, { sx: { fontWeight: 600, fontSize: "1.02rem", color: theme.palette.text.primary }, children: q.question }),
+            q.header && /* @__PURE__ */ jsx10(
+              Chip,
               {
-                onClick: () => {
-                  setSelected((p) => ({ ...p, [q.id]: opt.label }));
-                  setFreeform((p) => ({ ...p, [q.id]: "" }));
-                },
-                variant: isSel ? "contained" : "outlined",
-                color: "primary",
+                label: q.header,
+                size: "small",
                 sx: {
-                  justifyContent: "flex-start",
-                  textAlign: "left",
-                  textTransform: "none",
-                  py: 1,
-                  px: 1.5,
-                  borderColor: alpha4(theme.palette.primary.main, 0.4)
-                },
-                children: /* @__PURE__ */ jsxs7(Box7, { children: [
-                  /* @__PURE__ */ jsx10(Typography4, { sx: { fontWeight: 600, lineHeight: 1.3 }, children: opt.label }),
-                  opt.description && /* @__PURE__ */ jsx10(
-                    Typography4,
-                    {
-                      variant: "caption",
-                      sx: { color: isSel ? alpha4("#fff", 0.85) : theme.palette.text.secondary, display: "block" },
-                      children: opt.description
-                    }
-                  )
-                ] })
+                  height: 20,
+                  fontSize: "0.7rem",
+                  fontWeight: 600,
+                  color: theme.palette.text.secondary,
+                  bgcolor: alpha4(theme.palette.text.primary, 0.06)
+                }
+              }
+            )
+          ] }),
+          q.options && q.options.length > 0 && /* @__PURE__ */ jsx10(
+            Box7,
+            {
+              sx: {
+                display: "grid",
+                gridTemplateColumns: { xs: "1fr", sm: q.options.length > 1 ? "1fr 1fr" : "1fr" },
+                gap: 1
               },
-              opt.label
-            );
-          }) }),
+              children: q.options.map((opt) => {
+                const isSel = selected[q.id] === opt.label && !freeform[q.id]?.trim();
+                return /* @__PURE__ */ jsxs7(
+                  Box7,
+                  {
+                    component: "button",
+                    type: "button",
+                    onClick: () => handleOptionTap(q.id, opt.label),
+                    sx: {
+                      font: "inherit",
+                      textAlign: "left",
+                      cursor: "pointer",
+                      display: "flex",
+                      alignItems: "flex-start",
+                      gap: 1,
+                      px: 1.5,
+                      py: 1.25,
+                      borderRadius: 2,
+                      border: `1.5px solid ${isSel ? theme.palette.primary.main : theme.palette.divider}`,
+                      bgcolor: isSel ? alpha4(theme.palette.primary.main, 0.08) : "transparent",
+                      color: "inherit",
+                      transition: "border-color 0.15s ease, background-color 0.15s ease, transform 0.1s ease",
+                      "&:hover": {
+                        borderColor: theme.palette.primary.main,
+                        bgcolor: alpha4(theme.palette.primary.main, 0.05)
+                      },
+                      "&:active": { transform: "scale(0.99)" }
+                    },
+                    children: [
+                      /* @__PURE__ */ jsx10(
+                        Box7,
+                        {
+                          sx: {
+                            width: 18,
+                            height: 18,
+                            mt: "2px",
+                            borderRadius: "50%",
+                            flexShrink: 0,
+                            display: "flex",
+                            alignItems: "center",
+                            justifyContent: "center",
+                            border: `1.5px solid ${isSel ? theme.palette.primary.main : alpha4(theme.palette.text.secondary, 0.5)}`,
+                            bgcolor: isSel ? theme.palette.primary.main : "transparent",
+                            color: theme.palette.getContrastText(theme.palette.primary.main),
+                            transition: "all 0.15s ease"
+                          },
+                          children: isSel && /* @__PURE__ */ jsx10(CheckIcon3, { size: 12, strokeWidth: 3 })
+                        }
+                      ),
+                      /* @__PURE__ */ jsxs7(Box7, { sx: { minWidth: 0 }, children: [
+                        /* @__PURE__ */ jsx10(Typography4, { sx: { fontWeight: 600, lineHeight: 1.35, fontSize: "0.92rem" }, children: opt.label }),
+                        opt.description && /* @__PURE__ */ jsx10(
+                          Typography4,
+                          {
+                            variant: "caption",
+                            sx: { color: theme.palette.text.secondary, display: "block", mt: 0.25 },
+                            children: opt.description
+                          }
+                        )
+                      ] })
+                    ]
+                  },
+                  opt.label
+                );
+              })
+            }
+          ),
           q.allowFreeform !== false && /* @__PURE__ */ jsx10(
             TextField3,
             {
@@ -2233,17 +2376,45 @@ var AskUserCard = () => {
               placeholder: q.options && q.options.length ? "Or type your own answer\u2026" : "Type your answer\u2026",
               value: freeform[q.id] ?? "",
               onChange: (e) => setFreeform((p) => ({ ...p, [q.id]: e.target.value })),
-              sx: { mt: q.options && q.options.length ? 1.25 : 0 }
+              onKeyDown: (e) => {
+                if (e.key === "Enter" && !e.shiftKey && answerFor(q.id)) {
+                  e.preventDefault();
+                  if (allAnswered) handleSubmit();
+                }
+              },
+              sx: {
+                mt: q.options && q.options.length ? 1.25 : 0,
+                "& .MuiOutlinedInput-root": { borderRadius: 2 }
+              }
             }
           )
         ] }, q.id)),
-        /* @__PURE__ */ jsxs7(Box7, { sx: { display: "flex", justifyContent: "flex-end", gap: 1, mt: 2 }, children: [
-          /* @__PURE__ */ jsx10(Button2, { onClick: cancel, color: "inherit", sx: { textTransform: "none", color: theme.palette.text.secondary }, children: "Skip" }),
-          /* @__PURE__ */ jsx10(Button2, { onClick: handleSubmit, disabled: !allAnswered, variant: "contained", sx: { textTransform: "none" }, children: "Submit" })
+        /* @__PURE__ */ jsxs7(Box7, { sx: { display: "flex", alignItems: "center", justifyContent: "flex-end", gap: 1, mt: 2 }, children: [
+          /* @__PURE__ */ jsx10(
+            Button2,
+            {
+              onClick: cancel,
+              color: "inherit",
+              size: "small",
+              sx: { textTransform: "none", color: theme.palette.text.secondary },
+              children: "Skip \u2014 use your best judgment"
+            }
+          ),
+          (!isSingleQuestion || Object.values(freeform).some((v) => v.trim())) && /* @__PURE__ */ jsx10(
+            Button2,
+            {
+              onClick: handleSubmit,
+              disabled: !allAnswered,
+              variant: "contained",
+              size: "small",
+              sx: { textTransform: "none", borderRadius: 999, px: 2.5 },
+              children: "Answer"
+            }
+          )
         ] })
       ]
     }
-  ) });
+  ) }) });
 };
 var ask_user_card_default = AskUserCard;
 
@@ -4152,14 +4323,14 @@ TOOL USAGE PROTOCOL (conservative approach)
   * 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. PDF is NOT a supported format \u2014 if the user asks for a PDF, create a docx instead. 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.
+  * ask_user({"questions": [{"question": "...", "header": "Format", "options": [{"label": "Inline (Recommended)"}, {"label": "Download a file"}]}]}) - when the request is AMBIGUOUS or missing a detail you need to do a good job (unclear scope, audience, format, tone, which of several interpretations they meant), or you are blocked on a decision that is the user's to make and cannot resolve from the request, context, or sensible defaults. Renders clickable options the user answers in one step \u2014 ALWAYS prefer this over asking a clarifying question in prose and ending your turn. Give 1-4 questions, each with 2-4 short options; if one is clearly best, list it first and append " (Recommended)". The user may also type their own answer; act on it directly. Don't overuse it \u2014 when a sensible default exists, take it and note the assumption instead.
 - 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.
 - When a tool is truly needed, call exactly ONE tool that best matches the request.
 - Begin tool usage with a fenced code block: \`\`\`tool_code
 functionName({"param": "value"})
 \`\`\`
-- If you cannot answer with your knowledge and context, and no suitable tool exists, ask a clarifying question.
+- If you cannot answer with your knowledge and context, and no suitable tool exists, use ask_user to ask a clarifying question with clickable options.
 
 Examples of appropriate tool usage:
 
@@ -4214,8 +4385,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 +4407,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 +4456,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) {
@@ -6413,7 +6595,7 @@ import {
   TextField as TextField6,
   alpha as alpha7
 } from "@mui/material";
-import { ChevronDown as ExpandMoreIcon2, ChevronUp as ExpandLessIcon, Plus as AddIcon3, Folder as FolderIcon3, FolderOpen as FolderOpenIcon, Inbox as InboxIcon2, X as CloseIcon5, Check as CheckIcon3 } from "lucide-react";
+import { ChevronDown as ExpandMoreIcon2, ChevronUp as ExpandLessIcon, Plus as AddIcon3, Folder as FolderIcon3, FolderOpen as FolderOpenIcon, Inbox as InboxIcon2, X as CloseIcon5, Check as CheckIcon4 } from "lucide-react";
 import { useTheme as useTheme12 } from "@mui/material/styles";
 import { jsx as jsx15, jsxs as jsxs12 } from "react/jsx-runtime";
 var ProjectHeader = ({
@@ -6647,7 +6829,7 @@ var ProjectHeader = ({
                 commitRename();
               },
               sx: { color: theme.palette.success.main },
-              children: /* @__PURE__ */ jsx15(CheckIcon3, { size: 16 })
+              children: /* @__PURE__ */ jsx15(CheckIcon4, { size: 16 })
             }
           ) })
         ] }),
@@ -11498,4 +11680,4 @@ var chat_default = Chat;
 export {
   chat_default
 };
-//# sourceMappingURL=chunk-2WFTEJPS.mjs.map
+//# sourceMappingURL=chunk-RRSACJI2.mjs.map