@arbidocs/react 0.5.111 → 0.5.112

package/dist/chunk-XWPLOFDN.js

@@ -1,644 +0,0 @@
-import { arbiMutationKeys, arbiQueryKeys, useArbi } from './chunk-ZM6N7YZH.js';
-import { create } from 'zustand';
-import { useRef, useCallback, useSyncExternalStore } from 'react';
-import { useQueryClient, useMutation, useQuery } from '@tanstack/react-query';
-import { ArbiWorkspaceUnavailableError, tags } from '@arbidocs/sdk/browser';
-import { isAbortError } from '@arbidocs/client';
-
-// src/documents/documents-cache.ts
-var rowListeners = /* @__PURE__ */ new Set();
-function subscribeDocumentRows(listener) {
-  rowListeners.add(listener);
-  return () => {
-    rowListeners.delete(listener);
-  };
-}
-function getCachedDocuments(qc, workspaceId) {
-  if (!workspaceId) return void 0;
-  return qc.getQueryData(arbiQueryKeys.documents(workspaceId));
-}
-function cachedLists(qc) {
-  return qc.getQueriesData({ queryKey: arbiQueryKeys.allDocuments() }).flatMap(
-    ([key, data]) => typeof key[2] === "string" && Array.isArray(data) ? [[key[2], data]] : []
-  );
-}
-function diffRows(workspaceId, before, after) {
-  const beforeById = new Map(before.map((doc) => [doc.external_id, doc]));
-  const add = [];
-  const update2 = [];
-  const seen = /* @__PURE__ */ new Set();
-  for (const doc of after) {
-    seen.add(doc.external_id);
-    const previous = beforeById.get(doc.external_id);
-    if (!previous) add.push(doc);
-    else if (previous !== doc) update2.push(doc);
-  }
-  const remove = [...beforeById.keys()].filter((id) => !seen.has(id));
-  return { workspaceId, add, update: update2, remove };
-}
-function patchDocuments(qc, reducer, workspaceId) {
-  const lists = workspaceId ? (() => {
-    const current = getCachedDocuments(qc, workspaceId);
-    return current ? [[workspaceId, current]] : [];
-  })() : cachedLists(qc);
-  for (const [id, before] of lists) {
-    const after = reducer(before);
-    if (after === before) continue;
-    qc.setQueryData(arbiQueryKeys.documents(id), after);
-    const change = diffRows(id, before, after);
-    const beforeById = new Map(before.map((doc) => [doc.external_id, doc]));
-    if (change.update.some(
-      (doc) => doc.current_version !== beforeById.get(doc.external_id)?.current_version
-    ))
-      void qc.invalidateQueries({ queryKey: arbiQueryKeys.allDocumentVersions() });
-    if (change.add.length || change.update.length || change.remove.length)
-      for (const listener of rowListeners) listener(change);
-  }
-}
-function upsertDocuments(qc, documents, workspaceId) {
-  if (!documents.length) return;
-  const byId = new Map(documents.map((doc) => [doc.external_id, doc]));
-  const byWorkspace = /* @__PURE__ */ new Map();
-  for (const doc of documents) {
-    const ws = workspaceId ?? doc.workspace_ext_id;
-    byWorkspace.set(ws, [...byWorkspace.get(ws) ?? [], doc]);
-  }
-  for (const [ws, incoming] of byWorkspace) {
-    patchDocuments(
-      qc,
-      (old) => {
-        const known = new Set(old.map((doc) => doc.external_id));
-        return [
-          ...old.map((doc) => byId.get(doc.external_id) ?? doc),
-          ...incoming.filter((doc) => !known.has(doc.external_id))
-        ];
-      },
-      ws
-    );
-  }
-}
-function setCachedDocument(qc, doc) {
-  qc.setQueryData(arbiQueryKeys.document(doc.external_id), doc);
-}
-function patchCachedDocument(qc, docId, reducer) {
-  const current = qc.getQueryData(arbiQueryKeys.document(docId));
-  if (current) qc.setQueryData(arbiQueryKeys.document(docId), reducer(current));
-}
-function dropTagFromCachedDocuments(qc, tagId) {
-  const strip = (doc) => {
-    const doctags = doc.doctags?.filter((dt) => dt.tag_ext_id !== tagId);
-    return doctags && doctags.length !== doc.doctags?.length ? { ...doc, doctags } : doc;
-  };
-  const carriers = [];
-  patchDocuments(qc, (docs) => {
-    const next = docs.map(strip);
-    next.forEach((doc, i) => doc !== docs[i] && carriers.push(doc.external_id));
-    return carriers.length ? next : docs;
-  });
-  for (const id of carriers) patchCachedDocument(qc, id, strip);
-}
-function forgetDocuments(qc, docIds) {
-  for (const id of docIds) {
-    qc.removeQueries({ queryKey: arbiQueryKeys.document(id) });
-    qc.removeQueries({ queryKey: arbiQueryKeys.documentVersions(id) });
-    qc.removeQueries({ queryKey: arbiQueryKeys.parsedDocument(id) });
-  }
-}
-var useEntityExtractionStore = create(() => ({ docIds: [] }));
-function update(docIds, running) {
-  if (!docIds.length) return;
-  useEntityExtractionStore.setState((state) => {
-    const ids = new Set(state.docIds);
-    docIds.forEach((id) => running ? ids.add(id) : ids.delete(id));
-    return { docIds: [...ids] };
-  });
-}
-function startEntityExtraction(docIds) {
-  update(docIds, true);
-}
-function finishEntityExtraction(docIds) {
-  update(docIds, false);
-}
-function useEntityExtractionDocIds() {
-  return useEntityExtractionStore((state) => state.docIds);
-}
-var QUEUED_WRITE_BLOCKED_MESSAGE = "This change belongs to a workspace that can't be opened right now, so it was not sent. It is kept here until it can be.";
-function primeOnClient(client) {
-  return (workspaceId) => workspaceId === client.session.getState().selectedWorkspaceId ? void 0 : client.workspaceKeys.prime(workspaceId);
-}
-function primeOnArbi(arbi) {
-  return (workspaceId) => arbi.hasWorkspace && workspaceId === arbi.getWorkspaceId() ? void 0 : arbi.workspaces.open(workspaceId);
-}
-async function inWorkspace(prime, workspaceId, write) {
-  if (workspaceId) {
-    try {
-      await prime(workspaceId);
-    } catch (error) {
-      if (isAbortError(error)) throw error;
-      throw new ArbiWorkspaceUnavailableError(workspaceId, error, QUEUED_WRITE_BLOCKED_MESSAGE);
-    }
-  }
-  return write();
-}
-function isWorkspaceUnavailable(error) {
-  return error instanceof ArbiWorkspaceUnavailableError;
-}
-var QUEUED_WRITE_RETRY = {
-  retry: (failureCount, error) => isWorkspaceUnavailable(error) && failureCount < 8,
-  retryDelay: (attempt) => Math.min(1e3 * 2 ** attempt, 3e4)
-};
-function isBlockedOnWorkspace(mutation) {
-  return isWorkspaceUnavailable(mutation.state.error ?? mutation.state.failureReason);
-}
-function useCapturingWorkspace(mutation, capture) {
-  const captureRef = useRef(capture);
-  captureRef.current = capture;
-  const { mutate: rawMutate, mutateAsync: rawMutateAsync } = mutation;
-  const mutate = useCallback(
-    (variables, ...rest) => rawMutate(captureRef.current(variables), ...rest),
-    [rawMutate]
-  );
-  const mutateAsync = useCallback(
-    (variables, ...rest) => rawMutateAsync(
-      captureRef.current(variables),
-      ...rest
-    ),
-    [rawMutateAsync]
-  );
-  return { ...mutation, mutate, mutateAsync };
-}
-
-// src/tags.ts
-function workspaceOfTag(qc, tagId) {
-  for (const [key, tags] of qc.getQueriesData({
-    queryKey: arbiQueryKeys.allTags()
-  })) {
-    if (Array.isArray(tags) && tags.some((t) => t.external_id === tagId))
-      return key[2] ?? null;
-  }
-  return null;
-}
-function openWorkspaceId(arbi) {
-  return arbi.hasWorkspace ? arbi.getWorkspaceId() : null;
-}
-function patchWorkspaceTags(qc, workspaceId, reducer) {
-  const key = arbiQueryKeys.tags(workspaceId);
-  const current = qc.getQueryData(key);
-  if (!current) return;
-  const next = reducer(current);
-  if (next !== current) qc.setQueryData(key, next);
-}
-function invalidateTagTables(qc, workspaceId) {
-  return qc.invalidateQueries({ queryKey: arbiQueryKeys.tagLists(workspaceId) });
-}
-function invalidateTagData(qc, workspaceId) {
-  return Promise.all([
-    qc.invalidateQueries({ queryKey: arbiQueryKeys.tags(workspaceId) }),
-    qc.invalidateQueries({ queryKey: arbiQueryKeys.facts(workspaceId) })
-  ]);
-}
-function useInvalidateTagData() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useCallback(() => {
-    const workspaceId = openWorkspaceId(arbi);
-    return workspaceId ? invalidateTagData(qc, workspaceId) : Promise.resolve();
-  }, [arbi, qc]);
-}
-function settleCreatedTag(qc, created) {
-  const workspaceId = created.workspace_ext_id;
-  patchWorkspaceTags(
-    qc,
-    workspaceId,
-    (old) => old.some((t) => t.external_id === created.external_id) ? old : [...old, created]
-  );
-  return invalidateTagTables(qc, workspaceId);
-}
-function settleChangedTag(qc, tag) {
-  const workspaceId = tag.workspace_ext_id;
-  patchWorkspaceTags(
-    qc,
-    workspaceId,
-    (old) => old.map((t) => t.external_id === tag.external_id ? { ...t, ...tag } : t)
-  );
-  return invalidateTagTables(qc, workspaceId);
-}
-var replayableTagMutations = [
-  {
-    mutationKey: arbiMutationKeys.createTag(),
-    // In the workspace it was made in (named on the request), opened for this session first.
-    run: (client, input) => inWorkspace(primeOnClient(client), input.workspaceId, () => tags.createTag(client, input)),
-    settle: (qc, created) => settleCreatedTag(qc, created)
-  }
-];
-function useCreateTag() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  const capture = (input) => input.workspaceId || !arbi.hasWorkspace ? input : { ...input, workspaceId: arbi.getWorkspaceId() };
-  return useCapturingWorkspace(
-    useMutation({
-      mutationKey: arbiMutationKeys.createTag(),
-      mutationFn: (input) => inWorkspace(primeOnArbi(arbi), input.workspaceId, () => arbi.tags.create(input)),
-      onSuccess: (created) => settleCreatedTag(qc, created),
-      ...QUEUED_WRITE_RETRY
-    }),
-    capture
-  );
-}
-function useIsCreatingTag() {
-  const qc = useQueryClient();
-  return useCallback(
-    (name) => {
-      const wanted = name.trim().toLowerCase();
-      return qc.getMutationCache().findAll({
-        mutationKey: arbiMutationKeys.createTag(),
-        predicate: (m) => m.state.status === "pending" && m.state.variables?.name.trim().toLowerCase() === wanted
-      }).length > 0;
-    },
-    [qc]
-  );
-}
-function useUpdateTag() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationKey: arbiMutationKeys.updateTag(),
-    mutationFn: ({ tagId, change }) => arbi.tags.update(tagId, change),
-    onSuccess: (tag) => settleChangedTag(qc, tag)
-  });
-}
-function useSetTagValues() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationKey: arbiMutationKeys.updateTag(),
-    mutationFn: ({
-      tagId,
-      values
-    }) => arbi.tags.setValues(tagId, values),
-    onSuccess: (tag) => settleChangedTag(qc, tag)
-  });
-}
-function useDeleteTag() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationKey: arbiMutationKeys.deleteTag(),
-    mutationFn: (tagId) => arbi.tags.delete(tagId),
-    // The tag's own workspace (the cached list holding it), else the one open when it is sent: a
-    // switch while the delete runs must not patch the next workspace's list.
-    onMutate: (tagId) => ({ workspaceId: workspaceOfTag(qc, tagId) ?? openWorkspaceId(arbi) }),
-    onSuccess: (_data, tagId, context) => {
-      dropTagFromCachedDocuments(qc, tagId);
-      const workspaceId = context?.workspaceId;
-      if (!workspaceId) return;
-      patchWorkspaceTags(
-        qc,
-        workspaceId,
-        (old) => old.filter((t) => t.external_id !== tagId).map((t) => t.parent_ext_id === tagId ? { ...t, parent_ext_id: null } : t)
-      );
-      return Promise.all([
-        invalidateTagTables(qc, workspaceId),
-        qc.invalidateQueries({ queryKey: arbiQueryKeys.facts(workspaceId) })
-      ]);
-    }
-  });
-}
-function useExtractEntities() {
-  const arbi = useArbi();
-  return useMutation({
-    mutationKey: arbiMutationKeys.extractEntities(),
-    mutationFn: (docIds) => arbi.tags.extractEntities(docIds),
-    onSuccess: (data) => startEntityExtraction(data.doc_ext_ids)
-  });
-}
-function useTagKindSchema(kind, options) {
-  const arbi = useArbi();
-  const workspaceId = openWorkspaceId(arbi) ?? "";
-  return useQuery({
-    queryKey: arbiQueryKeys.tagKind(workspaceId, kind ?? ""),
-    queryFn: () => arbi.readInWorkspace(workspaceId, () => arbi.tagKinds.get(kind)),
-    enabled: (options?.enabled ?? true) && !!kind && !!workspaceId
-  });
-}
-function useUpdateTagKindSchema() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationKey: arbiMutationKeys.updateTagKindSchema(),
-    mutationFn: ({
-      kind,
-      fields
-    }) => arbi.tagKinds.update(kind, fields),
-    // The workspace open when it is sent (the server writes the open workspace's schema until
-    // the request can name it): a switch meanwhile must not rewrite the next workspace's columns.
-    onMutate: () => ({ workspaceId: openWorkspaceId(arbi) }),
-    onSuccess: (schema, _input, context) => {
-      const workspaceId = context?.workspaceId;
-      if (!workspaceId) return;
-      qc.setQueryData(arbiQueryKeys.tagKind(workspaceId, schema.kind), schema);
-      return invalidateTagData(qc, workspaceId);
-    }
-  });
-}
-var DOCUMENTS_FIRST_PAGE_SIZE = 500;
-var DOCUMENTS_PAGE_SIZE = 2e3;
-var DOCUMENTS_FIRST_PAGE_MEASURE = "arbi:document-list:page1";
-var FIRST_PAGE_MEASURE = DOCUMENTS_FIRST_PAGE_MEASURE;
-var FIRST_PAGE_START = "arbi:document-list:start";
-function mark(name) {
-  try {
-    performance.mark(name);
-  } catch {
-  }
-}
-function measure(name, start) {
-  try {
-    performance.measure(name, start);
-  } catch {
-  }
-}
-var backgroundReads = /* @__PURE__ */ new Map();
-var loadingListeners = /* @__PURE__ */ new Set();
-function emitLoading() {
-  for (const listener of loadingListeners) listener();
-}
-function subscribeLoading(listener) {
-  loadingListeners.add(listener);
-  return () => {
-    loadingListeners.delete(listener);
-  };
-}
-function isDocumentsLoading() {
-  return backgroundReads.size > 0;
-}
-function useDocumentsLoading() {
-  return useSyncExternalStore(subscribeLoading, isDocumentsLoading, () => false);
-}
-function stopDocumentsLoading(workspaceId) {
-  const controller = backgroundReads.get(workspaceId);
-  if (!controller) return;
-  controller.abort();
-  backgroundReads.delete(workspaceId);
-  emitLoading();
-}
-async function loadDocuments(qc, arbi, workspaceId, signal) {
-  let stopped = false;
-  for (const [id, controller2] of backgroundReads) {
-    if (id === workspaceId) continue;
-    controller2.abort();
-    backgroundReads.delete(id);
-    stopped = true;
-  }
-  if (stopped) emitLoading();
-  const controller = new AbortController();
-  signal?.addEventListener("abort", () => controller.abort(), { once: true });
-  const pages = arbi.documents.listPaginated({
-    workspaceId,
-    pageSize: DOCUMENTS_PAGE_SIZE,
-    firstPageSize: DOCUMENTS_FIRST_PAGE_SIZE,
-    signal: controller.signal
-  });
-  mark(FIRST_PAGE_START);
-  const first = await pages.next();
-  const firstPage = first.done ? [] : first.value;
-  measure(FIRST_PAGE_MEASURE, FIRST_PAGE_START);
-  if (first.done || firstPage.length < DOCUMENTS_FIRST_PAGE_SIZE) return firstPage;
-  if (backgroundReads.has(workspaceId)) {
-    controller.abort();
-    return firstPage;
-  }
-  backgroundReads.set(workspaceId, controller);
-  emitLoading();
-  void appendRemainingPages(qc, workspaceId, pages, controller.signal).finally(() => {
-    if (backgroundReads.get(workspaceId) !== controller) return;
-    backgroundReads.delete(workspaceId);
-    emitLoading();
-  });
-  return firstPage;
-}
-async function appendRemainingPages(qc, workspaceId, pages, signal) {
-  await Promise.resolve();
-  try {
-    for await (const page of pages) {
-      if (signal.aborted) return;
-      patchDocuments(
-        qc,
-        (old) => {
-          const known = new Set(old.map((doc) => doc.external_id));
-          const fresh = page.filter((doc) => !known.has(doc.external_id));
-          return fresh.length ? [...old, ...fresh] : old;
-        },
-        workspaceId
-      );
-    }
-  } catch (error) {
-    if (!signal.aborted) console.error(`Reading the documents of ${workspaceId} failed:`, error);
-  }
-}
-
-// src/documents/processing-order.ts
-var FINISHED_DOCUMENT_STATUSES = /* @__PURE__ */ new Set([
-  "completed",
-  "low-content",
-  "failed",
-  "skipped",
-  "empty"
-]);
-var IN_FLIGHT_DOCUMENT_STATUSES = /* @__PURE__ */ new Set([
-  "uploading",
-  "queued",
-  "parsing",
-  "encrypting",
-  "indexing",
-  "analysing"
-]);
-function ordered(value) {
-  return typeof value === "number";
-}
-function isStaleTaskUpdate(held, next) {
-  return !!held && ordered(held.seq) && ordered(next.seq) && next.seq <= held.seq;
-}
-function isRowNewer(row, task) {
-  if (!ordered(row.status_seq) || !ordered(task.seq)) return void 0;
-  return row.status_seq > task.seq;
-}
-function resolveDocumentStatus(row, task) {
-  const rowStatus = (row.status ?? "").toLowerCase();
-  const live = task?.status;
-  if (!task || !live) return rowStatus;
-  const rowNewer = isRowNewer(row, task);
-  if (rowNewer !== void 0) return rowNewer ? rowStatus || live : live;
-  if (IN_FLIGHT_DOCUMENT_STATUSES.has(live)) return live;
-  if (FINISHED_DOCUMENT_STATUSES.has(rowStatus)) return rowStatus;
-  return live;
-}
-function settledByRows(tasks, rows) {
-  const inFlight = /* @__PURE__ */ new Map();
-  for (const task of Object.values(tasks))
-    if (IN_FLIGHT_DOCUMENT_STATUSES.has(task.status)) inFlight.set(task.doc_ext_id, task);
-  if (!inFlight.size) return [];
-  const settled = [];
-  for (const row of rows) {
-    const task = inFlight.get(row.external_id);
-    if (!task) continue;
-    const rowStatus = (row.status ?? "").toLowerCase();
-    if (!FINISHED_DOCUMENT_STATUSES.has(rowStatus) || isRowNewer(row, task) === false) continue;
-    settled.push({ ...task, status: rowStatus, seq: row.status_seq ?? task.seq });
-  }
-  return settled;
-}
-var DISMISSED_TASK_TTL_MS = 10 * 60 * 1e3;
-var finished = (status) => FINISHED_DOCUMENT_STATUSES.has(status);
-var useArbiWsStore = create((set, get) => ({
-  status: "disconnected",
-  tasks: {},
-  dismissed: {},
-  lastTaskUpdate: null,
-  presence: {},
-  evicted: false,
-  setStatus: (status) => set({ status }),
-  setEvicted: (evicted) => set({ evicted }),
-  applyTaskUpdate: (msg) => {
-    const { tasks, dismissed } = get();
-    const dismissedAt = dismissed[msg.doc_ext_id];
-    if (dismissedAt !== void 0 && Date.now() - dismissedAt < DISMISSED_TASK_TTL_MS) return false;
-    const held = tasks[msg.doc_ext_id];
-    if (isStaleTaskUpdate(held, msg)) return false;
-    const now = Date.now();
-    const task = {
-      ...msg,
-      startTime: held?.startTime ?? now,
-      endTime: finished(msg.status) ? held?.endTime ?? now : void 0
-    };
-    set({ tasks: { ...tasks, [msg.doc_ext_id]: task }, lastTaskUpdate: msg });
-    return true;
-  },
-  putTask: (task) => set((s) => ({ tasks: { ...s.tasks, [task.doc_ext_id]: task } })),
-  clearTask: (docId) => set((s) => {
-    const tasks = { ...s.tasks };
-    delete tasks[docId];
-    return { tasks, dismissed: { ...s.dismissed, [docId]: Date.now() } };
-  }),
-  clearFinishedTasks: () => set((s) => ({
-    tasks: Object.fromEntries(Object.entries(s.tasks).filter(([, t]) => !finished(t.status)))
-  })),
-  settleTasks: (rows) => set((s) => {
-    const settled = settledByRows(s.tasks, rows);
-    if (!settled.length) return s;
-    const tasks = { ...s.tasks };
-    const now = Date.now();
-    for (const task of settled)
-      tasks[task.doc_ext_id] = { ...task, endTime: task.endTime ?? now };
-    return { tasks };
-  }),
-  applyPresence: (msg) => set((s) => ({ presence: { ...s.presence, [msg.user_id]: msg.status } })),
-  reset: () => set({ status: "disconnected", tasks: {}, dismissed: {}, lastTaskUpdate: null, presence: {} })
-}));
-function openWorkspaceId2(arbi) {
-  return arbi.hasWorkspace ? arbi.getWorkspaceId() : null;
-}
-function useDocuments(workspaceId, options) {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  const workspace = workspaceId ?? openWorkspaceId2(arbi) ?? "";
-  return useQuery({
-    queryKey: arbiQueryKeys.documents(workspace),
-    queryFn: ({ signal }) => {
-      if (openWorkspaceId2(arbi) !== workspace)
-        return Promise.reject(new Error("The selected workspace changed."));
-      return loadDocuments(qc, arbi, workspace, signal);
-    },
-    enabled: (options?.enabled ?? true) && !!workspace && arbi.hasWorkspace,
-    staleTime: Infinity,
-    refetchOnMount: false,
-    refetchOnReconnect: false,
-    refetchOnWindowFocus: false
-  });
-}
-var DOCUMENT_STATUS_POLL_MS = 4e3;
-function processingInterval(doc) {
-  return doc?.status && !FINISHED_DOCUMENT_STATUSES.has(doc.status) ? DOCUMENT_STATUS_POLL_MS : false;
-}
-function useDocument(docId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.document(docId ?? ""),
-    queryFn: async () => (await arbi.documents.get([docId]))[0] ?? null,
-    enabled: (options?.enabled ?? true) && !!docId && arbi.isLoggedIn,
-    refetchInterval: options?.pollWhileProcessing ? (query) => processingInterval(query.state.data) : false
-  });
-}
-var SIMILARITY_THRESHOLD = 0.98;
-function useSimilarDocuments(options) {
-  const arbi = useArbi();
-  const workspace = openWorkspaceId2(arbi) ?? "";
-  return useQuery({
-    queryKey: arbiQueryKeys.similarDocuments(workspace, SIMILARITY_THRESHOLD),
-    queryFn: () => arbi.readInWorkspace(
-      workspace,
-      () => arbi.documents.similar({ threshold: SIMILARITY_THRESHOLD })
-    ),
-    enabled: (options?.enabled ?? true) && !!workspace
-  });
-}
-var readSeq = 0;
-var latestReadByDoc = /* @__PURE__ */ new Map();
-async function refreshDocuments(qc, arbi, docIds) {
-  if (!docIds.length) return [];
-  const seq = ++readSeq;
-  for (const id of docIds) latestReadByDoc.set(id, seq);
-  try {
-    const read = await arbi.documents.get(docIds);
-    const fresh = read.filter((doc) => latestReadByDoc.get(doc.external_id) === seq).map((doc) => {
-      const live = useArbiWsStore.getState().tasks[doc.external_id];
-      return live && FINISHED_DOCUMENT_STATUSES.has(live.status) && doc.status !== live.status && isRowNewer(doc, live) !== true ? { ...doc, status: live.status } : doc;
-    });
-    if (!fresh.length) return [];
-    reconcileTagCounts(qc, fresh);
-    upsertDocuments(qc, fresh);
-    for (const doc of fresh) setCachedDocument(qc, doc);
-    return fresh;
-  } catch (error) {
-    console.error("Re-reading documents failed; the cached ones are kept:", error);
-    return [];
-  } finally {
-    for (const id of docIds) if (latestReadByDoc.get(id) === seq) latestReadByDoc.delete(id);
-  }
-}
-function useRefreshDocuments() {
-  const qc = useQueryClient();
-  const arbi = useArbi();
-  return useCallback((docIds) => refreshDocuments(qc, arbi, docIds), [qc, arbi]);
-}
-function reconcileTagCounts(qc, fresh) {
-  const byWorkspace = /* @__PURE__ */ new Map();
-  for (const doc of fresh) {
-    const cached = getCachedDocuments(qc, doc.workspace_ext_id)?.find(
-      (d) => d.external_id === doc.external_id
-    );
-    const before = new Set((cached?.doctags ?? []).map((dt) => dt.tag_ext_id));
-    const after = new Set((doc.doctags ?? []).map((dt) => dt.tag_ext_id));
-    const delta = byWorkspace.get(doc.workspace_ext_id) ?? /* @__PURE__ */ new Map();
-    for (const id of after) if (!before.has(id)) delta.set(id, (delta.get(id) ?? 0) + 1);
-    for (const id of before) if (!after.has(id)) delta.set(id, (delta.get(id) ?? 0) - 1);
-    byWorkspace.set(doc.workspace_ext_id, delta);
-  }
-  for (const [workspaceId, delta] of byWorkspace) adjustTagCounts(qc, workspaceId, delta);
-}
-function adjustTagCounts(qc, workspaceId, delta) {
-  if (![...delta.values()].some(Boolean)) return;
-  patchWorkspaceTags(
-    qc,
-    workspaceId,
-    (tags) => tags.map((tag) => {
-      const change = delta.get(tag.external_id);
-      return change ? { ...tag, doctag_count: Math.max(0, tag.doctag_count + change) } : tag;
-    })
-  );
-  void invalidateTagTables(qc, workspaceId);
-}
-
-export { DISMISSED_TASK_TTL_MS, DOCUMENTS_FIRST_PAGE_MEASURE, DOCUMENTS_FIRST_PAGE_SIZE, DOCUMENTS_PAGE_SIZE, FINISHED_DOCUMENT_STATUSES, IN_FLIGHT_DOCUMENT_STATUSES, QUEUED_WRITE_BLOCKED_MESSAGE, QUEUED_WRITE_RETRY, SIMILARITY_THRESHOLD, adjustTagCounts, dropTagFromCachedDocuments, finishEntityExtraction, forgetDocuments, getCachedDocuments, inWorkspace, invalidateTagData, invalidateTagTables, isBlockedOnWorkspace, isDocumentsLoading, isRowNewer, isStaleTaskUpdate, isWorkspaceUnavailable, patchCachedDocument, patchDocuments, patchWorkspaceTags, primeOnArbi, primeOnClient, refreshDocuments, replayableTagMutations, resolveDocumentStatus, setCachedDocument, settledByRows, startEntityExtraction, stopDocumentsLoading, subscribeDocumentRows, upsertDocuments, useArbiWsStore, useCapturingWorkspace, useCreateTag, useDeleteTag, useDocument, useDocuments, useDocumentsLoading, useEntityExtractionDocIds, useExtractEntities, useInvalidateTagData, useIsCreatingTag, useRefreshDocuments, useSetTagValues, useSimilarDocuments, useTagKindSchema, useUpdateTag, useUpdateTagKindSchema };
-//# sourceMappingURL=chunk-XWPLOFDN.js.map
-//# sourceMappingURL=chunk-XWPLOFDN.js.map