@arbidocs/sdk 0.3.74 → 0.3.76

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { m as ConfigStore, k as CliConfig, l as CliCredentials, C as ChatSession, X as SkippedFile, a0 as UploadResult, h as AuthHeaders, Y as UploadBatchResult, $ as UploadOptions, Z as UploadDirectOptions, _ as UploadDirectResult, V as SUPPORTED_EXTENSIONS, o as DmCryptoContext } from './browser-D_TPBZXY.cjs';
-export { A as AgentStepEvent, a as Arbi, b as ArbiApiError, c as ArbiError, e as ArbiOptions, f as ArtifactEvent, g as AuthContext, i as AuthenticatedClient, j as CitationSummary, n as ConnectOptions, D as DOC_TERMINAL_STATUSES, p as DocumentListFields, q as DocumentListOrder, r as DocumentWaiter, s as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, t as ListPaginatedOptions, M as MessageLevel, u as MessageMetadataPayload, v as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, w as ReconnectableWsConnection, x as ResolvedCitation, y as ResponseCompletedEvent, z as ResponseContentPartAddedEvent, B as ResponseCreatedEvent, E as ResponseFailedEvent, G as ResponseOutputItemAddedEvent, H as ResponseOutputItemDoneEvent, I as ResponseOutputTextDeltaEvent, J as ResponseOutputTextDoneEvent, K as ResponseUsage, S as SSEEvent, N as SSEStreamCallbacks, T as SSEStreamResult, U as SSEStreamStartData, W as SkillSummary, a1 as UserInfo, a2 as UserInputRequestEvent, a3 as UserMessageEvent, a4 as WorkspaceContext, a5 as WsConnection, a6 as agentconfig, a7 as assistant, a8 as authenticatedFetch, a9 as buildRetrievalChunkTool, aa as buildRetrievalFullContextTool, ab as buildRetrievalTocTool, ac as connectWebSocket, ad as connectWithReconnect, ae as consumeSSEStream, af as contacts, ag as conversations, ah as countCitations, ai as createAuthenticatedClient, aj as createDocumentWaiter, ak as dm, al as doctags, am as documents, an as files, ao as filterSkills, ap as formatAgentStepLabel, aq as formatFileSize, ar as formatStreamSummary, as as formatUserName, at as formatWorkspaceChoices, au as formatWsMessage, av as generateEncryptedWorkspaceKey, aw as generateNewWorkspaceKey, ax as getErrorCode, ay as getErrorMessage, az as getRawWorkspaceKey, aA as health, aB as parseSSEEvents, aC as parseSlashCommand, aD as parseSlashTokenInProgress, aE as performPasswordLogin, aF as performSigningKeyLogin, aG as performSsoDeviceFlowLogin, aH as requireData, aI as requireOk, aJ as resolveAuth, aK as resolveCitations, aL as resolveWorkspace, aM as responses, aN as selectWorkspace, aO as selectWorkspaceById, aP as settings, aQ as streamSSE, aR as stripCitationMarkdown, aS as summarizeCitations, aT as tags, aU as workspaces } from './browser-D_TPBZXY.cjs';
+import { m as ConfigStore, k as CliConfig, l as CliCredentials, C as ChatSession, X as SkippedFile, a0 as UploadResult, h as AuthHeaders, Y as UploadBatchResult, $ as UploadOptions, Z as UploadDirectOptions, _ as UploadDirectResult, V as SUPPORTED_EXTENSIONS, o as DmCryptoContext } from './browser-D06F2w6S.cjs';
+export { A as AgentStepEvent, a as Arbi, b as ArbiApiError, c as ArbiError, e as ArbiOptions, f as ArtifactEvent, g as AuthContext, i as AuthenticatedClient, j as CitationSummary, n as ConnectOptions, D as DOC_TERMINAL_STATUSES, p as DocumentListFields, q as DocumentListOrder, r as DocumentWaiter, s as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, t as ListPaginatedOptions, M as MessageLevel, u as MessageMetadataPayload, v as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, w as ReconnectableWsConnection, x as ResolvedCitation, y as ResponseCompletedEvent, z as ResponseContentPartAddedEvent, B as ResponseCreatedEvent, E as ResponseFailedEvent, G as ResponseOutputItemAddedEvent, H as ResponseOutputItemDoneEvent, I as ResponseOutputTextDeltaEvent, J as ResponseOutputTextDoneEvent, K as ResponseUsage, S as SSEEvent, N as SSEStreamCallbacks, T as SSEStreamResult, U as SSEStreamStartData, W as SkillSummary, a1 as UserInfo, a2 as UserInputRequestEvent, a3 as UserMessageEvent, a4 as WebSocketAuthError, a5 as WorkspaceContext, a6 as WsConnection, a7 as agentconfig, a8 as assistant, a9 as authenticatedFetch, aa as buildRetrievalChunkTool, ab as buildRetrievalFullContextTool, ac as buildRetrievalTocTool, ad as connectWebSocket, ae as connectWithReconnect, af as consumeSSEStream, ag as contacts, ah as conversations, ai as countCitations, aj as createAuthenticatedClient, ak as createDocumentWaiter, al as dm, am as doctags, an as documents, ao as files, ap as filterSkills, aq as formatAgentStepLabel, ar as formatFileSize, as as formatStreamSummary, at as formatUserName, au as formatWorkspaceChoices, av as formatWsMessage, aw as generateEncryptedWorkspaceKey, ax as generateNewWorkspaceKey, ay as getErrorCode, az as getErrorMessage, aA as getRawWorkspaceKey, aB as health, aC as parseSSEEvents, aD as parseSlashCommand, aE as parseSlashTokenInProgress, aF as performPasswordLogin, aG as performSigningKeyLogin, aH as performSsoDeviceFlowLogin, aI as requireData, aJ as requireOk, aK as resolveAuth, aL as resolveCitations, aM as resolveWorkspace, aN as responses, aO as selectWorkspace, aP as selectWorkspaceById, aQ as settings, aR as streamSSE, aS as stripCitationMarkdown, aT as summarizeCitations, aU as tags, aV as workspaces } from './browser-D06F2w6S.cjs';
 import { SessionStorageProvider, ArbiClient, components } from '@arbidocs/client';
 
 /**
@@ -634,10 +634,9 @@ declare class OpenClawOrchestrator implements Orchestrator {
  */
 
 interface DmListenerOptions {
-    /** Authenticated ArbiClient instance */
+    /** Authenticated ArbiClient instance. The WebSocket token and re-login are
+     *  both derived from this — it must already hold a valid session. */
     arbi: ArbiClient;
-    /** Access token for WebSocket auth */
-    accessToken: string;
     /** Base URL of the ARBI server */
     baseUrl: string;
     /** DM crypto context for encrypt/decrypt */

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { m as ConfigStore, k as CliConfig, l as CliCredentials, C as ChatSession, X as SkippedFile, a0 as UploadResult, h as AuthHeaders, Y as UploadBatchResult, $ as UploadOptions, Z as UploadDirectOptions, _ as UploadDirectResult, V as SUPPORTED_EXTENSIONS, o as DmCryptoContext } from './browser-D_TPBZXY.js';
-export { A as AgentStepEvent, a as Arbi, b as ArbiApiError, c as ArbiError, e as ArbiOptions, f as ArtifactEvent, g as AuthContext, i as AuthenticatedClient, j as CitationSummary, n as ConnectOptions, D as DOC_TERMINAL_STATUSES, p as DocumentListFields, q as DocumentListOrder, r as DocumentWaiter, s as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, t as ListPaginatedOptions, M as MessageLevel, u as MessageMetadataPayload, v as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, w as ReconnectableWsConnection, x as ResolvedCitation, y as ResponseCompletedEvent, z as ResponseContentPartAddedEvent, B as ResponseCreatedEvent, E as ResponseFailedEvent, G as ResponseOutputItemAddedEvent, H as ResponseOutputItemDoneEvent, I as ResponseOutputTextDeltaEvent, J as ResponseOutputTextDoneEvent, K as ResponseUsage, S as SSEEvent, N as SSEStreamCallbacks, T as SSEStreamResult, U as SSEStreamStartData, W as SkillSummary, a1 as UserInfo, a2 as UserInputRequestEvent, a3 as UserMessageEvent, a4 as WorkspaceContext, a5 as WsConnection, a6 as agentconfig, a7 as assistant, a8 as authenticatedFetch, a9 as buildRetrievalChunkTool, aa as buildRetrievalFullContextTool, ab as buildRetrievalTocTool, ac as connectWebSocket, ad as connectWithReconnect, ae as consumeSSEStream, af as contacts, ag as conversations, ah as countCitations, ai as createAuthenticatedClient, aj as createDocumentWaiter, ak as dm, al as doctags, am as documents, an as files, ao as filterSkills, ap as formatAgentStepLabel, aq as formatFileSize, ar as formatStreamSummary, as as formatUserName, at as formatWorkspaceChoices, au as formatWsMessage, av as generateEncryptedWorkspaceKey, aw as generateNewWorkspaceKey, ax as getErrorCode, ay as getErrorMessage, az as getRawWorkspaceKey, aA as health, aB as parseSSEEvents, aC as parseSlashCommand, aD as parseSlashTokenInProgress, aE as performPasswordLogin, aF as performSigningKeyLogin, aG as performSsoDeviceFlowLogin, aH as requireData, aI as requireOk, aJ as resolveAuth, aK as resolveCitations, aL as resolveWorkspace, aM as responses, aN as selectWorkspace, aO as selectWorkspaceById, aP as settings, aQ as streamSSE, aR as stripCitationMarkdown, aS as summarizeCitations, aT as tags, aU as workspaces } from './browser-D_TPBZXY.js';
+import { m as ConfigStore, k as CliConfig, l as CliCredentials, C as ChatSession, X as SkippedFile, a0 as UploadResult, h as AuthHeaders, Y as UploadBatchResult, $ as UploadOptions, Z as UploadDirectOptions, _ as UploadDirectResult, V as SUPPORTED_EXTENSIONS, o as DmCryptoContext } from './browser-D06F2w6S.js';
+export { A as AgentStepEvent, a as Arbi, b as ArbiApiError, c as ArbiError, e as ArbiOptions, f as ArtifactEvent, g as AuthContext, i as AuthenticatedClient, j as CitationSummary, n as ConnectOptions, D as DOC_TERMINAL_STATUSES, p as DocumentListFields, q as DocumentListOrder, r as DocumentWaiter, s as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, t as ListPaginatedOptions, M as MessageLevel, u as MessageMetadataPayload, v as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, w as ReconnectableWsConnection, x as ResolvedCitation, y as ResponseCompletedEvent, z as ResponseContentPartAddedEvent, B as ResponseCreatedEvent, E as ResponseFailedEvent, G as ResponseOutputItemAddedEvent, H as ResponseOutputItemDoneEvent, I as ResponseOutputTextDeltaEvent, J as ResponseOutputTextDoneEvent, K as ResponseUsage, S as SSEEvent, N as SSEStreamCallbacks, T as SSEStreamResult, U as SSEStreamStartData, W as SkillSummary, a1 as UserInfo, a2 as UserInputRequestEvent, a3 as UserMessageEvent, a4 as WebSocketAuthError, a5 as WorkspaceContext, a6 as WsConnection, a7 as agentconfig, a8 as assistant, a9 as authenticatedFetch, aa as buildRetrievalChunkTool, ab as buildRetrievalFullContextTool, ac as buildRetrievalTocTool, ad as connectWebSocket, ae as connectWithReconnect, af as consumeSSEStream, ag as contacts, ah as conversations, ai as countCitations, aj as createAuthenticatedClient, ak as createDocumentWaiter, al as dm, am as doctags, an as documents, ao as files, ap as filterSkills, aq as formatAgentStepLabel, ar as formatFileSize, as as formatStreamSummary, at as formatUserName, au as formatWorkspaceChoices, av as formatWsMessage, aw as generateEncryptedWorkspaceKey, ax as generateNewWorkspaceKey, ay as getErrorCode, az as getErrorMessage, aA as getRawWorkspaceKey, aB as health, aC as parseSSEEvents, aD as parseSlashCommand, aE as parseSlashTokenInProgress, aF as performPasswordLogin, aG as performSigningKeyLogin, aH as performSsoDeviceFlowLogin, aI as requireData, aJ as requireOk, aK as resolveAuth, aL as resolveCitations, aM as resolveWorkspace, aN as responses, aO as selectWorkspace, aP as selectWorkspaceById, aQ as settings, aR as streamSSE, aS as stripCitationMarkdown, aT as summarizeCitations, aU as tags, aV as workspaces } from './browser-D06F2w6S.js';
 import { SessionStorageProvider, ArbiClient, components } from '@arbidocs/client';
 
 /**
@@ -634,10 +634,9 @@ declare class OpenClawOrchestrator implements Orchestrator {
  */
 
 interface DmListenerOptions {
-    /** Authenticated ArbiClient instance */
+    /** Authenticated ArbiClient instance. The WebSocket token and re-login are
+     *  both derived from this — it must already hold a valid session. */
     arbi: ArbiClient;
-    /** Access token for WebSocket auth */
-    accessToken: string;
     /** Base URL of the ARBI server */
     baseUrl: string;
     /** DM crypto context for encrypt/decrypt */

package/dist/index.js

@@ -4399,6 +4399,12 @@ function formatStreamSummary(result, elapsedTime) {
 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 = buildWebSocketUrl(baseUrl);
@@ -4426,7 +4432,7 @@ function connectWebSocket(options) {
             resolve2({ close: () => ws.close() });
           } else {
             ws.close();
-            reject(new Error(`WebSocket auth failed: ${msg.reason || "unknown"}`));
+            reject(new WebSocketAuthError(msg.reason || "unknown"));
           }
         }
         return;
@@ -4449,52 +4455,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((resolve2) => {
-        reconnectTimer = setTimeout(resolve2, delay);
+        const timer = setTimeout(resolve2, delay);
+        cancelDelay = () => {
+          clearTimeout(timer);
+          resolve2();
+        };
       });
+      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;
     }
@@ -7114,10 +7134,14 @@ var OpenClawOrchestrator = class {
 
 // src/listen/dm-listener.ts
 async function startDmListener(options) {
-  const { arbi, accessToken, baseUrl, crypto, orchestrator, parentExtId, onLog, onError } = options;
+  const { arbi, baseUrl, crypto, orchestrator, parentExtId, onLog, onError } = options;
   const log = onLog ?? (() => {
   });
   const logError = onError ?? ((msg) => console.error(msg));
+  const initialToken = arbi.session.getState().accessToken;
+  if (!initialToken) {
+    throw new Error("No access token \u2014 run `arbi login` first.");
+  }
   const handleNotification = async (msg) => {
     if (msg.type !== "user_message") return;
     const senderExtId = msg.sender?.external_id;
@@ -7187,7 +7211,10 @@ async function startDmListener(options) {
   }
   const ws = await connectWithReconnect({
     baseUrl,
-    accessToken,
+    accessToken: initialToken,
+    getAccessToken: () => arbi.session.getState().accessToken,
+    refreshAuth: () => arbi.auth.relogin(),
+    maxRetries: Infinity,
     onMessage: (msg) => {
       if ("sender" in msg && "recipient" in msg) {
         const notification = msg;
@@ -7199,14 +7226,12 @@ async function startDmListener(options) {
       }
     },
     onReconnecting: (attempt, max) => {
-      log(`WebSocket reconnecting (${attempt}/${max})...`);
+      const limit = Number.isFinite(max) ? `/${max}` : "";
+      log(`WebSocket reconnecting (attempt ${attempt}${limit})...`);
     },
     onReconnected: () => {
       log("WebSocket reconnected");
     },
-    onReconnectFailed: () => {
-      logError("WebSocket reconnection failed \u2014 listener stopped");
-    },
     onClose: (_code, reason) => {
       log(`WebSocket closed: ${reason || "no reason"}`);
     }
@@ -7220,6 +7245,6 @@ async function startDmListener(options) {
   };
 }
 
-export { Arbi, ArbiApiError, ArbiError, ClaudeOrchestrator, DOC_TERMINAL_STATUSES, FileConfigStore, OpenClawOrchestrator, agentconfig_exports as agentconfig, agents_exports as agents, assistant_exports as assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts_exports as contacts, conversations_exports as conversations, countCitations, createAuthenticatedClient, createDocumentWaiter, device_flow_exports as deviceFlow, dm_exports as dm, doctags_exports as doctags, documents_exports as documents, documents_node_exports as documentsNode, files_exports as files, filterSkills, formatAgentStepLabel, formatFileSize, formatStreamSummary, formatUserName, formatWorkspaceChoices, formatWsMessage, generateEncryptedWorkspaceKey, generateNewWorkspaceKey, getErrorCode, getErrorMessage, getRawWorkspaceKey, health_exports as health, listen_exports as listen, parseSSEEvents, parseSlashCommand, parseSlashTokenInProgress, performPasswordLogin, performSigningKeyLogin, performSsoDeviceFlowLogin, projects_exports as projects, requireData, requireOk, resolveAuth, resolveCitations, resolveWorkspace, responses_exports as responses, selectWorkspace, selectWorkspaceById, sessions_exports as sessions, settings_exports as settings, startDmListener, streamSSE, stripCitationMarkdown, summarizeCitations, tags_exports as tags, workspaces_exports as workspaces };
+export { Arbi, ArbiApiError, ArbiError, ClaudeOrchestrator, DOC_TERMINAL_STATUSES, FileConfigStore, OpenClawOrchestrator, WebSocketAuthError, agentconfig_exports as agentconfig, agents_exports as agents, assistant_exports as assistant, authenticatedFetch, buildRetrievalChunkTool, buildRetrievalFullContextTool, buildRetrievalTocTool, connectWebSocket, connectWithReconnect, consumeSSEStream, contacts_exports as contacts, conversations_exports as conversations, countCitations, createAuthenticatedClient, createDocumentWaiter, device_flow_exports as deviceFlow, dm_exports as dm, doctags_exports as doctags, documents_exports as documents, documents_node_exports as documentsNode, files_exports as files, filterSkills, formatAgentStepLabel, formatFileSize, formatStreamSummary, formatUserName, formatWorkspaceChoices, formatWsMessage, generateEncryptedWorkspaceKey, generateNewWorkspaceKey, getErrorCode, getErrorMessage, getRawWorkspaceKey, health_exports as health, listen_exports as listen, parseSSEEvents, parseSlashCommand, parseSlashTokenInProgress, performPasswordLogin, performSigningKeyLogin, performSsoDeviceFlowLogin, projects_exports as projects, requireData, requireOk, resolveAuth, resolveCitations, resolveWorkspace, responses_exports as responses, selectWorkspace, selectWorkspaceById, sessions_exports as sessions, settings_exports as settings, startDmListener, streamSSE, stripCitationMarkdown, summarizeCitations, tags_exports as tags, workspaces_exports as workspaces };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map