@arbidocs/blocks 0.4.73 → 0.4.74

package/dist/connect/index.cjs

@@ -2,16 +2,16 @@
 'use strict';
 
 var chunkB3WCKBJ7_cjs = require('../chunk-B3WCKBJ7.cjs');
-var chunkWOYUXVVQ_cjs = require('../chunk-WOYUXVVQ.cjs');
+var chunkHITAN7JF_cjs = require('../chunk-HITAN7JF.cjs');
 var chunkMUMH3Z6K_cjs = require('../chunk-MUMH3Z6K.cjs');
-var chunkGTNUQBXI_cjs = require('../chunk-GTNUQBXI.cjs');
+var chunkP2RBE3FD_cjs = require('../chunk-P2RBE3FD.cjs');
 require('../chunk-DETZZBG2.cjs');
 require('../chunk-7D737FLE.cjs');
-var chunkQVBZESYQ_cjs = require('../chunk-QVBZESYQ.cjs');
 require('../chunk-ADOKTJHV.cjs');
-var chunkKJ3QEKOJ_cjs = require('../chunk-KJ3QEKOJ.cjs');
+var chunkQVBZESYQ_cjs = require('../chunk-QVBZESYQ.cjs');
+var chunkIPGKTUF7_cjs = require('../chunk-IPGKTUF7.cjs');
 require('../chunk-VGOFFCMT.cjs');
-require('../chunk-V3L5DONY.cjs');
+require('../chunk-B4SRYJLM.cjs');
 require('../chunk-7UTB4PHM.cjs');
 require('../chunk-PPCATBOY.cjs');
 var chunkKDG6FER6_cjs = require('../chunk-KDG6FER6.cjs');
@@ -188,13 +188,35 @@ 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] = react.useState(0);
+  const [prev, setPrev] = react.useState(missing);
+  if (prev !== missing) {
+    setPrev(missing);
+    setAttempt(0);
+  }
+  const refetchRef = react.useRef(refetch);
+  refetchRef.current = refetch;
+  react.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 = react$1.useArbi();
   const { ensureWorkspaceOpen } = chunkMUMH3Z6K_cjs.useConnection();
-  const { wsId } = chunkWOYUXVVQ_cjs.useWorkspaceDocs();
+  const { wsId } = chunkHITAN7JF_cjs.useWorkspaceDocs();
   const [threadId, setThreadId] = react.useState(options.conversationId);
   const [localInputs, setLocalInputs] = react.useState(
-    () => chunkGTNUQBXI_cjs.initialRunInputs(workflow, options.initialInputs)
+    () => chunkP2RBE3FD_cjs.initialRunInputs(workflow, options.initialInputs)
   );
   const [edits, setEdits] = react.useState({});
   const [localStates, setLocalStates] = react.useState({});
@@ -203,12 +225,12 @@ function useWorkflowThread(workflow, options = {}) {
   const conversations = react$1.useConversations(wsId, { enabled: !!threadId, live: true });
   const conversation = conversations.data?.find((c) => c.external_id === threadId);
   const steps = react.useMemo(
-    () => chunkGTNUQBXI_cjs.workflowSteps(workflow).length > 0 ? chunkGTNUQBXI_cjs.workflowSteps(workflow) : chunkGTNUQBXI_cjs.fromPlanSteps(conversation?.plan?.steps ?? []),
+    () => chunkP2RBE3FD_cjs.workflowSteps(workflow).length > 0 ? chunkP2RBE3FD_cjs.workflowSteps(workflow) : chunkP2RBE3FD_cjs.fromPlanSteps(conversation?.plan?.steps ?? []),
     [workflow, conversation?.plan?.steps]
   );
   const activity = react$1.useConversationActivity(threadId, { live: !!threadId });
   const serverStates = react.useMemo(
-    () => chunkGTNUQBXI_cjs.stepStatesFromActivity(activity.data ?? []),
+    () => chunkP2RBE3FD_cjs.stepStatesFromActivity(activity.data ?? []),
     [activity.data]
   );
   const working = steps.some(
@@ -218,16 +240,14 @@ function useWorkflowThread(workflow, options = {}) {
   const answers = react.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;
-  react.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 = react.useMemo(() => {
     if (!threadId) return localStates;
     const out = {};
@@ -262,7 +282,7 @@ function useWorkflowThread(workflow, options = {}) {
       if (!id) {
         await ensureWorkspaceOpen();
         const created = await createWork.mutateAsync({
-          title: chunkGTNUQBXI_cjs.runThreadTitle(workflow, values).slice(0, 60),
+          title: chunkP2RBE3FD_cjs.runThreadTitle(workflow, values).slice(0, 60),
           description: workflow.description || null,
           prompt: null,
           skill_ref: null,
@@ -274,7 +294,7 @@ function useWorkflowThread(workflow, options = {}) {
             criteria: [],
             max_passes: 3,
             documents: [],
-            steps: chunkGTNUQBXI_cjs.toPlanSteps(workflow, {
+            steps: chunkP2RBE3FD_cjs.toPlanSteps(workflow, {
               inputs: values,
               resolveWorkflow: options.resolveWorkflow,
               compose: options.compose
@@ -308,7 +328,7 @@ function useWorkflowThread(workflow, options = {}) {
     },
     [arbi, threadId, refresh]
   );
-  const view = chunkGTNUQBXI_cjs.useWorkflowView({
+  const view = chunkP2RBE3FD_cjs.useWorkflowView({
     steps,
     states,
     inputs,
@@ -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,
@@ -337,17 +358,21 @@ function ConnectedWorkflowRunner({
   const l = { ...DEFAULT_CONNECTED_WORKFLOW_RUNNER_LABELS, ...connectedLabels };
   const arbi = react$1.useArbi();
   const { ensureWorkspaceOpen } = chunkMUMH3Z6K_cjs.useConnection();
-  const { wsId } = chunkWOYUXVVQ_cjs.useWorkspaceDocs();
+  const { wsId } = chunkHITAN7JF_cjs.useWorkspaceDocs();
   const { data: docs = [], refetch } = react$1.useDocuments(wsId);
   const [uploading, setUploading] = react.useState(false);
   const { view, conversationId } = useWorkflowThread(workflow, options);
+  const [added, setAdded] = react.useState({});
   const onUpload = react.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");
+  react.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__ */ jsxRuntime.jsxs(
+        "p",
+        {
+          className: "flex items-center gap-2 text-sm text-muted-foreground",
+          "data-testid": `${testIdPrefix}-${step.id}-answer-loading`,
+          children: [
+            /* @__PURE__ */ jsxRuntime.jsx(lucideReact.Loader2, { className: "size-4 animate-spin" }),
+            " ",
+            l.loadingAnswer
+          ]
+        }
+      );
+    }
     return /* @__PURE__ */ jsxRuntime.jsx(
       react$1.AiAnswer,
       {
@@ -384,7 +435,7 @@ function ConnectedWorkflowRunner({
       }
     ) }),
     /* @__PURE__ */ jsxRuntime.jsx(
-      chunkGTNUQBXI_cjs.WorkflowRunner,
+      chunkP2RBE3FD_cjs.WorkflowRunner,
       {
         ...runnerProps,
         workflow,
@@ -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,
@@ -1009,7 +1068,7 @@ function mapWorkNotification(note) {
 function ActiveChat({ agentId, workspaceId }) {
   const chat = react$1.useArbiChat({ agentId, workspaceId });
   return /* @__PURE__ */ jsxRuntime.jsx(
-    chunkKJ3QEKOJ_cjs.ChatPage,
+    chunkIPGKTUF7_cjs.ChatPage,
     {
       chat: { ...chat, assistantMode: void 0, setAssistantMode: void 0 },
       documentsRequired: false,
@@ -1071,35 +1130,35 @@ function ConnectedAgentChatPanel({
 
 Object.defineProperty(exports, "ConnectedAssistantDock", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.ConnectedAssistantDock; }
+  get: function () { return chunkHITAN7JF_cjs.ConnectedAssistantDock; }
 });
 Object.defineProperty(exports, "redlineStats", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.redlineStats; }
+  get: function () { return chunkHITAN7JF_cjs.redlineStats; }
 });
 Object.defineProperty(exports, "textToArtifact", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.textToArtifact; }
+  get: function () { return chunkHITAN7JF_cjs.textToArtifact; }
 });
 Object.defineProperty(exports, "toRedlineMarkdown", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.toRedlineMarkdown; }
+  get: function () { return chunkHITAN7JF_cjs.toRedlineMarkdown; }
 });
 Object.defineProperty(exports, "useConnectedAiTask", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.useConnectedAiTask; }
+  get: function () { return chunkHITAN7JF_cjs.useConnectedAiTask; }
 });
 Object.defineProperty(exports, "useConnectedChat", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.useConnectedChat; }
+  get: function () { return chunkHITAN7JF_cjs.useConnectedChat; }
 });
 Object.defineProperty(exports, "useSemanticSearch", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.useSemanticSearch; }
+  get: function () { return chunkHITAN7JF_cjs.useSemanticSearch; }
 });
 Object.defineProperty(exports, "useWorkspaceDocs", {
   enumerable: true,
-  get: function () { return chunkWOYUXVVQ_cjs.useWorkspaceDocs; }
+  get: function () { return chunkHITAN7JF_cjs.useWorkspaceDocs; }
 });
 Object.defineProperty(exports, "ConnectionProvider", {
   enumerable: true,