@arbidocs/blocks 0.4.80 → 0.4.81

package/dist/connect/index.js

@@ -1,26 +1,26 @@
 "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-E367ZUHE.js';
-import { useWorkspaceDocs } from '../chunk-WDD255XQ.js';
-export { ConnectedAssistantDock, redlineStats, textToArtifact, toRedlineMarkdown, useConnectedAiTask, useConnectedChat, useSemanticSearch, useWorkspaceDocs } from '../chunk-WDD255XQ.js';
+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-XQDM6T6C.js';
+export { ConnectedAssistantDock, redlineStats, textToArtifact, toRedlineMarkdown, useConnectedAiTask, useConnectedChat, useSemanticSearch, useWorkspaceDocs } from '../chunk-XQDM6T6C.js';
 import { ConnectedNativeDocxEditor } from '../chunk-A24C6BNA.js';
 import { ConnectionProvider, useConnection } from '../chunk-OYMWV45N.js';
 export { ConnectionProvider, SESSION_EXPIRED_MESSAGE, isAuthenticated, useConnection } from '../chunk-OYMWV45N.js';
 import { SignDocumentDialog } from '../chunk-FE4ULGOR.js';
 import '../chunk-OPA2Y6YZ.js';
 import { TagMatrix, tagMatrixMarkdown } from '../chunk-SWSVPQNB.js';
-import { splitDocRef, WorkflowStepBar, stepLevels, WorkflowRunner, initialRunInputs, workflowSteps, fromPlanSteps, stepStatesFromActivity, toPlanSteps, runThreadTitle, useWorkflowView, stepFields } from '../chunk-QFA3NCWL.js';
+import { StreamedArtifactFrame } from '../chunk-LOJVUGBJ.js';
+import { splitDocRef, WorkflowStepBar, stepLevels, WorkflowRunner, initialRunInputs, workflowSteps, fromPlanSteps, stepStatesFromActivity, toPlanSteps, runThreadTitle, useWorkflowView, stepFields } from '../chunk-THCFZKTM.js';
 import '../chunk-EVUEIBNU.js';
 import '../chunk-XD2M6LIG.js';
-import { StreamedArtifactFrame } from '../chunk-LOJVUGBJ.js';
 import '../chunk-VILAPTCB.js';
-import { ChatPage } from '../chunk-PFA43QNX.js';
+import { ChatPage } from '../chunk-EWSEQEPL.js';
 import '../chunk-KZJAG7T3.js';
 import '../chunk-HOXKR26S.js';
 import '../chunk-AA3PT6KR.js';
 import '../chunk-NPLYHR2E.js';
+import { initials } from '../chunk-OXF2IFXG.js';
 import { ModalScaffold } from '../chunk-TX2ZICS7.js';
 import '../chunk-7F6CBP3Z.js';
-import { initials } from '../chunk-OXF2IFXG.js';
 import '../chunk-VUNV25KB.js';
 import { lazy, useState, Suspense, useEffect, useRef, useCallback, useMemo } from 'react';
 import { QueryClient, QueryClientProvider, useQueryClient, useQuery } from '@tanstack/react-query';
@@ -593,6 +593,7 @@ function ConnectedWorkflowRunner({
   emptyHint,
   connectedLabels,
   onOpenDocument,
+  onOpenCitation,
   onOpenDraft,
   onOpenMessage,
   actions: hostActions,
@@ -608,8 +609,9 @@ function ConnectedWorkflowRunner({
   const { ensureWorkspaceOpen, user } = useConnection();
   const { wsId } = useWorkspaceDocs();
   const { data: docs = [], refetch } = useDocuments(wsId);
-  const [uploading, setUploading] = useState(false);
+  const [sending, setSending] = useState([]);
   const [added, setAdded] = useState({});
+  const [addedNames, setAddedNames] = useState({});
   const addedRef = useRef(added);
   addedRef.current = added;
   const { view, conversationId, people } = useWorkflowThread(workflow, {
@@ -643,24 +645,86 @@ function ConnectedWorkflowRunner({
     }),
     [user?.externalId, onOpenDocument, onOpenDraft, onOpenMessage, hostActions, nameOfDoc, inThread]
   );
-  const onUpload = useCallback(
-    async (files2, step) => {
-      if (!files2.length) return;
-      setUploading(true);
+  const openCitation = useMemo(
+    () => onOpenCitation ?? (onOpenDocument && ((t) => {
+      const docId = t.documentId ?? t.passage?.documentId;
+      if (docId) onOpenDocument(docId, void 0, t.page ?? t.passage?.page ?? void 0);
+    })),
+    [onOpenCitation, onOpenDocument]
+  );
+  const uploadSeq = useRef(0);
+  const sendingRef = useRef(sending);
+  sendingRef.current = sending;
+  const send = useCallback(
+    async (batch) => {
+      if (!batch.length) return;
+      const keys = new Set(batch.map((p) => p.key));
+      const mark = (key, patch) => setSending(
+        (all) => patch === null ? all.filter((p) => p.key !== key) : all.map((p) => p.key === key ? { ...p, ...patch } : p)
+      );
+      setSending((all) => [
+        ...all.filter((p) => !keys.has(p.key)),
+        ...batch.map((p) => ({ ...p, status: "uploading", error: void 0 }))
+      ]);
       try {
         await ensureWorkspaceOpen();
-        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])) }));
+      } catch (e) {
+        for (const p of batch) mark(p.key, { status: "failed", error: uploadError(e) });
+        return;
+      }
+      for (const p of batch) {
+        try {
+          const { doc_ext_ids = [] } = await arbi.documents.uploadFile(p.file, p.file.name) ?? {};
+          if (!doc_ext_ids.length) throw new Error("No document came back");
+          setAdded((a) => ({
+            ...a,
+            ...Object.fromEntries(doc_ext_ids.map((id) => [id, p.stepId]))
+          }));
+          setAddedNames((n) => ({
+            ...n,
+            ...Object.fromEntries(doc_ext_ids.map((id) => [id, p.file.name]))
+          }));
+          mark(p.key, null);
+        } catch (e) {
+          mark(p.key, { status: "failed", error: uploadError(e) });
         }
-        await refetch();
-      } finally {
-        setUploading(false);
       }
+      await refetch();
     },
     [arbi, ensureWorkspaceOpen, refetch]
   );
-  const files = runnerFiles(docs, added, fileFilter);
+  const onUpload = useCallback(
+    (files2, step) => send(
+      files2.map((file) => ({
+        key: `upload-${++uploadSeq.current}`,
+        file,
+        stepId: step.id,
+        status: "uploading"
+      }))
+    ),
+    [send]
+  );
+  const onRetryUpload = useCallback(
+    (file) => {
+      const p = sendingRef.current.find((x) => x.key === file.id);
+      if (p) void send([p]);
+    },
+    [send]
+  );
+  const onRemoveUpload = useCallback(
+    (file) => setSending((all) => all.filter((p) => p.key !== file.id)),
+    []
+  );
+  const files = [
+    ...sending.map((p) => ({
+      id: p.key,
+      name: p.file.name,
+      status: p.status === "failed" ? "upload-failed" : "uploading",
+      error: p.error,
+      addedFor: p.stepId
+    })),
+    ...runnerFiles(docs, added, fileFilter, addedNames)
+  ];
   const reading = files.some((f) => f.addedFor && f.status === "processing");
   useEffect(() => {
     if (!reading) return;
@@ -729,8 +793,10 @@ function ConnectedWorkflowRunner({
         }
       ) : null,
       onUpload,
-      uploading,
+      uploading: sending.some((p) => p.status === "uploading"),
       files,
+      onRetryUpload,
+      onRemoveUpload,
       actions,
       nameOfDoc,
       testIdPrefix,
@@ -772,6 +838,7 @@ function ConnectedWorkflowRunner({
           workspaceId: wsId,
           stepTitle: (id) => view.steps.find((s) => s.id === id)?.title,
           targetMessageId: target,
+          onOpenCitation: openCitation,
           labels: { workflows: l.currentStep },
           aside: runner(true)
         }
@@ -800,13 +867,25 @@ function ConnectedWorkflowRunner({
   ] });
 }
 var READ_POLL_MS = 3e3;
-function runnerFiles(docs, added, fileFilter) {
-  return docs.filter((d) => !fileFilter || added[d.external_id] || fileFilter(d)).map((d) => ({
-    id: d.external_id,
-    name: d.file_name ?? d.external_id,
-    status: fileStatus(d.status),
-    addedFor: added[d.external_id]
-  }));
+function uploadError(e) {
+  return e instanceof Error && e.message ? e.message : String(e || "Upload failed");
+}
+function runnerFiles(docs, added, fileFilter, names = {}) {
+  const listed = new Set(docs.map((d) => d.external_id));
+  return [
+    ...docs.filter((d) => !fileFilter || added[d.external_id] || fileFilter(d)).map((d) => ({
+      id: d.external_id,
+      name: d.file_name ?? d.external_id,
+      status: fileStatus(d.status),
+      addedFor: added[d.external_id]
+    })),
+    ...Object.entries(added).filter(([id]) => !listed.has(id)).map(([id, stepId]) => ({
+      id,
+      name: names[id] ?? id,
+      status: "processing",
+      addedFor: stepId
+    }))
+  ];
 }
 function fileStatus(status) {
   const s = status?.toLowerCase();