@arbidocs/sdk 0.3.11 → 0.3.14

package/dist/browser.cjs

@@ -1,13 +1,6 @@
 'use strict';
 
 var client = require('@arbidocs/client');
-var fs = require('fs');
-var path = require('path');
-
-function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
-
-var fs__default = /*#__PURE__*/_interopDefault(fs);
-var path__default = /*#__PURE__*/_interopDefault(path);
 
 var __defProp = Object.defineProperty;
 var __export = (target, all) => {
@@ -42,7 +35,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
@@ -72,8 +82,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}`,
@@ -112,7 +122,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 };
 }
@@ -127,7 +141,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 };
 }
@@ -180,6 +199,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();
@@ -187,6 +210,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 = client.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 = client.base64ToBytes(creds.signingPrivateKeyBase64);
+    const serverSessionKey = client.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,
@@ -199,10 +248,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 = [];
@@ -230,11 +326,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) => {
@@ -278,6 +376,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?.();
     },
@@ -290,8 +391,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);
     },
@@ -342,11 +447,13 @@ async function streamSSE(response, callbacks = {}) {
     text,
     assistantMessageExtId,
     agentSteps,
+    toolCallCount,
     errors,
     userMessage,
     metadata,
     artifacts,
-    usage
+    usage,
+    context
   };
 }
 var consumeSSEStream = streamSSE;
@@ -470,16 +577,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");
 }
@@ -518,22 +643,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__default.default.readFileSync(filePath);
-  const fileName = path__default.default.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({
@@ -733,6 +867,8 @@ async function retrieve(options) {
     input: query,
     workspace_ext_id: workspaceId,
     stream: false,
+    background: false,
+    store: true,
     tools,
     ...model ? { model } : {}
   };
@@ -748,6 +884,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([])
@@ -1207,11 +1345,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)
@@ -1353,9 +1492,74 @@ var Arbi = class {
   }
 };
 
+// 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("");
+}
+
 exports.Arbi = Arbi;
 exports.ArbiApiError = ArbiApiError;
 exports.ArbiError = ArbiError;
+exports.LIFECYCLE_LABELS = LIFECYCLE_LABELS;
+exports.TOOL_LABELS = TOOL_LABELS;
 exports.agentconfig = agentconfig_exports;
 exports.assistant = assistant_exports;
 exports.authenticatedFetch = authenticatedFetch;
@@ -1372,6 +1576,7 @@ exports.dm = dm_exports;
 exports.doctags = doctags_exports;
 exports.documents = documents_exports;
 exports.files = files_exports;
+exports.formatAgentStepLabel = formatAgentStepLabel;
 exports.formatFileSize = formatFileSize;
 exports.formatUserName = formatUserName;
 exports.formatWorkspaceChoices = formatWorkspaceChoices;
@@ -1383,6 +1588,7 @@ exports.requireData = requireData;
 exports.requireOk = requireOk;
 exports.resolveAuth = resolveAuth;
 exports.resolveWorkspace = resolveWorkspace;
+exports.responses = responses_exports;
 exports.selectWorkspace = selectWorkspace;
 exports.selectWorkspaceById = selectWorkspaceById;
 exports.settings = settings_exports;