@arbidocs/react 0.4.11 → 0.4.13

package/dist/index.cjs

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkVZLGOA4A_cjs = require('./chunk-VZLGOA4A.cjs');
+var chunk3MFESAPI_cjs = require('./chunk-3MFESAPI.cjs');
 var chunkPGLV23SB_cjs = require('./chunk-PGLV23SB.cjs');
 var chunkTA3IZP7I_cjs = require('./chunk-TA3IZP7I.cjs');
 var chunkWBWXP2DN_cjs = require('./chunk-WBWXP2DN.cjs');
@@ -74,22 +74,38 @@ var storedThread = (conversationId) => ({
   isLoaded: false
 });
 var publicId = (t) => t.conversationId ?? t.key;
+var UNFINISHED_STATUSES = /* @__PURE__ */ new Set(["failed", "in_progress", "queued"]);
+var FAILED_MESSAGE = "The assistant request failed.";
+var UNFINISHED_MESSAGE = "The answer did not finish. Try again.";
+var isUnfinished = (m) => UNFINISHED_STATUSES.has(m.status);
 function messagesFromHistory(history) {
-  return history.filter((m) => m.role !== "system" && m.content.trim() !== "").map((m) => {
+  return history.filter(
+    (m) => m.role !== "system" && (m.content.trim() !== "" || m.role === "assistant" && isUnfinished(m))
+  ).map((m) => {
     if (m.role === "user") {
       return { id: m.external_id, role: "user", content: m.content, status: "complete" };
     }
     const citations = toChatCitations({ tools: m.tools });
+    const unfinished = isUnfinished(m);
     return {
       id: m.external_id,
       role: "assistant",
       content: m.content,
-      status: m.status === "failed" ? "error" : "complete",
-      ...m.status === "failed" ? { error: "The assistant request failed." } : {},
+      status: unfinished ? "error" : "complete",
+      ...unfinished ? { error: m.status === "failed" ? FAILED_MESSAGE : UNFINISHED_MESSAGE } : {},
+      ...unfinished ? {} : { responseId: m.external_id },
       ...citations.length > 0 ? { citations } : {}
     };
   });
 }
+function chainId(messages) {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const m = messages[i];
+    if (m.role === "assistant" && m.status === "complete" && m.responseId) return m.responseId;
+  }
+  return null;
+}
+var sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
 function titleFromQuestion(question, maxLength = 48) {
   const flat = question.replace(/\s+/g, " ").trim();
   if (flat.length <= maxLength) return flat;
@@ -102,15 +118,17 @@ var messageOf = (e, fallback) => e instanceof Error ? e.message : fallback;
 var isAbort = (e) => e instanceof DOMException && e.name === "AbortError";
 var TITLE_REFRESH_DELAYS_MS = [2500, 6e3, 12e3];
 function useArbiChat(options = {}) {
-  const { workspaceId, model, titleLength = 48 } = options;
+  const { workspaceId, model, titleLength = 48, streamRecovery } = options;
+  const recoveryIntervalMs = streamRecovery?.intervalMs ?? 3e3;
+  const recoveryTimeoutMs = streamRecovery?.timeoutMs ?? 12e4;
   const arbi = chunkZMTLGO3K_cjs.useArbi();
-  const { data: docs = [], isLoading: isLoadingDocuments, refetch } = chunkVZLGOA4A_cjs.useDocuments(workspaceId);
+  const { data: docs = [], isLoading: isLoadingDocuments, refetch } = chunk3MFESAPI_cjs.useDocuments(workspaceId);
   const {
     data: stored = [],
     isLoading: isLoadingThreads,
     refetch: refetchConversations
-  } = chunkVZLGOA4A_cjs.useConversations(workspaceId);
-  const query = chunkVZLGOA4A_cjs.useAssistantQuery();
+  } = chunk3MFESAPI_cjs.useConversations(workspaceId);
+  const query = chunk3MFESAPI_cjs.useAssistantQuery();
   const { mutateAsync } = query;
   const [threads, setThreads] = React.useState(() => [emptyThread()]);
   const [activeKey, setActiveKey] = React.useState(() => threads[0].key);
@@ -159,20 +177,21 @@ function useArbiChat(options = {}) {
     ];
   }, [threads, stored]);
   const needsLoad = !!active.conversationId && !active.isLoaded;
-  const loaded = chunkVZLGOA4A_cjs.useConversationThreads(workspaceId, active.conversationId ?? void 0, {
+  const loaded = chunk3MFESAPI_cjs.useConversationThreads(workspaceId, active.conversationId ?? void 0, {
     enabled: needsLoad
   });
   React.useEffect(() => {
     const data = loaded.data;
     if (!needsLoad || !data || data.conversation_ext_id !== active.conversationId) return;
     const primary = [...data.threads].sort((a, b) => b.history.length - a.history.length)[0];
+    const messages = messagesFromHistory(primary?.history ?? []);
     setThreads(
       (all) => all.map(
         (t) => t.key === active.key ? {
           ...t,
           isLoaded: true,
-          messages: messagesFromHistory(primary?.history ?? []),
-          responseId: primary?.leaf_message_ext_id ?? null
+          messages,
+          responseId: chainId(messages)
         } : t
       )
     );
@@ -211,6 +230,29 @@ function useArbiChat(options = {}) {
       (delay) => setTimeout(() => void refetchConversations(), delay)
     );
   }, [refetchConversations]);
+  const recoverStoredAnswer = React.useCallback(
+    async (conversationId, messageId, signal) => {
+      if (!conversationId || !messageId) return null;
+      const deadline = Date.now() + recoveryTimeoutMs;
+      let failedReads = 0;
+      for (; ; ) {
+        try {
+          const { threads: all } = await arbi.conversations.getThreads(conversationId);
+          const found = all.flatMap((t) => t.history).find((m) => m.external_id === messageId);
+          if (found?.status === "completed" && found.content.trim() !== "") {
+            return { kind: "completed", message: found };
+          }
+          if (found?.status === "failed") return { kind: "failed" };
+          failedReads = 0;
+        } catch {
+          if (++failedReads >= 3) return null;
+        }
+        if (signal.aborted || Date.now() + recoveryIntervalMs > deadline) return null;
+        await sleep(recoveryIntervalMs);
+      }
+    },
+    [arbi, recoveryIntervalMs, recoveryTimeoutMs]
+  );
   const ask = React.useCallback(
     async (thread, text) => {
       const userMessage = {
@@ -236,6 +278,17 @@ function useArbiChat(options = {}) {
       let buffer = "";
       let steps = 0;
       let streamError = null;
+      let conversationId = thread.conversationId;
+      let backendMessageId = null;
+      const revertChain = () => patchThread(thread.key, (t) => ({ ...t, responseId: chainId(t.messages) }));
+      const fail = (error2, content) => {
+        patchMessage(thread.key, answer.id, {
+          status: "error",
+          error: error2,
+          ...content !== void 0 ? { content } : {}
+        });
+        revertChain();
+      };
       try {
         const result = await mutateAsync({
           question: text,
@@ -245,6 +298,7 @@ function useArbiChat(options = {}) {
           signal: controller.signal,
           onStreamStart: (data) => {
             if (data.assistant_message_ext_id) {
+              backendMessageId = data.assistant_message_ext_id;
               patchThread(thread.key, (t) => ({
                 ...t,
                 responseId: data.assistant_message_ext_id ?? t.responseId
@@ -252,6 +306,7 @@ function useArbiChat(options = {}) {
             }
           },
           onUserMessage: (ev) => {
+            conversationId = conversationId ?? ev.conversation_ext_id;
             patchThread(thread.key, (t) => ({
               ...t,
               conversationId: t.conversationId ?? ev.conversation_ext_id
@@ -278,20 +333,43 @@ function useArbiChat(options = {}) {
         });
         const finalText = result.text || buffer;
         const failure = streamError || result.errors[0] || null;
+        const messageId = result.assistantMessageExtId ?? backendMessageId;
         if (failure && !finalText) {
-          patchMessage(thread.key, answer.id, { status: "error", error: failure });
+          fail(failure);
+        } else if (!result.completed && !failure) {
+          const stored2 = await recoverStoredAnswer(conversationId, messageId, controller.signal);
+          if (stored2?.kind === "completed") {
+            const citations = toChatCitations({ tools: stored2.message.tools });
+            patchMessage(thread.key, answer.id, {
+              status: "complete",
+              content: stored2.message.content,
+              stepCount: result.agentSteps.length || steps || void 0,
+              responseId: stored2.message.external_id,
+              ...citations.length > 0 ? { citations } : {}
+            });
+            patchThread(thread.key, (t) => ({ ...t, responseId: stored2.message.external_id }));
+          } else if (controller.signal.aborted) {
+            patchMessage(thread.key, answer.id, {
+              status: "complete",
+              content: finalText,
+              stopped: true
+            });
+          } else {
+            fail(
+              stored2?.kind === "failed" ? FAILED_MESSAGE : UNFINISHED_MESSAGE,
+              finalText || void 0
+            );
+          }
         } else {
           const citations = toChatCitations(result.metadata);
           patchMessage(thread.key, answer.id, {
             status: "complete",
             content: finalText,
             stepCount: result.agentSteps.length || steps || void 0,
+            ...messageId ? { responseId: messageId } : {},
             ...citations.length > 0 ? { citations } : {}
           });
-          patchThread(thread.key, (t) => ({
-            ...t,
-            responseId: result.assistantMessageExtId ?? t.responseId
-          }));
+          patchThread(thread.key, (t) => ({ ...t, responseId: messageId ?? t.responseId }));
         }
       } catch (e) {
         if (controller.signal.aborted || isAbort(e)) {
@@ -301,10 +379,7 @@ function useArbiChat(options = {}) {
             stopped: true
           });
         } else {
-          patchMessage(thread.key, answer.id, {
-            status: "error",
-            error: messageOf(e, "The assistant request failed.")
-          });
+          fail(messageOf(e, FAILED_MESSAGE));
         }
       } finally {
         if (abortRef.current === controller) abortRef.current = null;
@@ -312,7 +387,16 @@ function useArbiChat(options = {}) {
         refreshTitles();
       }
     },
-    [mutateAsync, patchMessage, patchThread, refreshTitles, selectedIds, model, titleLength]
+    [
+      mutateAsync,
+      patchMessage,
+      patchThread,
+      refreshTitles,
+      recoverStoredAnswer,
+      selectedIds,
+      model,
+      titleLength
+    ]
   );
   const send = React.useCallback(
     async (text) => {
@@ -331,7 +415,8 @@ function useArbiChat(options = {}) {
     const failed = messages[messages.length - 1];
     const question = messages[messages.length - 2];
     if (failed?.status !== "error" || question?.role !== "user") return;
-    const trimmed = { ...active, messages: messages.slice(0, -2) };
+    const rest = messages.slice(0, -2);
+    const trimmed = { ...active, messages: rest, responseId: chainId(rest) };
     patchThread(active.key, () => trimmed);
     await ask(trimmed, question.content);
   }, [active, ask, isRunning, patchThread]);
@@ -699,14 +784,14 @@ function useArbiDocumentSync(workspaceId) {
     if (!workspaceId) return;
     if (client.isMessageType(msg, "batch_complete")) {
       if (msg.workspace_ext_id !== workspaceId) return;
-      void queryClient.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.documents(workspaceId) });
-      void queryClient.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.tags(workspaceId) });
+      void queryClient.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.documents(workspaceId) });
+      void queryClient.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.tags(workspaceId) });
       return;
     }
     if (client.isMessageType(msg, "task_update")) {
       if (msg.workspace_ext_id !== workspaceId) return;
       if (TERMINAL_STATUSES.has(msg.status)) {
-        void queryClient.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.documents(workspaceId) });
+        void queryClient.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.documents(workspaceId) });
       }
     }
   });
@@ -835,7 +920,7 @@ function useShellWorkspaces(options) {
 }
 function useArbiShellWorkspaces(options = {}) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
-  const query = chunkVZLGOA4A_cjs.useWorkspaces();
+  const query = chunk3MFESAPI_cjs.useWorkspaces();
   const { select, ...rest } = options;
   const defaultSelect = React.useCallback(
     (workspace) => arbi.selectWorkspace(workspace.id),
@@ -854,7 +939,7 @@ function useCreateTask() {
   const qc = reactQuery.useQueryClient();
   return reactQuery.useMutation({
     mutationFn: (body) => arbi.tasks.create(body),
-    onSuccess: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.tasks() })
+    onSuccess: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.tasks() })
   });
 }
 function useUpdateTask() {
@@ -862,7 +947,7 @@ function useUpdateTask() {
   const qc = reactQuery.useQueryClient();
   return reactQuery.useMutation({
     mutationFn: ({ taskId, patch }) => arbi.tasks.update(taskId, patch),
-    onSuccess: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.tasks() })
+    onSuccess: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.tasks() })
   });
 }
 function useRunTask() {
@@ -871,8 +956,8 @@ function useRunTask() {
   return reactQuery.useMutation({
     mutationFn: (taskId) => arbi.tasks.runWithAuth(taskId),
     onSuccess: (_task, taskId) => Promise.all([
-      qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.tasks() }),
-      qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.taskRuns(taskId) })
+      qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.tasks() }),
+      qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.taskRuns(taskId) })
     ])
   });
 }
@@ -882,15 +967,15 @@ function useDeleteTask() {
   return reactQuery.useMutation({
     mutationFn: (taskId) => arbi.tasks.delete(taskId),
     onSuccess: (_result, taskId) => {
-      qc.removeQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.taskRuns(taskId) });
-      return qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.tasks() });
+      qc.removeQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.taskRuns(taskId) });
+      return qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.tasks() });
     }
   });
 }
 function useTaskRuns(taskId, options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.taskRuns(taskId ?? ""),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.taskRuns(taskId ?? ""),
     queryFn: () => arbi.tasks.listRuns(taskId),
     enabled: (options?.enabled ?? true) && !!taskId && arbi.isLoggedIn,
     refetchInterval: (q) => q.state.data?.some((r) => r.status === "running") ? 3e3 : false
@@ -901,13 +986,13 @@ function useCreateAgent() {
   const qc = reactQuery.useQueryClient();
   return reactQuery.useMutation({
     mutationFn: (name) => arbi.agents.create(name),
-    onSuccess: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agents() })
+    onSuccess: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agents() })
   });
 }
 function useAgentConfig(agentExtId, options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agentConfig(agentExtId),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agentConfig(agentExtId),
     queryFn: async () => agentExtId ? await arbi.agents.getConfig(agentExtId) : await arbi.agentConfig.get("default"),
     enabled: (options?.enabled ?? true) && arbi.isLoggedIn
   });
@@ -915,7 +1000,7 @@ function useAgentConfig(agentExtId, options) {
 function useAgentModels(options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.models(),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.models(),
     queryFn: async () => (await arbi.agentConfig.getModels()).models.map((m) => m.model_name),
     enabled: (options?.enabled ?? true) && arbi.isLoggedIn
   });
@@ -938,17 +1023,17 @@ function useSaveAgentConfig() {
       });
     },
     onSuccess: (_saved, { agentExtId }) => Promise.all([
-      qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.configs() }),
-      qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agentConfig(agentExtId) }),
+      qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.configs() }),
+      qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agentConfig(agentExtId) }),
       // The agent list shows the pinned persona/model, so it refreshes too.
-      qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agents() })
+      qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agents() })
     ])
   });
 }
 function useAgentWorkspaces(agentExtId, options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agentWorkspaces(agentExtId ?? ""),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agentWorkspaces(agentExtId ?? ""),
     queryFn: () => arbi.agents.listWorkspaces(agentExtId),
     enabled: (options?.enabled ?? true) && !!agentExtId && arbi.isLoggedIn
   });
@@ -969,7 +1054,7 @@ function useGrantAgentAccess() {
       if (!agent?.email) throw new Error("The agent's address is unknown.");
       return arbi.workspaces.grantAccess({ workspaceId, emails: [agent.email] });
     },
-    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agentWorkspaces(agentExtId) })
+    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agentWorkspaces(agentExtId) })
   });
 }
 function useRevokeAgentAccess() {
@@ -988,13 +1073,13 @@ function useRevokeAgentAccess() {
         if (previous && previous !== workspaceId) await arbi.selectWorkspace(previous);
       }
     },
-    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.agentWorkspaces(agentExtId) })
+    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.agentWorkspaces(agentExtId) })
   });
 }
 function useSkills(options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: [...chunkVZLGOA4A_cjs.arbiQueryKeys.skills(), options?.includeHidden ?? false],
+    queryKey: [...chunk3MFESAPI_cjs.arbiQueryKeys.skills(), options?.includeHidden ?? false],
     queryFn: () => arbi.assistant.listSkills({ includeHidden: options?.includeHidden }),
     enabled: (options?.enabled ?? true) && arbi.isLoggedIn,
     staleTime: 3e4
@@ -1003,7 +1088,7 @@ function useSkills(options) {
 function useContacts(options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.contacts(),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.contacts(),
     queryFn: () => arbi.contacts.list(),
     enabled: (options?.enabled ?? true) && arbi.isLoggedIn
   });
@@ -1023,7 +1108,7 @@ function useAddContacts() {
   const qc = reactQuery.useQueryClient();
   return reactQuery.useMutation({
     mutationFn: (emails) => arbi.contacts.add(emails),
-    onSuccess: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.contacts() })
+    onSuccess: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.contacts() })
   });
 }
 function useRemoveContacts() {
@@ -1031,7 +1116,7 @@ function useRemoveContacts() {
   const qc = reactQuery.useQueryClient();
   return reactQuery.useMutation({
     mutationFn: (contactIds) => arbi.contacts.remove(contactIds),
-    onSuccess: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.contacts() })
+    onSuccess: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.contacts() })
   });
 }
 function toProfileView(profile) {
@@ -1043,7 +1128,7 @@ function toProfileView(profile) {
 function useProfile(options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.profile(),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.profile(),
     queryFn: () => {
       const profile = arbi.user.profile();
       return profile ? toProfileView(profile) : null;
@@ -1066,8 +1151,8 @@ function useUpdateProfile() {
       return profile ? toProfileView(profile) : null;
     },
     onSuccess: (profile) => {
-      qc.setQueryData(chunkVZLGOA4A_cjs.arbiQueryKeys.profile(), profile);
-      return qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.workspaces() });
+      qc.setQueryData(chunk3MFESAPI_cjs.arbiQueryKeys.profile(), profile);
+      return qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.workspaces() });
     }
   });
 }
@@ -1152,7 +1237,7 @@ function useConfirmEmailChange() {
     mutationFn: ({ newEmail, code }) => arbi.account.confirmEmailChange(newEmail, code),
     onSuccess: () => {
       const profile = arbi.user.profile();
-      qc.setQueryData(chunkVZLGOA4A_cjs.arbiQueryKeys.profile(), profile ? toProfileView(profile) : null);
+      qc.setQueryData(chunk3MFESAPI_cjs.arbiQueryKeys.profile(), profile ? toProfileView(profile) : null);
     }
   });
 }
@@ -1321,7 +1406,7 @@ function toNotificationPreferences(stored, keys) {
 function useUserSettings(options) {
   const arbi = chunkZMTLGO3K_cjs.useArbi();
   return reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.userSettings(),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.userSettings(),
     queryFn: () => arbi.settings.get(),
     enabled: (options?.enabled ?? true) && arbi.isLoggedIn
   });
@@ -1334,18 +1419,18 @@ function useNotificationPreferences(keys, options) {
   const save = reactQuery.useMutation({
     mutationFn: (patch) => arbi.settings.update({ email_notifications: patch }),
     onMutate: async (patch) => {
-      await qc.cancelQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.userSettings() });
-      const previous = qc.getQueryData(chunkVZLGOA4A_cjs.arbiQueryKeys.userSettings());
+      await qc.cancelQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.userSettings() });
+      const previous = qc.getQueryData(chunk3MFESAPI_cjs.arbiQueryKeys.userSettings());
       qc.setQueryData(
-        chunkVZLGOA4A_cjs.arbiQueryKeys.userSettings(),
+        chunk3MFESAPI_cjs.arbiQueryKeys.userSettings(),
         (old) => old ? { ...old, email_notifications: { ...old.email_notifications, ...patch } } : old
       );
       return { previous };
     },
     onError: (_err, _patch, context) => {
-      qc.setQueryData(chunkVZLGOA4A_cjs.arbiQueryKeys.userSettings(), context?.previous);
+      qc.setQueryData(chunk3MFESAPI_cjs.arbiQueryKeys.userSettings(), context?.previous);
     },
-    onSettled: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.userSettings() })
+    onSettled: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.userSettings() })
   });
   const model = React.useMemo(() => toNotificationPreferences(stored, keys), [stored, keys]);
   const setAll = React.useCallback(
@@ -1371,7 +1456,7 @@ function useTerminateSession() {
   const qc = reactQuery.useQueryClient();
   return reactQuery.useMutation({
     mutationFn: (sessionId) => arbi.sessions.terminate(sessionId),
-    onSuccess: () => qc.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.sessions() })
+    onSuccess: () => qc.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.sessions() })
   });
 }
 function useLogoutAllSessions() {
@@ -1447,7 +1532,7 @@ function useDeleteAccountFlow({
   };
 }
 function useWorkspaceDocs(workspaceId) {
-  const { data: docs = [], isLoading, isError } = chunkVZLGOA4A_cjs.useDocuments(workspaceId);
+  const { data: docs = [], isLoading, isError } = chunk3MFESAPI_cjs.useDocuments(workspaceId);
   const docIds = React.useMemo(
     () => docs.map((d) => d.external_id).filter((id) => !!id),
     [docs]
@@ -1455,7 +1540,7 @@ function useWorkspaceDocs(workspaceId) {
   return { wsId: workspaceId, docs, docIds, isLoading, isError, ready: docIds.length > 0 };
 }
 function useAiTask() {
-  const mutation = chunkVZLGOA4A_cjs.useAssistantQuery();
+  const mutation = chunk3MFESAPI_cjs.useAssistantQuery();
   const [status, setStatus] = React.useState("idle");
   const [text, setText] = React.useState("");
   const [stepCount, setStepCount] = React.useState(0);
@@ -2047,8 +2132,8 @@ function DocumentReview({
   onClose,
   className
 }) {
-  const parsedQuery = chunkVZLGOA4A_cjs.useParsedDocument(docId, stage);
-  const tagsQuery = chunkVZLGOA4A_cjs.useTags(workspaceId);
+  const parsedQuery = chunk3MFESAPI_cjs.useParsedDocument(docId, stage);
+  const tagsQuery = chunk3MFESAPI_cjs.useTags(workspaceId);
   const tagNameById = React__default.default.useMemo(() => {
     const map = /* @__PURE__ */ new Map();
     for (const t of tagsQuery.data ?? []) {
@@ -2279,7 +2364,7 @@ function SemanticSearch({
   const { results, status, error, search, isRunning, query } = useSemanticSearch(docIds);
   const [value, setValue] = React.useState("");
   const [selected, setSelected] = React.useState(null);
-  const { data: docs = [] } = chunkVZLGOA4A_cjs.useDocuments(workspaceId, { enabled: !!workspaceId });
+  const { data: docs = [] } = chunk3MFESAPI_cjs.useDocuments(workspaceId, { enabled: !!workspaceId });
   const docMap = React.useMemo(() => {
     const map = /* @__PURE__ */ new Map();
     for (const d of docs) if (d.external_id) map.set(d.external_id, d);
@@ -3562,7 +3647,7 @@ function ActivityFeed({
   enabled,
   "data-testid": testId = "activity-feed"
 }) {
-  const { data, isLoading, isError } = chunkVZLGOA4A_cjs.useActivity({ limit, enabled });
+  const { data, isLoading, isError } = chunk3MFESAPI_cjs.useActivity({ limit, enabled });
   if (isLoading) {
     return /* @__PURE__ */ jsxRuntime.jsx("div", { className: chunkS6A6FSG7_cjs.cn("space-y-2", className), "data-testid": testId, "aria-busy": "true", children: Array.from({ length: 3 }).map((_, i) => /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-3 py-1.5", children: [
       /* @__PURE__ */ jsxRuntime.jsx("div", { className: "h-4 w-4 shrink-0 animate-pulse rounded-full bg-muted" }),
@@ -3686,7 +3771,7 @@ function useDirectMessages(options) {
     enabled
   });
   const contactsQuery = reactQuery.useQuery({
-    queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.contacts(),
+    queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.contacts(),
     queryFn: () => arbi.contacts.list(),
     enabled
   });
@@ -4721,10 +4806,10 @@ function DocumentTable({
   extraColumns,
   className
 }) {
-  const query = chunkVZLGOA4A_cjs.useDocuments(workspaceId, { enabled: !documents });
+  const query = chunk3MFESAPI_cjs.useDocuments(workspaceId, { enabled: !documents });
   const rows = documents ?? query.data ?? [];
   const isLoading = !documents && query.isLoading;
-  const tagsQuery = chunkVZLGOA4A_cjs.useTags(workspaceId);
+  const tagsQuery = chunk3MFESAPI_cjs.useTags(workspaceId);
   const tagNameById = React__default.default.useMemo(() => {
     const map = /* @__PURE__ */ new Map();
     for (const t of tagsQuery.data ?? []) {
@@ -4905,7 +4990,7 @@ function DocumentCardMenu({
   const invalidateDocuments = () => {
     const workspaceId = arbi.getWorkspaceId();
     if (workspaceId) {
-      void queryClient.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.documents(workspaceId) });
+      void queryClient.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.documents(workspaceId) });
     }
   };
   const download = reactQuery.useMutation({
@@ -5004,10 +5089,10 @@ function DocumentGallery({
   minTileWidth = 148,
   className
 }) {
-  const query = chunkVZLGOA4A_cjs.useDocuments(workspaceId, { enabled: !documents });
+  const query = chunk3MFESAPI_cjs.useDocuments(workspaceId, { enabled: !documents });
   const rows = documents ?? query.data ?? [];
   const isLoading = !documents && query.isLoading;
-  const tagsQuery = chunkVZLGOA4A_cjs.useTags(workspaceId);
+  const tagsQuery = chunk3MFESAPI_cjs.useTags(workspaceId);
   const tagNameById = React__default.default.useMemo(() => {
     const map = /* @__PURE__ */ new Map();
     for (const t of tagsQuery.data ?? []) {
@@ -5019,7 +5104,7 @@ function DocumentGallery({
     () => rows.map((d) => d.external_id).filter((id) => !!id),
     [rows]
   );
-  const thumbsQuery = chunkVZLGOA4A_cjs.useThumbnails(docIds);
+  const thumbsQuery = chunk3MFESAPI_cjs.useThumbnails(docIds);
   const thumbs = thumbsQuery.data ?? {};
   const storeSelected = useArbiUiStore((s) => s.selectedDocIds);
   const storeSetSelected = useArbiUiStore((s) => s.setSelectedDocIds);
@@ -5055,7 +5140,7 @@ function DocumentGallery({
       setRenamingId(null);
       const activeWorkspace = workspaceId ?? arbi.getWorkspaceId();
       if (activeWorkspace) {
-        void queryClient.invalidateQueries({ queryKey: chunkVZLGOA4A_cjs.arbiQueryKeys.documents(activeWorkspace) });
+        void queryClient.invalidateQueries({ queryKey: chunk3MFESAPI_cjs.arbiQueryKeys.documents(activeWorkspace) });
       }
     }
   });
@@ -5673,131 +5758,135 @@ function hasVideoTrack(blob) {
 
 Object.defineProperty(exports, "DEFAULT_USD_PER_CREDIT", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.DEFAULT_USD_PER_CREDIT; }
+  get: function () { return chunk3MFESAPI_cjs.DEFAULT_USD_PER_CREDIT; }
 });
 Object.defineProperty(exports, "arbiQueryKeys", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.arbiQueryKeys; }
+  get: function () { return chunk3MFESAPI_cjs.arbiQueryKeys; }
 });
 Object.defineProperty(exports, "cleanDraftMarkdown", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.cleanDraftMarkdown; }
+  get: function () { return chunk3MFESAPI_cjs.cleanDraftMarkdown; }
 });
 Object.defineProperty(exports, "creditsToUsd", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.creditsToUsd; }
+  get: function () { return chunk3MFESAPI_cjs.creditsToUsd; }
 });
 Object.defineProperty(exports, "formatCredits", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.formatCredits; }
+  get: function () { return chunk3MFESAPI_cjs.formatCredits; }
 });
 Object.defineProperty(exports, "isUsageMetered", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.isUsageMetered; }
+  get: function () { return chunk3MFESAPI_cjs.isUsageMetered; }
 });
 Object.defineProperty(exports, "promoteBoldTitle", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.promoteBoldTitle; }
+  get: function () { return chunk3MFESAPI_cjs.promoteBoldTitle; }
+});
+Object.defineProperty(exports, "repairParsedStructure", {
+  enumerable: true,
+  get: function () { return chunk3MFESAPI_cjs.repairParsedStructure; }
 });
 Object.defineProperty(exports, "repairParsedTitle", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.repairParsedTitle; }
+  get: function () { return chunk3MFESAPI_cjs.repairParsedTitle; }
 });
 Object.defineProperty(exports, "stripAgentPreamble", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.stripAgentPreamble; }
+  get: function () { return chunk3MFESAPI_cjs.stripAgentPreamble; }
 });
 Object.defineProperty(exports, "unescapeMarkdown", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.unescapeMarkdown; }
+  get: function () { return chunk3MFESAPI_cjs.unescapeMarkdown; }
 });
 Object.defineProperty(exports, "unwrapCitationLinks", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.unwrapCitationLinks; }
+  get: function () { return chunk3MFESAPI_cjs.unwrapCitationLinks; }
 });
 Object.defineProperty(exports, "useActivity", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useActivity; }
+  get: function () { return chunk3MFESAPI_cjs.useActivity; }
 });
 Object.defineProperty(exports, "useAgents", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useAgents; }
+  get: function () { return chunk3MFESAPI_cjs.useAgents; }
 });
 Object.defineProperty(exports, "useAssistantQuery", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useAssistantQuery; }
+  get: function () { return chunk3MFESAPI_cjs.useAssistantQuery; }
 });
 Object.defineProperty(exports, "useConfigs", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useConfigs; }
+  get: function () { return chunk3MFESAPI_cjs.useConfigs; }
 });
 Object.defineProperty(exports, "useConversationThreads", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useConversationThreads; }
+  get: function () { return chunk3MFESAPI_cjs.useConversationThreads; }
 });
 Object.defineProperty(exports, "useConversations", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useConversations; }
+  get: function () { return chunk3MFESAPI_cjs.useConversations; }
 });
 Object.defineProperty(exports, "useDeleteAgents", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useDeleteAgents; }
+  get: function () { return chunk3MFESAPI_cjs.useDeleteAgents; }
 });
 Object.defineProperty(exports, "useDocuments", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useDocuments; }
+  get: function () { return chunk3MFESAPI_cjs.useDocuments; }
 });
 Object.defineProperty(exports, "useFacts", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useFacts; }
+  get: function () { return chunk3MFESAPI_cjs.useFacts; }
 });
 Object.defineProperty(exports, "useHealth", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useHealth; }
+  get: function () { return chunk3MFESAPI_cjs.useHealth; }
 });
 Object.defineProperty(exports, "useParsedDocument", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useParsedDocument; }
+  get: function () { return chunk3MFESAPI_cjs.useParsedDocument; }
 });
 Object.defineProperty(exports, "useProjectInvoices", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useProjectInvoices; }
+  get: function () { return chunk3MFESAPI_cjs.useProjectInvoices; }
 });
 Object.defineProperty(exports, "useProjectUsage", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useProjectUsage; }
+  get: function () { return chunk3MFESAPI_cjs.useProjectUsage; }
 });
 Object.defineProperty(exports, "useProjects", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useProjects; }
+  get: function () { return chunk3MFESAPI_cjs.useProjects; }
 });
 Object.defineProperty(exports, "useProjectsUsage", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useProjectsUsage; }
+  get: function () { return chunk3MFESAPI_cjs.useProjectsUsage; }
 });
 Object.defineProperty(exports, "useSessions", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useSessions; }
+  get: function () { return chunk3MFESAPI_cjs.useSessions; }
 });
 Object.defineProperty(exports, "useTags", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useTags; }
+  get: function () { return chunk3MFESAPI_cjs.useTags; }
 });
 Object.defineProperty(exports, "useTasks", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useTasks; }
+  get: function () { return chunk3MFESAPI_cjs.useTasks; }
 });
 Object.defineProperty(exports, "useThumbnails", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useThumbnails; }
+  get: function () { return chunk3MFESAPI_cjs.useThumbnails; }
 });
 Object.defineProperty(exports, "useUserUsageToday", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useUserUsageToday; }
+  get: function () { return chunk3MFESAPI_cjs.useUserUsageToday; }
 });
 Object.defineProperty(exports, "useWorkspaces", {
   enumerable: true,
-  get: function () { return chunkVZLGOA4A_cjs.useWorkspaces; }
+  get: function () { return chunk3MFESAPI_cjs.useWorkspaces; }
 });
 Object.defineProperty(exports, "DEFAULT_NOTIFY_TONE", {
   enumerable: true,