@burtson-labs/bandit-engine 2.0.60 → 2.0.62

package/dist/index.js

@@ -19435,6 +19435,64 @@ var init_telemetry = __esm({
   }
 });
 
+// src/store/engineStore.ts
+var import_zustand15, STORAGE_KEY2, readStored, useEngineStore;
+var init_engineStore = __esm({
+  "src/store/engineStore.ts"() {
+    "use strict";
+    import_zustand15 = require("zustand");
+    init_packageSettingsStore();
+    init_authenticationService();
+    init_debugLogger();
+    STORAGE_KEY2 = "bandit.selectedEngine";
+    readStored = () => {
+      try {
+        return typeof window !== "undefined" ? window.localStorage.getItem(STORAGE_KEY2) : null;
+      } catch {
+        return null;
+      }
+    };
+    useEngineStore = (0, import_zustand15.create)((set, get) => ({
+      selectedEngine: readStored(),
+      engines: [],
+      loaded: false,
+      setSelectedEngine: (id) => {
+        set({ selectedEngine: id });
+        try {
+          window.localStorage.setItem(STORAGE_KEY2, 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
 var import_rxjs20, MEMORY_LIMIT, MIN_MEMORY_WORDS, MERGE_THRESHOLD, REJECT_ECHO_THRESHOLD, REJECT_DUPLICATE_THRESHOLD, CONTEXTUAL_DIVERGENCE_THRESHOLD, normalizeText, isStructurallyDuplicate, isAboutBandit, hasEngagementValue, isMemoryTooShortOrGeneric, isPersonalText, mergeMemory, isVoiceShifted, sanitizeMemory, sanitizeMemoryText, shouldAcceptMemory, isContextuallyDivergent, useMemoryEnhancer;
 var init_useMemoryEnhancer = __esm({
@@ -20177,12 +20235,12 @@ var init_useAIProvider = __esm({
     init_knowledgeStore();
     init_aiProviderStore();
     init_telemetry();
+    init_engineStore();
     init_conversationStore();
     init_useMemoryEnhancer();
     init_useVectorStore();
     init_embeddingService();
     init_useMoodEngine();
-    init_packageSettingsStore();
     init_prompts();
     init_preferencesStore();
     init_mcp();
@@ -20546,7 +20604,7 @@ var init_useAIProvider = __esm({
             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);
@@ -20919,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>");
@@ -20957,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 });
@@ -21005,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 = [];
@@ -24778,6 +24856,7 @@ var init_chat_app_bar = __esm({
     init_packageSettingsStore();
     init_useFeatures();
     init_conversationSyncStore();
+    init_engineStore();
     import_shallow2 = require("zustand/shallow");
     import_jsx_runtime24 = require("react/jsx-runtime");
     CDN_BASE = "https://cdn.burtson.ai/";
@@ -24827,6 +24906,7 @@ var init_chat_app_bar = __esm({
         menuText
       } = theme.palette.chat.appBar;
       const [modelAnchorEl, setModelAnchorEl] = (0, import_react31.useState)(null);
+      const [engineAnchorEl, setEngineAnchorEl] = (0, import_react31.useState)(null);
       const [voiceAnchorEl, setVoiceAnchorEl] = (0, import_react31.useState)(null);
       const [modalOpen, setModalOpen] = (0, import_react31.useState)(false);
       const [confirmModelChangeOpen, setConfirmModelChangeOpen] = (0, import_react31.useState)(false);
@@ -24947,6 +25027,16 @@ var init_chat_app_bar = __esm({
       const selectedModel = useModelStore((s) => s.selectedModel);
       const currentModel = useModelStore((s) => s.availableModels.find((m) => m.name === selectedModel));
       const currentAvatar = currentModel?.avatarBase64 || modelAvatars2[selectedModel] || banditHead3;
+      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";
+      (0, import_react31.useEffect)(() => {
+        useEngineStore.getState().fetchEngines();
+      }, []);
       const pendingModelAvatar = useModelStore.getState().availableModels.find((m) => m.name === pendingModel)?.avatarBase64 || modelAvatars2[pendingModel || ""] || banditHead3;
       const resolvedHomeUrl = preferences.homeUrl?.trim() || packageSettings?.homeUrl?.trim() || "";
       const homeTooltip = (() => {
@@ -25259,6 +25349,81 @@ var init_chat_app_bar = __esm({
                         )
                       }
                     ) }),
+                    /* @__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 ${engineDisplay}`,
+                        children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(PsychologyIcon, { fontSize: "small" })
+                      }
+                    ) }),
+                    /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(
+                      import_material23.Menu,
+                      {
+                        anchorEl: engineAnchorEl,
+                        open: Boolean(engineAnchorEl),
+                        onClose: () => setEngineAnchorEl(null),
+                        transformOrigin: { horizontal: "right", vertical: "top" },
+                        anchorOrigin: { horizontal: "right", vertical: "bottom" },
+                        children: [
+                          /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Typography, { variant: "overline", sx: { px: 2, color: theme.palette.text.secondary }, children: "Engine \xB7 base model" }),
+                          engines.length === 0 && /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.MenuItem, { disabled: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_material23.Typography, { 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__ */ (0, import_jsx_runtime24.jsxs)(
+                              import_material23.MenuItem,
+                              {
+                                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__ */ (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: 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)(
+                                    import_material23.Box,
+                                    {
+                                      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__ */ (0, import_jsx_runtime24.jsx)(
                       import_material23.Menu,
                       {