@burtson-labs/bandit-engine 2.0.60 → 2.0.62

package/dist/{chat-NLBCURUN.mjs → chat-SN3UNX23.mjs}

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-MFDMM5MS.mjs";
+} from "./chunk-Y4NDNAAR.mjs";
 import "./chunk-ONQMRE2G.mjs";
 import "./chunk-SRCCNBHF.mjs";
 import "./chunk-6QTTNYF2.mjs";
@@ -13,4 +13,4 @@ import "./chunk-BJTO5JO5.mjs";
 export {
   chat_default as default
 };
-//# sourceMappingURL=chat-NLBCURUN.mjs.map
+//# sourceMappingURL=chat-SN3UNX23.mjs.map

package/dist/{chunk-3AWAL2YH.mjs → chunk-O7JGT7HR.mjs}

@@ -9459,7 +9459,7 @@ var MCPToolsTabV2_default = MCPToolsTabV2;
 
 // src/management/management.tsx
 import { jsx as jsx13, jsxs as jsxs13 } from "react/jsx-runtime";
-var preloadChatPage = () => import("./chat-NLBCURUN.mjs");
+var preloadChatPage = () => import("./chat-SN3UNX23.mjs");
 var buildCapabilitiesUrl = (gatewayApiUrl) => {
   const trimmed = gatewayApiUrl.replace(/\/$/, "");
   if (trimmed.endsWith("/api")) {
@@ -10804,4 +10804,4 @@ export {
   useGatewayMemory,
   management_default
 };
-//# sourceMappingURL=chunk-3AWAL2YH.mjs.map
+//# sourceMappingURL=chunk-O7JGT7HR.mjs.map

package/dist/{chunk-MFDMM5MS.mjs → chunk-Y4NDNAAR.mjs}

@@ -1695,6 +1695,55 @@ function telemetryEndTurn(outcome) {
   void active?.endTurn(outcome);
 }
 
+// src/store/engineStore.ts
+import { create as create2 } from "zustand";
+var STORAGE_KEY = "bandit.selectedEngine";
+var readStored = () => {
+  try {
+    return typeof window !== "undefined" ? window.localStorage.getItem(STORAGE_KEY) : null;
+  } catch {
+    return null;
+  }
+};
+var useEngineStore = create2((set, get) => ({
+  selectedEngine: readStored(),
+  engines: [],
+  loaded: false,
+  setSelectedEngine: (id) => {
+    set({ selectedEngine: id });
+    try {
+      window.localStorage.setItem(STORAGE_KEY, id);
+    } catch {
+    }
+  },
+  getSelectedEngine: () => get().selectedEngine || usePackageSettingsStore.getState().settings?.defaultModel || "bandit-core",
+  fetchEngines: async () => {
+    const settings = usePackageSettingsStore.getState().settings;
+    const base = settings?.gatewayApiUrl?.replace(/\/$/, "") ?? "";
+    if (!base || settings?.playgroundMode || base.toLowerCase().startsWith("playground://")) {
+      set({ loaded: true });
+      return;
+    }
+    try {
+      const headers = { "Content-Type": "application/json" };
+      const token = authenticationService.getToken();
+      if (token) headers["Authorization"] = `Bearer ${token}`;
+      const res = await fetch(`${base}/models`, { headers });
+      const data = await res.json();
+      if (res.ok && Array.isArray(data?.models)) {
+        set({ engines: data.models, loaded: true });
+      } else {
+        set({ loaded: true });
+      }
+    } catch (error) {
+      debugLogger.error("Failed to fetch engines", {
+        error: error instanceof Error ? error.message : String(error)
+      });
+      set({ loaded: true });
+    }
+  }
+}));
+
 // src/chat/hooks/useMemoryEnhancer.tsx
 import { lastValueFrom, map as map2 } from "rxjs";
 var MEMORY_LIMIT = 100;
@@ -2747,7 +2796,7 @@ var useAIProvider = ({
         question: pendingQuestion,
         images: pendingImages
       });
-      const modelName = usePackageSettingsStore.getState().settings?.defaultModel || "bandit-core:4b-it-qat";
+      const modelName = useEngineStore.getState().getSelectedEngine();
       const CONFIG = modelConfigs[modelName] ?? modelConfigs["bandit-core:4b-it-qat"];
       const base64Images = imageList.map((img) => img.split(",")[1]);
       const latestEntries = history.slice(-CONFIG.historyMessages);
@@ -3120,6 +3169,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>");
@@ -3158,6 +3208,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 });
@@ -3206,6 +3261,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 = [];
@@ -7049,6 +7118,7 @@ var ChatAppBar = ({
     menuText
   } = theme.palette.chat.appBar;
   const [modelAnchorEl, setModelAnchorEl] = useState12(null);
+  const [engineAnchorEl, setEngineAnchorEl] = useState12(null);
   const [voiceAnchorEl, setVoiceAnchorEl] = useState12(null);
   const [modalOpen, setModalOpen] = useState12(false);
   const [confirmModelChangeOpen, setConfirmModelChangeOpen] = useState12(false);
@@ -7169,6 +7239,16 @@ var ChatAppBar = ({
   const selectedModel = useModelStore((s) => s.selectedModel);
   const currentModel = useModelStore((s) => s.availableModels.find((m) => m.name === selectedModel));
   const currentAvatar = currentModel?.avatarBase64 || modelAvatars[selectedModel] || banditHead;
+  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) || 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";
+  useEffect11(() => {
+    useEngineStore.getState().fetchEngines();
+  }, []);
   const pendingModelAvatar = useModelStore.getState().availableModels.find((m) => m.name === pendingModel)?.avatarBase64 || modelAvatars[pendingModel || ""] || banditHead;
   const resolvedHomeUrl = preferences.homeUrl?.trim() || packageSettings?.homeUrl?.trim() || "";
   const homeTooltip = (() => {
@@ -7481,6 +7561,81 @@ var ChatAppBar = ({
                     )
                   }
                 ) }),
+                /* @__PURE__ */ jsx13(Tooltip4, { title: `Engine \xB7 ${engineDisplay}`, arrow: true, children: /* @__PURE__ */ jsx13(
+                  IconButton9,
+                  {
+                    onClick: (e) => setEngineAnchorEl(e.currentTarget),
+                    sx: pillButtonStyles,
+                    "aria-label": `Change base model (engine). Currently ${engineDisplay}`,
+                    children: /* @__PURE__ */ jsx13(PsychologyIcon, { fontSize: "small" })
+                  }
+                ) }),
+                /* @__PURE__ */ jsxs10(
+                  Menu5,
+                  {
+                    anchorEl: engineAnchorEl,
+                    open: Boolean(engineAnchorEl),
+                    onClose: () => setEngineAnchorEl(null),
+                    transformOrigin: { horizontal: "right", vertical: "top" },
+                    anchorOrigin: { horizontal: "right", vertical: "bottom" },
+                    children: [
+                      /* @__PURE__ */ jsx13(Typography8, { variant: "overline", sx: { px: 2, color: theme.palette.text.secondary }, children: "Engine \xB7 base model" }),
+                      engines.length === 0 && /* @__PURE__ */ jsx13(MenuItem5, { disabled: true, children: /* @__PURE__ */ jsx13(Typography8, { variant: "body2", children: "No engines available" }) }),
+                      engines.map((engine) => {
+                        const badges = [
+                          engine.vision && "vision",
+                          engine.tools && "tools",
+                          engine.thinking && "thinking",
+                          engine.cloud && "cloud"
+                        ].filter(Boolean);
+                        return /* @__PURE__ */ jsxs10(
+                          MenuItem5,
+                          {
+                            selected: engine.id === resolvedEngineId,
+                            disabled: !engine.available,
+                            onClick: () => {
+                              useEngineStore.getState().setSelectedEngine(engine.id);
+                              setEngineAnchorEl(null);
+                            },
+                            sx: {
+                              display: "flex",
+                              flexDirection: "column",
+                              alignItems: "flex-start",
+                              gap: 0.5,
+                              py: 1,
+                              px: 2,
+                              maxWidth: 360,
+                              whiteSpace: "normal"
+                            },
+                            children: [
+                              /* @__PURE__ */ jsxs10(Box10, { sx: { display: "flex", alignItems: "center", gap: 1, width: "100%" }, children: [
+                                /* @__PURE__ */ jsx13(Typography8, { variant: "body2", sx: { fontWeight: 600, flex: 1 }, children: cleanEngineName(engine.displayName) }),
+                                engine.id === resolvedEngineId && /* @__PURE__ */ jsx13(Box10, { sx: { width: 8, height: 8, borderRadius: "50%", bgcolor: theme.palette.primary.main } })
+                              ] }),
+                              /* @__PURE__ */ jsx13(Typography8, { variant: "caption", sx: { color: theme.palette.text.secondary }, children: engine.available ? engine.description : engine.unavailableReason || "Unavailable" }),
+                              badges.length > 0 && /* @__PURE__ */ jsx13(Box10, { sx: { display: "flex", gap: 0.5, flexWrap: "wrap", mt: 0.25 }, children: badges.map((b) => /* @__PURE__ */ jsx13(
+                                Box10,
+                                {
+                                  sx: {
+                                    fontSize: "0.65rem",
+                                    px: 0.75,
+                                    py: 0.1,
+                                    borderRadius: 1,
+                                    bgcolor: theme.palette.primary.main + "22",
+                                    color: theme.palette.primary.main
+                                  },
+                                  children: b
+                                },
+                                b
+                              )) })
+                            ]
+                          },
+                          engine.id
+                        );
+                      })
+                    ]
+                  }
+                ),
                 /* @__PURE__ */ jsx13(
                   Menu5,
                   {
@@ -9586,4 +9741,4 @@ var chat_default = Chat;
 export {
   chat_default
 };
-//# sourceMappingURL=chunk-MFDMM5MS.mjs.map
+//# sourceMappingURL=chunk-Y4NDNAAR.mjs.map