@burtson-labs/bandit-engine 2.0.61 → 2.0.62

package/dist/index.mjs

@@ -1,22 +1,22 @@
 import {
   chat_default
-} from "./chunk-G7U2FNUK.mjs";
+} from "./chunk-Y4NDNAAR.mjs";
 import {
   chat_provider_default
 } from "./chunk-D55E6ZDV.mjs";
 import "./chunk-ONQMRE2G.mjs";
-import {
-  defineCustomElement
-} from "./chunk-IXIM7BNO.mjs";
 import {
   management_default,
   useGatewayHealth,
   useGatewayMemory,
   useGatewayModels
-} from "./chunk-SKMJ43NN.mjs";
+} from "./chunk-O7JGT7HR.mjs";
 import "./chunk-SRCCNBHF.mjs";
 import "./chunk-VTC6AIWY.mjs";
 import "./chunk-6QTTNYF2.mjs";
+import {
+  defineCustomElement
+} from "./chunk-IXIM7BNO.mjs";
 import {
   chat_modal_default
 } from "./chunk-PY7A3J5T.mjs";

package/dist/management/management.js

@@ -23216,6 +23216,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>");
@@ -23254,6 +23255,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 });
@@ -23302,6 +23308,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 = [];
@@ -27249,8 +27269,10 @@ var init_chat_app_bar = __esm({
       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_react53.useEffect)(() => {
         useEngineStore.getState().fetchEngines();
       }, []);
@@ -27566,16 +27588,13 @@ var init_chat_app_bar = __esm({
                         )
                       }
                     ) }),
-                    /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Tooltip, { title: `Engine: ${currentEngine?.displayName ?? effectiveEngineId}`, arrow: true, children: /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(
+                    /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Tooltip, { title: `Engine \xB7 ${engineDisplay}`, arrow: true, children: /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(
                       import_material43.IconButton,
                       {
                         onClick: (e) => setEngineAnchorEl(e.currentTarget),
                         sx: pillButtonStyles,
-                        "aria-label": `Change base model (engine). Currently ${effectiveEngineId}`,
-                        children: [
-                          currentEngine?.cloud ? /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(CloudDoneIcon, { fontSize: "small" }) : /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(CloudOffIcon, { fontSize: "small" }),
-                          /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.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_runtime42.jsx)(PsychologyIcon, { fontSize: "small" })
                       }
                     ) }),
                     /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(
@@ -27599,7 +27618,7 @@ var init_chat_app_bar = __esm({
                             return /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(
                               import_material43.MenuItem,
                               {
-                                selected: engine.id === effectiveEngineId,
+                                selected: engine.id === resolvedEngineId,
                                 disabled: !engine.available,
                                 onClick: () => {
                                   useEngineStore.getState().setSelectedEngine(engine.id);
@@ -27617,8 +27636,8 @@ var init_chat_app_bar = __esm({
                                 },
                                 children: [
                                   /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(import_material43.Box, { sx: { display: "flex", alignItems: "center", gap: 1, width: "100%" }, children: [
-                                    /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Typography, { variant: "body2", sx: { fontWeight: 600, flex: 1 }, children: engine.displayName }),
-                                    engine.id === effectiveEngineId && /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Box, { sx: { width: 8, height: 8, borderRadius: "50%", bgcolor: theme.palette.primary.main } })
+                                    /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Typography, { variant: "body2", sx: { fontWeight: 600, flex: 1 }, children: cleanEngineName(engine.displayName) }),
+                                    engine.id === resolvedEngineId && /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Box, { sx: { width: 8, height: 8, borderRadius: "50%", bgcolor: theme.palette.primary.main } })
                                   ] }),
                                   /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Typography, { variant: "caption", sx: { color: theme.palette.text.secondary }, children: engine.available ? engine.description : engine.unavailableReason || "Unavailable" }),
                                   badges.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Box, { sx: { display: "flex", gap: 0.5, flexWrap: "wrap", mt: 0.25 }, children: badges.map((b) => /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(