@arbidocs/sdk 0.4.2 → 0.4.4

package/dist/index.cjs

@@ -3496,6 +3496,12 @@ var ArbiError = class extends Error {
     this.name = "ArbiError";
   }
 };
+var ArbiSessionExpiredError = class extends ArbiError {
+  constructor(message = "Your session expired. Sign in again.") {
+    super(message);
+    this.name = "ArbiSessionExpiredError";
+  }
+};
 var ArbiApiError = class extends ArbiError {
   apiError;
   constructor(message, apiError) {
@@ -5219,6 +5225,7 @@ __export(workspaces_exports, {
   copyDocuments: () => copyDocuments,
   createWorkspace: () => createWorkspace,
   deleteWorkspaces: () => deleteWorkspaces,
+  grantTemporaryWorkspaceAccess: () => grantTemporaryWorkspaceAccess,
   listWorkspaceUsers: () => listWorkspaceUsers,
   listWorkspaces: () => listWorkspaces,
   removeWorkspaceUsers: () => removeWorkspaceUsers,
@@ -5259,14 +5266,33 @@ async function updateWorkspace(arbi, body) {
 async function listWorkspaceUsers(arbi) {
   return requireData(await arbi.fetch.GET("/v1/workspace/users"), "Failed to fetch workspace users");
 }
-async function addWorkspaceUsers(arbi, emails, role = "collaborator") {
+async function addWorkspaceUsers(arbi, emails, role = "collaborator", target) {
   return requireData(
     await arbi.fetch.POST("/v1/workspace/users", {
-      body: { emails, role }
+      body: target ? {
+        emails,
+        role,
+        workspace_ext_id: target.workspaceExtId,
+        workspace_key: target.workspaceKey
+      } : { emails, role }
     }),
     "Failed to add users"
   );
 }
+async function grantTemporaryWorkspaceAccess(arbi, params) {
+  return requireData(
+    await arbi.fetch.POST("/v1/workspace/users", {
+      body: {
+        emails: params.emails,
+        role: params.role ?? "collaborator",
+        workspace_ext_id: params.workspaceExtId,
+        workspace_key: params.workspaceKey,
+        session_pubkey_b64: params.sessionPubkeyB64
+      }
+    }),
+    "Failed to grant temporary access"
+  );
+}
 async function removeWorkspaceUsers(arbi, userIds) {
   requireOk(
     await arbi.fetch.DELETE("/v1/workspace/users", {
@@ -5662,9 +5688,11 @@ __export(dm_exports, {
 // src/operations/agents.ts
 var agents_exports = {};
 __export(agents_exports, {
+  authorizeAgent: () => authorizeAgent,
   createAgent: () => createAgent,
   deleteAgents: () => deleteAgents,
   getAgentConfig: () => getAgentConfig,
+  getAuthorizeLimits: () => getAuthorizeLimits,
   listAgentWorkspaces: () => listAgentWorkspaces,
   listAgents: () => listAgents
 });
@@ -5699,6 +5727,21 @@ async function deleteAgents(arbi, agentExtIds) {
     "Failed to delete agents"
   );
 }
+async function authorizeAgent(arbi, agentExtId, workspaceExtId, ttlSeconds) {
+  return requireData(
+    await arbi.fetch.POST("/v1/agents/{agent_ext_id}/authorize", {
+      params: { path: { agent_ext_id: agentExtId } },
+      body: { workspace_ext_id: workspaceExtId, ttl_seconds: ttlSeconds ?? null }
+    }),
+    "Failed to authorize agent"
+  );
+}
+async function getAuthorizeLimits(arbi) {
+  return requireData(
+    await arbi.fetch.GET("/v1/agents/authorize-limits"),
+    "Failed to fetch authorize limits"
+  );
+}
 
 // src/operations/dm.ts
 function createDmCryptoContext(arbi, signingPrivateKeyBytes, userExtId) {
@@ -6085,6 +6128,7 @@ __export(tasks_exports, {
   createTask: () => createTask,
   deleteTask: () => deleteTask,
   getTask: () => getTask,
+  listTaskRuns: () => listTaskRuns,
   listTasks: () => listTasks,
   runTask: () => runTask,
   updateTask: () => updateTask
@@ -6122,6 +6166,12 @@ async function runTask(arbi, taskId) {
     "Failed to run task"
   );
 }
+async function listTaskRuns(arbi, taskId) {
+  return requireData(
+    await arbi.fetch.GET("/v1/tasks/{task_id}/runs", { params: { path: { task_id: taskId } } }),
+    "Failed to list task runs"
+  );
+}
 
 // src/operations/facts.ts
 var facts_exports = {};
@@ -6191,11 +6241,23 @@ async function remindFact(arbi, factId, options = {}) {
 }
 
 // src/arbi.ts
+function tokenExpiresWithin(token, seconds) {
+  const payload = token?.split(".")[1];
+  if (!payload) return false;
+  try {
+    const json = atob(payload.replace(/-/g, "+").replace(/_/g, "/"));
+    const { exp } = JSON.parse(json);
+    return typeof exp === "number" && exp * 1e3 - Date.now() < seconds * 1e3;
+  } catch {
+    return false;
+  }
+}
 var Arbi = class {
   client = null;
   loginResult = null;
   currentWorkspaceId = null;
   dmCryptoContext = null;
+  sessionExpiredListeners = /* @__PURE__ */ new Set();
   options;
   constructor(options) {
     this.options = options;
@@ -6208,7 +6270,13 @@ var Arbi = class {
     this.client = client.createArbiClient({
       baseUrl: this.options.url,
       deploymentDomain: domain,
-      credentials: this.options.credentials ?? "omit"
+      credentials: this.options.credentials ?? "omit",
+      // A silent re-login rotates the server session key: keep ours in step, or the next
+      // workspace open would seal the key for a session the server no longer has.
+      onReloginSuccess: ({ serverSessionKey }) => {
+        if (this.loginResult) this.loginResult = { ...this.loginResult, serverSessionKey };
+      },
+      onSessionExpired: () => this.expireSession()
     });
     await this.client.crypto.initSodium();
     return this.client;
@@ -6245,6 +6313,22 @@ var Arbi = class {
     const signingPrivateKey = client.crypto.base64ToBytes(signingPrivateKeyBase64);
     this.loginResult = await client.auth.loginWithKey({ email, signingPrivateKey });
   }
+  /** A workspace's key, sealed to the caller's own session (what `/open` and `POST /workspace/users` take). */
+  async sealWorkspaceKey(workspaceId) {
+    const client = this.requireClient();
+    const lr2 = this.requireLogin();
+    const all = await listWorkspaces(client);
+    const ws = all.find((w2) => w2.external_id === workspaceId);
+    if (!ws || !ws.wrapped_key) {
+      throw new ArbiError(`Workspace ${workspaceId} not found or has no encryption key`);
+    }
+    return generateEncryptedWorkspaceKey(
+      client,
+      ws.wrapped_key,
+      lr2.serverSessionKey,
+      client.crypto.bytesToBase64(lr2.signingPrivateKey)
+    );
+  }
   /**
    * Select a workspace by ID. Fetches the workspace list, finds the matching
    * workspace, decrypts the wrapped key, and sets up auth headers.
@@ -6271,6 +6355,52 @@ var Arbi = class {
     });
     this.currentWorkspaceId = ws.external_id;
   }
+  /**
+   * Subscribe to "the session cannot be recovered". Fires when a request failed with 401 or
+   * "No active workspace" and neither a workspace re-open nor a silent re-login was possible
+   * (the server refused; an unreachable server does not fire). The local session is already
+   * cleared when it fires: show "Your session expired, sign in again". Returns an unsubscribe.
+   */
+  onSessionExpired(listener) {
+    this.sessionExpiredListeners.add(listener);
+    return () => this.sessionExpiredListeners.delete(listener);
+  }
+  expireSession() {
+    if (this.loginResult) {
+      this.loginResult.signingPrivateKey.fill(0);
+      this.loginResult.serverSessionKey.fill(0);
+    }
+    this.loginResult = null;
+    this.currentWorkspaceId = null;
+    this.dmCryptoContext = null;
+    this.client?.session.clear();
+    for (const listener of [...this.sessionExpiredListeners]) listener();
+  }
+  /**
+   * Run a raw-fetch operation (streaming, multipart: no middleware) and, when it fails because
+   * the token expired or the server lost the open workspace, recover once and run it again.
+   * `run` must read the auth headers itself so the retry carries the fresh token.
+   */
+  async recovering(run) {
+    if (this.client && tokenExpiresWithin(this.client.session.getState().accessToken, 30)) {
+      await this.client.auth.relogin();
+    }
+    try {
+      return await run();
+    } catch (error) {
+      const message = error instanceof Error ? error.message : "";
+      const noWorkspace = /no active workspace/i.test(message);
+      const unauthorised = /\b401\b|token has expired/i.test(message);
+      if (!(error instanceof ArbiApiError) || !(noWorkspace || unauthorised)) throw error;
+      const client = this.requireClient();
+      const recovered = noWorkspace && await client.auth.reopenWorkspace() || await client.auth.relogin() !== null;
+      if (!recovered) {
+        this.expireSession();
+        throw new ArbiSessionExpiredError();
+      }
+      return run();
+    }
+  }
   /** Log out and clear internal state. */
   async logout() {
     if (this.client) {
@@ -6381,6 +6511,30 @@ var Arbi = class {
     update: (body) => updateWorkspace(this.requireClient(), body),
     listUsers: () => listWorkspaceUsers(this.requireClient()),
     addUsers: (emails, role) => addWorkspaceUsers(this.requireClient(), emails, role),
+    /**
+     * Grant users a temporary key scoped to ONE session (identified by its public key), for
+     * example an agent's run session minted by `agents.authorize()`. Nothing standing is created:
+     * the grant expires with that session. Works for any workspace the caller holds a key for;
+     * it does not need to be the open one.
+     */
+    grantTemporaryAccess: async (params) => {
+      const client = this.requireClient();
+      return grantTemporaryWorkspaceAccess(client, {
+        emails: params.emails,
+        workspaceExtId: params.workspaceId,
+        workspaceKey: await this.sealWorkspaceKey(params.workspaceId),
+        sessionPubkeyB64: params.sessionPubkeyB64,
+        role: params.role
+      });
+    },
+    /**
+     * Add users (standing, permanent membership) to ANY workspace the caller holds a key for,
+     * without opening it. `addUsers` does the same for the currently open workspace.
+     */
+    grantAccess: async (params) => addWorkspaceUsers(this.requireClient(), params.emails, params.role, {
+      workspaceExtId: params.workspaceId,
+      workspaceKey: await this.sealWorkspaceKey(params.workspaceId)
+    }),
     removeUsers: (userIds) => removeWorkspaceUsers(this.requireClient(), userIds),
     setUserRole: (userIds, role) => setUserRole(this.requireClient(), userIds, role),
     /**
@@ -6437,16 +6591,18 @@ var Arbi = class {
     },
     uploadFile: (fileData, fileName, options) => {
       this.requireWorkspace();
-      return uploadFile(
-        this.getAuthHeaders(),
-        fileData,
-        fileName,
-        options?.folder ? { folder: options.folder } : void 0
+      return this.recovering(
+        () => 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),
-    getThumbnails: (docIds) => getThumbnails(this.getAuthHeaders(), docIds)
+    download: (docId) => this.recovering(() => downloadDocument(this.getAuthHeaders(), docId)),
+    getParsedContent: (docId, stage) => this.recovering(() => getParsedContent(this.getAuthHeaders(), docId, stage)),
+    getThumbnails: (docIds) => this.recovering(() => getThumbnails(this.getAuthHeaders(), docIds))
   };
   // ── Conversations ─────────────────────────────────────────────────────
   conversations = {
@@ -6465,16 +6621,17 @@ var Arbi = class {
      * Returns the accumulated result after the stream completes.
      */
     query: async (question, options) => {
-      const auth = this.getAuthHeaders();
       const workspaceId = this.requireWorkspace();
-      const response = await queryAssistant({
-        ...auth,
-        workspaceId,
-        question,
-        docIds: options.docIds,
-        previousResponseId: options.previousResponseId,
-        model: options.model
-      });
+      const response = await this.recovering(
+        () => queryAssistant({
+          ...this.getAuthHeaders(),
+          workspaceId,
+          question,
+          docIds: options.docIds,
+          previousResponseId: options.previousResponseId,
+          model: options.model
+        })
+      );
       const callbacks = {
         onToken: options.onToken,
         onStreamStart: options.onStreamStart,
@@ -6493,6 +6650,8 @@ var Arbi = class {
      * Respond to an agent's question during a human-in-the-loop workflow.
      */
     respond: (assistantMessageExtId, answer) => respondToAgent(this.requireClient(), assistantMessageExtId, answer),
+    /** The skills the caller can run (own + shared marketplace), metadata only. */
+    listSkills: (options) => listSkills(this.requireClient(), options),
     /**
      * Search documents without LLM generation (retrieval only).
      */
@@ -6638,8 +6797,10 @@ var Arbi = class {
     list: (options) => listFiles(this.requireClient(), options),
     get: (fileId) => getFile(this.requireClient(), fileId),
     delete: (fileId) => deleteFile(this.requireClient(), fileId),
-    upload: (fileData, fileName, purpose) => uploadFile2(this.getAuthHeaders(), fileData, fileName, purpose),
-    getContent: (fileId) => getFileContent(this.getAuthHeaders(), fileId)
+    upload: (fileData, fileName, purpose) => this.recovering(
+      () => uploadFile2(this.getAuthHeaders(), fileData, fileName, purpose)
+    ),
+    getContent: (fileId) => this.recovering(() => getFileContent(this.getAuthHeaders(), fileId))
   };
   // ── Agents ──────────────────────────────────────────────────────────
   agents = {
@@ -6647,7 +6808,28 @@ var Arbi = class {
     create: (name) => createAgent(this.requireClient(), name),
     getConfig: (agentExtId) => getAgentConfig(this.requireClient(), agentExtId),
     listWorkspaces: (agentExtId) => listAgentWorkspaces(this.requireClient(), agentExtId),
-    delete: (agentExtIds) => deleteAgents(this.requireClient(), agentExtIds)
+    delete: (agentExtIds) => deleteAgents(this.requireClient(), agentExtIds),
+    /** Mint a TTL'd run session for an agent; see `grantTemporaryAccess` to also deposit a key. */
+    authorize: (agentExtId, workspaceId, ttlSeconds) => authorizeAgent(this.requireClient(), agentExtId, workspaceId, ttlSeconds),
+    authorizeLimits: () => getAuthorizeLimits(this.requireClient()),
+    /**
+     * Give an agent temporary, session-scoped access to a workspace: mint its run session, then
+     * deposit the workspace key on it. Returns the session details (`ttl` in seconds).
+     */
+    grantTemporaryAccess: async (agentExtId, workspaceId, ttlSeconds) => {
+      const session = await authorizeAgent(
+        this.requireClient(),
+        agentExtId,
+        workspaceId,
+        ttlSeconds
+      );
+      await this.workspaces.grantTemporaryAccess({
+        workspaceId,
+        emails: [session.agent_email],
+        sessionPubkeyB64: session.session_pubkey_b64
+      });
+      return session;
+    }
   };
   // ── Scheduled tasks ───────────────────────────────────────────────────
   tasks = {
@@ -6656,7 +6838,26 @@ var Arbi = class {
     get: (taskId) => getTask(this.requireClient(), taskId),
     update: (taskId, body) => updateTask(this.requireClient(), taskId, body),
     delete: (taskId) => deleteTask(this.requireClient(), taskId),
-    run: (taskId) => runTask(this.requireClient(), taskId)
+    run: (taskId) => runTask(this.requireClient(), taskId),
+    listRuns: (taskId) => listTaskRuns(this.requireClient(), taskId),
+    /**
+     * Run a task and, when the run is parked waiting for workspace access (a temporary
+     * authorization tier), grant the agent's run session a temporary key and run it again.
+     * A task that needs no approval returns from the first call already running.
+     */
+    runWithAuth: async (taskId) => {
+      const client = this.requireClient();
+      const first = await runTask(client, taskId);
+      if (!first.pending_auth_request_id || !first.pending_auth_session_pubkey || !first.pending_auth_agent_email || !first.workspace_ext_id) {
+        return first;
+      }
+      await this.workspaces.grantTemporaryAccess({
+        workspaceId: first.workspace_ext_id,
+        emails: [first.pending_auth_agent_email],
+        sessionPubkeyB64: first.pending_auth_session_pubkey
+      });
+      return runTask(client, taskId);
+    }
   };
   // ── Workspace facts ─────────────────────────────────────────────────
   facts = {
@@ -7824,6 +8025,7 @@ async function startDmListener(options) {
 exports.Arbi = Arbi;
 exports.ArbiApiError = ArbiApiError;
 exports.ArbiError = ArbiError;
+exports.ArbiSessionExpiredError = ArbiSessionExpiredError;
 exports.ClaudeOrchestrator = ClaudeOrchestrator;
 exports.DOC_TERMINAL_STATUSES = DOC_TERMINAL_STATUSES;
 exports.FileConfigStore = FileConfigStore;