@arbidocs/sdk 0.3.46 → 0.3.48

package/dist/browser.d.cts

@@ -1,2 +1,2 @@
-export { i as AgentStepEvent, j as Arbi, k as ArbiApiError, l as ArbiError, m as ArbiOptions, n as ArtifactEvent, o as AuthContext, A as AuthHeaders, p as AuthenticatedClient, c as ChatSession, q as CitationSummary, a as CliConfig, b as CliCredentials, C as ConfigStore, r as ConnectOptions, y as MessageMetadataPayload, z as MessageQueuedEvent, O as OutputTokensDetails, Q as QueryOptions, R as ReconnectOptions, B as ReconnectableWsConnection, E as ResolvedCitation, G as ResponseCompletedEvent, H as ResponseContentPartAddedEvent, I as ResponseCreatedEvent, J as ResponseFailedEvent, K as ResponseOutputItemAddedEvent, N as ResponseOutputItemDoneEvent, P as ResponseOutputTextDeltaEvent, T as ResponseOutputTextDoneEvent, V as ResponseUsage, W as SSEEvent, X as SSEStreamCallbacks, Y as SSEStreamResult, Z as SSEStreamStartData, _ as UserInfo, $ as UserInputRequestEvent, a0 as UserMessageEvent, a1 as WorkspaceContext, a2 as WsConnection, a3 as agentconfig, a4 as assistant, a5 as authenticatedFetch, a6 as buildRetrievalChunkTool, a7 as buildRetrievalFullContextTool, a8 as buildRetrievalTocTool, a9 as connectWebSocket, aa as connectWithReconnect, ab as consumeSSEStream, ac as contacts, ad as conversations, ae as countCitations, af as createAuthenticatedClient, ah as dm, ai as doctags, aj as documents, ak as files, al as formatAgentStepLabel, am as formatFileSize, ao as formatUserName, ap as formatWorkspaceChoices, au as getErrorMessage, aw as health, ax as parseSSEEvents, ay as performPasswordLogin, aB as requireData, aC as requireOk, aD as resolveAuth, aE as resolveCitations, aF as resolveWorkspace, aG as responses, aH as selectWorkspace, aI as selectWorkspaceById, aJ as settings, aK as streamSSE, aL as stripCitationMarkdown, aM as summarizeCitations, aN as tags, aO as workspaces } from './browser-HVCa8Ph5.cjs';
+export { i as AgentStepEvent, j as Arbi, k as ArbiApiError, l as ArbiError, m as ArbiOptions, n as ArtifactEvent, o as AuthContext, A as AuthHeaders, p as AuthenticatedClient, c as ChatSession, q as CitationSummary, a as CliConfig, b as CliCredentials, C as ConfigStore, r as ConnectOptions, y as MessageMetadataPayload, z as MessageQueuedEvent, O as OutputTokensDetails, Q as QueryOptions, R as ReconnectOptions, B as ReconnectableWsConnection, E as ResolvedCitation, G as ResponseCompletedEvent, H as ResponseContentPartAddedEvent, I as ResponseCreatedEvent, J as ResponseFailedEvent, K as ResponseOutputItemAddedEvent, N as ResponseOutputItemDoneEvent, P as ResponseOutputTextDeltaEvent, T as ResponseOutputTextDoneEvent, V as ResponseUsage, W as SSEEvent, X as SSEStreamCallbacks, Y as SSEStreamResult, Z as SSEStreamStartData, _ as UserInfo, $ as UserInputRequestEvent, a0 as UserMessageEvent, a1 as WorkspaceContext, a2 as WsConnection, a3 as agentconfig, a4 as assistant, a5 as authenticatedFetch, a6 as buildRetrievalChunkTool, a7 as buildRetrievalFullContextTool, a8 as buildRetrievalTocTool, a9 as connectWebSocket, aa as connectWithReconnect, ab as consumeSSEStream, ac as contacts, ad as conversations, ae as countCitations, af as createAuthenticatedClient, ah as dm, ai as doctags, aj as documents, ak as files, al as formatAgentStepLabel, am as formatFileSize, ao as formatUserName, ap as formatWorkspaceChoices, au as getErrorMessage, av as getRawWorkspaceKey, aw as health, ax as parseSSEEvents, ay as performPasswordLogin, aB as requireData, aC as requireOk, aD as resolveAuth, aE as resolveCitations, aF as resolveWorkspace, aG as responses, aH as selectWorkspace, aI as selectWorkspaceById, aJ as settings, aK as streamSSE, aL as stripCitationMarkdown, aM as summarizeCitations, aN as tags, aO as workspaces } from './browser-EMgd4bP4.cjs';
 import '@arbidocs/client';

package/dist/browser.d.ts

@@ -1,2 +1,2 @@
-export { i as AgentStepEvent, j as Arbi, k as ArbiApiError, l as ArbiError, m as ArbiOptions, n as ArtifactEvent, o as AuthContext, A as AuthHeaders, p as AuthenticatedClient, c as ChatSession, q as CitationSummary, a as CliConfig, b as CliCredentials, C as ConfigStore, r as ConnectOptions, y as MessageMetadataPayload, z as MessageQueuedEvent, O as OutputTokensDetails, Q as QueryOptions, R as ReconnectOptions, B as ReconnectableWsConnection, E as ResolvedCitation, G as ResponseCompletedEvent, H as ResponseContentPartAddedEvent, I as ResponseCreatedEvent, J as ResponseFailedEvent, K as ResponseOutputItemAddedEvent, N as ResponseOutputItemDoneEvent, P as ResponseOutputTextDeltaEvent, T as ResponseOutputTextDoneEvent, V as ResponseUsage, W as SSEEvent, X as SSEStreamCallbacks, Y as SSEStreamResult, Z as SSEStreamStartData, _ as UserInfo, $ as UserInputRequestEvent, a0 as UserMessageEvent, a1 as WorkspaceContext, a2 as WsConnection, a3 as agentconfig, a4 as assistant, a5 as authenticatedFetch, a6 as buildRetrievalChunkTool, a7 as buildRetrievalFullContextTool, a8 as buildRetrievalTocTool, a9 as connectWebSocket, aa as connectWithReconnect, ab as consumeSSEStream, ac as contacts, ad as conversations, ae as countCitations, af as createAuthenticatedClient, ah as dm, ai as doctags, aj as documents, ak as files, al as formatAgentStepLabel, am as formatFileSize, ao as formatUserName, ap as formatWorkspaceChoices, au as getErrorMessage, aw as health, ax as parseSSEEvents, ay as performPasswordLogin, aB as requireData, aC as requireOk, aD as resolveAuth, aE as resolveCitations, aF as resolveWorkspace, aG as responses, aH as selectWorkspace, aI as selectWorkspaceById, aJ as settings, aK as streamSSE, aL as stripCitationMarkdown, aM as summarizeCitations, aN as tags, aO as workspaces } from './browser-HVCa8Ph5.js';
+export { i as AgentStepEvent, j as Arbi, k as ArbiApiError, l as ArbiError, m as ArbiOptions, n as ArtifactEvent, o as AuthContext, A as AuthHeaders, p as AuthenticatedClient, c as ChatSession, q as CitationSummary, a as CliConfig, b as CliCredentials, C as ConfigStore, r as ConnectOptions, y as MessageMetadataPayload, z as MessageQueuedEvent, O as OutputTokensDetails, Q as QueryOptions, R as ReconnectOptions, B as ReconnectableWsConnection, E as ResolvedCitation, G as ResponseCompletedEvent, H as ResponseContentPartAddedEvent, I as ResponseCreatedEvent, J as ResponseFailedEvent, K as ResponseOutputItemAddedEvent, N as ResponseOutputItemDoneEvent, P as ResponseOutputTextDeltaEvent, T as ResponseOutputTextDoneEvent, V as ResponseUsage, W as SSEEvent, X as SSEStreamCallbacks, Y as SSEStreamResult, Z as SSEStreamStartData, _ as UserInfo, $ as UserInputRequestEvent, a0 as UserMessageEvent, a1 as WorkspaceContext, a2 as WsConnection, a3 as agentconfig, a4 as assistant, a5 as authenticatedFetch, a6 as buildRetrievalChunkTool, a7 as buildRetrievalFullContextTool, a8 as buildRetrievalTocTool, a9 as connectWebSocket, aa as connectWithReconnect, ab as consumeSSEStream, ac as contacts, ad as conversations, ae as countCitations, af as createAuthenticatedClient, ah as dm, ai as doctags, aj as documents, ak as files, al as formatAgentStepLabel, am as formatFileSize, ao as formatUserName, ap as formatWorkspaceChoices, au as getErrorMessage, av as getRawWorkspaceKey, aw as health, ax as parseSSEEvents, ay as performPasswordLogin, aB as requireData, aC as requireOk, aD as resolveAuth, aE as resolveCitations, aF as resolveWorkspace, aG as responses, aH as selectWorkspace, aI as selectWorkspaceById, aJ as settings, aK as streamSSE, aL as stripCitationMarkdown, aM as summarizeCitations, aN as tags, aO as workspaces } from './browser-EMgd4bP4.js';
 import '@arbidocs/client';

package/dist/browser.js

@@ -3634,6 +3634,23 @@ async function generateEncryptedWorkspaceKey(arbi, wrappedKey, serverSessionKey,
   const workspaceKey = sealedBoxDecrypt(wrappedKey, encryptionKeyPair.secretKey);
   return sealKeyForSession(workspaceKey, serverSessionKey);
 }
+async function getRawWorkspaceKey(arbi, workspaceId, signingPrivateKeyBase64) {
+  const { data: workspaces, error } = await arbi.fetch.GET("/v1/user/workspaces");
+  if (error || !workspaces) {
+    throw new ArbiError("Failed to fetch workspaces");
+  }
+  const ws = workspaces.find((w2) => w2.external_id === workspaceId);
+  if (!ws || !ws.wrapped_key) {
+    throw new ArbiError(`Workspace ${workspaceId} not found or has no encryption key`);
+  }
+  const signingPrivateKey = base64ToBytes(signingPrivateKeyBase64);
+  const ed25519PublicKey = signingPrivateKey.slice(32, 64);
+  const encryptionKeyPair = deriveEncryptionKeypairFromSigning({
+    publicKey: ed25519PublicKey,
+    secretKey: signingPrivateKey
+  });
+  return sealedBoxDecrypt(ws.wrapped_key, encryptionKeyPair.secretKey);
+}
 async function selectWorkspaceById(arbi, workspaceId, serverSessionKey, signingPrivateKeyBase64) {
   const { data: workspaces, error } = await arbi.fetch.GET("/v1/user/workspaces");
   if (error || !workspaces) {
@@ -4103,6 +4120,7 @@ __export(documents_exports, {
   listAll: () => listAll,
   listDocuments: () => listDocuments,
   listPaginated: () => listPaginated,
+  mimeFromName: () => mimeFromName,
   sanitizeFolderPath: () => sanitizeFolderPath,
   updateDocuments: () => updateDocuments,
   uploadDocumentsDirect: () => uploadDocumentsDirect,
@@ -4157,10 +4175,17 @@ async function listDocuments(arbi) {
   return requireData(await arbi.fetch.GET("/v1/document/list"), "Failed to fetch documents");
 }
 async function* listPaginated(arbi, options = {}) {
-  const { pageSize = 5e3, order = "id_asc", fields, signal, lookahead = 3 } = options;
-  const fetchPage = async (pageOffset) => {
+  const {
+    pageSize = 5e3,
+    firstPageSize,
+    order = "id_asc",
+    fields,
+    signal,
+    lookahead = 1
+  } = options;
+  const fetchPage = async (pageOffset, limit) => {
     const query = {
-      limit: pageSize,
+      limit,
       offset: pageOffset,
       order
     };
@@ -4183,19 +4208,21 @@ async function* listPaginated(arbi, options = {}) {
   const queue = [];
   const tryEnqueue = () => {
     while (!done && queue.length < depth && issued < MAX_PAGES) {
-      const p2 = fetchPage(nextOffsetToIssue);
-      p2.catch(() => {
+      const limit = issued === 0 && firstPageSize !== void 0 ? firstPageSize : pageSize;
+      const promise = fetchPage(nextOffsetToIssue, limit);
+      promise.catch(() => {
       });
-      queue.push(p2);
-      nextOffsetToIssue += pageSize;
+      queue.push({ limit, promise });
+      nextOffsetToIssue += limit;
       issued++;
     }
   };
   tryEnqueue();
   while (queue.length > 0 && !signal?.aborted) {
-    const page = await queue.shift();
+    const { limit, promise } = queue.shift();
+    const page = await promise;
     if (signal?.aborted) return;
-    const isShort = page.length < pageSize;
+    const isShort = page.length < limit;
     if (isShort) done = true;
     tryEnqueue();
     if (page.length > 0) {
@@ -4243,11 +4270,11 @@ async function updateDocuments(arbi, documents) {
     "Failed to update documents"
   );
 }
-async function uploadUrl(arbi, urls, workspaceId, shared = false) {
+async function uploadUrl(arbi, urls, shared = false) {
   return requireData(
     await arbi.fetch.POST("/v1/document/upload-url", {
       params: {
-        query: { urls, workspace_ext_id: workspaceId, shared }
+        query: { urls, shared }
       }
     }),
     "Failed to upload from URLs"
@@ -4260,29 +4287,31 @@ async function getParsedContent(auth, docId, stage) {
   });
   return res.json();
 }
-async function uploadFile(auth, workspaceId, fileData, fileName, options) {
+async function uploadFile(auth, fileData, fileName, options) {
   const formData = new FormData();
   formData.append("files", fileData, fileName);
-  const params = new URLSearchParams({ workspace_ext_id: workspaceId });
+  const params = new URLSearchParams();
   if (options?.folder) params.set("folder", sanitizeFolderPath(options.folder));
   if (options?.configExtId) params.set("config_ext_id", options.configExtId);
+  const qs = params.toString();
   const res = await authenticatedFetch({
     ...auth,
-    path: `/v1/document/upload?${params.toString()}`,
+    path: qs ? `/v1/document/upload?${qs}` : `/v1/document/upload`,
     method: "POST",
     body: formData
   });
   return res.json();
 }
-async function uploadFiles(auth, workspaceId, files, options) {
+async function uploadFiles(auth, files, options) {
   const formData = new FormData();
   for (const f2 of files) formData.append("files", f2.data, f2.name);
-  const params = new URLSearchParams({ workspace_ext_id: workspaceId });
+  const params = new URLSearchParams();
   if (options?.folder) params.set("folder", sanitizeFolderPath(options.folder));
   if (options?.configExtId) params.set("config_ext_id", options.configExtId);
+  const qs = params.toString();
   const res = await authenticatedFetch({
     ...auth,
-    path: `/v1/document/upload?${params.toString()}`,
+    path: qs ? `/v1/document/upload?${qs}` : `/v1/document/upload`,
     method: "POST",
     body: formData
   });
@@ -4294,6 +4323,68 @@ async function downloadDocument(auth, docId) {
     path: `/v1/document/${docId}/download`
   });
 }
+function mimeFromName(name) {
+  const dot = name.lastIndexOf(".");
+  const ext = dot === -1 ? "" : name.slice(dot).toLowerCase();
+  switch (ext) {
+    case ".pdf":
+      return "application/pdf";
+    case ".txt":
+      return "text/plain";
+    case ".md":
+      return "text/markdown";
+    case ".html":
+    case ".htm":
+      return "text/html";
+    case ".doc":
+      return "application/msword";
+    case ".docx":
+      return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+    case ".xls":
+      return "application/vnd.ms-excel";
+    case ".xlsx":
+      return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
+    case ".ppt":
+      return "application/vnd.ms-powerpoint";
+    case ".pptx":
+      return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
+    case ".eml":
+      return "message/rfc822";
+    default:
+      return void 0;
+  }
+}
+async function presignedPut(url, body, onProgress) {
+  const XhrGlobal = globalThis.XMLHttpRequest;
+  if (XhrGlobal) {
+    await new Promise((resolve, reject) => {
+      const xhr = new XhrGlobal();
+      xhr.open("PUT", url);
+      if (onProgress) {
+        xhr.upload.addEventListener("progress", (e) => {
+          if (e.lengthComputable) onProgress(e.loaded, e.total);
+        });
+      }
+      xhr.addEventListener("load", () => {
+        if (xhr.status >= 200 && xhr.status < 300) {
+          resolve();
+        } else {
+          const snippet = (xhr.responseText || "").slice(0, 200);
+          reject(new Error(`Presigned PUT failed: ${xhr.status} ${snippet}`));
+        }
+      });
+      xhr.addEventListener("error", () => reject(new Error("Presigned PUT network error")));
+      xhr.addEventListener("abort", () => reject(new Error("Presigned PUT aborted")));
+      xhr.send(body);
+    });
+    return;
+  }
+  const putRes = await fetch(url, { method: "PUT", body });
+  if (!putRes.ok) {
+    const snippet = await putRes.text().catch(() => "");
+    throw new Error(`Presigned PUT failed: ${putRes.status} ${snippet.slice(0, 200)}`);
+  }
+}
 async function uploadDocumentsDirect(arbi, workspaceKey, files, options) {
   if (files.length === 0) return { doc_ext_ids: [], skipped: [] };
   const SECRETBOX_ENVELOPE_OVERHEAD = 40;
@@ -4326,6 +4417,11 @@ async function uploadDocumentsDirect(arbi, workspaceKey, files, options) {
   }
   const skipped = [];
   const toCommit = [];
+  const indexByName = /* @__PURE__ */ new Map();
+  files.forEach((f2, i2) => {
+    if (!indexByName.has(f2.name)) indexByName.set(f2.name, []);
+    indexByName.get(f2.name).push(i2);
+  });
   for (const item of items) {
     if (item.status !== "uploading" || !item.upload_url || !item.doc_ext_id) {
       skipped.push({
@@ -4335,21 +4431,34 @@ async function uploadDocumentsDirect(arbi, workspaceKey, files, options) {
       continue;
     }
     const queue = bytesByName.get(item.file_name);
-    if (!queue || queue.length === 0) {
+    const idxQueue = indexByName.get(item.file_name);
+    if (!queue || queue.length === 0 || !idxQueue || idxQueue.length === 0) {
       throw new Error(`upload-init returned slot for unknown file: ${item.file_name}`);
     }
     const raw = queue.shift();
+    const fileIndex = idxQueue.shift();
     const ciphertext = encryptFile(raw, workspaceKey);
-    const putRes = await fetch(item.upload_url, {
-      method: "PUT",
-      body: ciphertext
-    });
-    if (!putRes.ok) {
-      const body = await putRes.text().catch(() => "");
+    const total = ciphertext.length;
+    try {
+      await presignedPut(item.upload_url, ciphertext, (loaded, putTotal) => {
+        options?.onBytesProgress?.({
+          fileIndex,
+          fileName: item.file_name,
+          loaded,
+          total: putTotal
+        });
+      });
+    } catch (err) {
       throw new Error(
-        `Presigned PUT for ${item.doc_ext_id} failed: ${putRes.status} ${body.slice(0, 200)}`
+        `Presigned PUT for ${item.doc_ext_id} failed: ${err instanceof Error ? err.message : String(err)}`
       );
     }
+    options?.onBytesProgress?.({
+      fileIndex,
+      fileName: item.file_name,
+      loaded: total,
+      total
+    });
     toCommit.push(item.doc_ext_id);
   }
   if (toCommit.length === 0) {
@@ -5190,19 +5299,19 @@ var Arbi = class {
     get: (externalIds) => getDocuments(this.requireClient(), externalIds),
     delete: (externalIds) => deleteDocuments(this.requireClient(), externalIds),
     update: (documents) => updateDocuments(this.requireClient(), documents),
-    uploadUrl: (urls, shared, workspaceId) => uploadUrl(
-      this.requireClient(),
-      urls,
-      workspaceId ?? this.requireWorkspace(),
-      shared
-    ),
-    uploadFile: (fileData, fileName, options) => uploadFile(
-      this.getAuthHeaders(),
-      options?.workspaceId ?? this.requireWorkspace(),
-      fileData,
-      fileName,
-      options?.folder ? { folder: options.folder } : void 0
-    ),
+    uploadUrl: (urls, shared) => {
+      this.requireWorkspace();
+      return uploadUrl(this.requireClient(), urls, shared);
+    },
+    uploadFile: (fileData, fileName, options) => {
+      this.requireWorkspace();
+      return uploadFile(
+        this.getAuthHeaders(),
+        fileData,
+        fileName,
+        options?.folder ? { folder: options.folder } : void 0
+      );
+    },
     download: (docId) => downloadDocument(this.getAuthHeaders(), docId),
     getParsedContent: (docId, stage) => getParsedContent(this.getAuthHeaders(), docId, stage)
   };
@@ -5415,6 +5524,6 @@ function extractResponseText(response) {
   return parts.join("");
 }
 
-export { Arbi, ArbiApiError, ArbiError, agentconfig_exports as agentconfig, assistant_exports as assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts_exports as contacts, conversations_exports as conversations, countCitations, createAuthenticatedClient, dm_exports as dm, doctags_exports as doctags, documents_exports as documents, files_exports as files, formatAgentStepLabel, formatFileSize, formatUserName, formatWorkspaceChoices, getErrorMessage, health_exports as health, parseSSEEvents, performPasswordLogin, requireData, requireOk, resolveAuth, resolveCitations, resolveWorkspace, responses_exports as responses, selectWorkspace, selectWorkspaceById, settings_exports as settings, streamSSE, stripCitationMarkdown, summarizeCitations, tags_exports as tags, workspaces_exports as workspaces };
+export { Arbi, ArbiApiError, ArbiError, agentconfig_exports as agentconfig, assistant_exports as assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts_exports as contacts, conversations_exports as conversations, countCitations, createAuthenticatedClient, dm_exports as dm, doctags_exports as doctags, documents_exports as documents, files_exports as files, formatAgentStepLabel, formatFileSize, formatUserName, formatWorkspaceChoices, getErrorMessage, getRawWorkspaceKey, health_exports as health, parseSSEEvents, performPasswordLogin, requireData, requireOk, resolveAuth, resolveCitations, resolveWorkspace, responses_exports as responses, selectWorkspace, selectWorkspaceById, settings_exports as settings, streamSSE, stripCitationMarkdown, summarizeCitations, tags_exports as tags, workspaces_exports as workspaces };
 //# sourceMappingURL=browser.js.map
 //# sourceMappingURL=browser.js.map