@burtson-labs/bandit-engine 2.0.61 → 2.0.62

package/dist/index.js

@@ -20977,6 +20977,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 +21016,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 +21069,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 = [];
@@ -25010,8 +25030,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_react31.useEffect)(() => {
         useEngineStore.getState().fetchEngines();
       }, []);
@@ -25327,16 +25349,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)(PsychologyIcon, { fontSize: "small" })
                       }
                     ) }),
                     /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(
@@ -25360,7 +25379,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 +25397,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)(