@arbidocs/blocks 0.4.73 → 0.4.75

package/dist/connect/index.js

@@ -1,30 +1,30 @@
 "use client";
 import { SettingsSection, ProfileCard, AvatarPicker, IdentityVerificationCard, IdentityVerificationDialog, RequireVerified, SettingsCard, SettingsActionRow, PasswordChangeDialog, EMAIL_NOTIFICATION_KEYS, EMAIL_NOTIFICATION_CATEGORIES, NotificationCategoryList, toNotificationCategories, toActiveSessionViews, SettingsRow, ActiveSessionsDialog, LogoutDialog, toLogoutSessionViews, SettingsDangerZone, DeleteAccountDialog, toDeleteBlockerViews } from '../chunk-GJV2G4FC.js';
-import { useWorkspaceDocs } from '../chunk-ORA2XE7U.js';
-export { ConnectedAssistantDock, redlineStats, textToArtifact, toRedlineMarkdown, useConnectedAiTask, useConnectedChat, useSemanticSearch, useWorkspaceDocs } from '../chunk-ORA2XE7U.js';
+import { useWorkspaceDocs } from '../chunk-MK6NJQGF.js';
+export { ConnectedAssistantDock, redlineStats, textToArtifact, toRedlineMarkdown, useConnectedAiTask, useConnectedChat, useSemanticSearch, useWorkspaceDocs } from '../chunk-MK6NJQGF.js';
 import { ConnectionProvider, useConnection } from '../chunk-OYMWV45N.js';
 export { ConnectionProvider, SESSION_EXPIRED_MESSAGE, isAuthenticated, useConnection } from '../chunk-OYMWV45N.js';
-import { WorkflowRunner, initialRunInputs, workflowSteps, fromPlanSteps, stepStatesFromActivity, toPlanSteps, runThreadTitle, useWorkflowView } from '../chunk-2P3T23C7.js';
-import '../chunk-EVUEIBNU.js';
-import '../chunk-XD2M6LIG.js';
 import { StreamedArtifactFrame } from '../chunk-LOJVUGBJ.js';
 import '../chunk-VILAPTCB.js';
-import { ChatPage } from '../chunk-G5DP27DU.js';
+import { WorkflowRunner, initialRunInputs, workflowSteps, fromPlanSteps, stepStatesFromActivity, toPlanSteps, runThreadTitle, useWorkflowView } from '../chunk-YWYVIQ7T.js';
+import '../chunk-EVUEIBNU.js';
+import '../chunk-XD2M6LIG.js';
+import { ChatPage } from '../chunk-UIWLDSTF.js';
 import '../chunk-KZJAG7T3.js';
-import '../chunk-3VQYJBOI.js';
+import '../chunk-HOXKR26S.js';
 import '../chunk-AA3PT6KR.js';
 import '../chunk-NPLYHR2E.js';
 import { initials } from '../chunk-OXF2IFXG.js';
 import '../chunk-TX2ZICS7.js';
 import '../chunk-7F6CBP3Z.js';
 import '../chunk-VUNV25KB.js';
-import { lazy, useState, Suspense, useCallback, useMemo, useEffect } from 'react';
+import { lazy, useState, Suspense, useCallback, useEffect, useMemo, useRef } from 'react';
 import { QueryClient, QueryClientProvider, useQueryClient, useQuery } from '@tanstack/react-query';
 import { ArbiProvider, useConversations, useRunConversation, useUploadSkillFiles, ArtifactCanvas, useArbi, useDocuments, useWorkspaceUsers, useWorkspaces, useShellWorkspaces, useProfile, useUpdateProfile, useRequestEmailChange, useConfirmEmailChange, useNameEditor, useAvatarCrop, useEmailChange, ImageCropModal, useIdentityVerification, usePasswordChange, usePasswordChangeForm, useNotificationPreferences, useSessions, useCurrentSessionId, useTerminateSession, useLogoutAllSessions, useDeleteAccount, useDeleteAccountFlow, useArbiWebSocket, notify, useWorkNotifications, useAgentSession, useStartAgentSession, useCloseAgentSession, AiAnswer, ArbiWebSocketProvider, useCreateWork, useActOnStep, useConversationActivity, useConversationThreads, useArbiChat } from '@arbidocs/react';
 import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { Button, Label, Input, cn } from '@arbidocs/react/ui';
 import { useStreamedArtifact } from '@arbidocs/react/artifacts';
-import { ExternalLink, UserCircle, KeyRound, Radio } from 'lucide-react';
+import { ExternalLink, UserCircle, KeyRound, Radio, Loader2 } from 'lucide-react';
 import { useNotify } from '@arbidocs/react/shell';
 
 function ConnectionBoundWebSocket({ children }) {
@@ -188,6 +188,28 @@ function formValues(step, inputs) {
   if (step.kind !== "form") return void 0;
   return Object.fromEntries(step.inputs.map((f) => [f.name, inputs[f.name] ?? ""]));
 }
+var ANSWER_RETRY_CAP = 16;
+function answerRetryDelay(attempt) {
+  return attempt === 0 ? 0 : Math.min(1e3 * 2 ** (attempt - 1), 15e3);
+}
+function useAnswerRetry(missing, refetch) {
+  const [attempt, setAttempt] = useState(0);
+  const [prev, setPrev] = useState(missing);
+  if (prev !== missing) {
+    setPrev(missing);
+    setAttempt(0);
+  }
+  const refetchRef = useRef(refetch);
+  refetchRef.current = refetch;
+  useEffect(() => {
+    if (!missing || attempt >= ANSWER_RETRY_CAP) return;
+    const t = setTimeout(() => {
+      void refetchRef.current();
+      setAttempt((a) => a + 1);
+    }, answerRetryDelay(attempt));
+    return () => clearTimeout(t);
+  }, [missing, attempt]);
+}
 function useWorkflowThread(workflow, options = {}) {
   const arbi = useArbi();
   const { ensureWorkspaceOpen } = useConnection();
@@ -218,16 +240,14 @@ function useWorkflowThread(workflow, options = {}) {
   const answers = useMemo(() => {
     const byId = /* @__PURE__ */ new Map();
     for (const t of threads.data?.threads ?? [])
-      for (const m of t.history) if (m.role === "assistant") byId.set(m.external_id, m.content);
+      for (const m of t.history)
+        if (m.role === "assistant" && m.content?.trim()) byId.set(m.external_id, m.content);
     return byId;
   }, [threads.data]);
-  const answerMissing = Object.values(serverStates).some(
-    (st) => st.status === "done" && !!st.responseId && !answers.has(st.responseId)
-  );
-  const refetchThreads = threads.refetch;
-  useEffect(() => {
-    if (answerMissing) void refetchThreads();
-  }, [answerMissing, activity.data, refetchThreads]);
+  const missing = Object.values(serverStates).flatMap(
+    (st) => st.status === "done" && st.responseId && !answers.has(st.responseId) ? [st.responseId] : []
+  ).sort().join(" ");
+  useAnswerRetry(missing, threads.refetch);
   const states = useMemo(() => {
     if (!threadId) return localStates;
     const out = {};
@@ -323,7 +343,8 @@ function useWorkflowThread(workflow, options = {}) {
 var DEFAULT_CONNECTED_WORKFLOW_RUNNER_LABELS = {
   notify: "Ask to approve",
   notified: "Notified.",
-  openThread: "Open the thread"
+  openThread: "Open the thread",
+  loadingAnswer: "Fetching the answer\u2026"
 };
 function ConnectedWorkflowRunner({
   workflow,
@@ -341,13 +362,17 @@ function ConnectedWorkflowRunner({
   const { data: docs = [], refetch } = useDocuments(wsId);
   const [uploading, setUploading] = useState(false);
   const { view, conversationId } = useWorkflowThread(workflow, options);
+  const [added, setAdded] = useState({});
   const onUpload = useCallback(
-    async (files) => {
-      if (!files.length) return;
+    async (files2, step) => {
+      if (!files2.length) return;
       setUploading(true);
       try {
         await ensureWorkspaceOpen();
-        for (const f of files) await arbi.documents.uploadFile(f, f.name);
+        for (const f of files2) {
+          const { doc_ext_ids = [] } = await arbi.documents.uploadFile(f, f.name) ?? {};
+          setAdded((a) => ({ ...a, ...Object.fromEntries(doc_ext_ids.map((id) => [id, step.id])) }));
+        }
         await refetch();
       } finally {
         setUploading(false);
@@ -355,9 +380,35 @@ function ConnectedWorkflowRunner({
     },
     [arbi, ensureWorkspaceOpen, refetch]
   );
+  const files = docs.map((d) => ({
+    id: d.external_id,
+    name: d.file_name ?? d.external_id,
+    status: fileStatus(d.status),
+    addedFor: added[d.external_id]
+  }));
+  const reading = files.some((f) => f.addedFor && f.status === "processing");
+  useEffect(() => {
+    if (!reading) return;
+    const t = setInterval(() => void refetch(), READ_POLL_MS);
+    return () => clearInterval(t);
+  }, [reading, refetch]);
   const renderAgentOutput = (step, state) => {
     const status = state?.status;
     if (!state?.text && status !== "done") return null;
+    if (!state?.text && state?.responseId) {
+      return /* @__PURE__ */ jsxs(
+        "p",
+        {
+          className: "flex items-center gap-2 text-sm text-muted-foreground",
+          "data-testid": `${testIdPrefix}-${step.id}-answer-loading`,
+          children: [
+            /* @__PURE__ */ jsx(Loader2, { className: "size-4 animate-spin" }),
+            " ",
+            l.loadingAnswer
+          ]
+        }
+      );
+    }
     return /* @__PURE__ */ jsx(
       AiAnswer,
       {
@@ -401,12 +452,20 @@ function ConnectedWorkflowRunner({
         ) : null,
         onUpload,
         uploading,
-        files: docs.map((d) => ({ id: d.external_id, name: d.file_name ?? d.external_id })),
+        files,
         testIdPrefix
       }
     )
   ] });
 }
+var READ_POLL_MS = 3e3;
+function fileStatus(status) {
+  const s = status?.toLowerCase();
+  if (!s) return void 0;
+  if (s === "failed") return "failed";
+  if (s === "completed" || s === "empty" || s === "low-content") return "ready";
+  return "processing";
+}
 function ReviewNotify({
   conversationId,
   workspaceId,