@arbidocs/sdk 0.3.109 → 0.3.111

package/dist/index.cjs

@@ -4465,7 +4465,7 @@ var WebSocketAuthError = class extends Error {
   }
 };
 function connectWebSocket(options) {
-  const { baseUrl, accessToken, onMessage, onClose } = options;
+  const { baseUrl, accessToken, onMessage, onClose, onAuthenticated } = options;
   const url = client.buildWebSocketUrl(baseUrl);
   return new Promise((resolve2, reject) => {
     const ws = new WebSocket(url);
@@ -4488,6 +4488,7 @@ function connectWebSocket(options) {
           clearTimeout(timeout);
           if (msg.success) {
             authenticated = true;
+            onAuthenticated?.(msg);
             resolve2({ close: () => ws.close() });
           } else {
             ws.close();
@@ -4521,6 +4522,8 @@ async function connectWithReconnect(options) {
     onReconnectFailed,
     onClose,
     onMessage,
+    onAuthenticated,
+    shouldReconnect,
     baseUrl,
     accessToken
   } = options;
@@ -4535,10 +4538,11 @@ async function connectWithReconnect(options) {
       baseUrl,
       accessToken: token,
       onMessage,
+      onAuthenticated,
       onClose: (code, reason) => {
         currentConnection = null;
         onClose?.(code, reason);
-        if (!closed) void scheduleReconnect();
+        if (!closed && (shouldReconnect?.(code, reason) ?? true)) void scheduleReconnect();
       }
     });
   };
@@ -4569,7 +4573,14 @@ async function connectWithReconnect(options) {
     }
     if (!closed) onReconnectFailed?.();
   };
-  currentConnection = await connectOnce();
+  try {
+    currentConnection = await connectOnce();
+  } catch (err) {
+    if (!(err instanceof WebSocketAuthError) || !refreshAuth) throw err;
+    await refreshAuth().catch(() => {
+    });
+    currentConnection = await connectOnce();
+  }
   return {
     close: () => {
       closed = true;
@@ -5336,6 +5347,7 @@ __export(assistant_exports, {
   buildRetrievalFullContextTool: () => buildRetrievalFullContextTool,
   buildRetrievalTocTool: () => buildRetrievalTocTool,
   filterSkills: () => filterSkills,
+  listCommands: () => listCommands,
   listSkills: () => listSkills,
   parseSlashCommand: () => parseSlashCommand,
   parseSlashTokenInProgress: () => parseSlashTokenInProgress,
@@ -5443,6 +5455,13 @@ async function listSkills(arbi, options) {
   );
   return result.skills;
 }
+async function listCommands(arbi) {
+  const result = requireData(
+    await arbi.fetch.GET("/v1/assistant/commands", {}),
+    "Failed to list commands"
+  );
+  return result.commands;
+}
 function parseSlashCommand(input) {
   const m2 = /^\s*\/([a-zA-Z0-9_-]+)(?:\s+([\s\S]*))?$/.exec(input);
   if (!m2) return null;
@@ -6102,6 +6121,19 @@ var Arbi = class {
   get hasWorkspace() {
     return this.currentWorkspaceId !== null;
   }
+  /**
+   * Live connection info for opening a raw transport (e.g. a WebSocket).
+   *
+   * Reads the access token from the session on each call, so a token refreshed
+   * elsewhere is reflected on the next (re)connect. Returns `null` when there is
+   * no client/token yet (not logged in) rather than throwing, so callers can
+   * poll for readiness instead of guarding against exceptions.
+   */
+  getConnectionInfo() {
+    const accessToken = this.client?.session.getState().accessToken;
+    if (!accessToken) return null;
+    return { baseUrl: this.options.url, accessToken };
+  }
   // ── Auth headers (for raw fetch operations) ───────────────────────────
   getAuthHeaders() {
     const client = this.requireClient();