@burtson-labs/bandit-engine 2.0.61 → 2.0.63

package/dist/management/management.mjs

@@ -1,11 +1,11 @@
 import {
   management_default
-} from "../chunk-SKMJ43NN.mjs";
-import "../chunk-SRCCNBHF.mjs";
+} from "../chunk-5NSOHUCR.mjs";
+import "../chunk-E4Q7BHBQ.mjs";
 import "../chunk-VTC6AIWY.mjs";
 import "../chunk-6QTTNYF2.mjs";
-import "../chunk-PY7A3J5T.mjs";
-import "../chunk-5WQMMCZQ.mjs";
+import "../chunk-UQ4VQSWS.mjs";
+import "../chunk-NCMSNXB2.mjs";
 import "../chunk-EUBVBTB3.mjs";
 import "../chunk-IPMTNREZ.mjs";
 import "../chunk-H3BYFEIE.mjs";

package/dist/modals/chat-modal/chat-modal.js

@@ -11509,7 +11509,6 @@ var import_rehype_raw = __toESM(require("rehype-raw"));
 var import_rehype_sanitize2 = __toESM(require("rehype-sanitize"));
 init_debugLogger();
 var import_material8 = require("@mui/material");
-init_modelStore();
 
 // src/modals/chat-modal/ai-response-action-bar.tsx
 var import_react13 = require("react");
@@ -12409,99 +12408,9 @@ var AiResponseActionsBar = ({ text }) => {
 };
 var ai_response_action_bar_default = AiResponseActionsBar;
 
-// src/services/branding/brandingService.ts
-init_indexedDBService();
-init_debugLogger();
-var BrandingService = class {
-  DB_NAME = "banditConfig";
-  DB_VERSION = 1;
-  STORE_NAME = "config";
-  STORE_CONFIGS = [{ name: "config", keyPath: "id" }];
-  async getBranding() {
-    try {
-      const data = await indexedDBService_default.get(
-        this.DB_NAME,
-        this.DB_VERSION,
-        this.STORE_NAME,
-        "main",
-        this.STORE_CONFIGS
-      );
-      if (data?.branding && (data.branding.logoBase64 || data.branding.hasTransparentLogo !== void 0 || data.branding.brandingText)) {
-        return {
-          logoBase64: data.branding.logoBase64,
-          hasTransparentLogo: data.branding.hasTransparentLogo,
-          brandingText: data.branding.brandingText
-        };
-      }
-      return null;
-    } catch (error) {
-      throw error;
-    }
-  }
-  async setBrandingFromConfig(config) {
-    try {
-      const existing = await indexedDBService_default.get(
-        this.DB_NAME,
-        this.DB_VERSION,
-        this.STORE_NAME,
-        "main",
-        this.STORE_CONFIGS
-      ) || { id: "main" };
-      if (existing.branding?.userSaved) {
-        debugLogger.debug("\u{1F6E1}\uFE0F BrandingService: User branding detected, blocking CDN config override");
-        return;
-      }
-      const hasUserBranding = existing.branding && (existing.branding.logoBase64 || existing.branding.brandingText || existing.branding.theme || existing.branding.hasTransparentLogo !== void 0);
-      if (hasUserBranding) {
-        debugLogger.debug("\u{1F6E1}\uFE0F BrandingService: Detected user branding data, blocking CDN config override");
-        return;
-      }
-      debugLogger.debug("\u{1F527} BrandingService: No user branding detected, applying CDN config");
-      const branding = config.branding || {};
-      const updated = {
-        ...existing,
-        branding: {
-          ...existing.branding,
-          ...branding
-        }
-      };
-      await indexedDBService_default.put(
-        this.DB_NAME,
-        this.DB_VERSION,
-        this.STORE_NAME,
-        updated,
-        this.STORE_CONFIGS
-      );
-    } catch (error) {
-      throw error;
-    }
-  }
-};
-var brandingService = new BrandingService();
-var brandingService_default = brandingService;
-
 // src/modals/chat-modal/ai-response-text-field.tsx
 var import_jsx_runtime9 = require("react/jsx-runtime");
 var brainIcon = "https://cdn.burtson.ai/images/brain-icon.png";
-var 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"
-};
-var banditHead2 = `https://cdn.burtson.ai/images/bandit-head.png`;
-var resolveAvatar = (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}`;
-};
 var normalizeTables = (markdown2) => {
   const lines = markdown2.split("\n");
   const output = [];
@@ -12847,15 +12756,6 @@ var AIResponseTextField = ({
     const timeout3 = setTimeout(() => setShowMemoryUpdated(false), 3e3);
     return () => clearTimeout(timeout3);
   }, [memoryUpdated]);
-  const selectedModel = useModelStore((state) => state.selectedModel);
-  const [userAvatar, setUserAvatar] = (0, import_react14.useState)(banditHead2);
-  (0, import_react14.useEffect)(() => {
-    const fetchBranding = async () => {
-      const branding = await brandingService_default.getBranding();
-      setUserAvatar(branding?.logoBase64 || banditHead2);
-    };
-    fetchBranding();
-  }, []);
   const theme = (0, import_styles2.useTheme)();
   const chatResponse = theme.palette.chat?.response;
   const sanitizeMarkdown = (raw) => {
@@ -13224,283 +13124,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_runtime9.jsxs)(
-            import_material8.Box,
-            {
-              sx: {
-                display: "flex",
-                flexDirection: "column",
-                alignItems: {
-                  xs: "flex-start",
-                  sm: "flex-end"
+          /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(import_material8.Box, { sx: { display: "flex", justifyContent: "flex-end", mb: 2.5 }, children: /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)(import_material8.Box, { sx: { maxWidth: { xs: "100%", sm: "85%" }, minWidth: 0 }, children: [
+            /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
+              import_material8.Box,
+              {
+                sx: {
+                  bgcolor: chatResponse.userBubble || (0, import_styles2.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_runtime9.jsxs)(
-                  import_material8.Box,
+                children: /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.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_runtime9.jsx)(
-                        import_material8.Typography,
-                        {
-                          variant: "caption",
-                          sx: { color: chatResponse.modelLabel || "#888", mt: 1, fontStyle: "italic" },
-                          children: "You said"
-                        }
-                      )
-                    ]
+                    children: question
                   }
-                ),
-                /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)(import_material8.Box, { sx: { flex: 1 }, children: [
-                  /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.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_runtime9.jsx)(import_material8.Box, { sx: { display: "flex", gap: 1, mt: 1, flexWrap: "wrap" }, children: images.map((img, i) => /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.jsx)(import_material8.Box, { sx: { borderBottom: `1px solid ${theme.palette.divider}`, my: 2 } }),
-          /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.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_runtime9.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_runtime9.jsxs)(
-                  import_material8.Box,
-                  {
-                    sx: {
-                      display: "flex",
-                      flexDirection: "column",
-                      alignItems: "center",
-                      mr: 2,
-                      position: "relative"
-                    },
-                    children: [
-                      /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
-                        import_material8.Avatar,
-                        {
-                          src: resolveAvatar(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_runtime9.jsx)(import_material8.Box, { sx: { display: { xs: "none", sm: "block" } }, children: /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)(
-                        import_material8.Typography,
-                        {
-                          variant: "caption",
-                          sx: { color: chatResponse.modelLabel || "#888", mt: 1, fontStyle: "italic" },
-                          children: [
-                            selectedModel,
-                            " says"
-                          ]
-                        }
-                      ) })
-                    ]
-                  }
-                ),
-                /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
-                  import_material8.Box,
-                  {
-                    sx: {
-                      display: { xs: "flex", sm: "none" },
-                      justifyContent: "right",
-                      width: "100%",
-                      mt: -1,
-                      mb: 1
-                    },
-                    children: /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)(
-                      import_material8.Typography,
-                      {
-                        variant: "caption",
-                        sx: { fontStyle: "italic", color: chatResponse.modelLabel || "#888" },
-                        children: [
-                          selectedModel,
-                          " says"
-                        ]
-                      }
-                    )
-                  }
-                ),
-                /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)(import_material8.Box, { sx: { position: "relative", width: "100%" }, children: [
-                  cancelled && /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.jsx)(import_material8.Typography, { variant: "caption", sx: { fontStyle: "italic", opacity: 0.85 }, children: "Cancelled by you" })
-                    }
-                  ),
-                  showMemoryUpdated && /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.jsx)(
-                          "img",
-                          {
-                            src: brainIcon,
-                            alt: "Memory",
-                            style: { width: 18, height: 18 }
-                          }
-                        ),
-                        /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
-                          import_material8.Typography,
-                          {
-                            variant: "caption",
-                            sx: { color: chatResponse.memoryText || "#2e7d32", fontStyle: "italic" },
-                            children: "Bandit added to memory"
-                          }
-                        )
-                      ]
-                    }
-                  ),
+              i
+            )) })
+          ] }) }),
+          /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)(import_material8.Box, { sx: { position: "relative", width: "100%", maxWidth: { xs: "100%", sm: "768px" } }, children: [
+            cancelled && /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
+              import_material8.Box,
+              {
+                sx: {
+                  position: "absolute",
+                  top: -22,
+                  left: 0,
+                  display: "flex",
+                  alignItems: "center",
+                  gap: 1,
+                  zIndex: 1
+                },
+                children: /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(import_material8.Typography, { variant: "caption", sx: { fontStyle: "italic", opacity: 0.85 }, children: "Cancelled by you" })
+              }
+            ),
+            showMemoryUpdated && /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.jsx)("img", { src: brainIcon, alt: "Memory", style: { width: 16, height: 16 } }),
                   /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.jsx)(import_material8.Box, { sx: { width: "100%", maxWidth: "100%" }, children: typeof response === "string" ? /* @__PURE__ */ (0, import_jsx_runtime9.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_react14.default.isValidElement(response) ? response : /* @__PURE__ */ (0, import_jsx_runtime9.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_runtime9.jsx)(import_material8.Box, { sx: { width: "100%", maxWidth: "100%" }, children: typeof response === "string" ? /* @__PURE__ */ (0, import_jsx_runtime9.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_react14.default.isValidElement(response) ? response : /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(import_material8.Typography, { color: "error", children: "\u26A0\uFE0F Invalid AI response" }) })
+          ] }),
           !!(responseText || typeof response === "string" && response) && /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(ai_response_action_bar_default, { text: responseText || response }),
           displaySourceFiles && displaySourceFiles.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime9.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", {
@@ -18169,6 +17894,79 @@ var useNotificationService = () => {
 
 // src/modals/chat-modal/chat-modal.tsx
 init_aiProviderStore();
+
+// src/services/branding/brandingService.ts
+init_indexedDBService();
+init_debugLogger();
+var BrandingService = class {
+  DB_NAME = "banditConfig";
+  DB_VERSION = 1;
+  STORE_NAME = "config";
+  STORE_CONFIGS = [{ name: "config", keyPath: "id" }];
+  async getBranding() {
+    try {
+      const data = await indexedDBService_default.get(
+        this.DB_NAME,
+        this.DB_VERSION,
+        this.STORE_NAME,
+        "main",
+        this.STORE_CONFIGS
+      );
+      if (data?.branding && (data.branding.logoBase64 || data.branding.hasTransparentLogo !== void 0 || data.branding.brandingText)) {
+        return {
+          logoBase64: data.branding.logoBase64,
+          hasTransparentLogo: data.branding.hasTransparentLogo,
+          brandingText: data.branding.brandingText
+        };
+      }
+      return null;
+    } catch (error) {
+      throw error;
+    }
+  }
+  async setBrandingFromConfig(config) {
+    try {
+      const existing = await indexedDBService_default.get(
+        this.DB_NAME,
+        this.DB_VERSION,
+        this.STORE_NAME,
+        "main",
+        this.STORE_CONFIGS
+      ) || { id: "main" };
+      if (existing.branding?.userSaved) {
+        debugLogger.debug("\u{1F6E1}\uFE0F BrandingService: User branding detected, blocking CDN config override");
+        return;
+      }
+      const hasUserBranding = existing.branding && (existing.branding.logoBase64 || existing.branding.brandingText || existing.branding.theme || existing.branding.hasTransparentLogo !== void 0);
+      if (hasUserBranding) {
+        debugLogger.debug("\u{1F6E1}\uFE0F BrandingService: Detected user branding data, blocking CDN config override");
+        return;
+      }
+      debugLogger.debug("\u{1F527} BrandingService: No user branding detected, applying CDN config");
+      const branding = config.branding || {};
+      const updated = {
+        ...existing,
+        branding: {
+          ...existing.branding,
+          ...branding
+        }
+      };
+      await indexedDBService_default.put(
+        this.DB_NAME,
+        this.DB_VERSION,
+        this.STORE_NAME,
+        updated,
+        this.STORE_CONFIGS
+      );
+    } catch (error) {
+      throw error;
+    }
+  }
+};
+var brandingService = new BrandingService();
+var brandingService_default = brandingService;
+
+// src/modals/chat-modal/chat-modal.tsx
 init_debugLogger();
 
 // src/theme/banditTheme.ts
@@ -20540,7 +20338,7 @@ var FULL_SCREEN_THRESHOLD = 100;
 var MIN_WINDOWED_HEIGHT = 400;
 var MAX_WINDOWED_HEIGHT = 600;
 var CDN_BASE = "https://cdn.burtson.ai/";
-var banditHead3 = `${CDN_BASE}/images/bandit-head.png`;
+var banditHead2 = `${CDN_BASE}/images/bandit-head.png`;
 var modelAvatars = {
   "Bandit-Core": `${CDN_BASE}/avatars/core-avatar.png`,
   "Bandit-Muse": `${CDN_BASE}/avatars/muse-avatar.png`,
@@ -20630,7 +20428,7 @@ var ChatModal = ({
   const provider = useAIProviderStore((state) => state.provider);
   const notificationService2 = useNotificationService();
   const currentModel = availableModels.find((m) => m.name === selectedModel);
-  const currentAvatar = currentModel?.avatarBase64 || modelAvatars[selectedModel] || banditHead3;
+  const currentAvatar = currentModel?.avatarBase64 || modelAvatars[selectedModel] || banditHead2;
   const removeImage = (index) => {
     setPastedImages((prev) => prev.filter((_, i) => i !== index));
   };
@@ -20950,7 +20748,7 @@ var ChatModal = ({
                                   /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
                                     import_material17.Avatar,
                                     {
-                                      src: model.avatarBase64 || modelAvatars[model.name] || banditHead3,
+                                      src: model.avatarBase64 || modelAvatars[model.name] || banditHead2,
                                       alt: model.name,
                                       sx: {
                                         width: 28,