@arbidocs/sdk 0.4.2 → 0.4.3

package/dist/browser.d.cts

@@ -1,2 +1,2 @@
-export { A as AgentControlStateEvent, a as AgentStepDeltaEvent, b as AgentStepEvent, c as Arbi, d as ArbiApiError, e as ArbiError, f as ArbiErrorEvent, g as ArbiOptions, h as ArtifactEvent, i as AuthContext, j as AuthHeaders, k as AuthenticatedClient, C as ChatSession, l as CitationSummary, m as CliConfig, n as CliCredentials, o as CommandDescriptor, p as ConfigStore, q as ConnectOptions, r as CreateFactRequest, s as CreateFactsRequest, t as CreateTaskRequest, u as DeferredInterjectionEvent, F as FactResponse, E as FactSearchRequest, G as FactSearchResponse, I as ListFactsOptions, K as MessageMetadataPayload, N as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as ProbeAnswerEvent, R as ProjectInvoicesResponse, S as ProjectUsageResponse, T as QueryOptions, U as ReconnectOptions, V as ReconnectableWsConnection, W as ResolvedCitation, Y as ResponseCompletedEvent, Z as ResponseContentPartAddedEvent, _ as ResponseCreatedEvent, $ as ResponseFailedEvent, a0 as ResponseOutputItemAddedEvent, a1 as ResponseOutputItemDoneEvent, a2 as ResponseOutputTextDeltaEvent, a3 as ResponseOutputTextDoneEvent, a4 as ResponseUsage, a5 as SSEEvent, a6 as SSEStreamCallbacks, a7 as SSEStreamResult, a8 as SSEStreamStartData, aa as SkillSummary, ac as TaskResponse, ad as TokenBudgetContext, ae as UpdateFactRequest, af as UpdateTaskRequest, al as UserDailyUsageResponse, am as UserInfo, an as UserInputRequestEvent, ao as UserMessageEvent, ap as WebSocketAuthError, aq as WorkspaceContext, ar as WsConnection, as as agentconfig, at as assistant, au as authenticatedFetch, aw as buildRetrievalChunkTool, ax as buildRetrievalFullContextTool, ay as buildRetrievalTocTool, az 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 facts, aL as files, aM as filterSkills, aN as formatAgentStepLabel, aO as formatFileSize, aP as formatItemLabel, aR as formatUserName, aS as formatWorkspaceChoices, aX as getErrorMessage, aY as getRawWorkspaceKey, aZ as health, a_ as parseSSEEvents, a$ as parseSlashCommand, b0 as parseSlashTokenInProgress, b1 as performPasswordLogin, b4 as projects, b5 as requireData, b6 as requireOk, b7 as resolveAuth, b8 as resolveCitations, b9 as resolveWorkspace, ba as responses, bb as selectWorkspace, bc as selectWorkspaceById, bd as settings, be as streamSSE, bf as stripCitationMarkdown, bg as summarizeCitations, bh as tags, bi as tasks, bj as workspaces } from './browser-U90kdb7T.cjs';
+export { A as AgentControlStateEvent, a as AgentStepDeltaEvent, b as AgentStepEvent, c as Arbi, d as ArbiApiError, e as ArbiError, f as ArbiErrorEvent, g as ArbiOptions, i as ArtifactEvent, j as AuthContext, k as AuthHeaders, l as AuthenticatedClient, C as ChatSession, m as CitationSummary, n as CliConfig, o as CliCredentials, p as CommandDescriptor, q as ConfigStore, r as ConnectOptions, s as CreateFactRequest, t as CreateFactsRequest, u as CreateTaskRequest, v as DeferredInterjectionEvent, F as FactResponse, G as FactSearchRequest, H as FactSearchResponse, J as ListFactsOptions, N as MessageMetadataPayload, O as MessageQueuedEvent, P as OutputTokensDetails, Q as ParsedSlashCommand, R as ProbeAnswerEvent, S as ProjectInvoicesResponse, T as ProjectUsageResponse, U as QueryOptions, V as ReconnectOptions, W as ReconnectableWsConnection, X as ResolvedCitation, Z as ResponseCompletedEvent, _ as ResponseContentPartAddedEvent, $ as ResponseCreatedEvent, a0 as ResponseFailedEvent, a1 as ResponseOutputItemAddedEvent, a2 as ResponseOutputItemDoneEvent, a3 as ResponseOutputTextDeltaEvent, a4 as ResponseOutputTextDoneEvent, a5 as ResponseUsage, a6 as SSEEvent, a7 as SSEStreamCallbacks, a8 as SSEStreamResult, a9 as SSEStreamStartData, ab as SkillSummary, ad as TaskResponse, ae as TokenBudgetContext, af as UpdateFactRequest, ag as UpdateTaskRequest, am as UserDailyUsageResponse, an as UserInfo, ao as UserInputRequestEvent, ap as UserMessageEvent, aq as WebSocketAuthError, ar as WorkspaceContext, as as WsConnection, at as agentconfig, au as assistant, av as authenticatedFetch, ax as buildRetrievalChunkTool, ay as buildRetrievalFullContextTool, az as buildRetrievalTocTool, aA as connectWebSocket, aB as connectWithReconnect, aC as consumeSSEStream, aD as contacts, aE as conversations, aF as countCitations, aG as createAuthenticatedClient, aI as dm, aJ as doctags, aK as documents, aL as facts, aM as files, aN as filterSkills, aO as formatAgentStepLabel, aP as formatFileSize, aQ as formatItemLabel, aS as formatUserName, aT as formatWorkspaceChoices, aY as getErrorMessage, aZ as getRawWorkspaceKey, a_ as health, a$ as parseSSEEvents, b0 as parseSlashCommand, b1 as parseSlashTokenInProgress, b2 as performPasswordLogin, b5 as projects, b6 as requireData, b7 as requireOk, b8 as resolveAuth, b9 as resolveCitations, ba as resolveWorkspace, bb as responses, bc as selectWorkspace, bd as selectWorkspaceById, be as settings, bf as streamSSE, bg as stripCitationMarkdown, bh as summarizeCitations, bi as tags, bj as tasks, bk as workspaces } from './browser-4VHbRx1J.cjs';
 import '@arbidocs/client';

package/dist/browser.d.ts

@@ -1,2 +1,2 @@
-export { A as AgentControlStateEvent, a as AgentStepDeltaEvent, b as AgentStepEvent, c as Arbi, d as ArbiApiError, e as ArbiError, f as ArbiErrorEvent, g as ArbiOptions, h as ArtifactEvent, i as AuthContext, j as AuthHeaders, k as AuthenticatedClient, C as ChatSession, l as CitationSummary, m as CliConfig, n as CliCredentials, o as CommandDescriptor, p as ConfigStore, q as ConnectOptions, r as CreateFactRequest, s as CreateFactsRequest, t as CreateTaskRequest, u as DeferredInterjectionEvent, F as FactResponse, E as FactSearchRequest, G as FactSearchResponse, I as ListFactsOptions, K as MessageMetadataPayload, N as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as ProbeAnswerEvent, R as ProjectInvoicesResponse, S as ProjectUsageResponse, T as QueryOptions, U as ReconnectOptions, V as ReconnectableWsConnection, W as ResolvedCitation, Y as ResponseCompletedEvent, Z as ResponseContentPartAddedEvent, _ as ResponseCreatedEvent, $ as ResponseFailedEvent, a0 as ResponseOutputItemAddedEvent, a1 as ResponseOutputItemDoneEvent, a2 as ResponseOutputTextDeltaEvent, a3 as ResponseOutputTextDoneEvent, a4 as ResponseUsage, a5 as SSEEvent, a6 as SSEStreamCallbacks, a7 as SSEStreamResult, a8 as SSEStreamStartData, aa as SkillSummary, ac as TaskResponse, ad as TokenBudgetContext, ae as UpdateFactRequest, af as UpdateTaskRequest, al as UserDailyUsageResponse, am as UserInfo, an as UserInputRequestEvent, ao as UserMessageEvent, ap as WebSocketAuthError, aq as WorkspaceContext, ar as WsConnection, as as agentconfig, at as assistant, au as authenticatedFetch, aw as buildRetrievalChunkTool, ax as buildRetrievalFullContextTool, ay as buildRetrievalTocTool, az 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 facts, aL as files, aM as filterSkills, aN as formatAgentStepLabel, aO as formatFileSize, aP as formatItemLabel, aR as formatUserName, aS as formatWorkspaceChoices, aX as getErrorMessage, aY as getRawWorkspaceKey, aZ as health, a_ as parseSSEEvents, a$ as parseSlashCommand, b0 as parseSlashTokenInProgress, b1 as performPasswordLogin, b4 as projects, b5 as requireData, b6 as requireOk, b7 as resolveAuth, b8 as resolveCitations, b9 as resolveWorkspace, ba as responses, bb as selectWorkspace, bc as selectWorkspaceById, bd as settings, be as streamSSE, bf as stripCitationMarkdown, bg as summarizeCitations, bh as tags, bi as tasks, bj as workspaces } from './browser-U90kdb7T.js';
+export { A as AgentControlStateEvent, a as AgentStepDeltaEvent, b as AgentStepEvent, c as Arbi, d as ArbiApiError, e as ArbiError, f as ArbiErrorEvent, g as ArbiOptions, i as ArtifactEvent, j as AuthContext, k as AuthHeaders, l as AuthenticatedClient, C as ChatSession, m as CitationSummary, n as CliConfig, o as CliCredentials, p as CommandDescriptor, q as ConfigStore, r as ConnectOptions, s as CreateFactRequest, t as CreateFactsRequest, u as CreateTaskRequest, v as DeferredInterjectionEvent, F as FactResponse, G as FactSearchRequest, H as FactSearchResponse, J as ListFactsOptions, N as MessageMetadataPayload, O as MessageQueuedEvent, P as OutputTokensDetails, Q as ParsedSlashCommand, R as ProbeAnswerEvent, S as ProjectInvoicesResponse, T as ProjectUsageResponse, U as QueryOptions, V as ReconnectOptions, W as ReconnectableWsConnection, X as ResolvedCitation, Z as ResponseCompletedEvent, _ as ResponseContentPartAddedEvent, $ as ResponseCreatedEvent, a0 as ResponseFailedEvent, a1 as ResponseOutputItemAddedEvent, a2 as ResponseOutputItemDoneEvent, a3 as ResponseOutputTextDeltaEvent, a4 as ResponseOutputTextDoneEvent, a5 as ResponseUsage, a6 as SSEEvent, a7 as SSEStreamCallbacks, a8 as SSEStreamResult, a9 as SSEStreamStartData, ab as SkillSummary, ad as TaskResponse, ae as TokenBudgetContext, af as UpdateFactRequest, ag as UpdateTaskRequest, am as UserDailyUsageResponse, an as UserInfo, ao as UserInputRequestEvent, ap as UserMessageEvent, aq as WebSocketAuthError, ar as WorkspaceContext, as as WsConnection, at as agentconfig, au as assistant, av as authenticatedFetch, ax as buildRetrievalChunkTool, ay as buildRetrievalFullContextTool, az as buildRetrievalTocTool, aA as connectWebSocket, aB as connectWithReconnect, aC as consumeSSEStream, aD as contacts, aE as conversations, aF as countCitations, aG as createAuthenticatedClient, aI as dm, aJ as doctags, aK as documents, aL as facts, aM as files, aN as filterSkills, aO as formatAgentStepLabel, aP as formatFileSize, aQ as formatItemLabel, aS as formatUserName, aT as formatWorkspaceChoices, aY as getErrorMessage, aZ as getRawWorkspaceKey, a_ as health, a$ as parseSSEEvents, b0 as parseSlashCommand, b1 as parseSlashTokenInProgress, b2 as performPasswordLogin, b5 as projects, b6 as requireData, b7 as requireOk, b8 as resolveAuth, b9 as resolveCitations, ba as resolveWorkspace, bb as responses, bc as selectWorkspace, bd as selectWorkspaceById, be as settings, bf as streamSSE, bg as stripCitationMarkdown, bh as summarizeCitations, bi as tags, bj as tasks, bk as workspaces } from './browser-4VHbRx1J.js';
 import '@arbidocs/client';

package/dist/browser.js

@@ -3481,6 +3481,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) {
@@ -5617,11 +5623,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;
@@ -5634,7 +5652,13 @@ var Arbi = class {
     this.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;
@@ -5697,6 +5721,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) {
@@ -5863,16 +5933,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 = {
@@ -5891,16 +5963,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,
@@ -6064,8 +6137,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 = {