@neta-art/cohub 2.14.0 → 2.14.1

package/dist/chunks/websocket.d.ts

@@ -2411,7 +2411,9 @@ type WebsocketClientOptions = {
   pingIntervalMs?: number;
   pongTimeoutMs?: number;
   debug?: boolean;
-  getAccessToken?: () => Promise<string | null> | string | null;
+  getAccessToken?: (options?: {
+    forceRefresh?: boolean;
+  }) => Promise<string | null> | string | null;
   WebSocketImpl?: WebSocketConstructor;
 };
 type WebsocketClientState = "idle" | "connecting" | "reconnecting" | "open" | "closed";
@@ -2496,6 +2498,9 @@ declare class WebsocketClient {
   private reconnectTimer;
   private reconnectTimerResolver;
   private reconnectAttempt;
+  private authReconnectAttempted;
+  private forceRefreshOnNextAuth;
+  private retryAuthClose;
   private manuallyClosed;
   private connectPromise;
   private authWaiter;

package/dist/chunks/websocket.js

@@ -66,12 +66,13 @@ const normalizeOptions = (options = {}) => ({
 	debug: options.debug === true
 });
 const formatCloseMessage = (code, reason) => `WebSocket closed: ${code ?? 0} ${reason || ""}`.trim();
+const AUTH_CLOSE_CODE = 4003;
+const AUTH_CLOSE_REASON = "authentication failed";
 const isRetryableCloseCode = (code) => {
 	if (code === 1e3) return false;
-	if (code === 4003) return false;
+	if (code === AUTH_CLOSE_CODE) return false;
 	return true;
 };
-const AUTH_CLOSE_REASON = "authentication failed";
 const PATCH_STREAM_BUFFER_MAX_PENDING = 128;
 const isRecord = (value) => Boolean(value && typeof value === "object" && !Array.isArray(value));
 const isRealtimeCompactFrame = (value) => {
@@ -144,6 +145,9 @@ var WebsocketClient = class {
 	reconnectTimer = null;
 	reconnectTimerResolver = null;
 	reconnectAttempt = 0;
+	authReconnectAttempted = false;
+	forceRefreshOnNextAuth = false;
+	retryAuthClose = false;
 	manuallyClosed = false;
 	connectPromise = null;
 	authWaiter = null;
@@ -223,12 +227,18 @@ var WebsocketClient = class {
 					resolveOnce();
 				} catch (error) {
 					const authError = error instanceof Error ? error : /* @__PURE__ */ new Error("authentication failed");
+					const recoverable = Boolean(this.getAccessToken && this.autoReconnect && !this.manuallyClosed && !this.authReconnectAttempted);
+					this.retryAuthClose = recoverable;
+					if (recoverable) {
+						this.authReconnectAttempted = true;
+						this.forceRefreshOnNextAuth = true;
+					}
 					this.emit("error", {
 						error: authError,
-						recoverable: false
+						recoverable
 					});
 					rejectOnce(authError);
-					ws.close(4003, AUTH_CLOSE_REASON);
+					ws.close(AUTH_CLOSE_CODE, AUTH_CLOSE_REASON);
 				}
 			};
 			ws.onmessage = (event) => {
@@ -249,7 +259,9 @@ var WebsocketClient = class {
 				this.patchStreamBuffers.clear();
 				const closeError = new Error(formatCloseMessage(event.code, event.reason));
 				this.rejectAuthWaiter(closeError);
-				const willReconnect = !this.manuallyClosed && this.autoReconnect && isRetryableCloseCode(event.code);
+				const retryAuthClose = event.code === AUTH_CLOSE_CODE && this.retryAuthClose;
+				this.retryAuthClose = false;
+				const willReconnect = !this.manuallyClosed && this.autoReconnect && (retryAuthClose || isRetryableCloseCode(event.code));
 				this.log("closed", {
 					code: event.code,
 					reason: event.reason,
@@ -276,6 +288,9 @@ var WebsocketClient = class {
 		this.ws?.close(code, reason);
 		this.ws = null;
 		this.connectPromise = null;
+		this.authReconnectAttempted = false;
+		this.forceRefreshOnNextAuth = false;
+		this.retryAuthClose = false;
 		this.compactStreamContexts.clear();
 		this.patchStreamBuffers.clear();
 		for (const state of this.roomSubscriptions.values()) {
@@ -399,7 +414,9 @@ var WebsocketClient = class {
 		ws.send(JSON.stringify(event));
 	}
 	async authenticate() {
-		const token = this.getAccessToken ? await this.getAccessToken() : null;
+		const forceRefresh = this.forceRefreshOnNextAuth;
+		this.forceRefreshOnNextAuth = false;
+		const token = this.getAccessToken ? await this.getAccessToken(forceRefresh ? { forceRefresh: true } : void 0) : null;
 		if (!token) throw new WebsocketAuthError("missing access token");
 		const waiter = this.createAuthWaiter();
 		this.send({
@@ -411,6 +428,7 @@ var WebsocketClient = class {
 		});
 		await waiter.promise;
 		await this.restoreRoomSubscriptions();
+		this.authReconnectAttempted = false;
 	}
 	flushRoomSubscriptions() {
 		if (this.ws?.readyState !== WebSocket.OPEN) return;
@@ -808,7 +826,7 @@ var WebsocketClient = class {
 		await this.connect().catch((error) => {
 			this.emit("error", {
 				error,
-				recoverable: true
+				recoverable: this.state === "reconnecting"
 			});
 		});
 	}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neta-art/cohub",
-  "version": "2.14.0",
+  "version": "2.14.1",
   "description": "Cohub SDK for spaces, sessions, checkpoints, and realtime agent collaboration.",
   "license": "Apache-2.0",
   "private": false,