@burtson-labs/bandit-engine 2.0.61 → 2.0.62

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

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-G7U2FNUK.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-E23BSMK5.mjs.map
+//# sourceMappingURL=chat-SN3UNX23.mjs.map

package/dist/{chunk-SKMJ43NN.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-E23BSMK5.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-SKMJ43NN.mjs.map
+//# sourceMappingURL=chunk-O7JGT7HR.mjs.map

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

@@ -3169,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>");
@@ -3207,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 });
@@ -3255,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 = [];
@@ -7222,8 +7242,10 @@ var ChatAppBar = ({
   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";
   useEffect11(() => {
     useEngineStore.getState().fetchEngines();
   }, []);
@@ -7539,16 +7561,13 @@ var ChatAppBar = ({
                     )
                   }
                 ) }),
-                /* @__PURE__ */ jsx13(Tooltip4, { title: `Engine: ${currentEngine?.displayName ?? effectiveEngineId}`, arrow: true, children: /* @__PURE__ */ jsxs10(
+                /* @__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 ${effectiveEngineId}`,
-                    children: [
-                      currentEngine?.cloud ? /* @__PURE__ */ jsx13(CloudDoneIcon, { fontSize: "small" }) : /* @__PURE__ */ jsx13(CloudOffIcon, { fontSize: "small" }),
-                      /* @__PURE__ */ jsx13(Typography8, { variant: "caption", sx: { ml: 0.75, fontWeight: 600, whiteSpace: "nowrap" }, children: engineLabel })
-                    ]
+                    "aria-label": `Change base model (engine). Currently ${engineDisplay}`,
+                    children: /* @__PURE__ */ jsx13(PsychologyIcon, { fontSize: "small" })
                   }
                 ) }),
                 /* @__PURE__ */ jsxs10(
@@ -7572,7 +7591,7 @@ var ChatAppBar = ({
                         return /* @__PURE__ */ jsxs10(
                           MenuItem5,
                           {
-                            selected: engine.id === effectiveEngineId,
+                            selected: engine.id === resolvedEngineId,
                             disabled: !engine.available,
                             onClick: () => {
                               useEngineStore.getState().setSelectedEngine(engine.id);
@@ -7590,8 +7609,8 @@ var ChatAppBar = ({
                             },
                             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: engine.displayName }),
-                                engine.id === effectiveEngineId && /* @__PURE__ */ jsx13(Box10, { sx: { width: 8, height: 8, borderRadius: "50%", bgcolor: theme.palette.primary.main } })
+                                /* @__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(
@@ -9722,4 +9741,4 @@ var chat_default = Chat;
 export {
   chat_default
 };
-//# sourceMappingURL=chunk-G7U2FNUK.mjs.map
+//# sourceMappingURL=chunk-Y4NDNAAR.mjs.map