@burtson-labs/bandit-engine 2.0.61 → 2.0.63

package/dist/index.js

@@ -13559,7 +13559,7 @@ var init_ai_response_action_bar = __esm({
 });
 
 // src/modals/chat-modal/ai-response-text-field.tsx
-var import_react15, import_styles4, import_react_markdown, import_remark_gfm, import_rehype_raw, import_rehype_sanitize2, import_material8, import_jsx_runtime11, brainIcon, avatarFilenames, banditHead2, resolveAvatar2, normalizeTables, isNonEmptyString, toDateSafe, toNumberArray, MarkdownCodeBlock, AIResponseTextField, ai_response_text_field_default;
+var import_react15, import_styles4, import_react_markdown, import_remark_gfm, import_rehype_raw, import_rehype_sanitize2, import_material8, import_jsx_runtime11, brainIcon, normalizeTables, isNonEmptyString, toDateSafe, toNumberArray, MarkdownCodeBlock, AIResponseTextField, ai_response_text_field_default;
 var init_ai_response_text_field = __esm({
   "src/modals/chat-modal/ai-response-text-field.tsx"() {
     "use strict";
@@ -13574,33 +13574,12 @@ var init_ai_response_text_field = __esm({
     import_rehype_sanitize2 = __toESM(require("rehype-sanitize"));
     init_debugLogger();
     import_material8 = require("@mui/material");
-    init_modelStore();
     init_lowlight();
     init_markdownRendering();
     init_lucide_icons();
     init_ai_response_action_bar();
-    init_brandingService();
     import_jsx_runtime11 = require("react/jsx-runtime");
     brainIcon = "https://cdn.burtson.ai/images/brain-icon.png";
-    avatarFilenames = {
-      "Bandit-Core": "core-avatar.png",
-      "Bandit-Muse": "muse-avatar.png",
-      "Bandit-Logic": "logic-avatar.png",
-      "Bandit-D1VA": "d1va-avatar.png",
-      "Bandit-Exec": "exec-avatar.png",
-      "default": "bandit-head.png"
-    };
-    banditHead2 = `https://cdn.burtson.ai/images/bandit-head.png`;
-    resolveAvatar2 = (selectedModel) => {
-      const model = useModelStore.getState().availableModels.find(
-        (m) => m.name === selectedModel
-      );
-      if (model?.avatarBase64) {
-        return model.avatarBase64;
-      }
-      const avatarFilename = avatarFilenames[selectedModel] || avatarFilenames["default"];
-      return `https://cdn.burtson.ai/avatars/${avatarFilename}`;
-    };
     normalizeTables = (markdown2) => {
       const lines = markdown2.split("\n");
       const output = [];
@@ -13946,15 +13925,6 @@ var init_ai_response_text_field = __esm({
         const timeout3 = setTimeout(() => setShowMemoryUpdated(false), 3e3);
         return () => clearTimeout(timeout3);
       }, [memoryUpdated]);
-      const selectedModel = useModelStore((state) => state.selectedModel);
-      const [userAvatar, setUserAvatar] = (0, import_react15.useState)(banditHead2);
-      (0, import_react15.useEffect)(() => {
-        const fetchBranding = async () => {
-          const branding = await brandingService_default.getBranding();
-          setUserAvatar(branding?.logoBase64 || banditHead2);
-        };
-        fetchBranding();
-      }, []);
       const theme = (0, import_styles4.useTheme)();
       const chatResponse = theme.palette.chat?.response;
       const sanitizeMarkdown = (raw) => {
@@ -14323,283 +14293,108 @@ ${sourcesMarkdownList.join("\n")}`;
               alignSelf: "stretch",
               display: "flex",
               flexDirection: "column",
-              bgcolor: chatResponse.containerBackground,
-              color: chatResponse.aiText || "#fff",
-              p: isMobile ? 1 : 2,
-              borderRadius: "4px",
-              userSelect: "text",
-              border: "1px solid " + (chatResponse.aiBorder || "#ccc"),
-              boxShadow: "0 0 6px rgba(0,0,0,0.3)"
+              color: chatResponse.aiText || theme.palette.text.primary,
+              px: isMobile ? 0.5 : 2,
+              py: 1,
+              userSelect: "text"
             },
             children: [
-              /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                import_material8.Box,
-                {
-                  sx: {
-                    display: "flex",
-                    flexDirection: "column",
-                    alignItems: {
-                      xs: "flex-start",
-                      sm: "flex-end"
+              /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { display: "flex", justifyContent: "flex-end", mb: 2.5 }, children: /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(import_material8.Box, { sx: { maxWidth: { xs: "100%", sm: "85%" }, minWidth: 0 }, children: [
+                /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
+                  import_material8.Box,
+                  {
+                    sx: {
+                      bgcolor: chatResponse.userBubble || (0, import_styles4.alpha)(theme.palette.common.white, 0.07),
+                      borderRadius: "18px",
+                      px: 2,
+                      py: 1.25,
+                      wordBreak: "break-word",
+                      whiteSpace: "pre-wrap",
+                      overflowWrap: "break-word"
                     },
-                    gap: 2,
-                    mb: 2,
-                    justifyContent: "flex-end"
-                  },
-                  children: [
-                    /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                      import_material8.Box,
+                    children: /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
+                      import_material8.Typography,
                       {
                         sx: {
-                          display: "flex",
-                          flexDirection: "column",
-                          alignItems: "center"
+                          color: theme.palette.text.primary,
+                          lineHeight: 1.5,
+                          wordBreak: "break-word",
+                          whiteSpace: "pre-wrap"
                         },
-                        children: [
-                          /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                            import_material8.Avatar,
-                            {
-                              alt: "You",
-                              src: userAvatar,
-                              sx: {
-                                display: { xs: "none", sm: "flex" },
-                                width: { sm: 72 },
-                                height: { sm: 72 },
-                                bgcolor: chatResponse.userAvatarBackground || (theme.palette.mode === "dark" ? "transparent" : "#eee"),
-                                color: "#fff",
-                                fontWeight: "bold",
-                                fontSize: { sm: "1.1rem" },
-                                border: "2px solid #a78bfa",
-                                boxShadow: "0 0 8px rgba(167, 139, 250, 0.3)",
-                                transform: "scaleX(1)"
-                              }
-                            }
-                          ),
-                          /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                            import_material8.Typography,
-                            {
-                              variant: "caption",
-                              sx: { color: chatResponse.modelLabel || "#888", mt: 1, fontStyle: "italic" },
-                              children: "You said"
-                            }
-                          )
-                        ]
+                        children: question
                       }
-                    ),
-                    /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(import_material8.Box, { sx: { flex: 1 }, children: [
-                      /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                        import_material8.Box,
-                        {
-                          sx: {
-                            display: "inline-block",
-                            textAlign: "left",
-                            bgcolor: chatResponse.userBubble || "#1f1f1f",
-                            borderRadius: "4px",
-                            px: isMobile ? 1 : 2,
-                            py: 1.5,
-                            width: isMobile ? "100%" : "fit-content",
-                            maxWidth: "100%",
-                            border: "1px solid " + (chatResponse.aiBorder || "#444"),
-                            wordBreak: "break-word",
-                            whiteSpace: "pre-wrap",
-                            overflowWrap: "break-word",
-                            mt: { xs: 0.5, sm: 0.25 }
-                          },
-                          children: /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                            import_material8.Typography,
-                            {
-                              sx: {
-                                color: chatResponse.userText || "#6C9AC5",
-                                fontStyle: "italic",
-                                wordBreak: "break-word",
-                                whiteSpace: "pre-wrap"
-                              },
-                              children: question
-                            }
-                          )
-                        }
-                      ),
-                      images && images.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { display: "flex", gap: 1, mt: 1, flexWrap: "wrap" }, children: images.map((img, i) => /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                        import_material8.Avatar,
-                        {
-                          src: img,
-                          variant: "rounded",
-                          onClick: () => setOpenImage(img),
-                          sx: {
-                            width: 64,
-                            height: 64,
-                            borderRadius: 2,
-                            cursor: "pointer",
-                            "&:hover": { boxShadow: `0 0 0 2px ${theme.palette.primary.main}` }
-                          }
-                        },
-                        i
-                      )) })
-                    ] })
-                  ]
-                }
-              ),
-              /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { borderBottom: `1px solid ${theme.palette.divider}`, my: 2 } }),
-              /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                import_material8.Box,
-                {
-                  sx: {
-                    display: "flex",
-                    flexDirection: "column",
-                    alignItems: {
-                      xs: "center",
-                      sm: "flex-start"
-                    },
-                    gap: 2,
-                    mb: 2
+                    )
+                  }
+                ),
+                images && images.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { display: "flex", gap: 1, mt: 1, flexWrap: "wrap", justifyContent: "flex-end" }, children: images.map((img, i) => /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
+                  import_material8.Avatar,
+                  {
+                    src: img,
+                    variant: "rounded",
+                    onClick: () => setOpenImage(img),
+                    sx: {
+                      width: 64,
+                      height: 64,
+                      borderRadius: 2,
+                      cursor: "pointer",
+                      "&:hover": { boxShadow: `0 0 0 2px ${theme.palette.primary.main}` }
+                    }
                   },
-                  children: [
-                    typeof response === "string" && response.trim() !== "" && /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                      import_material8.Box,
-                      {
-                        sx: {
-                          display: "flex",
-                          flexDirection: "column",
-                          alignItems: "center",
-                          mr: 2,
-                          position: "relative"
-                        },
-                        children: [
-                          /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                            import_material8.Avatar,
-                            {
-                              src: resolveAvatar2(selectedModel),
-                              alt: selectedModel,
-                              sx: {
-                                display: { xs: "none", sm: "flex" },
-                                width: 72,
-                                height: 72,
-                                border: "1px solid #888",
-                                boxShadow: "0 0 6px rgba(136, 136, 136, 0.4)",
-                                filter: "brightness(1.6)"
-                              }
-                            }
-                          ),
-                          /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { display: { xs: "none", sm: "block" } }, children: /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                            import_material8.Typography,
-                            {
-                              variant: "caption",
-                              sx: { color: chatResponse.modelLabel || "#888", mt: 1, fontStyle: "italic" },
-                              children: [
-                                selectedModel,
-                                " says"
-                              ]
-                            }
-                          ) })
-                        ]
-                      }
-                    ),
-                    /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                      import_material8.Box,
-                      {
-                        sx: {
-                          display: { xs: "flex", sm: "none" },
-                          justifyContent: "right",
-                          width: "100%",
-                          mt: -1,
-                          mb: 1
-                        },
-                        children: /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                          import_material8.Typography,
-                          {
-                            variant: "caption",
-                            sx: { fontStyle: "italic", color: chatResponse.modelLabel || "#888" },
-                            children: [
-                              selectedModel,
-                              " says"
-                            ]
-                          }
-                        )
-                      }
-                    ),
-                    /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(import_material8.Box, { sx: { position: "relative", width: "100%" }, children: [
-                      cancelled && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                        import_material8.Box,
-                        {
-                          sx: {
-                            position: "absolute",
-                            top: -24,
-                            left: 0,
-                            display: "flex",
-                            alignItems: "center",
-                            gap: 1,
-                            pl: 1,
-                            zIndex: 1
-                          },
-                          children: /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Typography, { variant: "caption", sx: { fontStyle: "italic", opacity: 0.85 }, children: "Cancelled by you" })
-                        }
-                      ),
-                      showMemoryUpdated && /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
-                        import_material8.Box,
-                        {
-                          sx: {
-                            position: "absolute",
-                            top: -24,
-                            right: 0,
-                            display: "flex",
-                            alignItems: "center",
-                            gap: 1,
-                            pr: 1,
-                            animation: "fadeOut 0.3s ease-in 2.7s forwards",
-                            zIndex: 1
-                          },
-                          children: [
-                            /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                              "img",
-                              {
-                                src: brainIcon,
-                                alt: "Memory",
-                                style: { width: 18, height: 18 }
-                              }
-                            ),
-                            /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                              import_material8.Typography,
-                              {
-                                variant: "caption",
-                                sx: { color: chatResponse.memoryText || "#2e7d32", fontStyle: "italic" },
-                                children: "Bandit added to memory"
-                              }
-                            )
-                          ]
-                        }
-                      ),
+                  i
+                )) })
+              ] }) }),
+              /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(import_material8.Box, { sx: { position: "relative", width: "100%", maxWidth: { xs: "100%", sm: "768px" } }, children: [
+                cancelled && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
+                  import_material8.Box,
+                  {
+                    sx: {
+                      position: "absolute",
+                      top: -22,
+                      left: 0,
+                      display: "flex",
+                      alignItems: "center",
+                      gap: 1,
+                      zIndex: 1
+                    },
+                    children: /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Typography, { variant: "caption", sx: { fontStyle: "italic", opacity: 0.85 }, children: "Cancelled by you" })
+                  }
+                ),
+                showMemoryUpdated && /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
+                  import_material8.Box,
+                  {
+                    sx: {
+                      position: "absolute",
+                      top: -22,
+                      right: 0,
+                      display: "flex",
+                      alignItems: "center",
+                      gap: 1,
+                      animation: "fadeOut 0.3s ease-in 2.7s forwards",
+                      zIndex: 1
+                    },
+                    children: [
+                      /* @__PURE__ */ (0, import_jsx_runtime11.jsx)("img", { src: brainIcon, alt: "Memory", style: { width: 16, height: 16 } }),
                       /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                        import_material8.Box,
+                        import_material8.Typography,
                         {
-                          sx: {
-                            bgcolor: chatResponse.aiBubble ?? "#2f2f2f",
-                            borderRadius: "4px",
-                            px: isMobile ? 1 : 1.5,
-                            // Reduced padding on mobile
-                            py: 1.25,
-                            width: "100%",
-                            maxWidth: isMobile ? "100%" : "768px",
-                            // Full width on mobile
-                            border: "1px solid " + (chatResponse.aiBorder || "#ccc"),
-                            wordBreak: "break-word",
-                            alignSelf: "flex-start",
-                            mt: { xs: 0.5, sm: 0.25 }
-                          },
-                          children: /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { width: "100%", maxWidth: "100%" }, children: typeof response === "string" ? /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
-                            import_react_markdown.default,
-                            {
-                              remarkPlugins: [import_remark_gfm.default],
-                              rehypePlugins: [import_rehype_raw.default, [import_rehype_sanitize2.default, markdownSanitizeSchema]],
-                              components,
-                              children: enrichedMarkdown ?? sanitizeMarkdown(response)
-                            }
-                          ) : import_react15.default.isValidElement(response) ? response : /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Typography, { color: "error", children: "\u26A0\uFE0F Invalid AI response" }) })
+                          variant: "caption",
+                          sx: { color: chatResponse.memoryText || "#2e7d32", fontStyle: "italic" },
+                          children: "Bandit added to memory"
                         }
                       )
-                    ] })
-                  ]
-                }
-              ),
+                    ]
+                  }
+                ),
+                /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { width: "100%", maxWidth: "100%" }, children: typeof response === "string" ? /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
+                  import_react_markdown.default,
+                  {
+                    remarkPlugins: [import_remark_gfm.default],
+                    rehypePlugins: [import_rehype_raw.default, [import_rehype_sanitize2.default, markdownSanitizeSchema]],
+                    components,
+                    children: enrichedMarkdown ?? sanitizeMarkdown(response)
+                  }
+                ) : import_react15.default.isValidElement(response) ? response : /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Typography, { color: "error", children: "\u26A0\uFE0F Invalid AI response" }) })
+              ] }),
               !!(responseText || typeof response === "string" && response) && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(ai_response_action_bar_default, { text: responseText || response }),
               displaySourceFiles && displaySourceFiles.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material8.Box, { sx: { mt: 1.5, display: "flex", gap: 2, flexWrap: "wrap", justifyContent: "flex-start" }, children: displaySourceFiles.map((doc, idx) => {
                 debugLogger.debug("Rendering DocumentCard in AI response", {
@@ -20977,6 +20772,7 @@ ${protocol}`;
           let fullMessage = "";
           let latestDisplayMessage = "";
           let sawToolBlock = false;
+          const nativeToolCalls = [];
           const stripThinking = (text) => {
             let result = text.replace(/<think>[\s\S]*?<\/think>/g, "");
             const openIdx = result.indexOf("<think>");
@@ -21015,6 +20811,11 @@ ${protocol}`;
           const sub = stream.subscribe({
             next: (data) => {
               if (!data?.message?.content && !data?.message?.tool_calls) return;
+              if (Array.isArray(data.message.tool_calls) && data.message.tool_calls.length > 0) {
+                nativeToolCalls.push(...data.message.tool_calls);
+                sawToolBlock = true;
+                clearFlushTimer();
+              }
               if (data.message.content) {
                 fullMessage += data.message.content;
                 telemetryEvent("tool_loop:llm_chunk", { chunk: data.message.content });
@@ -21063,6 +20864,20 @@ ${protocol}`;
                 if (!sawToolBlock) {
                   flushNow();
                 }
+                if (nativeToolCalls.length > 0 && !/```(?:tool_code|TOOL_CODE)/.test(fullMessage)) {
+                  for (const raw of nativeToolCalls) {
+                    const tc = raw;
+                    const fn = tc.function?.name ?? tc.name;
+                    if (!fn) continue;
+                    const rawArgs = tc.function?.arguments ?? tc.arguments ?? {};
+                    const argStr = typeof rawArgs === "string" ? rawArgs : JSON.stringify(rawArgs ?? {});
+                    fullMessage += `
+
+\`\`\`tool_code
+${fn}(${argStr})
+\`\`\``;
+                  }
+                }
                 const toolCallMatches = fullMessage.match(/```(?:tool_code|TOOL_CODE)\s*\n([^`]+)\n```/gi);
                 let enhancedMessage = fullMessage;
                 const summarizableResults = [];
@@ -24816,7 +24631,7 @@ var init_enhanced_mobile_conversations_modal = __esm({
 });
 
 // src/chat/chat-app-bar.tsx
-var import_material22, import_react31, import_material23, import_react_router_dom, import_shallow2, import_jsx_runtime24, CDN_BASE, banditHead3, modelAvatars2, ChatAppBar, chat_app_bar_default;
+var import_material22, import_react31, import_material23, import_react_router_dom, import_shallow2, import_jsx_runtime24, CDN_BASE, banditHead2, modelAvatars2, ChatAppBar, chat_app_bar_default;
 var init_chat_app_bar = __esm({
   "src/chat/chat-app-bar.tsx"() {
     "use strict";
@@ -24840,7 +24655,7 @@ var init_chat_app_bar = __esm({
     import_shallow2 = require("zustand/shallow");
     import_jsx_runtime24 = require("react/jsx-runtime");
     CDN_BASE = "https://cdn.burtson.ai/";
-    banditHead3 = `${CDN_BASE}/images/bandit-head.png`;
+    banditHead2 = `${CDN_BASE}/images/bandit-head.png`;
     modelAvatars2 = {
       "Bandit-Core": `${CDN_BASE}/avatars/core-avatar.png`,
       "Bandit-Muse": `${CDN_BASE}/avatars/muse-avatar.png`,
@@ -25006,16 +24821,18 @@ var init_chat_app_bar = __esm({
       };
       const selectedModel = useModelStore((s) => s.selectedModel);
       const currentModel = useModelStore((s) => s.availableModels.find((m) => m.name === selectedModel));
-      const currentAvatar = currentModel?.avatarBase64 || modelAvatars2[selectedModel] || banditHead3;
+      const currentAvatar = currentModel?.avatarBase64 || modelAvatars2[selectedModel] || banditHead2;
       const engines = useEngineStore((s) => s.engines);
       const selectedEngine = useEngineStore((s) => s.selectedEngine);
       const effectiveEngineId = selectedEngine || usePackageSettingsStore.getState().settings?.defaultModel || "bandit-core";
-      const currentEngine = engines.find((e) => e.id === effectiveEngineId);
-      const engineLabel = currentEngine?.displayName?.replace(/^Bandit /, "") || "Engine";
+      const currentEngine = engines.find((e) => e.id === effectiveEngineId) || engines.find((e) => effectiveEngineId.startsWith(e.id + ":"));
+      const resolvedEngineId = currentEngine?.id ?? effectiveEngineId;
+      const cleanEngineName = (name) => (name || "").replace(/\s*\([^)]*\)\s*$/, "").trim();
+      const engineDisplay = cleanEngineName(currentEngine?.displayName) || "Engine";
       (0, import_react31.useEffect)(() => {
         useEngineStore.getState().fetchEngines();
       }, []);
-      const pendingModelAvatar = useModelStore.getState().availableModels.find((m) => m.name === pendingModel)?.avatarBase64 || modelAvatars2[pendingModel || ""] || banditHead3;
+      const pendingModelAvatar = useModelStore.getState().availableModels.find((m) => m.name === pendingModel)?.avatarBase64 || modelAvatars2[pendingModel || ""] || banditHead2;
       const resolvedHomeUrl = preferences.homeUrl?.trim() || packageSettings?.homeUrl?.trim() || "";
       const homeTooltip = (() => {
         if (!resolvedHomeUrl) return "Home";
@@ -25327,16 +25144,13 @@ var init_chat_app_bar = __esm({
                         )
                       }
                     ) }),
-                    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Tooltip, { title: `Engine: ${currentEngine?.displayName ?? effectiveEngineId}`, arrow: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(
+                    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Tooltip, { title: `Engine \xB7 ${engineDisplay}`, arrow: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
                       import_material23.IconButton,
                       {
                         onClick: (e) => setEngineAnchorEl(e.currentTarget),
                         sx: pillButtonStyles,
-                        "aria-label": `Change base model (engine). Currently ${effectiveEngineId}`,
-                        children: [
-                          currentEngine?.cloud ? /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(CloudDoneIcon, { fontSize: "small" }) : /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(CloudOffIcon, { fontSize: "small" }),
-                          /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Typography, { variant: "caption", sx: { ml: 0.75, fontWeight: 600, whiteSpace: "nowrap" }, children: engineLabel })
-                        ]
+                        "aria-label": `Change base model (engine). Currently ${engineDisplay}`,
+                        children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(AutoAwesomeIcon, { fontSize: "small" })
                       }
                     ) }),
                     /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(
@@ -25360,7 +25174,7 @@ var init_chat_app_bar = __esm({
                             return /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(
                               import_material23.MenuItem,
                               {
-                                selected: engine.id === effectiveEngineId,
+                                selected: engine.id === resolvedEngineId,
                                 disabled: !engine.available,
                                 onClick: () => {
                                   useEngineStore.getState().setSelectedEngine(engine.id);
@@ -25378,8 +25192,8 @@ var init_chat_app_bar = __esm({
                                 },
                                 children: [
                                   /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(import_material23.Box, { sx: { display: "flex", alignItems: "center", gap: 1, width: "100%" }, children: [
-                                    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Typography, { variant: "body2", sx: { fontWeight: 600, flex: 1 }, children: engine.displayName }),
-                                    engine.id === effectiveEngineId && /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Box, { sx: { width: 8, height: 8, borderRadius: "50%", bgcolor: theme.palette.primary.main } })
+                                    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Typography, { variant: "body2", sx: { fontWeight: 600, flex: 1 }, children: cleanEngineName(engine.displayName) }),
+                                    engine.id === resolvedEngineId && /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Box, { sx: { width: 8, height: 8, borderRadius: "50%", bgcolor: theme.palette.primary.main } })
                                   ] }),
                                   /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Typography, { variant: "caption", sx: { color: theme.palette.text.secondary }, children: engine.available ? engine.description : engine.unavailableReason || "Unavailable" }),
                                   badges.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Box, { sx: { display: "flex", gap: 0.5, flexWrap: "wrap", mt: 0.25 }, children: badges.map((b) => /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
@@ -25491,7 +25305,7 @@ var init_chat_app_bar = __esm({
                               /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
                                 import_material22.Avatar,
                                 {
-                                  src: model.avatarBase64 || modelAvatars2[model.name] || banditHead3,
+                                  src: model.avatarBase64 || modelAvatars2[model.name] || banditHead2,
                                   alt: model.name,
                                   sx: {
                                     width: 28,
@@ -30854,7 +30668,7 @@ var import_material30 = require("@mui/material");
 init_lucide_icons();
 var import_jsx_runtime31 = require("react/jsx-runtime");
 var banditaiLogo = "https://cdn.burtson.ai/logos/bandit-ai-logo.png";
-var banditHead4 = "https://cdn.burtson.ai/images/bandit-head.png";
+var banditHead3 = "https://cdn.burtson.ai/images/bandit-head.png";
 var ModalHeader = ({
   fullScreen,
   setFullScreen,
@@ -30868,7 +30682,7 @@ var ModalHeader = ({
 }) => {
   const theme = (0, import_material30.useTheme)();
   const isMobile = (0, import_material30.useMediaQuery)((theme2) => theme2.breakpoints.down("sm"));
-  const displayLogo = logo && logo !== banditaiLogo ? logo : banditHead4;
+  const displayLogo = logo && logo !== banditaiLogo ? logo : banditHead3;
   const [isHovering, setIsHovering] = (0, import_react38.useState)(false);
   const buttonStyles = {
     transition: "all 0.15s ease-out",
@@ -32863,7 +32677,7 @@ init_lucide_icons();
 var import_jsx_runtime35 = require("react/jsx-runtime");
 var FULL_SCREEN_THRESHOLD = 100;
 var CDN_BASE2 = "https://cdn.burtson.ai/";
-var banditHead5 = `${CDN_BASE2}/images/bandit-head.png`;
+var banditHead4 = `${CDN_BASE2}/images/bandit-head.png`;
 var modelAvatars3 = {
   "Bandit-Core": `${CDN_BASE2}/avatars/core-avatar.png`,
   "Bandit-Muse": `${CDN_BASE2}/avatars/muse-avatar.png`,
@@ -32953,7 +32767,7 @@ var ChatModal = ({
   const provider = useAIProviderStore((state) => state.provider);
   const notificationService2 = useNotificationService();
   const currentModel = availableModels.find((m) => m.name === selectedModel);
-  const currentAvatar = currentModel?.avatarBase64 || modelAvatars3[selectedModel] || banditHead5;
+  const currentAvatar = currentModel?.avatarBase64 || modelAvatars3[selectedModel] || banditHead4;
   const removeImage = (index) => {
     setPastedImages((prev) => prev.filter((_, i) => i !== index));
   };
@@ -33273,7 +33087,7 @@ var ChatModal = ({
                                   /* @__PURE__ */ (0, import_jsx_runtime35.jsx)(
                                     import_material35.Avatar,
                                     {
-                                      src: model.avatarBase64 || modelAvatars3[model.name] || banditHead5,
+                                      src: model.avatarBase64 || modelAvatars3[model.name] || banditHead4,
                                       alt: model.name,
                                       sx: {
                                         width: 28,
@@ -42741,7 +42555,7 @@ var Management = () => {
   const isMobile = (0, import_useMediaQuery2.default)("(max-width:900px)");
   const [sidebarOpen, setSidebarOpen] = (0, import_react58.useState)(false);
   const getOptimalFabLogo = async () => {
-    const banditHead7 = "https://cdn.burtson.ai/images/bandit-head.png";
+    const banditHead6 = "https://cdn.burtson.ai/images/bandit-head.png";
     try {
       const subdomain = window.location.hostname.split(".")[0];
       const faviconUrl = `https://cdn.burtson.ai/favicons/${subdomain}/favicon.png`;
@@ -42758,10 +42572,10 @@ var Management = () => {
       if (branding?.logoBase64) {
         return branding.logoBase64;
       }
-      return banditHead7;
+      return banditHead6;
     } catch (error) {
       debugLogger.error("Failed to get optimal FAB logo", { error });
-      return banditHead7;
+      return banditHead6;
     }
   };
   const {
@@ -42780,8 +42594,8 @@ var Management = () => {
     setHasTransparentLogo
   } = useModelStore();
   const [modalOpen, setModalOpen] = (0, import_react58.useState)(false);
-  const banditHead6 = "https://cdn.burtson.ai/images/bandit-head.png";
-  const [fabLogo, setFabLogo] = (0, import_react58.useState)(banditHead6);
+  const banditHead5 = "https://cdn.burtson.ai/images/bandit-head.png";
+  const [fabLogo, setFabLogo] = (0, import_react58.useState)(banditHead5);
   const [tabIndex, setTabIndex] = (0, import_react58.useState)(4);
   const [logoFile, setLogoFile] = (0, import_react58.useState)(null);
   const [logoBase64, setLogoBase64] = (0, import_react58.useState)(null);