@burtson-labs/bandit-engine 2.0.158 → 2.0.160

package/dist/management/management.js

@@ -1585,7 +1585,7 @@ function emitAuthTokenChanged(token) {
     detail: { token }
   }));
 }
-var TOKEN_KEY2, AUTH_TOKEN_CHANGED_EVENT, AuthenticationService, authenticationService2;
+var TOKEN_KEY2, AUTH_TOKEN_CHANGED_EVENT, AuthenticationService, authenticationService;
 var init_authenticationService = __esm({
   "src/services/auth/authenticationService.ts"() {
     "use strict";
@@ -1638,7 +1638,7 @@ var init_authenticationService = __esm({
         }
       }
     };
-    authenticationService2 = new AuthenticationService();
+    authenticationService = new AuthenticationService();
   }
 });
 
@@ -1729,7 +1729,7 @@ var init_gateway_service = __esm({
           return;
         }
         try {
-          authenticationService2.clearToken();
+          authenticationService.clearToken();
         } catch {
         }
         if (typeof window !== "undefined" && typeof window.dispatchEvent === "function") {
@@ -3077,7 +3077,7 @@ var init_playground_provider = __esm({
     PlaygroundProvider = class {
       config;
       constructor(config) {
-        this.config = { type: "playground" /* PLAYGROUND */, ...config };
+        this.config = { ...config, type: config.type ?? "playground" /* PLAYGROUND */ };
       }
       chat(request) {
         const lastUserMessage = [...request.messages].reverse().find((m) => m.role === "user")?.content ?? "";
@@ -8140,7 +8140,7 @@ function buildUrl(path) {
 }
 function buildHeaders() {
   const headers = { "Content-Type": "application/json" };
-  const token = authenticationService2.getToken();
+  const token = authenticationService.getToken();
   if (token) {
     headers["Authorization"] = `Bearer ${token}`;
   }
@@ -8288,7 +8288,7 @@ function resolveAuthIdentity(token) {
   if (!token) {
     return null;
   }
-  const claims = authenticationService2.parseJwtClaims(token);
+  const claims = authenticationService.parseJwtClaims(token);
   if (claims?.sub) {
     return claims.sub;
   }
@@ -8821,7 +8821,7 @@ var init_conversationSyncStore = __esm({
       async initialize() {
         ensureTrackersInitialized();
         const gatewayUrl = usePackageSettingsStore.getState().settings?.gatewayApiUrl;
-        const token = authenticationService2.getToken();
+        const token = authenticationService.getToken();
         const tokenIdentity = resolveAuthIdentity(token);
         const current = get();
         const storedIdentity = getStoredSyncIdentity();
@@ -8947,10 +8947,10 @@ var init_conversationSyncStore = __esm({
           });
           set({
             hasLoadedPreference: true,
-            initializedForToken: authenticationService2.getToken(),
-            initializedForIdentity: resolveAuthIdentity(authenticationService2.getToken())
+            initializedForToken: authenticationService.getToken(),
+            initializedForIdentity: resolveAuthIdentity(authenticationService.getToken())
           });
-          setStoredSyncIdentity(resolveAuthIdentity(authenticationService2.getToken()));
+          setStoredSyncIdentity(resolveAuthIdentity(authenticationService.getToken()));
           if (enabled) {
             set({ hasCompletedInitialUpload: false });
           }
@@ -8990,10 +8990,10 @@ var init_conversationSyncStore = __esm({
           });
           set({
             hasLoadedPreference: true,
-            initializedForToken: authenticationService2.getToken(),
-            initializedForIdentity: resolveAuthIdentity(authenticationService2.getToken())
+            initializedForToken: authenticationService.getToken(),
+            initializedForIdentity: resolveAuthIdentity(authenticationService.getToken())
           });
-          setStoredSyncIdentity(resolveAuthIdentity(authenticationService2.getToken()));
+          setStoredSyncIdentity(resolveAuthIdentity(authenticationService.getToken()));
           if (preference.syncEnabled && preference.isAdvancedVectorFeaturesEnabled) {
             await get().runSync({ force: true });
           }
@@ -9033,7 +9033,7 @@ var init_conversationSyncStore = __esm({
           debugLogger.error("conversationSyncStore: runSync error - missing gateway URL");
           return;
         }
-        const token = authenticationService2.getToken();
+        const token = authenticationService.getToken();
         if (!token) {
           set({ status: "error", lastError: "Authentication required to sync conversations." });
           debugLogger.error("conversationSyncStore: runSync error - missing auth token");
@@ -9966,7 +9966,7 @@ var init_useKnowledgeStore = __esm({
     init_indexedDBService();
     import_uuid5 = require("uuid");
     import_mammoth = __toESM(require("mammoth"));
-    pdfjsLib = __toESM(require("pdfjs-dist/legacy/build/pdf"));
+    pdfjsLib = __toESM(require("pdfjs-dist/legacy/build/pdf.mjs"));
     init_embeddingService();
     init_debugLogger();
     init_vectorDatabaseService();
@@ -11177,7 +11177,7 @@ var init_knowledge_file_modal = __esm({
     import_material7 = require("@mui/material");
     init_useVectorStore();
     init_packageSettingsStore();
-    pdfjsLib2 = __toESM(require("pdfjs-dist/legacy/build/pdf"));
+    pdfjsLib2 = __toESM(require("pdfjs-dist/legacy/build/pdf.mjs"));
     init_lowlight();
     init_markdownRendering();
     init_debugLogger();
@@ -11968,7 +11968,7 @@ var init_voiceService = __esm({
           debugLogger.error("Gateway API URL not configured");
           throw new Error("Gateway API URL not configured");
         }
-        const token = authenticationService2.getToken();
+        const token = authenticationService.getToken();
         if (!token) {
           debugLogger.warn("No JWT token available - skipping voice models fetch");
           const emptyModels = {
@@ -11978,7 +11978,7 @@ var init_voiceService = __esm({
           };
           return emptyModels;
         }
-        if (authenticationService2.isTokenExpired(token)) {
+        if (authenticationService.isTokenExpired(token)) {
           debugLogger.warn("JWT token is expired - skipping voice models fetch");
           const emptyModels = {
             models: [],
@@ -12067,7 +12067,7 @@ var init_tts_client = __esm({
     init_packageSettingsStore();
     init_debugLogger();
     getOrAppendAuthHeader = (existing = {}) => {
-      const token = authenticationService2.getToken();
+      const token = authenticationService.getToken();
       if (token) {
         existing["Authorization"] = `Bearer ${token}`;
         return existing;
@@ -12758,8 +12758,8 @@ var init_ai_response_action_bar = __esm({
       const { preferences } = usePreferencesStore();
       const { settings: packageSettings } = usePackageSettingsStore();
       (0, import_react14.useEffect)(() => {
-        const isAuthenticated = authenticationService2.isAuthenticated();
-        const token = authenticationService2.getToken();
+        const isAuthenticated = authenticationService.isAuthenticated();
+        const token = authenticationService.getToken();
         if (!initialized || availableVoices.length === 0) {
           if (token && isAuthenticated) {
             debugLogger.debug("AI Action Bar: Voices not initialized or unavailable, attempting to load from API");
@@ -13603,11 +13603,12 @@ ${sourcesMarkdownList.join("\n")}`;
         tr({ children, node: _node, ...tableRowProps }) {
           return /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(import_material9.TableRow, { ...tableRowProps, children });
         },
-        th({ children, node: _node, ...tableCellProps }) {
+        th({ children, node: _node, align, ...tableCellProps }) {
           return /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(
             import_material9.TableCell,
             {
               ...tableCellProps,
+              align: align === "char" ? void 0 : align,
               sx: {
                 fontWeight: "bold",
                 color: theme.palette.text.primary,
@@ -13620,11 +13621,12 @@ ${sourcesMarkdownList.join("\n")}`;
             }
           );
         },
-        td({ children, node: _node, ...tableCellProps }) {
+        td({ children, node: _node, align, ...tableCellProps }) {
           return /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(
             import_material9.TableCell,
             {
               ...tableCellProps,
+              align: align === "char" ? void 0 : align,
               sx: {
                 color: theme.palette.text.primary,
                 fontSize: "0.875rem",
@@ -14701,7 +14703,7 @@ var init_memory_modal = __esm({
                     color: (0, import_styles3.alpha)(theme.palette.text.secondary, 0.7),
                     maxWidth: 280
                   }, children: selectedTab === "user" ? shouldUseVectorForMemories ? "Start by adding something you'd like me to remember. Your memories will be kept in a private workspace for quick recall." : "Start by adding something you'd like me to remember about you or your preferences." : shouldUseVectorForMemories ? "Auto memories are created automatically based on our conversations and kept in your private workspace for better answers." : "Auto memories are created automatically based on our conversations." })
-                ] }) : filteredMemories.map((memory) => {
+                ] }) : filteredMemories.map((memory, memoryIndex) => {
                   const isSelected = selectedIds.includes(memory.id);
                   return /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(import_react17.default.Fragment, { children: [
                     /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material10.List, { sx: { listStyle: "none", pl: 0, m: 0 }, children: /* @__PURE__ */ (0, import_jsx_runtime11.jsxs)(
@@ -14901,7 +14903,7 @@ var init_memory_modal = __esm({
                         ]
                       }
                     ) }),
-                    filteredMemories.indexOf(memory) < filteredMemories.length - 1 && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material10.Divider, { sx: { bgcolor: (0, import_styles3.alpha)(appBar.border, 0.5), my: 0.5 } })
+                    memoryIndex < filteredMemories.length - 1 && /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(import_material10.Divider, { sx: { bgcolor: (0, import_styles3.alpha)(appBar.border, 0.5), my: 0.5 } })
                   ] }, memory.id);
                 })
               }
@@ -15613,7 +15615,7 @@ var init_FeedbackModal = __esm({
           }
           const gatewayService = new GatewayService(
             gatewayUrl,
-            () => authenticationService2.getToken(),
+            () => authenticationService.getToken(),
             feedbackEmail || settings?.feedbackEmail
           );
           const feedbackRequest = {
@@ -19242,7 +19244,7 @@ ${listMarkdown}`;
           children
         }
       );
-      const CodeRenderer = ({ inline, className, children, ...props }) => {
+      const CodeRenderer = ({ node: _node, className, children, ...props }) => {
         const match = /language-([\w-]+)/.exec(className || "");
         const requestedLanguage = match?.[1]?.toLowerCase() ?? "";
         const codeText = String(children).replace(/\n$/, "");
@@ -19576,7 +19578,7 @@ var init_mcpServersService = __esm({
     };
     authHeaders2 = () => {
       const headers = { "Content-Type": "application/json" };
-      const token = authenticationService2.getToken();
+      const token = authenticationService.getToken();
       if (token) headers["Authorization"] = `Bearer ${token}`;
       return headers;
     };
@@ -20773,7 +20775,7 @@ var init_engineStore = __esm({
         }
         try {
           const headers = { "Content-Type": "application/json" };
-          const token = authenticationService2.getToken();
+          const token = authenticationService.getToken();
           if (token) headers["Authorization"] = `Bearer ${token}`;
           const res = await fetch(`${base}/models`, { headers });
           const data = await res.json();
@@ -21092,7 +21094,7 @@ var init_stt_client = __esm({
           filename,
           url: `${gatewayUrl}/stt/transcribe`
         });
-        const token = authenticationService2.getToken();
+        const token = authenticationService.getToken();
         const response = await fetch(`${gatewayUrl}/stt/transcribe`, {
           method: "POST",
           headers: {
@@ -21212,6 +21214,109 @@ var init_transcriber = __esm({
   }
 });
 
+// src/chat/composer-submission.ts
+var isPromiseLike, awaitAcceptance, buildSubmission, isImeComposing;
+var init_composer_submission = __esm({
+  "src/chat/composer-submission.ts"() {
+    "use strict";
+    isPromiseLike = (value) => !!value && typeof value.then === "function";
+    awaitAcceptance = async (ack) => {
+      if (ack === false) return false;
+      if (!isPromiseLike(ack)) return true;
+      try {
+        return await ack !== false;
+      } catch {
+        return false;
+      }
+    };
+    buildSubmission = (text, images, documents) => {
+      const typed = text.trim();
+      const base = typed || (images.length === 1 ? "[\u{1F4F7} You attached an image]" : images.length > 1 ? `[\u{1F4F7} You attached ${images.length} images]` : "");
+      const fileSummary = documents.length > 0 ? `[\u{1F4CE} Attached ${documents.length} file${documents.length > 1 ? "s" : ""}: ${documents.map((d) => d.fileName).join(", ")}]` : "";
+      const displayQuestion = [base || (fileSummary ? "[\u{1F4E4} You submitted content without text]" : ""), fileSummary].filter(Boolean).join("\n\n");
+      const prompt = [text, ...documents.map((d) => d.block)].filter(Boolean).join("\n\n");
+      return { prompt, images: [...images], displayQuestion };
+    };
+    isImeComposing = (event) => event.nativeEvent?.isComposing === true || event.keyCode === 229;
+  }
+});
+
+// src/chat/document-import.ts
+var CODE_EXTENSIONS, TEXT_EXTENSIONS, isCodeFileName, isPlainTextFile, CONTROL_CHARS, INVISIBLE_FORMATTING, stripInvisible, neutralizeRoleMarkers, prepareDocumentText, formatDocumentBlock, readText, readPdfText, readDocument;
+var init_document_import = __esm({
+  "src/chat/document-import.ts"() {
+    "use strict";
+    CODE_EXTENSIONS = [".js", ".ts", ".tsx", ".json", ".py", ".java", ".c", ".cpp", ".cs"];
+    TEXT_EXTENSIONS = [".md", ".txt", ...CODE_EXTENSIONS];
+    isCodeFileName = (name) => {
+      const lower = name.toLowerCase();
+      return CODE_EXTENSIONS.some((ext) => lower.endsWith(ext));
+    };
+    isPlainTextFile = (name, mimeType = "") => mimeType.startsWith("text/") || TEXT_EXTENSIONS.some((ext) => name.toLowerCase().endsWith(ext));
+    CONTROL_CHARS = /[\u0000-\u0008\u000B\u000C\u000E-\u001F\u007F-\u009F]/g;
+    INVISIBLE_FORMATTING = /[\u202A-\u202E\u2066-\u2069\u200B\u2060\uFEFF]/g;
+    stripInvisible = (text) => text.normalize("NFC").replace(CONTROL_CHARS, "").replace(INVISIBLE_FORMATTING, "");
+    neutralizeRoleMarkers = (text) => text.replace(/<\s*\/?\s*script\b[^>]*>/gi, "").replace(/(^|\n)([ \t]*)\/?(system|assistant|user)\s*:/gi, "$1$2[removed-role]");
+    prepareDocumentText = (text, fileName) => {
+      const visible = stripInvisible(text);
+      return (isCodeFileName(fileName) ? visible : neutralizeRoleMarkers(visible)).trim();
+    };
+    formatDocumentBlock = (fileName, text) => `\u{1F4C4} ${fileName}
+${prepareDocumentText(text, fileName)}`;
+    readText = (file) => typeof file.text === "function" ? file.text() : new Promise((resolve, reject) => {
+      const reader = new FileReader();
+      reader.onload = () => resolve(String(reader.result ?? ""));
+      reader.onerror = () => reject(reader.error ?? new Error("It couldn't be read."));
+      reader.readAsText(file);
+    });
+    readPdfText = async (file, maxPages = 10) => {
+      const pdfjsLib3 = await import("pdfjs-dist/legacy/build/pdf.mjs");
+      pdfjsLib3.GlobalWorkerOptions.workerSrc = "https://cdn.burtson.ai/scripts/pdf.worker.js";
+      const pdf = await pdfjsLib3.getDocument({ data: await file.arrayBuffer() }).promise;
+      const pages = await Promise.all(
+        Array.from({ length: Math.min(pdf.numPages, maxPages) }, async (_, i) => {
+          try {
+            const page = await pdf.getPage(i + 1);
+            const content = await page.getTextContent();
+            return content.items.map((item) => typeof item?.str === "string" ? item.str : "").join(" ");
+          } catch {
+            return "";
+          }
+        })
+      );
+      return pages.join("\n\n");
+    };
+    readDocument = async (file) => {
+      const name = file.name.toLowerCase();
+      try {
+        let text;
+        if (isPlainTextFile(name, file.type)) {
+          text = await readText(file);
+        } else if (name.endsWith(".docx")) {
+          const mammoth2 = await import("mammoth");
+          text = (await mammoth2.extractRawText({ arrayBuffer: await file.arrayBuffer() })).value;
+        } else if (name.endsWith(".pdf")) {
+          text = await readPdfText(file);
+        } else {
+          return { ok: false, fileName: file.name, reason: "This file type can't be read yet." };
+        }
+        const block = formatDocumentBlock(file.name, text);
+        if (block === `\u{1F4C4} ${file.name}
+`) {
+          return { ok: false, fileName: file.name, reason: "No readable text was found in it." };
+        }
+        return { ok: true, fileName: file.name, block };
+      } catch (error) {
+        return {
+          ok: false,
+          fileName: file.name,
+          reason: error instanceof Error && error.message ? error.message : "It couldn't be read."
+        };
+      }
+    };
+  }
+});
+
 // src/store/voiceModeStore.ts
 var import_zustand16, useVoiceModeStore;
 var init_voiceModeStore = __esm({
@@ -21266,6 +21371,8 @@ var init_chat_input = __esm({
     init_packageSettingsStore();
     init_transcriber();
     init_memory_modal();
+    init_composer_submission();
+    init_document_import();
     import_styles24 = require("@mui/material/styles");
     init_debugLogger();
     init_brandingService();
@@ -21434,102 +21541,60 @@ var init_chat_input = __esm({
       const removeImage = (index) => {
         setPastedImages((prev) => prev.filter((_, i) => i !== index));
       };
+      const latestDraftRef = (0, import_react53.useRef)({ text: inputValue, images: pastedImages, files: fileInputs });
+      latestDraftRef.current = { text: inputValue, images: pastedImages, files: fileInputs };
+      const submittingRef = (0, import_react53.useRef)(false);
+      const [sendError, setSendError] = (0, import_react53.useState)(null);
       const handleSubmit = async () => {
-        if (inputValue.trim() === "" && pastedImages.length === 0 && fileInputs.length === 0)
-          return;
-        const tempImages = [...pastedImages];
-        const escapePromptInjection = (text) => {
-          return text.replace(/<\s*\/?script.*?>/gi, "").replace(/```.*?```/gs, "").replace(/(?<=\n|^)\/?(system|assistant|user):/gi, "[removed-role]");
-        };
-        const getFileText = async (file) => {
-          const isCodeFile2 = (name2) => name2.endsWith(".js") || name2.endsWith(".ts") || name2.endsWith(".tsx") || name2.endsWith(".json") || name2.endsWith(".py") || name2.endsWith(".java") || name2.endsWith(".c") || name2.endsWith(".cpp") || name2.endsWith(".cs");
-          const sanitize = (text, name2) => isCodeFile2(name2) ? text.trim() : text.replace(/[^\x20-\x7E\r\n\t]/g, "").trim();
-          const name = file.name.toLowerCase();
-          try {
-            if (file.type.startsWith("text/") || name.endsWith(".md") || name.endsWith(".txt") || name.endsWith(".js") || name.endsWith(".ts") || name.endsWith(".tsx") || name.endsWith(".json") || name.endsWith(".py") || name.endsWith(".java") || name.endsWith(".c") || name.endsWith(".cpp") || name.endsWith(".cs")) {
-              const text = await file.text();
-              if (name.endsWith(".txt") || name.endsWith(".md") || name.endsWith(".docx") || name.endsWith(".pdf")) {
-                return `\u{1F4C4} ${file.name}
-${sanitize(escapePromptInjection(text), file.name)}`;
-              } else {
-                return `\u{1F4C4} ${file.name}
-${sanitize(text, file.name)}`;
-              }
-            }
-            if (name.endsWith(".docx")) {
-              const mammoth2 = await import("mammoth");
-              const arrayBuffer = await file.arrayBuffer();
-              const { value } = await mammoth2.extractRawText({ arrayBuffer });
-              return `\u{1F4C4} ${file.name}
-${sanitize(escapePromptInjection(value), file.name)}`;
-            }
-            if (name.endsWith(".pdf")) {
-              const pdfjsLib3 = await import("pdfjs-dist/legacy/build/pdf");
-              pdfjsLib3.GlobalWorkerOptions.workerSrc = "https://cdn.burtson.ai/scripts/pdf.worker.js";
-              const buffer = await file.arrayBuffer();
-              const loadingTask = pdfjsLib3.getDocument({ data: buffer });
-              try {
-                const pdf = await loadingTask.promise;
-                debugLogger.info("PDF loaded", { numPages: pdf.numPages });
-                const maxPages = Math.min(pdf.numPages, 10);
-                const pages = await Promise.all(
-                  Array.from({ length: maxPages }, async (_, i) => {
-                    try {
-                      const page = await pdf.getPage(i + 1);
-                      const content = await page.getTextContent();
-                      return content.items.map((item) => {
-                        if (item && typeof item === "object" && "str" in item && typeof item.str === "string") {
-                          return item.str;
-                        }
-                        return "";
-                      }).join(" ");
-                    } catch (err) {
-                      debugLogger.error(`Failed to read page ${i + 1}`, { error: err });
-                      return "";
-                    }
-                  })
-                );
-                const joined = pages.join("\n\n");
-                debugLogger.info("Extracted PDF content", { contentPreview: joined.slice(0, 500) });
-                return `\u{1F4C4} ${file.name}
-${sanitize(
-                  escapePromptInjection(joined),
-                  file.name
-                )}`;
-              } catch (error) {
-                debugLogger.error("PDF load failed", {
-                  error: error instanceof Error ? error.message : String(error),
-                  fileName: file.name
-                });
-                return "";
-              }
+        if (submittingRef.current) return;
+        if (inputValue.trim() === "" && pastedImages.length === 0 && fileInputs.length === 0) return;
+        submittingRef.current = true;
+        const draft = { text: inputValue, images: [...pastedImages], files: [...fileInputs] };
+        try {
+          const reads = await Promise.all(draft.files.map(readDocument));
+          const failed = reads.filter((r) => !r.ok);
+          if (failed.length > 0) {
+            setSendError(
+              failed.map((f) => `Couldn't attach ${f.fileName}: ${f.reason}`).join(" ")
+            );
+            return;
+          }
+          const documents = reads.filter((r) => r.ok);
+          const submission = buildSubmission(draft.text, draft.images, documents);
+          debugLogger.debug("Composer submission", {
+            documents: documents.map((d) => d.fileName),
+            promptLength: submission.prompt.length
+          });
+          setSendError(null);
+          setInputValue("");
+          setPastedImages([]);
+          setFileInputs([]);
+          if (fileInputRef.current) fileInputRef.current.value = "";
+          inputRef.current?.blur();
+          const accepted = await awaitAcceptance(
+            onSend(submission.prompt, submission.images, submission.displayQuestion)
+          );
+          if (!accepted) {
+            const now = latestDraftRef.current;
+            const untouched = now.text.trim() === "" && now.images.length === 0 && now.files.length === 0;
+            if (untouched) {
+              setInputValue(draft.text);
+              setPastedImages(draft.images);
+              setFileInputs(draft.files);
+              setSendError("Your message wasn't sent. It's back in the box so you can try again.");
+            } else {
+              setSendError("Your last message wasn't sent.");
             }
-            return "";
-          } catch {
-            return "";
           }
-        };
-        const fileTexts = await Promise.all(fileInputs.map(getFileText));
-        debugLogger.debug("All extracted file texts", { fileTexts });
-        const fileContent = fileTexts.filter(Boolean).join("\n\n");
-        const tempQuestionBase = inputValue.trim() || (tempImages.length === 1 ? "[\u{1F4F7} You attached an image]" : tempImages.length > 1 ? `[\u{1F4F7} You attached ${tempImages.length} images]` : "");
-        const fileDisplaySummary = fileInputs.length > 0 ? `[\u{1F4CE} Attached ${fileInputs.length} file${fileInputs.length > 1 ? "s" : ""}: ${fileInputs.map((f) => f.name).join(", ")}]` : "";
-        const displayQuestion = [
-          tempQuestionBase || (fileDisplaySummary ? "[\u{1F4E4} You submitted content without text]" : ""),
-          fileDisplaySummary
-        ].filter(Boolean).join("\n\n");
-        const fullPrompt = [inputValue, fileContent].filter(Boolean).join("\n\n");
-        setInputValue("");
-        setPastedImages([]);
-        setFileInputs([]);
-        onSend(fullPrompt, tempImages, displayQuestion);
-        inputRef.current?.blur();
-        fileInputRef.current.value = "";
+        } finally {
+          submittingRef.current = false;
+        }
       };
       const handleKeyPress = (e) => {
+        if (isImeComposing(e)) return;
         if (e.key === "Enter" && !e.shiftKey) {
           e.preventDefault();
-          handleSubmit();
+          void handleSubmit();
         }
       };
       const handleTranscriptionCompleted = (transcribed) => {
@@ -21647,6 +21712,15 @@ ${sanitize(
                     gap: isMobile ? compactMobile ? 0.75 : 1 : 0.5
                   },
                   children: [
+                    sendError && /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(
+                      import_material40.Typography,
+                      {
+                        role: "alert",
+                        variant: "caption",
+                        sx: { color: theme.palette.error.main, px: 0.5, pt: isMobile ? 0 : 0.5 },
+                        children: sendError
+                      }
+                    ),
                     /* @__PURE__ */ (0, import_jsx_runtime40.jsxs)(
                       import_material40.Box,
                       {
@@ -21692,7 +21766,10 @@ ${sanitize(
                                   import_material40.IconButton,
                                   {
                                     size: "small",
-                                    onClick: () => setFileInputs((prev) => prev.filter((_, i) => i !== idx)),
+                                    onClick: () => {
+                                      setSendError(null);
+                                      setFileInputs((prev) => prev.filter((_, i) => i !== idx));
+                                    },
                                     sx: { ml: 0.5, color: theme.palette.mode === "dark" ? "#aaa" : "#444" },
                                     children: /* @__PURE__ */ (0, import_jsx_runtime40.jsx)(CloseIcon, { sx: { fontSize: 14 } })
                                   }
@@ -22700,7 +22777,7 @@ function syncTelemetry() {
     const settings = usePackageSettingsStore.getState().settings;
     const cfg = resolveTelemetryConfig({
       telemetry: settings?.telemetry,
-      banditApiKey: authenticationService2.getToken() ?? void 0
+      banditApiKey: authenticationService.getToken() ?? void 0
     });
     active = cfg ? new TelemetryExporter(cfg) : null;
   } catch {
@@ -23351,7 +23428,7 @@ var init_mcpService = __esm({
         }
         if (tool.mcpServerId) {
           const base2 = settings.gatewayApiUrl.replace(/\/$/, "");
-          const token2 = authenticationService2.getToken();
+          const token2 = authenticationService.getToken();
           const controller2 = new AbortController();
           const timeoutId2 = setTimeout(() => controller2.abort(), 3e4);
           try {
@@ -23398,7 +23475,7 @@ var init_mcpService = __esm({
             "Content-Type": "application/json"
           }
         };
-        const token = authenticationService2.getToken();
+        const token = authenticationService.getToken();
         if (token) {
           requestOptions.headers = {
             ...requestOptions.headers,
@@ -23498,6 +23575,7 @@ var init_useAIProvider = __esm({
   "src/chat/hooks/useAIProvider.tsx"() {
     "use strict";
     init_debugLogger();
+    init_authenticationService();
     import_react58 = require("react");
     init_knowledgeStore();
     init_aiProviderStore();
@@ -23799,8 +23877,11 @@ var init_useAIProvider = __esm({
       history,
       addHistory,
       inputRef,
-      onError
+      onError,
+      onTurnOutcome
     }) => {
+      const onTurnOutcomeRef = (0, import_react58.useRef)(onTurnOutcome);
+      onTurnOutcomeRef.current = onTurnOutcome;
       const currentSubRef = (0, import_react58.useRef)(null);
       const lastPartialRef = (0, import_react58.useRef)({
         text: "",
@@ -23846,10 +23927,14 @@ var init_useAIProvider = __esm({
       };
       const runAIProvider = (0, import_react58.useCallback)(
         async (systemPrompt, question, images) => {
+          let turnHadOutput = false;
+          const reportOutcome = (outcome) => onTurnOutcomeRef.current?.(outcome);
           if (!provider) {
             debugLogger.error("No AI provider configured");
             overrideComponentStatus("Error");
-            if (onError) onError(new Error("No AI provider configured"));
+            const error = new Error("No AI provider configured");
+            if (onError) onError(error);
+            reportOutcome({ kind: "failed", hadOutput: false, error });
             return;
           }
           const userPreferences = preferences;
@@ -24003,7 +24088,7 @@ Use this information to personalize your responses whenever relevant.` : "";
                         s3Url: resolvedId,
                         isUserContent: result.isUserContent,
                         isTeamContent: result.isTeamContent,
-                        contentSource: result.contentSource,
+                        contentSource: result.contentSource === "user" || result.contentSource === "team" ? result.contentSource : void 0,
                         originalFileName: result.originalFileName || result.filename || result.name,
                         // Preserve original data for downstream checks (e.g. vector modal detection)
                         _vectorResult: result
@@ -24358,6 +24443,10 @@ ${protocol}`;
           const sub = stream.subscribe({
             next: (data) => {
               if (!data?.message?.content && !data?.message?.tool_calls) return;
+              if (!turnHadOutput) {
+                turnHadOutput = true;
+                reportOutcome({ kind: "output" });
+              }
               if (Array.isArray(data.message.tool_calls) && data.message.tool_calls.length > 0) {
                 nativeToolCalls.push(...data.message.tool_calls);
                 sawToolBlock = true;
@@ -24392,13 +24481,14 @@ ${protocol}`;
             },
             error: (err) => {
               debugLogger.error("Stream error:", err);
+              reportOutcome({ kind: "failed", hadOutput: turnHadOutput, error: err });
               stopWorking();
               overrideComponentStatus("Idle");
               setIsSubmitting(false);
               setIsStreaming(false);
               if (/lifetime validation|token.*(expired|invalid)|unauthorized|\b401\b/i.test(err.message || "")) {
-                const { replaceLastAnswer: replaceLastAnswer2 } = useConversationStore.getState();
-                replaceLastAnswer2(
+                const { replaceLastAnswer: replaceLastAnswer3 } = useConversationStore.getState();
+                replaceLastAnswer3(
                   "_Your session ended \u2014 sign back in to continue._",
                   lastPartialRef.current.images,
                   false,
@@ -24425,8 +24515,8 @@ ${protocol}`;
                   scope: rateLimit.scope,
                   frontier: rateLimit.frontier
                 });
-                const { replaceLastAnswer: replaceLastAnswer2 } = useConversationStore.getState();
-                replaceLastAnswer2(
+                const { replaceLastAnswer: replaceLastAnswer3 } = useConversationStore.getState();
+                replaceLastAnswer3(
                   rateLimit.scope === "weekly" ? "_You've reached your weekly limit. Upgrade to keep chatting._" : "_You've hit your session limit. It resets soon \u2014 or upgrade for higher limits._",
                   lastPartialRef.current.images,
                   false,
@@ -24447,10 +24537,16 @@ ${protocol}`;
               if (partial && !partial.trim().endsWith("\u2026") && !partial.trim().endsWith("...")) {
                 partial = partial.trimEnd() + " \u2026";
               }
+              const { replaceLastAnswer: replaceLastAnswer2 } = useConversationStore.getState();
               if (partial) {
-                const { replaceLastAnswer: replaceLastAnswer2 } = useConversationStore.getState();
                 replaceLastAnswer2(partial, lastPartialRef.current.images, false, lastPartialRef.current.usedDocs, true);
                 setResponse(partial);
+              } else {
+                const { conversations: convs, currentId: convId } = useConversationStore.getState();
+                const last = convs.find((c) => c.id === convId)?.history.at(-1);
+                if (last?.answer === "...") {
+                  replaceLastAnswer2("_No response came back. Try sending again._", lastPartialRef.current.images, false, lastPartialRef.current.usedDocs, true);
+                }
               }
               setStreamBuffer("");
               setIsThinking?.(false);
@@ -24462,6 +24558,7 @@ ${protocol}`;
               }
             },
             complete: async () => {
+              reportOutcome({ kind: "completed" });
               try {
                 setIsThinking?.(false);
                 telemetryEvent("tool_loop:llm_response", { responseLength: fullMessage.length });
@@ -24728,7 +24825,7 @@ _This link is temporary and expires in about ${mins} minutes._`;
                         }
                       } else {
                         const data = result.data ?? {};
-                        const informative = [data.message, data.details, data.error, result.error].map((value) => typeof value === "string" ? value.trim() : void 0).filter((value) => Boolean(value) && value.toLowerCase() !== "n/a");
+                        const informative = [data.message, data.details, data.error, result.error].map((value) => typeof value === "string" ? value.trim() : "").filter((value) => value.length > 0 && value.toLowerCase() !== "n/a");
                         resultText = informative.length ? informative.join(" \u2014 ") : `I couldn't complete that request: ${result.error || "Unknown error"}.`;
                       }
                       enhancedMessage = enhancedMessage.replace(placeholderToken, resultText);
@@ -25075,6 +25172,7 @@ ${inlineImageBlocks.join("\n\n")}` : "");
           } catch {
           }
           currentSubRef.current = null;
+          onTurnOutcomeRef.current?.({ kind: "cancelled" });
           let partial = lastPartialRef.current.text;
           if (partial && !partial.trim().endsWith("\u2026") && !partial.trim().endsWith("...")) {
             partial = partial.trimEnd() + " \u2026";
@@ -25293,7 +25391,7 @@ var init_useNetworkStatus = __esm({
         if (connection) {
           const updateConnectionInfo = () => {
             const slowConnections = ["slow-2g", "2g"];
-            const isSlowConnection = slowConnections.includes(connection.effectiveType);
+            const isSlowConnection = slowConnections.includes(connection.effectiveType ?? "");
             setNetworkStatus((prev) => ({
               ...prev,
               isSlowConnection,
@@ -27743,7 +27841,7 @@ var init_conversation_drawer = __esm({
                           title: "Sign out",
                           onClick: (event) => {
                             event.stopPropagation();
-                            authenticationService2.clearToken();
+                            authenticationService.clearToken();
                             window.location.href = "/login";
                           },
                           sx: {
@@ -28870,7 +28968,7 @@ var init_enhanced_mobile_conversations_modal = __esm({
                             title: "Sign out",
                             onClick: (event) => {
                               event.stopPropagation();
-                              authenticationService2.clearToken();
+                              authenticationService.clearToken();
                               window.location.href = "/login";
                             },
                             sx: {
@@ -29690,7 +29788,8 @@ var init_query_suggestion_picker = __esm({
                 }
               },
               onClick: () => {
-                onSend(prompt, []);
+                void Promise.resolve(onSend(prompt, [])).catch(() => {
+                });
                 setHasSentPrompt(true);
               },
               children: /* @__PURE__ */ (0, import_jsx_runtime50.jsx)(
@@ -29751,7 +29850,7 @@ var init_UnderReview = __esm({
       const theme = (0, import_material52.useTheme)();
       const navigate = (0, import_react_router_dom5.useNavigate)();
       const handleResetSession = () => {
-        authenticationService2.clearToken();
+        authenticationService.clearToken();
         navigate("/login");
       };
       return /* @__PURE__ */ (0, import_jsx_runtime51.jsxs)(
@@ -30432,8 +30531,8 @@ var init_chat2 = __esm({
       const notificationService2 = useNotificationService();
       const [selectedTheme, setSelectedTheme] = (0, import_react79.useState)(null);
       const [themeLoading, setThemeLoading] = (0, import_react79.useState)(true);
-      const token = authenticationService2.getToken();
-      const claims = token ? authenticationService2.parseJwtClaims(token) : null;
+      const token = authenticationService.getToken();
+      const claims = token ? authenticationService.parseJwtClaims(token) : null;
       const baseTheme = themeMap_default[selectedTheme ?? "bandit-dark"] || banditDarkTheme;
       const banditTheme = (0, import_react79.useMemo)(() => {
         return (0, import_styles34.createTheme)(baseTheme, {
@@ -30506,7 +30605,7 @@ var init_chat2 = __esm({
       } = useTTS();
       (0, import_react79.useEffect)(() => {
         const timer = setTimeout(() => {
-          const isAuthenticated = authenticationService2.isAuthenticated();
+          const isAuthenticated = authenticationService.isAuthenticated();
           if (!initialized || availableVoices.length === 0) {
             if (token && isAuthenticated) {
               debugLogger.debug("Chat: Voices not initialized or unavailable, attempting to load from API");
@@ -30519,7 +30618,7 @@ var init_chat2 = __esm({
         return () => clearTimeout(timer);
       }, [initialized, availableVoices.length, loadVoicesFromAPI, token]);
       (0, import_react79.useEffect)(() => {
-        const isAuthenticated = authenticationService2.isAuthenticated();
+        const isAuthenticated = authenticationService.isAuthenticated();
         if (packageSettings?.gatewayApiUrl && availableVoices.length === 0 && !initialized) {
           if (token && isAuthenticated) {
             debugLogger.debug("Chat: Gateway URL available and no voices loaded, attempting to load from API");
@@ -31134,7 +31233,16 @@ var init_chat2 = __esm({
           if (rafId) cancelAnimationFrame(rafId);
         };
       }, [_hasHydrated, chatContainerEl, getScrollState, history.length]);
+      const sendAckRef = (0, import_react79.useRef)(null);
+      const settleSendAck = (0, import_react79.useCallback)((outcome) => {
+        const ack = sendAckRef.current;
+        if (!ack) return;
+        sendAckRef.current = null;
+        if (outcome.kind === "failed" && !outcome.hadOutput) ack.reject(outcome.error);
+        else ack.resolve();
+      }, []);
       const aiProvider = useAIProvider({
+        onTurnOutcome: settleSendAck,
         overrideComponentStatus: setComponentStatus,
         setIsSubmitting,
         setIsStreaming,
@@ -31185,6 +31293,26 @@ var init_chat2 = __esm({
       }, [ttsStop, isStreaming, handleStop]);
       const handleSend = (0, import_react79.useCallback)(
         (question, images, displayQuestion) => {
+          sendAckRef.current?.resolve();
+          const acknowledged = new Promise((resolve, reject) => {
+            sendAckRef.current = { resolve, reject };
+          });
+          const runProvider = (systemPrompt2, providerImages2) => {
+            aiProvider(systemPrompt2, question, providerImages2).catch((error) => {
+              debugLogger.error("AI provider failed before streaming", {
+                error: error instanceof Error ? error.message : String(error)
+              });
+              setIsStreaming(false);
+              setResponseStarted(false);
+              setIsSubmitting(false);
+              setPendingMessage(null);
+              const { conversations: convs, currentId: convId, replaceLastAnswer } = useConversationStore.getState();
+              if (convs.find((c) => c.id === convId)?.history.at(-1)?.answer === "...") {
+                replaceLastAnswer("_No response came back. Try sending again._", void 0, false, void 0, true);
+              }
+              settleSendAck({ kind: "failed", hadOutput: false, error });
+            });
+          };
           const requestStartTime = trackRequestStart();
           const questionForDisplay = displayQuestion || question;
           const pendingImages = images.length > 0 ? [...images] : void 0;
@@ -31223,10 +31351,15 @@ var init_chat2 = __esm({
                   rawQuestion: question
                 });
                 const providerImages2 = pendingImages ? [...pendingImages] : [];
-                aiProvider(systemPrompt, question, providerImages2);
+                runProvider(systemPrompt, providerImages2);
               }, 0);
+            }).catch((error) => {
+              debugLogger.error("Starting a new conversation failed", {
+                error: error instanceof Error ? error.message : String(error)
+              });
+              settleSendAck({ kind: "failed", hadOutput: false, error });
             });
-            return;
+            return acknowledged;
           }
           if (existing.name === "New Conversation") {
             generateName(question).then((newName) => {
@@ -31274,10 +31407,12 @@ var init_chat2 = __esm({
           }
           setResponse("");
           const providerImages = pendingImages ? [...pendingImages] : [];
-          aiProvider(systemPrompt, question, providerImages);
+          runProvider(systemPrompt, providerImages);
+          return acknowledged;
         },
         [
           aiProvider,
+          settleSendAck,
           chatContainerRef,
           generateName,
           getScrollState,
@@ -31301,7 +31436,8 @@ var init_chat2 = __esm({
           }
           setInputValue("");
           setPastedImages([]);
-          handleSend(cleaned, [], cleaned);
+          handleSend(cleaned, [], cleaned).catch(() => {
+          });
         },
         [handleSend, setInputValue, setPastedImages]
       );
@@ -31879,12 +32015,12 @@ var init_chat2 = __esm({
         ossConfigured,
         isOSSMode: isOSSMode(),
         tier: featureFlag.getEvaluation()?.tier,
-        hasToken: authenticationService2.isAuthenticated(),
+        hasToken: authenticationService.isAuthenticated(),
         playgroundBypassAuth,
         isPlaygroundRoute,
         bypassAuth
       });
-      if (!allowUnauthenticated && !bypassAuth && !authenticationService2.isAuthenticated()) {
+      if (!allowUnauthenticated && !bypassAuth && !authenticationService.isAuthenticated()) {
         debugLogger.debug("User is not authenticated, redirecting to login");
         return /* @__PURE__ */ (0, import_jsx_runtime54.jsx)(import_react_router_dom6.Navigate, { to: "/login", replace: true });
       }
@@ -36704,7 +36840,7 @@ var useGatewayClient = () => {
     }
     return new GatewayService(
       gatewayUrl,
-      () => authenticationService2.getToken(),
+      () => authenticationService.getToken(),
       feedbackEmail
     );
   }, [gatewayUrl, feedbackEmail]);
@@ -40553,7 +40689,7 @@ var buildHeaders2 = () => {
   const headers = {
     "Content-Type": "application/json"
   };
-  const token = authenticationService2.getToken();
+  const token = authenticationService.getToken();
   if (token) {
     headers.Authorization = `Bearer ${token}`;
   }
@@ -43152,7 +43288,7 @@ var AIProviderInitService = class _AIProviderInitService {
         type: "ollama" /* OLLAMA */,
         baseUrl: settings.ollamaUrl,
         tokenFactory: () => {
-          let token = authenticationService2.getToken();
+          let token = authenticationService.getToken();
           if (!token) {
             token = localStorage.getItem("authToken");
           }
@@ -43193,7 +43329,7 @@ var AIProviderInitService = class _AIProviderInitService {
             type: "ollama" /* OLLAMA */,
             baseUrl: settings.ollamaUrl || "http://localhost:11434",
             tokenFactory: () => {
-              let token = authenticationService2.getToken();
+              let token = authenticationService.getToken();
               if (!token) {
                 token = localStorage.getItem("authToken");
               }
@@ -43224,7 +43360,7 @@ var AIProviderInitService = class _AIProviderInitService {
             type: "ollama" /* OLLAMA */,
             baseUrl: "http://localhost:11434",
             tokenFactory: () => {
-              let token = authenticationService2.getToken();
+              let token = authenticationService.getToken();
               if (!token) {
                 token = localStorage.getItem("authToken");
               }
@@ -43314,7 +43450,7 @@ var AIProviderInitService = class _AIProviderInitService {
         return config;
       }
       config.tokenFactory = () => {
-        let token = authenticationService2.getToken();
+        let token = authenticationService.getToken();
         if (!token && typeof localStorage !== "undefined") {
           try {
             token = localStorage.getItem("authToken");
@@ -43573,7 +43709,6 @@ var ProviderTab = () => {
         usePackageSettingsStore.setState({ settings: updatedSettings });
       }
       setProviderConfig(applyDefaultModel(normalizedConfig));
-      setIsProviderConfigOpen(false);
       showMessage("Provider configuration saved and switched successfully!", "success");
       debugLogger.info("Provider configuration saved and switched", { type: normalizedConfig.type });
     } catch (error) {
@@ -43961,7 +44096,7 @@ function buildUrl3(path) {
 }
 function authHeaders() {
   const headers = { "Content-Type": "application/json" };
-  const token = authenticationService2.getToken();
+  const token = authenticationService.getToken();
   if (token) headers["Authorization"] = `Bearer ${token}`;
   return headers;
 }
@@ -45306,7 +45441,7 @@ var Management = () => {
     let isActive = true;
     const loadCapabilities = async () => {
       try {
-        const token = authToken ?? authenticationService2.getToken();
+        const token = authToken ?? authenticationService.getToken();
         const headers = {
           "Content-Type": "application/json"
         };
@@ -45442,7 +45577,7 @@ var Management = () => {
     "admin",
     "platform_admin"
   ]);
-  const userRoles = (authenticationService2.parseJwtClaims(authenticationService2.getToken() ?? "")?.roles ?? []).map((r) => r.toLowerCase());
+  const userRoles = (authenticationService.parseJwtClaims(authenticationService.getToken() ?? "")?.roles ?? []).map((r) => r.toLowerCase());
   const isWorkspaceAdmin = userRoles.some(
     (r) => platformAdminRoles.has(r) || r.endsWith("-admin")
   );