@arbidocs/blocks 0.4.79 → 0.4.81

package/dist/connect/index.d.ts

@@ -9,7 +9,7 @@ export { ConnectedTaskAccessRequest } from '../connection/ConnectedTaskAccessReq
 export { ConnectedSkillUpload } from '../connection/ConnectedSkillUpload';
 export { useConnectedAiTask, useConnectedChat, ConnectedAssistantDock, useWorkspaceDocs, useSemanticSearch, toRedlineMarkdown, redlineStats, textToArtifact, type AiTaskStatus, type AiRunOptions, type SearchChunk, type ArtifactEvent, } from '../connection/ai';
 export { StreamedArtifact, type StreamedArtifactProps } from '../connection/StreamedArtifact';
-export { ConnectedWorkflowRunner, DEFAULT_CONNECTED_WORKFLOW_RUNNER_LABELS, type ConnectedWorkflowRunnerProps, type ConnectedWorkflowRunnerLabels, } from '../connection/ConnectedWorkflowRunner';
+export { ConnectedWorkflowRunner, DEFAULT_CONNECTED_WORKFLOW_RUNNER_LABELS, type ConnectedWorkflowRunnerProps, type ConnectedWorkflowRunnerLabels, type WorkflowRunnerDocument, } from '../connection/ConnectedWorkflowRunner';
 export { WorkflowRunThread, type WorkflowRunThreadProps } from '../connection/WorkflowRunThread';
 export { ConnectedTagMatrix, type ConnectedTagMatrixProps } from '../connection/ConnectedTagMatrix';
 export { useConnectedShellWorkspaces, type UseConnectedShellWorkspacesOptions, } from '../connection/useConnectedShellWorkspaces';

package/dist/connect/index.js

@@ -1,5 +1,5 @@
 "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-O35LRB3P.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';
@@ -7,19 +7,19 @@ 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-CWAZNXC5.js';
-import { splitDocRef, WorkflowStepBar, stepLevels, WorkflowRunner, initialRunInputs, workflowSteps, fromPlanSteps, stepStatesFromActivity, toPlanSteps, runThreadTitle, useWorkflowView, stepFields } from '../chunk-6GXF62XN.js';
+import { TagMatrix, tagMatrixMarkdown } from '../chunk-SWSVPQNB.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 { ModalScaffold } from '../chunk-TX2ZICS7.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 '../chunk-VUNV25KB.js';
 import { lazy, useState, Suspense, useEffect, useRef, useCallback, useMemo } from 'react';
@@ -593,11 +593,13 @@ function ConnectedWorkflowRunner({
   emptyHint,
   connectedLabels,
   onOpenDocument,
+  onOpenCitation,
   onOpenDraft,
   onOpenMessage,
   actions: hostActions,
   dialogLabels,
   layout = "thread",
+  fileFilter,
   testIdPrefix = "workflow-run",
   ...runnerProps
 }) {
@@ -607,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, {
@@ -642,29 +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 = docs.map((d) => ({
-    id: d.external_id,
-    name: d.file_name ?? d.external_id,
-    status: fileStatus(d.status),
-    addedFor: added[d.external_id]
-  }));
+  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;
@@ -733,8 +793,10 @@ function ConnectedWorkflowRunner({
         }
       ) : null,
       onUpload,
-      uploading,
+      uploading: sending.some((p) => p.status === "uploading"),
       files,
+      onRetryUpload,
+      onRemoveUpload,
       actions,
       nameOfDoc,
       testIdPrefix,
@@ -776,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)
         }
@@ -804,6 +867,26 @@ function ConnectedWorkflowRunner({
   ] });
 }
 var READ_POLL_MS = 3e3;
+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();
   if (!s) return void 0;