@arbidocs/sdk 0.3.10 → 0.3.13

package/dist/browser.d.cts

@@ -1,2 +1,2 @@
-export { AgentStepEvent, Arbi, ArbiApiError, ArbiError, ArbiOptions, ArtifactEvent, AuthContext, AuthHeaders, AuthenticatedClient, ChatSession, CliConfig, CliCredentials, ConfigStore, ConnectOptions, MessageMetadataPayload, OutputTokensDetails, QueryOptions, ReconnectOptions, ReconnectableWsConnection, ResponseCompletedEvent, ResponseContentPartAddedEvent, ResponseCreatedEvent, ResponseFailedEvent, ResponseOutputItemAddedEvent, ResponseOutputItemDoneEvent, ResponseOutputTextDeltaEvent, ResponseOutputTextDoneEvent, ResponseUsage, SSEEvent, SSEStreamCallbacks, SSEStreamResult, SSEStreamStartData, UserInfo, UserInputRequestEvent, UserMessageEvent, WorkspaceContext, WsConnection, agentconfig, assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts, conversations, createAuthenticatedClient, dm, doctags, documents, files, formatFileSize, formatUserName, formatWorkspaceChoices, getErrorMessage, health, parseSSEEvents, performPasswordLogin, requireData, requireOk, resolveAuth, resolveWorkspace, selectWorkspace, selectWorkspaceById, settings, streamSSE, tags, workspaces } from './index.cjs';
+export { f as AgentStepEvent, g as Arbi, h as ArbiApiError, i as ArbiError, j as ArbiOptions, k as ArtifactEvent, l as AuthContext, A as AuthHeaders, m as AuthenticatedClient, c as ChatSession, a as CliConfig, b as CliCredentials, C as ConfigStore, n as ConnectOptions, L as LIFECYCLE_LABELS, p as MessageMetadataPayload, O as OutputTokensDetails, Q as QueryOptions, R as ReconnectOptions, q as ReconnectableWsConnection, r as ResponseCompletedEvent, s as ResponseContentPartAddedEvent, t as ResponseCreatedEvent, u as ResponseFailedEvent, v as ResponseOutputItemAddedEvent, w as ResponseOutputItemDoneEvent, x as ResponseOutputTextDeltaEvent, y as ResponseOutputTextDoneEvent, z as ResponseUsage, S as SSEEvent, B as SSEStreamCallbacks, E as SSEStreamResult, G as SSEStreamStartData, T as TOOL_LABELS, H as UserInfo, I as UserInputRequestEvent, J as UserMessageEvent, W as WorkspaceContext, K as WsConnection, N as agentconfig, P as assistant, V as authenticatedFetch, X as buildRetrievalChunkTool, Y as buildRetrievalFullContextTool, Z as buildRetrievalTocTool, _ as connectWebSocket, $ as connectWithReconnect, a0 as consumeSSEStream, a1 as contacts, a2 as conversations, a3 as createAuthenticatedClient, a5 as dm, a6 as doctags, a7 as documents, a8 as files, a9 as formatAgentStepLabel, aa as formatFileSize, ab as formatUserName, ac as formatWorkspaceChoices, ag as getErrorMessage, ah as health, ai as parseSSEEvents, aj as performPasswordLogin, ak as requireData, al as requireOk, am as resolveAuth, an as resolveWorkspace, ao as selectWorkspace, ap as selectWorkspaceById, aq as settings, ar as streamSSE, as as tags, at as workspaces } from './browser-BszYf6lU.cjs';
 import '@arbidocs/client';

package/dist/browser.d.ts

@@ -1,2 +1,2 @@
-export { AgentStepEvent, Arbi, ArbiApiError, ArbiError, ArbiOptions, ArtifactEvent, AuthContext, AuthHeaders, AuthenticatedClient, ChatSession, CliConfig, CliCredentials, ConfigStore, ConnectOptions, MessageMetadataPayload, OutputTokensDetails, QueryOptions, ReconnectOptions, ReconnectableWsConnection, ResponseCompletedEvent, ResponseContentPartAddedEvent, ResponseCreatedEvent, ResponseFailedEvent, ResponseOutputItemAddedEvent, ResponseOutputItemDoneEvent, ResponseOutputTextDeltaEvent, ResponseOutputTextDoneEvent, ResponseUsage, SSEEvent, SSEStreamCallbacks, SSEStreamResult, SSEStreamStartData, UserInfo, UserInputRequestEvent, UserMessageEvent, WorkspaceContext, WsConnection, agentconfig, assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts, conversations, createAuthenticatedClient, dm, doctags, documents, files, formatFileSize, formatUserName, formatWorkspaceChoices, getErrorMessage, health, parseSSEEvents, performPasswordLogin, requireData, requireOk, resolveAuth, resolveWorkspace, selectWorkspace, selectWorkspaceById, settings, streamSSE, tags, workspaces } from './index.js';
+export { f as AgentStepEvent, g as Arbi, h as ArbiApiError, i as ArbiError, j as ArbiOptions, k as ArtifactEvent, l as AuthContext, A as AuthHeaders, m as AuthenticatedClient, c as ChatSession, a as CliConfig, b as CliCredentials, C as ConfigStore, n as ConnectOptions, L as LIFECYCLE_LABELS, p as MessageMetadataPayload, O as OutputTokensDetails, Q as QueryOptions, R as ReconnectOptions, q as ReconnectableWsConnection, r as ResponseCompletedEvent, s as ResponseContentPartAddedEvent, t as ResponseCreatedEvent, u as ResponseFailedEvent, v as ResponseOutputItemAddedEvent, w as ResponseOutputItemDoneEvent, x as ResponseOutputTextDeltaEvent, y as ResponseOutputTextDoneEvent, z as ResponseUsage, S as SSEEvent, B as SSEStreamCallbacks, E as SSEStreamResult, G as SSEStreamStartData, T as TOOL_LABELS, H as UserInfo, I as UserInputRequestEvent, J as UserMessageEvent, W as WorkspaceContext, K as WsConnection, N as agentconfig, P as assistant, V as authenticatedFetch, X as buildRetrievalChunkTool, Y as buildRetrievalFullContextTool, Z as buildRetrievalTocTool, _ as connectWebSocket, $ as connectWithReconnect, a0 as consumeSSEStream, a1 as contacts, a2 as conversations, a3 as createAuthenticatedClient, a5 as dm, a6 as doctags, a7 as documents, a8 as files, a9 as formatAgentStepLabel, aa as formatFileSize, ab as formatUserName, ac as formatWorkspaceChoices, ag as getErrorMessage, ah as health, ai as parseSSEEvents, aj as performPasswordLogin, ak as requireData, al as requireOk, am as resolveAuth, an as resolveWorkspace, ao as selectWorkspace, ap as selectWorkspaceById, aq as settings, ar as streamSSE, as as tags, at as workspaces } from './browser-BszYf6lU.js';
 import '@arbidocs/client';

package/dist/browser.js

@@ -1,6 +1,4 @@
 import { createArbiClient, base64ToBytes, deriveEncryptionKeypairFromSigning, sealedBoxDecrypt, createWorkspaceKeyHeader, buildWebSocketUrl, createAuthMessage, parseServerMessage, isMessageType } from '@arbidocs/client';
-import fs from 'fs';
-import path from 'path';
 
 var __defProp = Object.defineProperty;
 var __export = (target, all) => {
@@ -35,17 +33,55 @@ function requireOk(result, message) {
   }
 }
 function getErrorMessage(err) {
-  return err instanceof Error ? err.message : String(err);
+  if (!(err instanceof Error)) return String(err);
+  const rootCause = getDeepestCause(err);
+  if (rootCause !== err) {
+    const rootMsg = rootCause.message || String(rootCause);
+    if (err.message && !err.message.includes(rootMsg)) {
+      return `${err.message}: ${rootMsg}`;
+    }
+  }
+  return err.message;
+}
+function getDeepestCause(err) {
+  let current = err;
+  const seen = /* @__PURE__ */ new Set();
+  while (current.cause instanceof Error && !seen.has(current.cause)) {
+    seen.add(current);
+    current = current.cause;
+  }
+  return current;
 }
 
 // src/fetch.ts
-var STATUS_MESSAGES = {
+var STATUS_FALLBACKS = {
   401: "Token has expired. Please run: arbi login",
+  403: "Access denied. You may not have permission for this workspace.",
+  404: "Resource not found.",
   503: "The selected LLM is not responding. Please retry or select another model."
 };
+async function extractErrorMessage(res) {
+  let bodyDetail;
+  try {
+    const text = await res.text();
+    if (text) {
+      try {
+        const json = JSON.parse(text);
+        bodyDetail = json.detail || json.message || json.error || void 0;
+      } catch {
+        if (text.length < 200) bodyDetail = text;
+      }
+    }
+  } catch {
+  }
+  if (bodyDetail) return `Request failed (${res.status}): ${bodyDetail}`;
+  const fallback = STATUS_FALLBACKS[res.status];
+  if (fallback) return fallback;
+  return `Request failed: ${res.status} ${res.statusText}`;
+}
 async function authenticatedFetch(options) {
-  const { baseUrl, accessToken, workspaceKeyHeader, path: path2, method, body, headers } = options;
-  const res = await fetch(`${baseUrl}${path2}`, {
+  const { baseUrl, accessToken, workspaceKeyHeader, path, method, body, headers } = options;
+  const res = await fetch(`${baseUrl}${path}`, {
     method: method ?? (body ? "POST" : "GET"),
     headers: {
       Authorization: `Bearer ${accessToken}`,
@@ -55,10 +91,8 @@ async function authenticatedFetch(options) {
     body
   });
   if (!res.ok) {
-    const knownMessage = STATUS_MESSAGES[res.status];
-    if (knownMessage) throw new Error(knownMessage);
-    const text = await res.text().catch(() => "");
-    throw new Error(`Request failed: ${res.status} ${text}`);
+    const message = await extractErrorMessage(res);
+    throw new ArbiApiError(message, { status: res.status, statusText: res.statusText });
   }
   return res;
 }
@@ -86,7 +120,11 @@ async function createAuthenticatedClient(config, creds, store) {
   });
   store.saveCredentials({
     ...creds,
-    serverSessionKeyBase64: arbi.crypto.bytesToBase64(loginResult.serverSessionKey)
+    serverSessionKeyBase64: arbi.crypto.bytesToBase64(loginResult.serverSessionKey),
+    accessToken: void 0,
+    workspaceKeyHeader: void 0,
+    workspaceId: void 0,
+    tokenTimestamp: void 0
   });
   return { arbi, loginResult };
 }
@@ -101,7 +139,12 @@ async function performPasswordLogin(config, email, password, store) {
   store.saveCredentials({
     email,
     signingPrivateKeyBase64: arbi.crypto.bytesToBase64(loginResult.signingPrivateKey),
-    serverSessionKeyBase64: arbi.crypto.bytesToBase64(loginResult.serverSessionKey)
+    serverSessionKeyBase64: arbi.crypto.bytesToBase64(loginResult.serverSessionKey),
+    // Clear any cached workspace tokens — new session key invalidates them
+    accessToken: void 0,
+    workspaceKeyHeader: void 0,
+    workspaceId: void 0,
+    tokenTimestamp: void 0
   });
   return { arbi, loginResult, config };
 }
@@ -154,6 +197,10 @@ async function resolveAuth(store) {
   const { arbi, loginResult } = await createAuthenticatedClient(config, creds, store);
   return { arbi, loginResult, config };
 }
+var TOKEN_MAX_AGE_MS = 50 * 60 * 1e3;
+function isCachedTokenValid(creds, workspaceId) {
+  return !!(creds.accessToken && creds.workspaceKeyHeader && creds.workspaceId === workspaceId && creds.tokenTimestamp && Date.now() - new Date(creds.tokenTimestamp).getTime() < TOKEN_MAX_AGE_MS);
+}
 async function resolveWorkspace(store, workspaceOpt) {
   const config = store.requireConfig();
   const creds = store.requireCredentials();
@@ -161,6 +208,32 @@ async function resolveWorkspace(store, workspaceOpt) {
   if (!workspaceId) {
     throw new ArbiError("No workspace selected. Run: arbi workspace select <id>");
   }
+  if (isCachedTokenValid(creds, workspaceId)) {
+    const arbi2 = createArbiClient({
+      baseUrl: config.baseUrl,
+      deploymentDomain: config.deploymentDomain,
+      credentials: "omit"
+    });
+    await arbi2.crypto.initSodium();
+    arbi2.session.setSelectedWorkspace(workspaceId);
+    arbi2.session.setAccessToken(creds.accessToken);
+    arbi2.session.setCachedWorkspaceHeader(workspaceId, creds.workspaceKeyHeader);
+    const signingPrivateKey = base64ToBytes(creds.signingPrivateKeyBase64);
+    const serverSessionKey = base64ToBytes(creds.serverSessionKeyBase64);
+    const loginResult2 = {
+      accessToken: creds.accessToken,
+      signingPrivateKey,
+      serverSessionKey
+    };
+    return {
+      arbi: arbi2,
+      loginResult: loginResult2,
+      config,
+      workspaceId,
+      accessToken: creds.accessToken,
+      workspaceKeyHeader: creds.workspaceKeyHeader
+    };
+  }
   const { arbi, loginResult } = await createAuthenticatedClient(config, creds, store);
   await selectWorkspaceById(
     arbi,
@@ -173,10 +246,57 @@ async function resolveWorkspace(store, workspaceOpt) {
   if (!accessToken || !workspaceKeyHeader) {
     throw new ArbiError("Authentication error \u2014 missing token or workspace key");
   }
+  store.saveCredentials({
+    ...store.requireCredentials(),
+    accessToken,
+    workspaceKeyHeader,
+    workspaceId,
+    tokenTimestamp: (/* @__PURE__ */ new Date()).toISOString()
+  });
   return { arbi, loginResult, config, workspaceId, accessToken, workspaceKeyHeader };
 }
 
 // src/sse.ts
+var TOOL_LABELS = {
+  search_documents: "Searching documents",
+  get_document_passages: "Reading document",
+  get_table_of_contents: "Getting table of contents",
+  view_document_pages: "Viewing document pages",
+  get_full_document: "Reading full document",
+  web_search: "Searching the web",
+  read_url: "Reading web pages",
+  ask_user: "Asking user",
+  compaction: "Compacting conversation",
+  personal_agent: "Running agent",
+  create_artifact: "Creating artifact",
+  create_plan: "Creating plan",
+  save_skill: "Saving skill",
+  run_code: "Running code"
+};
+var LIFECYCLE_LABELS = {
+  evaluation: "Evaluating results",
+  answering: "Writing answer",
+  reviewing: "Reviewing answer",
+  planning: "Planning",
+  tool_progress: "Working"
+};
+function formatAgentStepLabel(step) {
+  if (step.focus) return step.focus;
+  const detail = step.detail;
+  if (step.step === "tool_progress" && detail && detail.length > 0) {
+    const toolName = detail[0].tool;
+    const label = toolName && TOOL_LABELS[toolName] || LIFECYCLE_LABELS.tool_progress;
+    const message = detail[0].message;
+    return message ? `${label}: ${message}` : label;
+  }
+  if (step.step) {
+    return LIFECYCLE_LABELS[step.step] || step.step;
+  }
+  if (detail && detail.length > 0 && detail[0].tool) {
+    return TOOL_LABELS[detail[0].tool] || detail[0].tool;
+  }
+  return "";
+}
 function parseSSEEvents(chunk, buffer) {
   const combined = buffer + chunk;
   const events = [];
@@ -247,6 +367,11 @@ async function streamSSE(response, callbacks = {}) {
         usage = data.response.usage;
         callbacks.onUsage?.(data.response.usage);
       }
+      if (data.response?.metadata) {
+        const meta = data.response.metadata;
+        metadata = meta;
+        callbacks.onMetadata?.(meta);
+      }
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
       callbacks.onComplete?.();
     },
@@ -259,8 +384,8 @@ async function streamSSE(response, callbacks = {}) {
     // ARBI-specific events (dot-prefixed from server)
     "arbi.agent_step": (raw) => {
       const data = JSON.parse(raw);
-      const focus = data.focus || data.step || "";
-      if (focus) agentSteps.push(focus);
+      const label = formatAgentStepLabel(data);
+      if (label) agentSteps.push(label);
       callbacks.onAgentStep?.(data);
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
     },
@@ -439,16 +564,34 @@ function formatUserName(user) {
 // src/operations/documents.ts
 var documents_exports = {};
 __export(documents_exports, {
+  SUPPORTED_EXTENSIONS: () => SUPPORTED_EXTENSIONS,
   deleteDocuments: () => deleteDocuments,
   downloadDocument: () => downloadDocument,
   getDocuments: () => getDocuments,
   getParsedContent: () => getParsedContent,
   listDocuments: () => listDocuments,
+  sanitizeFolderPath: () => sanitizeFolderPath,
   updateDocuments: () => updateDocuments,
   uploadFile: () => uploadFile,
-  uploadLocalFile: () => uploadLocalFile,
+  uploadFiles: () => uploadFiles,
   uploadUrl: () => uploadUrl
 });
+var SUPPORTED_EXTENSIONS = /* @__PURE__ */ new Set([
+  ".pdf",
+  ".txt",
+  ".md",
+  ".html",
+  ".doc",
+  ".docx",
+  ".rtf",
+  ".ppt",
+  ".pptx",
+  ".xls",
+  ".xlsx"
+]);
+function sanitizeFolderPath(folderPath) {
+  return folderPath.replace(/[^a-zA-Z0-9_\-/]/g, "_").replace(/_{2,}/g, "_");
+}
 async function listDocuments(arbi) {
   return requireData(await arbi.fetch.GET("/v1/document/list"), "Failed to fetch documents");
 }
@@ -487,22 +630,31 @@ async function getParsedContent(auth, docId, stage) {
   });
   return res.json();
 }
-async function uploadFile(auth, workspaceId, fileData, fileName) {
+async function uploadFile(auth, workspaceId, fileData, fileName, options) {
   const formData = new FormData();
   formData.append("files", fileData, fileName);
+  const params = new URLSearchParams({ workspace_ext_id: workspaceId });
+  if (options?.folder) params.set("folder", sanitizeFolderPath(options.folder));
   const res = await authenticatedFetch({
     ...auth,
-    path: `/v1/document/upload?workspace_ext_id=${workspaceId}`,
+    path: `/v1/document/upload?${params.toString()}`,
     method: "POST",
     body: formData
   });
   return res.json();
 }
-async function uploadLocalFile(auth, workspaceId, filePath) {
-  const fileBuffer = fs.readFileSync(filePath);
-  const fileName = path.basename(filePath);
-  const result = await uploadFile(auth, workspaceId, new Blob([fileBuffer]), fileName);
-  return { ...result, fileName };
+async function uploadFiles(auth, workspaceId, files, options) {
+  const formData = new FormData();
+  for (const f of files) formData.append("files", f.data, f.name);
+  const params = new URLSearchParams({ workspace_ext_id: workspaceId });
+  if (options?.folder) params.set("folder", sanitizeFolderPath(options.folder));
+  const res = await authenticatedFetch({
+    ...auth,
+    path: `/v1/document/upload?${params.toString()}`,
+    method: "POST",
+    body: formData
+  });
+  return res.json();
 }
 async function downloadDocument(auth, docId) {
   return authenticatedFetch({
@@ -702,6 +854,8 @@ async function retrieve(options) {
     input: query,
     workspace_ext_id: workspaceId,
     stream: false,
+    background: false,
+    store: true,
     tools,
     ...model ? { model } : {}
   };
@@ -717,6 +871,8 @@ async function queryAssistant(options) {
     input: question,
     workspace_ext_id: workspaceId,
     stream: true,
+    background: false,
+    store: true,
     tools: {
       retrieval_chunk: buildRetrievalChunkTool(docIds),
       retrieval_full_context: buildRetrievalFullContextTool([])
@@ -1176,11 +1332,12 @@ var Arbi = class {
       workspaceId ?? this.requireWorkspace(),
       shared
     ),
-    uploadFile: (fileData, fileName, workspaceId) => uploadFile(
+    uploadFile: (fileData, fileName, options) => uploadFile(
       this.getAuthHeaders(),
-      workspaceId ?? this.requireWorkspace(),
+      options?.workspaceId ?? this.requireWorkspace(),
       fileData,
-      fileName
+      fileName,
+      options?.folder ? { folder: options.folder } : void 0
     ),
     download: (docId) => downloadDocument(this.getAuthHeaders(), docId),
     getParsedContent: (docId, stage) => getParsedContent(this.getAuthHeaders(), docId, stage)
@@ -1322,6 +1479,6 @@ var Arbi = class {
   }
 };
 
-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, createAuthenticatedClient, dm_exports as dm, doctags_exports as doctags, documents_exports as documents, files_exports as files, formatFileSize, formatUserName, formatWorkspaceChoices, getErrorMessage, health_exports as health, parseSSEEvents, performPasswordLogin, requireData, requireOk, resolveAuth, resolveWorkspace, selectWorkspace, selectWorkspaceById, settings_exports as settings, streamSSE, tags_exports as tags, workspaces_exports as workspaces };
+export { Arbi, ArbiApiError, ArbiError, LIFECYCLE_LABELS, TOOL_LABELS, agentconfig_exports as agentconfig, assistant_exports as assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts_exports as contacts, conversations_exports as conversations, 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, resolveWorkspace, selectWorkspace, selectWorkspaceById, settings_exports as settings, streamSSE, tags_exports as tags, workspaces_exports as workspaces };
 //# sourceMappingURL=browser.js.map
 //# sourceMappingURL=browser.js.map