@burtson-labs/bandit-engine 2.0.59 → 2.0.61

package/dist/management/management.js

@@ -19448,7 +19448,11 @@ ${listMarkdown}`;
           href,
           target: "_blank",
           rel: "noopener noreferrer",
-          style: { color: theme.palette.primary.main, textDecoration: "underline" },
+          style: {
+            color: theme.palette.info?.main ?? theme.palette.primary.main,
+            textDecoration: "underline",
+            textUnderlineOffset: "2px"
+          },
           ...props,
           children
         }
@@ -21346,6 +21350,388 @@ ${sanitize(
   }
 });
 
+// src/services/telemetry/otlpExporter.ts
+function redactSecretsString(s) {
+  return s.replace(/\b(?:sk|tvly|ghp|gho|pk|rk)[-_][A-Za-z0-9]{8,}\b/gi, "[redacted]").replace(/\beyJ[A-Za-z0-9_-]{8,}\.[A-Za-z0-9_-]+\.[A-Za-z0-9_-]+/g, "[redacted-jwt]").replace(/\bBearer\s+[A-Za-z0-9._-]{8,}/gi, "Bearer [redacted]");
+}
+function resolveTelemetryConfig(opts) {
+  if (!opts.telemetry?.enabled) return null;
+  const endpoint = (opts.telemetry.endpoint ?? "https://otlp.burtson.ai").replace(/\/+$/, "");
+  const headers = { ...opts.telemetry.headers ?? {} };
+  const hasAuth = Object.keys(headers).some((k) => k.toLowerCase() === "authorization");
+  if (!hasAuth && opts.banditApiKey) {
+    headers["Authorization"] = `Bearer ${opts.banditApiKey}`;
+  }
+  const mode = opts.telemetry.mode ?? "metrics+traces";
+  return { endpoint, headers, mode, serviceName: opts.telemetry.serviceName ?? "bandit-web" };
+}
+function toAttrs(rec) {
+  const out = [];
+  for (const [key, v] of Object.entries(rec)) {
+    if (v === void 0 || v === null || v === "") continue;
+    if (typeof v === "boolean") out.push({ key, value: { boolValue: v } });
+    else if (typeof v === "number")
+      out.push({ key, value: Number.isInteger(v) ? { intValue: String(v) } : { doubleValue: v } });
+    else out.push({ key, value: { stringValue: v } });
+  }
+  return out;
+}
+function hex(bytes) {
+  const arr = new Uint8Array(bytes);
+  if (webCrypto?.getRandomValues) webCrypto.getRandomValues(arr);
+  return Array.from(arr, (b) => b.toString(16).padStart(2, "0")).join("");
+}
+function histogramPoint(value, bounds, attrs, startMs, endMs) {
+  const counts = new Array(bounds.length + 1).fill(0);
+  let idx = bounds.findIndex((b) => value <= b);
+  if (idx === -1) idx = bounds.length;
+  counts[idx] = 1;
+  return {
+    attributes: toAttrs(attrs),
+    startTimeUnixNano: nano(startMs),
+    timeUnixNano: nano(endMs),
+    count: "1",
+    sum: value,
+    bucketCounts: counts.map(String),
+    explicitBounds: bounds
+  };
+}
+function sumPoint(value, attrs, startMs, endMs) {
+  return {
+    attributes: toAttrs(attrs),
+    startTimeUnixNano: nano(startMs),
+    timeUnixNano: nano(endMs),
+    asInt: String(Math.round(value))
+  };
+}
+var webCrypto, nano, TTFT_BUCKETS, DURATION_BUCKETS, clip, TelemetryExporter;
+var init_otlpExporter = __esm({
+  "src/services/telemetry/otlpExporter.ts"() {
+    "use strict";
+    init_debugLogger();
+    webCrypto = globalThis.crypto;
+    nano = (ms) => String(Math.round(ms * 1e6));
+    TTFT_BUCKETS = [0.1, 0.25, 0.5, 1, 2, 5, 10, 30];
+    DURATION_BUCKETS = [0.5, 1, 2, 5, 10, 30, 60, 120, 300];
+    clip = (s, n = 120) => redactSecretsString(s.slice(0, n)).slice(0, n);
+    TelemetryExporter = class {
+      cfg;
+      now;
+      traceId = "";
+      turn = null;
+      llm = null;
+      llmFirstChunkMs = 0;
+      openTools = [];
+      completedSpans = [];
+      model = "";
+      turnChunkChars = 0;
+      turnTokens = 0;
+      ttftSeconds = null;
+      constructor(cfg, opts) {
+        this.cfg = cfg;
+        this.now = opts?.now ?? (() => Date.now());
+      }
+      startTurn(goal, model) {
+        this.traceId = hex(16);
+        this.model = model;
+        this.turnChunkChars = 0;
+        this.turnTokens = 0;
+        this.ttftSeconds = null;
+        this.llm = null;
+        this.llmFirstChunkMs = 0;
+        this.openTools = [];
+        this.completedSpans = [];
+        this.turn = {
+          spanId: hex(8),
+          name: "agent.turn",
+          startMs: this.now(),
+          attrs: { "gen_ai.request.model": model, "bandit.turn.goal": clip(goal, 160) }
+        };
+      }
+      /** Fed from the chat turn lifecycle. Best-effort; swallows bad payloads. */
+      onEvent(type, payload) {
+        if (!this.turn) return;
+        try {
+          const p = payload ?? {};
+          switch (type) {
+            case "tool_loop:llm_start":
+              this.llm = {
+                spanId: hex(8),
+                parentSpanId: this.turn.spanId,
+                name: "llm.generate",
+                startMs: this.now(),
+                attrs: { "gen_ai.request.model": this.model }
+              };
+              this.llmFirstChunkMs = 0;
+              break;
+            case "tool_loop:llm_chunk": {
+              const chunk = typeof p.chunk === "string" ? p.chunk : "";
+              if (this.llm && this.llmFirstChunkMs === 0 && chunk.length > 0) {
+                this.llmFirstChunkMs = this.now();
+                const ttft = (this.llmFirstChunkMs - this.llm.startMs) / 1e3;
+                if (this.ttftSeconds === null) this.ttftSeconds = ttft;
+                this.llm.attrs["bandit.llm.ttft_seconds"] = ttft;
+              }
+              this.turnChunkChars += chunk.length;
+              this.turnTokens = Math.floor(this.turnChunkChars / 4);
+              break;
+            }
+            case "tool_loop:llm_response":
+              if (this.llm) {
+                this.llm.endMs = this.now();
+                if (typeof p.responseLength === "number") this.llm.attrs["bandit.llm.response_chars"] = p.responseLength;
+                this.completedSpans.push(this.llm);
+                this.llm = null;
+              }
+              break;
+            case "tool_loop:tool_execute": {
+              const name = typeof p.name === "string" ? p.name : "tool";
+              const params = p.params ?? {};
+              const primary = params.query ?? params.url ?? params.prompt ?? params.topic ?? "";
+              this.openTools.push({
+                spanId: hex(8),
+                parentSpanId: this.turn.spanId,
+                name: `tool.${name}`,
+                startMs: this.now(),
+                attrs: { "bandit.tool.name": name, "bandit.tool.primary": primary ? clip(primary) : void 0 }
+              });
+              break;
+            }
+            case "tool_loop:tool_result":
+            case "tool_loop:tool_error": {
+              const name = typeof p.name === "string" ? p.name : void 0;
+              const span = this.takeOpenTool(name);
+              if (span) {
+                span.endMs = this.now();
+                if (type === "tool_loop:tool_error" || p.isError === true) span.error = "tool error";
+                this.completedSpans.push(span);
+              }
+              break;
+            }
+          }
+        } catch {
+        }
+      }
+      takeOpenTool(name) {
+        if (name) {
+          for (let i = this.openTools.length - 1; i >= 0; i -= 1) {
+            if (this.openTools[i].name === `tool.${name}`) return this.openTools.splice(i, 1)[0];
+          }
+        }
+        return this.openTools.shift();
+      }
+      /** Close the turn, build OTLP traces + metrics, and flush. Never rejects. */
+      async endTurn(outcome) {
+        const turn = this.turn;
+        if (!turn) return;
+        this.turn = null;
+        const end = this.now();
+        if (this.llm && !this.llm.endMs) {
+          this.llm.endMs = end;
+          this.completedSpans.push(this.llm);
+          this.llm = null;
+        }
+        for (const t of this.openTools.splice(0)) {
+          t.endMs = end;
+          t.error = t.error ?? "incomplete";
+          this.completedSpans.push(t);
+        }
+        turn.endMs = end;
+        if (outcome?.error) turn.error = outcome.error;
+        const traceId = this.traceId;
+        const spans = [turn, ...this.completedSpans];
+        const jobs = [];
+        if (this.cfg.mode === "metrics+traces") jobs.push(this.post("/v1/traces", this.buildTraces(traceId, spans)));
+        jobs.push(this.post("/v1/metrics", this.buildMetrics(turn)));
+        try {
+          await Promise.all(jobs);
+        } catch {
+        }
+      }
+      buildTraces(traceId, spans) {
+        return {
+          resourceSpans: [
+            {
+              resource: { attributes: toAttrs({ "service.name": this.cfg.serviceName }) },
+              scopeSpans: [
+                {
+                  scope: { name: this.cfg.serviceName },
+                  spans: spans.map((s) => ({
+                    traceId,
+                    spanId: s.spanId,
+                    parentSpanId: s.parentSpanId,
+                    name: s.name,
+                    kind: 1,
+                    startTimeUnixNano: nano(s.startMs),
+                    endTimeUnixNano: nano(s.endMs ?? s.startMs),
+                    attributes: toAttrs(s.attrs),
+                    status: s.error ? { code: 2, message: s.error.slice(0, 200) } : { code: 1 }
+                  }))
+                }
+              ]
+            }
+          ]
+        };
+      }
+      buildMetrics(turn) {
+        const start = turn.startMs;
+        const end = turn.endMs ?? this.now();
+        const metrics = [];
+        if (this.turnTokens > 0) {
+          metrics.push({
+            name: "bandit.llm.tokens",
+            sum: {
+              aggregationTemporality: 1,
+              isMonotonic: true,
+              dataPoints: [sumPoint(this.turnTokens, { type: "output", "gen_ai.request.model": this.model }, start, end)]
+            }
+          });
+        }
+        if (this.ttftSeconds !== null) {
+          metrics.push({
+            name: "bandit.llm.ttft",
+            unit: "s",
+            histogram: {
+              aggregationTemporality: 1,
+              dataPoints: [histogramPoint(this.ttftSeconds, TTFT_BUCKETS, { "gen_ai.request.model": this.model }, start, end)]
+            }
+          });
+        }
+        metrics.push({
+          name: "bandit.turn.duration",
+          unit: "s",
+          histogram: {
+            aggregationTemporality: 1,
+            dataPoints: [histogramPoint((end - start) / 1e3, DURATION_BUCKETS, { "gen_ai.request.model": this.model }, start, end)]
+          }
+        });
+        return {
+          resourceMetrics: [
+            {
+              resource: { attributes: toAttrs({ "service.name": this.cfg.serviceName }) },
+              scopeMetrics: [{ scope: { name: this.cfg.serviceName }, metrics }]
+            }
+          ]
+        };
+      }
+      async post(path, body) {
+        const doFetch = globalThis.fetch;
+        if (!doFetch) return;
+        const ctrl = new AbortController();
+        const timer = setTimeout(() => ctrl.abort(), 4e3);
+        try {
+          await doFetch(`${this.cfg.endpoint}${path}`, {
+            method: "POST",
+            headers: { "Content-Type": "application/json", ...this.cfg.headers },
+            body: JSON.stringify(body),
+            signal: ctrl.signal
+          });
+        } catch (e) {
+          debugLogger.debug("[telemetry] OTLP post failed", {
+            path,
+            error: e instanceof Error ? e.message : String(e)
+          });
+        } finally {
+          clearTimeout(timer);
+        }
+      }
+    };
+  }
+});
+
+// src/services/telemetry/index.ts
+function syncTelemetry() {
+  try {
+    const settings = usePackageSettingsStore.getState().settings;
+    const cfg = resolveTelemetryConfig({
+      telemetry: settings?.telemetry,
+      banditApiKey: authenticationService.getToken() ?? void 0
+    });
+    active = cfg ? new TelemetryExporter(cfg) : null;
+  } catch {
+    active = null;
+  }
+  return active !== null;
+}
+function telemetryStartTurn(goal, model) {
+  active?.startTurn(goal, model);
+}
+function telemetryEvent(type, payload) {
+  active?.onEvent(type, payload);
+}
+function telemetryEndTurn(outcome) {
+  void active?.endTurn(outcome);
+}
+var active;
+var init_telemetry = __esm({
+  "src/services/telemetry/index.ts"() {
+    "use strict";
+    init_otlpExporter();
+    init_packageSettingsStore();
+    init_authenticationService();
+    init_otlpExporter();
+    active = null;
+  }
+});
+
+// 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_rxjs23, 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({
@@ -22087,12 +22473,13 @@ var init_useAIProvider = __esm({
     import_react44 = require("react");
     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();
@@ -22456,7 +22843,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);
@@ -22851,6 +23238,9 @@ ${protocol}`;
               setStreamBuffer(latestDisplayMessage);
             }, delay);
           };
+          syncTelemetry();
+          telemetryStartTurn(question, modelName);
+          telemetryEvent("tool_loop:llm_start");
           const stream = provider.chat(request);
           const initialPlaceholderQuestion = lastEntry?.question;
           lastPartialRef.current = { text: "", images: [...imageList], usedDocs, question };
@@ -22864,7 +23254,10 @@ ${protocol}`;
           const sub = stream.subscribe({
             next: (data) => {
               if (!data?.message?.content && !data?.message?.tool_calls) return;
-              if (data.message.content) fullMessage += data.message.content;
+              if (data.message.content) {
+                fullMessage += data.message.content;
+                telemetryEvent("tool_loop:llm_chunk", { chunk: data.message.content });
+              }
               const inThinkBlock = /<think>/.test(fullMessage) && !/<think>[\s\S]*<\/think>/.test(fullMessage);
               setIsThinking?.(inThinkBlock);
               const visibleMessage = stripThinking(fullMessage);
@@ -22896,6 +23289,7 @@ ${protocol}`;
               setIsThinking?.(false);
               setPendingMessage(null);
               setLogoVisible(false);
+              telemetryEndTurn({ error: err?.message || "stream error" });
               if (onError) {
                 onError(err);
               }
@@ -22903,6 +23297,7 @@ ${protocol}`;
             complete: async () => {
               try {
                 setIsThinking?.(false);
+                telemetryEvent("tool_loop:llm_response", { responseLength: fullMessage.length });
                 latestDisplayMessage = stripThinking(fullMessage);
                 if (!sawToolBlock) {
                   flushNow();
@@ -22911,6 +23306,7 @@ ${protocol}`;
                 let enhancedMessage = fullMessage;
                 const summarizableResults = [];
                 const inlineImageBlocks = [];
+                const collectedSources = [];
                 if (toolCallMatches && toolCallMatches.length > 0 && mcpToolsAvailable) {
                   debugLogger.info("Detected tool calls in AI response", {
                     toolCallCount: toolCallMatches.length,
@@ -22945,10 +23341,21 @@ ${protocol}`;
                       });
                       const placeholderToken = `<<TOOL_LOADING_${functionName}_${Math.random().toString(36).slice(2)}>>`;
                       enhancedMessage = enhancedMessage.replace(match, placeholderToken);
+                      clearFlushTimer();
+                      const toolStatus = functionName === "web_search" || functionName === "web-search" ? "Searching the web\u2026" : functionName === "web_fetch" || functionName === "web-fetch" ? "Reading the page\u2026" : functionName === "image_generation" || functionName === "image-generation" ? "Generating the image\u2026" : "Working on it\u2026";
+                      const toolPreamble = stripToolBlocks(fullMessage).trim();
+                      setStreamBuffer(toolPreamble ? `${toolPreamble}
+
+_${toolStatus}_` : `_${toolStatus}_`);
+                      telemetryEvent("tool_loop:tool_execute", { name: functionName, params: parsedParams });
                       const result = await executeMCPTool({
                         toolName: functionName,
                         parameters: parsedParams
                       });
+                      telemetryEvent(result.success ? "tool_loop:tool_result" : "tool_loop:tool_error", {
+                        name: functionName,
+                        isError: !result.success
+                      });
                       let resultText = "";
                       if (result.success) {
                         if (functionName === "web_search" || functionName === "web-search") {
@@ -22962,18 +23369,16 @@ ${protocol}`;
                             blocks.push(
                               items.slice(0, 6).map((item, index) => {
                                 const title = item.title?.trim() || "Untitled";
-                                const url = item.url?.trim();
+                                const url = item.url?.trim() || "";
                                 const snippet = item.content?.trim();
-                                let line = `${index + 1}. **${title}**`;
-                                if (url) line += `
-   ${url}`;
+                                if (url) collectedSources.push({ title, url });
+                                let line = url ? `${index + 1}. [${title}](${url})` : `${index + 1}. ${title}`;
                                 if (snippet) {
                                   const truncated = snippet.length > 300 ? `${snippet.slice(0, 300)}\u2026` : snippet;
-                                  line += `
-   ${truncated}`;
+                                  line += ` \u2014 ${truncated}`;
                                 }
                                 return line;
-                              }).join("\n\n")
+                              }).join("\n")
                             );
                           }
                           resultText = blocks.length ? blocks.join("\n\n") : `No results found${search.query ? ` for "${search.query}"` : ""}.`;
@@ -23055,7 +23460,7 @@ ${r.output}`).join("\n\n");
 
 ${toolResultsText}
 
-Using these results together with your own knowledge, answer my original question concisely and in a natural, well-formatted way. Cite source URLs inline when you rely on them. Do NOT output tool_code or call any tools again.`
+Using these results together with your own knowledge, answer my original question concisely and in a natural, well-formatted way. Reference sources by name where relevant, but do NOT paste raw URLs or a list of links \u2014 a clean Sources section is added automatically below your answer. Do NOT output tool_code or call any tools again.`
                         }
                       ];
                       const summaryRequest = {
@@ -23069,7 +23474,7 @@ Using these results together with your own knowledge, answer my original questio
                       setStreamBuffer(
                         summaryPreamble ? `${summaryPreamble}
 
-_Working on it\u2026_` : "_Working on it\u2026_"
+_Writing the answer\u2026_` : "_Writing the answer\u2026_"
                       );
                       const summaryText = await new Promise((resolve) => {
                         let acc = "";
@@ -23110,7 +23515,18 @@ _Working on it\u2026_` : "_Working on it\u2026_"
                         }, 3e4);
                       });
                       if (summaryText.trim()) {
-                        enhancedMessage = summaryText + (inlineImageBlocks.length ? `
+                        const sourcesMd = collectedSources.length ? `
+
+**Sources**
+${collectedSources.slice(0, 6).map((s) => {
+                          let domain = s.url;
+                          try {
+                            domain = new URL(s.url).hostname.replace(/^www\./, "");
+                          } catch {
+                          }
+                          return `- [${s.title || domain}](${s.url}) \u2014 ${domain}`;
+                        }).join("\n")}` : "";
+                        enhancedMessage = summaryText + sourcesMd + (inlineImageBlocks.length ? `
 
 ${inlineImageBlocks.join("\n\n")}` : "");
                       }
@@ -23159,6 +23575,7 @@ ${inlineImageBlocks.join("\n\n")}` : "");
                 }
                 setInputValue("");
                 setPastedImages([]);
+                telemetryEndTurn();
                 setTimeout(() => {
                   clearFlushTimer();
                   setPendingMessage(null);
@@ -23174,6 +23591,7 @@ ${inlineImageBlocks.join("\n\n")}` : "");
                 overrideComponentStatus("Idle");
                 setIsSubmitting(false);
                 setIsStreaming(false);
+                telemetryEndTurn({ error: e instanceof Error ? e.message : String(e) });
               }
             }
           });
@@ -26032,10 +26450,10 @@ var init_enhanced_mobile_conversations_modal = __esm({
       (0, import_react52.useEffect)(() => {
         setDeletedConversationIds((prev) => {
           let changed = false;
-          const active = new Set(conversations.map((conv) => conv.id));
+          const active2 = new Set(conversations.map((conv) => conv.id));
           const next = /* @__PURE__ */ new Set();
           prev.forEach((id) => {
-            if (active.has(id)) {
+            if (active2.has(id)) {
               next.add(id);
             } else {
               changed = true;
@@ -26657,6 +27075,7 @@ var init_chat_app_bar = __esm({
     init_packageSettingsStore();
     init_useFeatures();
     init_conversationSyncStore();
+    init_engineStore();
     import_shallow2 = require("zustand/shallow");
     import_jsx_runtime42 = require("react/jsx-runtime");
     CDN_BASE2 = "https://cdn.burtson.ai/";
@@ -26706,6 +27125,7 @@ var init_chat_app_bar = __esm({
         menuText
       } = theme.palette.chat.appBar;
       const [modelAnchorEl, setModelAnchorEl] = (0, import_react53.useState)(null);
+      const [engineAnchorEl, setEngineAnchorEl] = (0, import_react53.useState)(null);
       const [voiceAnchorEl, setVoiceAnchorEl] = (0, import_react53.useState)(null);
       const [modalOpen, setModalOpen] = (0, import_react53.useState)(false);
       const [confirmModelChangeOpen, setConfirmModelChangeOpen] = (0, import_react53.useState)(false);
@@ -26826,6 +27246,14 @@ 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 || modelAvatars3[selectedModel] || banditHead5;
+      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";
+      (0, import_react53.useEffect)(() => {
+        useEngineStore.getState().fetchEngines();
+      }, []);
       const pendingModelAvatar = useModelStore.getState().availableModels.find((m) => m.name === pendingModel)?.avatarBase64 || modelAvatars3[pendingModel || ""] || banditHead5;
       const resolvedHomeUrl = preferences.homeUrl?.trim() || packageSettings?.homeUrl?.trim() || "";
       const homeTooltip = (() => {
@@ -27138,6 +27566,84 @@ 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)(
+                      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 })
+                        ]
+                      }
+                    ) }),
+                    /* @__PURE__ */ (0, import_jsx_runtime42.jsxs)(
+                      import_material43.Menu,
+                      {
+                        anchorEl: engineAnchorEl,
+                        open: Boolean(engineAnchorEl),
+                        onClose: () => setEngineAnchorEl(null),
+                        transformOrigin: { horizontal: "right", vertical: "top" },
+                        anchorOrigin: { horizontal: "right", vertical: "bottom" },
+                        children: [
+                          /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.Typography, { variant: "overline", sx: { px: 2, color: theme.palette.text.secondary }, children: "Engine \xB7 base model" }),
+                          engines.length === 0 && /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.MenuItem, { disabled: true, children: /* @__PURE__ */ (0, import_jsx_runtime42.jsx)(import_material43.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_runtime42.jsxs)(
+                              import_material43.MenuItem,
+                              {
+                                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__ */ (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: "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)(
+                                    import_material43.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_runtime42.jsx)(
                       import_material43.Menu,
                       {