@arbidocs/sdk 0.3.11 → 0.3.14

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 responses, ap as selectWorkspace, aq as selectWorkspaceById, ar as settings, as as streamSSE, at as tags, au as workspaces } from './browser-DtwXhEkG.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 responses, ap as selectWorkspace, aq as selectWorkspaceById, ar as settings, as as streamSSE, at as tags, au as workspaces } from './browser-DtwXhEkG.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,7 +33,24 @@ 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
@@ -65,8 +80,8 @@ async function extractErrorMessage(res) {
   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}`,
@@ -105,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 };
 }
@@ -120,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 };
 }
@@ -173,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();
@@ -180,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,
@@ -192,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 = [];
@@ -223,11 +324,13 @@ async function streamSSE(response, callbacks = {}) {
   let text = "";
   let assistantMessageExtId = null;
   const agentSteps = [];
+  let toolCallCount = 0;
   const errors = [];
   const artifacts = [];
   let userMessage = null;
   let metadata = null;
   let usage = null;
+  let context = null;
   const eventHandlers = {
     // OpenAI Responses API events (dot-separated names from server)
     "response.created": (raw) => {
@@ -271,6 +374,9 @@ async function streamSSE(response, callbacks = {}) {
         metadata = meta;
         callbacks.onMetadata?.(meta);
       }
+      if (data.context) {
+        context = data.context;
+      }
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
       callbacks.onComplete?.();
     },
@@ -283,8 +389,12 @@ 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);
+      const detail = data.detail;
+      if (detail && Array.isArray(detail)) {
+        toolCallCount += detail.filter((d) => d.tool).length;
+      }
       callbacks.onAgentStep?.(data);
       if (data.t != null) callbacks.onElapsedTime?.(data.t);
     },
@@ -335,11 +445,13 @@ async function streamSSE(response, callbacks = {}) {
     text,
     assistantMessageExtId,
     agentSteps,
+    toolCallCount,
     errors,
     userMessage,
     metadata,
     artifacts,
-    usage
+    usage,
+    context
   };
 }
 var consumeSSEStream = streamSSE;
@@ -463,16 +575,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");
 }
@@ -511,22 +641,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({
@@ -726,6 +865,8 @@ async function retrieve(options) {
     input: query,
     workspace_ext_id: workspaceId,
     stream: false,
+    background: false,
+    store: true,
     tools,
     ...model ? { model } : {}
   };
@@ -741,6 +882,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([])
@@ -1200,11 +1343,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)
@@ -1346,6 +1490,69 @@ 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 };
+// src/operations/responses.ts
+var responses_exports = {};
+__export(responses_exports, {
+  extractResponseText: () => extractResponseText,
+  getResponse: () => getResponse,
+  submitBackgroundQuery: () => submitBackgroundQuery
+});
+async function submitBackgroundQuery(options) {
+  const { workspaceId, question, docIds, previousResponseId, model, ...auth } = options;
+  const tools = {};
+  if (docIds.length > 0) {
+    tools.retrieval_chunk = {
+      name: "retrieval_chunk",
+      description: "retrieval chunk",
+      tool_args: { doc_ext_ids: docIds },
+      tool_responses: {}
+    };
+    tools.retrieval_full_context = {
+      name: "retrieval_full_context",
+      description: "retrieval full context",
+      tool_args: { doc_ext_ids: [] },
+      tool_responses: {}
+    };
+  }
+  const body = {
+    input: question,
+    workspace_ext_id: workspaceId,
+    stream: false,
+    background: true,
+    store: true,
+    tools,
+    ...previousResponseId ? { previous_response_id: previousResponseId } : {},
+    ...model ? { model } : {}
+  };
+  const res = await authenticatedFetch({
+    ...auth,
+    path: "/v1/responses",
+    method: "POST",
+    body: JSON.stringify(body),
+    headers: { "Content-Type": "application/json" }
+  });
+  return await res.json();
+}
+async function getResponse(auth, responseId) {
+  const res = await authenticatedFetch({
+    ...auth,
+    path: `/v1/responses/${responseId}`,
+    method: "GET"
+  });
+  return await res.json();
+}
+function extractResponseText(response) {
+  const parts = [];
+  for (const msg of response.output) {
+    for (const item of msg.content) {
+      if ("type" in item && item.type === "output_text" && "text" in item) {
+        parts.push(item.text);
+      }
+    }
+  }
+  return parts.join("");
+}
+
+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, responses_exports as responses, selectWorkspace, selectWorkspaceById, settings_exports as settings, streamSSE, tags_exports as tags, workspaces_exports as workspaces };
 //# sourceMappingURL=browser.js.map
 //# sourceMappingURL=browser.js.map