@burtson-labs/bandit-engine 2.0.60 → 2.0.61

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

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-MFDMM5MS.mjs";
+} from "./chunk-G7U2FNUK.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-E23BSMK5.mjs.map

package/dist/{chunk-MFDMM5MS.mjs → chunk-G7U2FNUK.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);
@@ -7049,6 +7098,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 +7219,14 @@ 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);
+  const engineLabel = currentEngine?.displayName?.replace(/^Bandit /, "") || "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 +7539,84 @@ var ChatAppBar = ({
                     )
                   }
                 ) }),
+                /* @__PURE__ */ jsx13(Tooltip4, { title: `Engine: ${currentEngine?.displayName ?? effectiveEngineId}`, arrow: true, children: /* @__PURE__ */ jsxs10(
+                  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 })
+                    ]
+                  }
+                ) }),
+                /* @__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 === effectiveEngineId,
+                            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: engine.displayName }),
+                                engine.id === effectiveEngineId && /* @__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 +9722,4 @@ var chat_default = Chat;
 export {
   chat_default
 };
-//# sourceMappingURL=chunk-MFDMM5MS.mjs.map
+//# sourceMappingURL=chunk-G7U2FNUK.mjs.map