@arbidocs/sdk 0.3.74 → 0.3.76

package/dist/{browser-D_TPBZXY.d.cts → browser-D06F2w6S.d.cts}

@@ -445,6 +445,16 @@ declare const consumeSSEStream: typeof streamSSE;
  * Authenticates via SDK helpers, routes typed messages to the consumer.
  */
 
+/**
+ * Thrown when a WebSocket connection is rejected for authentication reasons
+ * (the server returned `auth_result { success: false }`, or no token was
+ * available to send). The reconnect loop uses this to decide whether to
+ * re-authenticate before the next attempt, versus treating the failure as a
+ * transient network drop.
+ */
+declare class WebSocketAuthError extends Error {
+    constructor(reason: string);
+}
 interface WsConnection {
     close: () => void;
 }
@@ -455,8 +465,28 @@ interface ConnectOptions {
     onClose?: (code: number, reason: string) => void;
 }
 interface ReconnectOptions extends ConnectOptions {
+    /**
+     * Max consecutive reconnect attempts after a drop before giving up and
+     * calling `onReconnectFailed`. Pass `Infinity` for an unattended daemon that
+     * must keep trying indefinitely (backoff still caps at {@link MAX_BACKOFF_MS}).
+     * Default 10.
+     */
     maxRetries?: number;
     initialDelayMs?: number;
+    /**
+     * Resolve the access token before each (re)connect attempt. Defaults to the
+     * static `accessToken`. Provide this so a token refreshed elsewhere (e.g. by
+     * auto-relogin on a REST 401) is picked up on reconnect instead of reusing a
+     * stale token captured when the listener first started.
+     */
+    getAccessToken?: () => string | null | Promise<string | null>;
+    /**
+     * Re-authenticate when a reconnect attempt is rejected for auth reasons
+     * (expired/invalid token — see {@link WebSocketAuthError}). Should refresh the
+     * token in whatever store `getAccessToken` reads from; the return value is
+     * ignored. Best-effort: errors are swallowed and normal backoff continues.
+     */
+    refreshAuth?: () => Promise<string | null>;
     onReconnecting?: (attempt: number, maxRetries: number) => void;
     onReconnected?: () => void;
     onReconnectFailed?: () => void;
@@ -475,7 +505,15 @@ declare function connectWebSocket(options: ConnectOptions): Promise<WsConnection
  * Connect WebSocket with automatic reconnection on disconnect.
  *
  * Initial connection throws on failure (same as connectWebSocket).
- * After successful auth, disconnects trigger exponential backoff reconnection.
+ * After successful auth, disconnects trigger exponential backoff reconnection:
+ * - a fresh token is resolved before every attempt (see `getAccessToken`), so a
+ *   token rotated elsewhere is picked up rather than reusing a stale one;
+ * - an attempt rejected for auth reasons triggers `refreshAuth` (re-login) so
+ *   the next attempt uses a new token;
+ * - retries continue up to `maxRetries` (pass `Infinity` for a daemon that must
+ *   never give up). Only after exhausting a finite limit is `onReconnectFailed`
+ *   called.
+ *
  * Call close() on the returned handle to stop reconnection attempts.
  */
 declare function connectWithReconnect(options: ReconnectOptions): Promise<ReconnectableWsConnection>;
@@ -4364,4 +4402,4 @@ declare namespace responses {
   export { type responses_SubmitBackgroundQueryOptions as SubmitBackgroundQueryOptions, responses_extractResponseText as extractResponseText, responses_getResponse as getResponse, responses_submitBackgroundQuery as submitBackgroundQuery };
 }
 
-export { type UploadOptions as $, type AgentStepEvent as A, type ResponseCreatedEvent as B, type ChatSession as C, DOC_TERMINAL_STATUSES as D, type ResponseFailedEvent as E, type FormattedWsMessage as F, type ResponseOutputItemAddedEvent as G, type ResponseOutputItemDoneEvent as H, type ResponseOutputTextDeltaEvent as I, type ResponseOutputTextDoneEvent as J, type ResponseUsage as K, type ListAllOptions as L, type MessageLevel as M, type SSEStreamCallbacks as N, type OutputTokensDetails as O, type ParsedSlashCommand as P, type QueryOptions as Q, type ReconnectOptions as R, type SSEEvent as S, type SSEStreamResult as T, type SSEStreamStartData as U, SUPPORTED_EXTENSIONS as V, type SkillSummary as W, type SkippedFile as X, type UploadBatchResult as Y, type UploadDirectOptions as Z, type UploadDirectResult as _, Arbi as a, type UploadResult as a0, type UserInfo as a1, type UserInputRequestEvent as a2, type UserMessageEvent as a3, type WorkspaceContext as a4, type WsConnection as a5, agentconfig as a6, assistant as a7, authenticatedFetch as a8, buildRetrievalChunkTool as a9, health as aA, parseSSEEvents as aB, parseSlashCommand as aC, parseSlashTokenInProgress as aD, performPasswordLogin as aE, performSigningKeyLogin as aF, performSsoDeviceFlowLogin as aG, requireData as aH, requireOk as aI, resolveAuth as aJ, resolveCitations as aK, resolveWorkspace as aL, responses as aM, selectWorkspace as aN, selectWorkspaceById as aO, settings as aP, streamSSE as aQ, stripCitationMarkdown as aR, summarizeCitations as aS, tags as aT, workspaces as aU, buildRetrievalFullContextTool as aa, buildRetrievalTocTool as ab, connectWebSocket as ac, connectWithReconnect as ad, consumeSSEStream as ae, contacts as af, conversations as ag, countCitations as ah, createAuthenticatedClient as ai, createDocumentWaiter as aj, dm as ak, doctags as al, documents as am, files as an, filterSkills as ao, formatAgentStepLabel as ap, formatFileSize as aq, formatStreamSummary as ar, formatUserName as as, formatWorkspaceChoices as at, formatWsMessage as au, generateEncryptedWorkspaceKey as av, generateNewWorkspaceKey as aw, getErrorCode as ax, getErrorMessage as ay, getRawWorkspaceKey as az, ArbiApiError as b, ArbiError as c, type ArbiErrorEvent as d, type ArbiOptions as e, type ArtifactEvent as f, type AuthContext as g, type AuthHeaders as h, type AuthenticatedClient as i, type CitationSummary as j, type CliConfig as k, type CliCredentials as l, type ConfigStore as m, type ConnectOptions as n, type DmCryptoContext as o, type DocumentListFields as p, type DocumentListOrder as q, type DocumentWaiter as r, type DocumentWaiterOptions as s, type ListPaginatedOptions as t, type MessageMetadataPayload$1 as u, type MessageQueuedEvent as v, type ReconnectableWsConnection as w, type ResolvedCitation as x, type ResponseCompletedEvent as y, type ResponseContentPartAddedEvent as z };
+export { type UploadOptions as $, type AgentStepEvent as A, type ResponseCreatedEvent as B, type ChatSession as C, DOC_TERMINAL_STATUSES as D, type ResponseFailedEvent as E, type FormattedWsMessage as F, type ResponseOutputItemAddedEvent as G, type ResponseOutputItemDoneEvent as H, type ResponseOutputTextDeltaEvent as I, type ResponseOutputTextDoneEvent as J, type ResponseUsage as K, type ListAllOptions as L, type MessageLevel as M, type SSEStreamCallbacks as N, type OutputTokensDetails as O, type ParsedSlashCommand as P, type QueryOptions as Q, type ReconnectOptions as R, type SSEEvent as S, type SSEStreamResult as T, type SSEStreamStartData as U, SUPPORTED_EXTENSIONS as V, type SkillSummary as W, type SkippedFile as X, type UploadBatchResult as Y, type UploadDirectOptions as Z, type UploadDirectResult as _, Arbi as a, type UploadResult as a0, type UserInfo as a1, type UserInputRequestEvent as a2, type UserMessageEvent as a3, WebSocketAuthError as a4, type WorkspaceContext as a5, type WsConnection as a6, agentconfig as a7, assistant as a8, authenticatedFetch as a9, getRawWorkspaceKey as aA, health as aB, parseSSEEvents as aC, parseSlashCommand as aD, parseSlashTokenInProgress as aE, performPasswordLogin as aF, performSigningKeyLogin as aG, performSsoDeviceFlowLogin as aH, requireData as aI, requireOk as aJ, resolveAuth as aK, resolveCitations as aL, resolveWorkspace as aM, responses as aN, selectWorkspace as aO, selectWorkspaceById as aP, settings as aQ, streamSSE as aR, stripCitationMarkdown as aS, summarizeCitations as aT, tags as aU, workspaces as aV, buildRetrievalChunkTool as aa, buildRetrievalFullContextTool as ab, buildRetrievalTocTool as ac, connectWebSocket as ad, connectWithReconnect as ae, consumeSSEStream as af, contacts as ag, conversations as ah, countCitations as ai, createAuthenticatedClient as aj, createDocumentWaiter as ak, dm as al, doctags as am, documents as an, files as ao, filterSkills as ap, formatAgentStepLabel as aq, formatFileSize as ar, formatStreamSummary as as, formatUserName as at, formatWorkspaceChoices as au, formatWsMessage as av, generateEncryptedWorkspaceKey as aw, generateNewWorkspaceKey as ax, getErrorCode as ay, getErrorMessage as az, ArbiApiError as b, ArbiError as c, type ArbiErrorEvent as d, type ArbiOptions as e, type ArtifactEvent as f, type AuthContext as g, type AuthHeaders as h, type AuthenticatedClient as i, type CitationSummary as j, type CliConfig as k, type CliCredentials as l, type ConfigStore as m, type ConnectOptions as n, type DmCryptoContext as o, type DocumentListFields as p, type DocumentListOrder as q, type DocumentWaiter as r, type DocumentWaiterOptions as s, type ListPaginatedOptions as t, type MessageMetadataPayload$1 as u, type MessageQueuedEvent as v, type ReconnectableWsConnection as w, type ResolvedCitation as x, type ResponseCompletedEvent as y, type ResponseContentPartAddedEvent as z };

package/dist/{browser-D_TPBZXY.d.ts → browser-D06F2w6S.d.ts}

@@ -445,6 +445,16 @@ declare const consumeSSEStream: typeof streamSSE;
  * Authenticates via SDK helpers, routes typed messages to the consumer.
  */
 
+/**
+ * Thrown when a WebSocket connection is rejected for authentication reasons
+ * (the server returned `auth_result { success: false }`, or no token was
+ * available to send). The reconnect loop uses this to decide whether to
+ * re-authenticate before the next attempt, versus treating the failure as a
+ * transient network drop.
+ */
+declare class WebSocketAuthError extends Error {
+    constructor(reason: string);
+}
 interface WsConnection {
     close: () => void;
 }
@@ -455,8 +465,28 @@ interface ConnectOptions {
     onClose?: (code: number, reason: string) => void;
 }
 interface ReconnectOptions extends ConnectOptions {
+    /**
+     * Max consecutive reconnect attempts after a drop before giving up and
+     * calling `onReconnectFailed`. Pass `Infinity` for an unattended daemon that
+     * must keep trying indefinitely (backoff still caps at {@link MAX_BACKOFF_MS}).
+     * Default 10.
+     */
     maxRetries?: number;
     initialDelayMs?: number;
+    /**
+     * Resolve the access token before each (re)connect attempt. Defaults to the
+     * static `accessToken`. Provide this so a token refreshed elsewhere (e.g. by
+     * auto-relogin on a REST 401) is picked up on reconnect instead of reusing a
+     * stale token captured when the listener first started.
+     */
+    getAccessToken?: () => string | null | Promise<string | null>;
+    /**
+     * Re-authenticate when a reconnect attempt is rejected for auth reasons
+     * (expired/invalid token — see {@link WebSocketAuthError}). Should refresh the
+     * token in whatever store `getAccessToken` reads from; the return value is
+     * ignored. Best-effort: errors are swallowed and normal backoff continues.
+     */
+    refreshAuth?: () => Promise<string | null>;
     onReconnecting?: (attempt: number, maxRetries: number) => void;
     onReconnected?: () => void;
     onReconnectFailed?: () => void;
@@ -475,7 +505,15 @@ declare function connectWebSocket(options: ConnectOptions): Promise<WsConnection
  * Connect WebSocket with automatic reconnection on disconnect.
  *
  * Initial connection throws on failure (same as connectWebSocket).
- * After successful auth, disconnects trigger exponential backoff reconnection.
+ * After successful auth, disconnects trigger exponential backoff reconnection:
+ * - a fresh token is resolved before every attempt (see `getAccessToken`), so a
+ *   token rotated elsewhere is picked up rather than reusing a stale one;
+ * - an attempt rejected for auth reasons triggers `refreshAuth` (re-login) so
+ *   the next attempt uses a new token;
+ * - retries continue up to `maxRetries` (pass `Infinity` for a daemon that must
+ *   never give up). Only after exhausting a finite limit is `onReconnectFailed`
+ *   called.
+ *
  * Call close() on the returned handle to stop reconnection attempts.
  */
 declare function connectWithReconnect(options: ReconnectOptions): Promise<ReconnectableWsConnection>;
@@ -4364,4 +4402,4 @@ declare namespace responses {
   export { type responses_SubmitBackgroundQueryOptions as SubmitBackgroundQueryOptions, responses_extractResponseText as extractResponseText, responses_getResponse as getResponse, responses_submitBackgroundQuery as submitBackgroundQuery };
 }
 
-export { type UploadOptions as $, type AgentStepEvent as A, type ResponseCreatedEvent as B, type ChatSession as C, DOC_TERMINAL_STATUSES as D, type ResponseFailedEvent as E, type FormattedWsMessage as F, type ResponseOutputItemAddedEvent as G, type ResponseOutputItemDoneEvent as H, type ResponseOutputTextDeltaEvent as I, type ResponseOutputTextDoneEvent as J, type ResponseUsage as K, type ListAllOptions as L, type MessageLevel as M, type SSEStreamCallbacks as N, type OutputTokensDetails as O, type ParsedSlashCommand as P, type QueryOptions as Q, type ReconnectOptions as R, type SSEEvent as S, type SSEStreamResult as T, type SSEStreamStartData as U, SUPPORTED_EXTENSIONS as V, type SkillSummary as W, type SkippedFile as X, type UploadBatchResult as Y, type UploadDirectOptions as Z, type UploadDirectResult as _, Arbi as a, type UploadResult as a0, type UserInfo as a1, type UserInputRequestEvent as a2, type UserMessageEvent as a3, type WorkspaceContext as a4, type WsConnection as a5, agentconfig as a6, assistant as a7, authenticatedFetch as a8, buildRetrievalChunkTool as a9, health as aA, parseSSEEvents as aB, parseSlashCommand as aC, parseSlashTokenInProgress as aD, performPasswordLogin as aE, performSigningKeyLogin as aF, performSsoDeviceFlowLogin as aG, requireData as aH, requireOk as aI, resolveAuth as aJ, resolveCitations as aK, resolveWorkspace as aL, responses as aM, selectWorkspace as aN, selectWorkspaceById as aO, settings as aP, streamSSE as aQ, stripCitationMarkdown as aR, summarizeCitations as aS, tags as aT, workspaces as aU, buildRetrievalFullContextTool as aa, buildRetrievalTocTool as ab, connectWebSocket as ac, connectWithReconnect as ad, consumeSSEStream as ae, contacts as af, conversations as ag, countCitations as ah, createAuthenticatedClient as ai, createDocumentWaiter as aj, dm as ak, doctags as al, documents as am, files as an, filterSkills as ao, formatAgentStepLabel as ap, formatFileSize as aq, formatStreamSummary as ar, formatUserName as as, formatWorkspaceChoices as at, formatWsMessage as au, generateEncryptedWorkspaceKey as av, generateNewWorkspaceKey as aw, getErrorCode as ax, getErrorMessage as ay, getRawWorkspaceKey as az, ArbiApiError as b, ArbiError as c, type ArbiErrorEvent as d, type ArbiOptions as e, type ArtifactEvent as f, type AuthContext as g, type AuthHeaders as h, type AuthenticatedClient as i, type CitationSummary as j, type CliConfig as k, type CliCredentials as l, type ConfigStore as m, type ConnectOptions as n, type DmCryptoContext as o, type DocumentListFields as p, type DocumentListOrder as q, type DocumentWaiter as r, type DocumentWaiterOptions as s, type ListPaginatedOptions as t, type MessageMetadataPayload$1 as u, type MessageQueuedEvent as v, type ReconnectableWsConnection as w, type ResolvedCitation as x, type ResponseCompletedEvent as y, type ResponseContentPartAddedEvent as z };
+export { type UploadOptions as $, type AgentStepEvent as A, type ResponseCreatedEvent as B, type ChatSession as C, DOC_TERMINAL_STATUSES as D, type ResponseFailedEvent as E, type FormattedWsMessage as F, type ResponseOutputItemAddedEvent as G, type ResponseOutputItemDoneEvent as H, type ResponseOutputTextDeltaEvent as I, type ResponseOutputTextDoneEvent as J, type ResponseUsage as K, type ListAllOptions as L, type MessageLevel as M, type SSEStreamCallbacks as N, type OutputTokensDetails as O, type ParsedSlashCommand as P, type QueryOptions as Q, type ReconnectOptions as R, type SSEEvent as S, type SSEStreamResult as T, type SSEStreamStartData as U, SUPPORTED_EXTENSIONS as V, type SkillSummary as W, type SkippedFile as X, type UploadBatchResult as Y, type UploadDirectOptions as Z, type UploadDirectResult as _, Arbi as a, type UploadResult as a0, type UserInfo as a1, type UserInputRequestEvent as a2, type UserMessageEvent as a3, WebSocketAuthError as a4, type WorkspaceContext as a5, type WsConnection as a6, agentconfig as a7, assistant as a8, authenticatedFetch as a9, getRawWorkspaceKey as aA, health as aB, parseSSEEvents as aC, parseSlashCommand as aD, parseSlashTokenInProgress as aE, performPasswordLogin as aF, performSigningKeyLogin as aG, performSsoDeviceFlowLogin as aH, requireData as aI, requireOk as aJ, resolveAuth as aK, resolveCitations as aL, resolveWorkspace as aM, responses as aN, selectWorkspace as aO, selectWorkspaceById as aP, settings as aQ, streamSSE as aR, stripCitationMarkdown as aS, summarizeCitations as aT, tags as aU, workspaces as aV, buildRetrievalChunkTool as aa, buildRetrievalFullContextTool as ab, buildRetrievalTocTool as ac, connectWebSocket as ad, connectWithReconnect as ae, consumeSSEStream as af, contacts as ag, conversations as ah, countCitations as ai, createAuthenticatedClient as aj, createDocumentWaiter as ak, dm as al, doctags as am, documents as an, files as ao, filterSkills as ap, formatAgentStepLabel as aq, formatFileSize as ar, formatStreamSummary as as, formatUserName as at, formatWorkspaceChoices as au, formatWsMessage as av, generateEncryptedWorkspaceKey as aw, generateNewWorkspaceKey as ax, getErrorCode as ay, getErrorMessage as az, ArbiApiError as b, ArbiError as c, type ArbiErrorEvent as d, type ArbiOptions as e, type ArtifactEvent as f, type AuthContext as g, type AuthHeaders as h, type AuthenticatedClient as i, type CitationSummary as j, type CliConfig as k, type CliCredentials as l, type ConfigStore as m, type ConnectOptions as n, type DmCryptoContext as o, type DocumentListFields as p, type DocumentListOrder as q, type DocumentWaiter as r, type DocumentWaiterOptions as s, type ListPaginatedOptions as t, type MessageMetadataPayload$1 as u, type MessageQueuedEvent as v, type ReconnectableWsConnection as w, type ResolvedCitation as x, type ResponseCompletedEvent as y, type ResponseContentPartAddedEvent as z };

package/dist/browser.cjs

@@ -3986,6 +3986,12 @@ async function streamSSE(response, callbacks = {}) {
 var consumeSSEStream = streamSSE;
 var AUTH_TIMEOUT_MS = 1e4;
 var MAX_BACKOFF_MS = 3e4;
+var WebSocketAuthError = class extends Error {
+  constructor(reason) {
+    super(`WebSocket auth failed: ${reason}`);
+    this.name = "WebSocketAuthError";
+  }
+};
 function connectWebSocket(options) {
   const { baseUrl, accessToken, onMessage, onClose } = options;
   const url = client.buildWebSocketUrl(baseUrl);
@@ -4013,7 +4019,7 @@ function connectWebSocket(options) {
             resolve({ close: () => ws.close() });
           } else {
             ws.close();
-            reject(new Error(`WebSocket auth failed: ${msg.reason || "unknown"}`));
+            reject(new WebSocketAuthError(msg.reason || "unknown"));
           }
         }
         return;
@@ -4036,52 +4042,66 @@ async function connectWithReconnect(options) {
   const {
     maxRetries = 10,
     initialDelayMs = 1e3,
+    getAccessToken,
+    refreshAuth,
     onReconnecting,
     onReconnected,
     onReconnectFailed,
     onClose,
-    ...connectOpts
+    onMessage,
+    baseUrl,
+    accessToken
   } = options;
+  const resolveToken = getAccessToken ?? (() => accessToken);
   let closed = false;
-  let reconnectTimer = null;
+  let cancelDelay = null;
   let currentConnection = null;
+  const connectOnce = async () => {
+    const token = await resolveToken();
+    if (!token) throw new WebSocketAuthError("no access token available");
+    return connectWebSocket({
+      baseUrl,
+      accessToken: token,
+      onMessage,
+      onClose: (code, reason) => {
+        currentConnection = null;
+        onClose?.(code, reason);
+        if (!closed) void scheduleReconnect();
+      }
+    });
+  };
   const scheduleReconnect = async () => {
     for (let attempt = 1; attempt <= maxRetries; attempt++) {
       if (closed) return;
       const delay = Math.min(initialDelayMs * Math.pow(2, attempt - 1), MAX_BACKOFF_MS);
       onReconnecting?.(attempt, maxRetries);
       await new Promise((resolve) => {
-        reconnectTimer = setTimeout(resolve, delay);
+        const timer = setTimeout(resolve, delay);
+        cancelDelay = () => {
+          clearTimeout(timer);
+          resolve();
+        };
       });
+      cancelDelay = null;
       if (closed) return;
       try {
-        currentConnection = await connectWebSocket({
-          ...connectOpts,
-          onClose: (code, reason) => {
-            currentConnection = null;
-            onClose?.(code, reason);
-            if (!closed) scheduleReconnect();
-          }
-        });
+        currentConnection = await connectOnce();
         onReconnected?.();
         return;
-      } catch {
+      } catch (err) {
+        if (err instanceof WebSocketAuthError && refreshAuth) {
+          await refreshAuth().catch(() => {
+          });
+        }
       }
     }
     if (!closed) onReconnectFailed?.();
   };
-  currentConnection = await connectWebSocket({
-    ...connectOpts,
-    onClose: (code, reason) => {
-      currentConnection = null;
-      onClose?.(code, reason);
-      if (!closed) scheduleReconnect();
-    }
-  });
+  currentConnection = await connectOnce();
   return {
     close: () => {
       closed = true;
-      if (reconnectTimer) clearTimeout(reconnectTimer);
+      cancelDelay?.();
       currentConnection?.close();
       currentConnection = null;
     }
@@ -5630,6 +5650,7 @@ function extractResponseText(response) {
 exports.Arbi = Arbi;
 exports.ArbiApiError = ArbiApiError;
 exports.ArbiError = ArbiError;
+exports.WebSocketAuthError = WebSocketAuthError;
 exports.agentconfig = agentconfig_exports;
 exports.assistant = assistant_exports;
 exports.authenticatedFetch = authenticatedFetch;