@arbidocs/react 0.4.10 → 0.4.12

package/dist/index.d.cts

@@ -82,6 +82,8 @@ interface ArbiChatMessage {
     citations?: ArbiChatCitation[];
     /** The reader stopped the answer; `content` is what had streamed. */
     stopped?: boolean;
+    /** The backend id of a finished assistant answer: what a follow-up chains from. */
+    responseId?: string;
 }
 interface ArbiChatThread {
     id: string;
@@ -100,6 +102,15 @@ interface UseArbiChatOptions {
     model?: string;
     /** Longest auto-title in characters (used until the backend has titled the conversation). Default 48. */
     titleLength?: number;
+    /**
+     * When the answer stream ends without the terminal `response.completed` event (a dropped
+     * connection, a stalled run), the stored message is polled this often (ms) and for at most
+     * `timeoutMs` before the turn is shown as failed with a Retry. Default 3000 / 120000.
+     */
+    streamRecovery?: {
+        intervalMs?: number;
+        timeoutMs?: number;
+    };
 }
 interface ArbiChat {
     threads: ArbiChatThread[];

package/dist/index.d.ts

@@ -82,6 +82,8 @@ interface ArbiChatMessage {
     citations?: ArbiChatCitation[];
     /** The reader stopped the answer; `content` is what had streamed. */
     stopped?: boolean;
+    /** The backend id of a finished assistant answer: what a follow-up chains from. */
+    responseId?: string;
 }
 interface ArbiChatThread {
     id: string;
@@ -100,6 +102,15 @@ interface UseArbiChatOptions {
     model?: string;
     /** Longest auto-title in characters (used until the backend has titled the conversation). Default 48. */
     titleLength?: number;
+    /**
+     * When the answer stream ends without the terminal `response.completed` event (a dropped
+     * connection, a stalled run), the stored message is polled this often (ms) and for at most
+     * `timeoutMs` before the turn is shown as failed with a Retry. Default 3000 / 120000.
+     */
+    streamRecovery?: {
+        intervalMs?: number;
+        timeoutMs?: number;
+    };
 }
 interface ArbiChat {
     threads: ArbiChatThread[];

package/dist/index.js

@@ -1,10 +1,7 @@
 import { formatRelativeTime } from './chunk-FSMIN5VX.js';
 export { DEFAULT_NOTIFY_TONE, NotificationBell, Toaster, formatRelativeTime, notify, seedNotifications, selectUnread, toast, useNotify } from './chunk-FSMIN5VX.js';
-import { useDocuments, useConversations, useAssistantQuery, useConversationThreads, arbiQueryKeys, useWorkspaces, useParsedDocument, useTags, useActivity, useThumbnails } from './chunk-UV5ZEUNV.js';
-export { DEFAULT_USD_PER_CREDIT, arbiQueryKeys, cleanDraftMarkdown, creditsToUsd, formatCredits, isUsageMetered, promoteBoldTitle, repairParsedTitle, stripAgentPreamble, unescapeMarkdown, unwrapCitationLinks, useActivity, useAgents, useAssistantQuery, useConfigs, useConversationThreads, useConversations, useDeleteAgents, useDocuments, useFacts, useHealth, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSessions, useTags, useTasks, useThumbnails, useUserUsageToday, useWorkspaces } from './chunk-UV5ZEUNV.js';
-import { AudioViewer, ImageViewer, VideoViewer } from './chunk-HDCRPPBW.js';
-import { useArbi } from './chunk-J4MX2W4H.js';
-export { ArbiProvider, useArbi } from './chunk-J4MX2W4H.js';
+import { useDocuments, useConversations, useAssistantQuery, useConversationThreads, arbiQueryKeys, useWorkspaces, useParsedDocument, useTags, useActivity, useThumbnails } from './chunk-QFELRTTG.js';
+export { DEFAULT_USD_PER_CREDIT, arbiQueryKeys, cleanDraftMarkdown, creditsToUsd, formatCredits, isUsageMetered, promoteBoldTitle, repairParsedTitle, stripAgentPreamble, unescapeMarkdown, unwrapCitationLinks, useActivity, useAgents, useAssistantQuery, useConfigs, useConversationThreads, useConversations, useDeleteAgents, useDocuments, useFacts, useHealth, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSessions, useTags, useTasks, useThumbnails, useUserUsageToday, useWorkspaces } from './chunk-QFELRTTG.js';
 import { MarkMenu, AiMarkdown, Card, Button, Badge, TagEntity, Input, FILE_TYPE_COLORS, getEntityIcon, UserEntity, getEntityColorStyles, AgentBadge, Textarea, Label, Slider, Dialog, DialogContent, DialogHeader, DialogTitle, DialogFooter, DocumentEntity, DocumentCard, useAudioRecorder, MicWaveformIndicator, IconToggle, classifyMediaError, formatElapsed, PassageCard } from './chunk-NVMQLJI4.js';
 export { AgentBadge, AiMarkdown, ArbiBotHeadIcon, AudIcon, Avatar, AvatarFallback, AvatarImage, BASE_COLORS, Badge, Button, Card, CardContent, CardDescription, CardFooter, CardHeader, CardTitle, Checkbox, Collapsible, CollapsibleContent, CollapsibleTrigger, Command, CommandDialog, CommandEmpty, CommandGroup, CommandInput, CommandItem, CommandList, CommandSeparator, CommandShortcut, Dialog, DialogClose, DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogOverlay, DialogPortal, DialogTitle, DialogTrigger, DictateProvider, DocumentCard, DocumentEntity, ENTITY_MAPPINGS, EmlIcon, EntityIcon, EntityPill, FILE_TYPE_COLORS, HoverCard, HoverCardContent, HoverCardTrigger, IconToggle, ImgIcon, InlineMicButton, Input, Label, MarkMenu, MicRecordingOverlay, MicWaveformIndicator, NEUTRAL_GRAY, PLACEHOLDER_COLOR, PassageCard, PdfIcon, Popover, PopoverAnchor, PopoverContent, PopoverTrigger, PresenceLabel, Progress, ProjectEntity, RadioGroup, RadioGroupItem, RagDot, RelevanceDot, Select, SelectContent, SelectGroup, SelectItem, SelectLabel, SelectScrollDownButton, SelectScrollUpButton, SelectSeparator, SelectTrigger, SelectValue, Separator, Slider, StatusPill, Switch, TAG_TYPE_ICONS, TAG_TYPE_LABELS, Table, TableBody, TableCaption, TableCell, TableFooter, TableHead, TableHeader, TableRow, Tabs, TabsContent, TabsList, TabsTrigger, TagEntity, Textarea, Toggle, ToggleGroup, ToggleGroupItem, Tooltip, TooltipContent, TooltipProvider, TooltipTrigger, UserEntity, UserMenu, VISIBILITY_ICONS, VidIcon, VisibilityIcon, badgeVariants, buttonVariants, cardVariants, configureEntityRegistry, configureRagLabels, configureStatusRegistry, formatElapsed, getDocIcon, getEntityColor, getEntityColorStyles, getEntityIcon, getEntityLabel, getStatusVariant, getTagTypeIcon, getTagTypeLabel, getWorkspaceColor, toggleVariants, useAudioRecorder } from './chunk-NVMQLJI4.js';
 import { ScrollArea, DropdownMenu, DropdownMenuTrigger, DropdownMenuContent, DropdownMenuItem, DropdownMenuSeparator } from './chunk-N5SVBDLC.js';
@@ -13,6 +10,9 @@ import { normalizeCitationMarkup, MarkdownRenderer } from './chunk-TGGQGFS3.js';
 export { MarkdownRenderer, MarkdownStreamingContext, closeOpenEmphasis, markdownBaseComponents, normalizeCitationMarkup, splitMarkdownBlocks, trimOpenCitation } from './chunk-TGGQGFS3.js';
 import { cn } from './chunk-SN5YV6FF.js';
 export { cn } from './chunk-SN5YV6FF.js';
+import { AudioViewer, ImageViewer, VideoViewer } from './chunk-HDCRPPBW.js';
+import { useArbi } from './chunk-J4MX2W4H.js';
+export { ArbiProvider, useArbi } from './chunk-J4MX2W4H.js';
 import React, { memo, useState, createContext, useMemo, useCallback, useEffect, useRef, useLayoutEffect, useContext, useId, Fragment as Fragment$1 } from 'react';
 import { resolveCitations, summarizeCitations, connectWithReconnect } from '@arbidocs/sdk/browser';
 export { Arbi, ArbiError } from '@arbidocs/sdk/browser';
@@ -75,22 +75,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;
@@ -103,7 +119,9 @@ 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 = useArbi();
   const { data: docs = [], isLoading: isLoadingDocuments, refetch } = useDocuments(workspaceId);
   const {
@@ -167,13 +185,14 @@ function useArbiChat(options = {}) {
     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
       )
     );
@@ -212,6 +231,29 @@ function useArbiChat(options = {}) {
       (delay) => setTimeout(() => void refetchConversations(), delay)
     );
   }, [refetchConversations]);
+  const recoverStoredAnswer = 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 = useCallback(
     async (thread, text) => {
       const userMessage = {
@@ -237,6 +279,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,
@@ -246,6 +299,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
@@ -253,6 +307,7 @@ function useArbiChat(options = {}) {
             }
           },
           onUserMessage: (ev) => {
+            conversationId = conversationId ?? ev.conversation_ext_id;
             patchThread(thread.key, (t) => ({
               ...t,
               conversationId: t.conversationId ?? ev.conversation_ext_id
@@ -279,20 +334,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)) {
@@ -302,10 +380,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;
@@ -313,7 +388,16 @@ function useArbiChat(options = {}) {
         refreshTitles();
       }
     },
-    [mutateAsync, patchMessage, patchThread, refreshTitles, selectedIds, model, titleLength]
+    [
+      mutateAsync,
+      patchMessage,
+      patchThread,
+      refreshTitles,
+      recoverStoredAnswer,
+      selectedIds,
+      model,
+      titleLength
+    ]
   );
   const send = useCallback(
     async (text) => {
@@ -332,7 +416,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]);