@langchain/langgraph-sdk 1.9.20 → 1.9.22

package/dist/client/stream/transport/websocket.cjs

@@ -1,10 +1,29 @@
+const require_error = require("../error.cjs");
 const require_queue = require("./queue.cjs");
 const require_utils = require("./utils.cjs");
 //#region src/client/stream/transport/websocket.ts
+const WEB_SOCKET_CONNECTING = 0;
+const WEB_SOCKET_OPEN = 1;
+const WEB_SOCKET_CLOSED = 3;
+/**
+* Exponential backoff with jitter for WebSocket reconnect. Mirrors
+* {@link streamWithRetry} in `utils/stream.ts` (capped at 5s + 1s jitter).
+*/
+function webSocketReconnectDelayMs(attempt) {
+	return Math.min(1e3 * 2 ** (attempt - 1), 5e3) + Math.random() * 1e3;
+}
 /**
 * Transport adapter that speaks the thread-centric protocol over a
-* bidirectional WebSocket. Bound to a specific `threadId` — the socket
-* connects to `ws://.../threads/:thread_id/stream/events`.
+* bidirectional WebSocket. Bound to a `threadId` at construction or later
+* via {@link setThreadId} — the socket connects to
+* `ws://.../threads/:thread_id/stream/events`.
+*
+* On unexpected disconnect the adapter reconnects with exponential
+* backoff (see {@link ProtocolWebSocketTransportOptions.maxReconnectAttempts}).
+* The server replays buffered events on the new socket; the SDK
+* deduplicates by `event_id`. {@link ProtocolWebSocketTransportOptions.onReconnected}
+* runs after each successful reconnect so `ThreadStream` can re-issue
+* `subscription.subscribe` commands.
 */
 var ProtocolWebSocketTransportAdapter = class {
 	threadId;
@@ -13,30 +32,63 @@ var ProtocolWebSocketTransportAdapter = class {
 	defaultHeaders;
 	onRequest;
 	webSocketFactory;
-	streamUrl;
+	paths;
+	maxReconnectAttempts;
+	onReconnect;
+	reconnectDelayMs;
+	onReconnected;
 	pending = /* @__PURE__ */ new Map();
 	socket = null;
 	closed = false;
 	intentionalClose = false;
+	reconnectInFlight = null;
 	constructor(options) {
 		this.apiUrl = options.apiUrl;
-		this.threadId = options.threadId;
+		this.threadId = options.threadId ?? "";
 		this.defaultHeaders = options.defaultHeaders;
 		this.onRequest = options.onRequest;
 		this.webSocketFactory = options.webSocketFactory ?? ((url) => new WebSocket(url));
-		this.streamUrl = options.paths?.stream ?? `/threads/${this.threadId}/stream/events`;
+		this.paths = options.paths;
+		this.maxReconnectAttempts = options.maxReconnectAttempts ?? 5;
+		this.onReconnect = options.onReconnect;
+		this.onReconnected = options.onReconnected;
+		this.reconnectDelayMs = options.reconnectDelayMs ?? webSocketReconnectDelayMs;
+	}
+	/** {@inheritDoc TransportAdapter.setThreadId} */
+	setThreadId(threadId) {
+		if (threadId === this.threadId) return;
+		if (this.reconnectInFlight != null || this.socket != null && this.socket.readyState !== WEB_SOCKET_CLOSED) throw new Error("Protocol WebSocket transport cannot be rebound to a different thread while the socket is open. Close the current stream and create a new WebSocket transport for the new thread.");
+		this.threadId = threadId;
+	}
+	/**
+	* Socket URL derives from the currently-bound thread so a single adapter
+	* can follow {@link setThreadId} re-binds; the next {@link open} connects
+	* to the new thread. A fixed `paths.stream` string overrides the default.
+	*/
+	get streamUrl() {
+		return require_utils.resolveProtocolPath(this.paths?.stream, this.threadId, (id) => `/threads/${id}/stream/events`);
+	}
+	/**
+	* Register a callback invoked after each successful reconnect. Used
+	* by {@link ThreadStream} to re-send active `subscription.subscribe`
+	* commands.
+	*/
+	setOnReconnected(handler) {
+		this.onReconnected = handler;
 	}
 	async open() {
-		if (this.socket != null) return;
+		if (this.closed) throw new Error("Protocol WebSocket transport is closed.");
+		if (this.socket?.readyState === WEB_SOCKET_OPEN) return;
+		if (this.socket != null) {
+			this.#detachSocket(this.socket);
+			this.socket = null;
+		}
 		this.assertBrowserSafeTransportConfig();
 		const wsUrl = require_utils.toWebSocketUrl(require_utils.toAbsoluteUrl(this.apiUrl, this.streamUrl).toString());
 		const socket = this.webSocketFactory(wsUrl);
 		this.socket = socket;
-		this.closed = false;
 		this.intentionalClose = false;
-		socket.addEventListener("message", this.handleMessage);
-		socket.addEventListener("close", this.handleClose);
-		socket.addEventListener("error", this.handleSocketError);
+		this.#attachSocket(socket);
 		await new Promise((resolve, reject) => {
 			const onOpen = () => {
 				cleanup();
@@ -80,8 +132,9 @@ var ProtocolWebSocketTransportAdapter = class {
 		const socket = this.socket;
 		this.socket = null;
 		if (!socket) return;
+		this.#detachSocket(socket);
 		await new Promise((resolve) => {
-			if (socket.readyState === WebSocket.CLOSED) {
+			if (socket.readyState === WEB_SOCKET_CLOSED) {
 				resolve();
 				return;
 			}
@@ -90,7 +143,7 @@ var ProtocolWebSocketTransportAdapter = class {
 				resolve();
 			};
 			socket.addEventListener("close", onClose, { once: true });
-			if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) socket.close();
+			if (socket.readyState === WEB_SOCKET_OPEN || socket.readyState === WEB_SOCKET_CONNECTING) socket.close();
 			else resolve();
 		});
 	}
@@ -98,8 +151,12 @@ var ProtocolWebSocketTransportAdapter = class {
 		if (require_utils.hasHeaders(this.defaultHeaders) || this.onRequest != null) throw new Error("Browser WebSocket protocol transport does not support defaultHeaders or onRequest hooks. Supply a custom protocolWebSocketFactory if you need custom WebSocket setup.");
 	}
 	async sendCommand(command) {
-		const socket = this.socket;
-		if (socket == null || socket.readyState !== WebSocket.OPEN) throw new Error("Protocol WebSocket is not open.");
+		let socket = this.socket;
+		if (this.reconnectInFlight != null && (socket == null || socket.readyState !== WEB_SOCKET_OPEN)) {
+			await this.reconnectInFlight.catch(() => void 0);
+			socket = this.socket;
+		}
+		if (socket == null || socket.readyState !== WEB_SOCKET_OPEN) throw new Error("Protocol WebSocket is not open.");
 		return await new Promise((resolve, reject) => {
 			this.pending.set(command.id, {
 				resolve,
@@ -113,6 +170,16 @@ var ProtocolWebSocketTransportAdapter = class {
 			}
 		});
 	}
+	#attachSocket(socket) {
+		socket.addEventListener("message", this.handleMessage);
+		socket.addEventListener("close", this.handleClose);
+		socket.addEventListener("error", this.handleSocketError);
+	}
+	#detachSocket(socket) {
+		socket.removeEventListener("message", this.handleMessage);
+		socket.removeEventListener("close", this.handleClose);
+		socket.removeEventListener("error", this.handleSocketError);
+	}
 	handleMessage = (event) => {
 		let payload;
 		try {
@@ -131,25 +198,62 @@ var ProtocolWebSocketTransportAdapter = class {
 		if (require_utils.isRecord(payload) && payload.type === "event") this.queue.push(payload);
 	};
 	handleClose = () => {
+		const socket = this.socket;
+		if (socket != null) this.#detachSocket(socket);
 		this.socket = null;
 		if (this.intentionalClose || this.closed) {
 			this.queue.close();
 			return;
 		}
-		const error = /* @__PURE__ */ new Error("Protocol WebSocket closed unexpectedly.");
-		for (const { reject } of this.pending.values()) reject(error);
-		this.pending.clear();
-		this.queue.close(error);
+		this.#handleUnexpectedDisconnect(/* @__PURE__ */ new Error("Protocol WebSocket closed unexpectedly."));
 	};
 	handleSocketError = () => {
 		if (this.closed || this.intentionalClose) return;
-		const error = /* @__PURE__ */ new Error("Protocol WebSocket encountered an error.");
+		this.#handleUnexpectedDisconnect(/* @__PURE__ */ new Error("Protocol WebSocket encountered an error."));
+	};
+	#handleUnexpectedDisconnect(cause) {
+		const error = require_utils.toError(cause);
 		for (const { reject } of this.pending.values()) reject(error);
 		this.pending.clear();
-		this.queue.close(error);
-	};
+		if (this.maxReconnectAttempts <= 0) {
+			this.queue.close(error);
+			return;
+		}
+		this.#scheduleReconnect(cause);
+	}
+	#scheduleReconnect(cause) {
+		if (this.closed || this.intentionalClose) return;
+		if (this.reconnectInFlight != null) return;
+		this.reconnectInFlight = this.#runReconnectLoop(cause).finally(() => {
+			this.reconnectInFlight = null;
+		});
+	}
+	async #runReconnectLoop(initialCause) {
+		let lastError = initialCause;
+		for (let attempt = 1; attempt <= this.maxReconnectAttempts; attempt += 1) {
+			if (this.closed || this.intentionalClose) return;
+			this.onReconnect?.({
+				attempt,
+				cause: lastError
+			});
+			const delay = this.reconnectDelayMs(attempt);
+			if (delay > 0) await new Promise((resolve) => {
+				setTimeout(resolve, delay);
+			});
+			if (this.closed || this.intentionalClose) return;
+			try {
+				await this.open();
+				if (this.onReconnected) await this.onReconnected();
+				return;
+			} catch (error) {
+				lastError = error;
+			}
+		}
+		this.queue.close(new require_error.MaxWebSocketReconnectAttemptsError(this.maxReconnectAttempts, lastError));
+	}
 };
 //#endregion
 exports.ProtocolWebSocketTransportAdapter = ProtocolWebSocketTransportAdapter;
+exports.webSocketReconnectDelayMs = webSocketReconnectDelayMs;
 
 //# sourceMappingURL=websocket.cjs.map

package/dist/client/stream/transport/websocket.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"websocket.cjs","names":["AsyncQueue","toWebSocketUrl","toAbsoluteUrl","hasHeaders","toError","isRecord"],"sources":["../../../../src/client/stream/transport/websocket.ts"],"sourcesContent":["import { AsyncQueue } from \"./queue.js\";\nimport type {\n  Message,\n  Command,\n  CommandResponse,\n  ErrorResponse,\n} from \"@langchain/protocol\";\n\nimport {\n  toAbsoluteUrl,\n  toWebSocketUrl,\n  isRecord,\n  hasHeaders,\n  toError,\n} from \"./utils.js\";\nimport type {\n  HeaderValue,\n  ProtocolRequestHook,\n  PendingResponse,\n  ProtocolWebSocketTransportOptions,\n} from \"./types.js\";\nimport type { TransportAdapter } from \"../transport.js\";\n\n/**\n * Transport adapter that speaks the thread-centric protocol over a\n * bidirectional WebSocket. Bound to a specific `threadId` — the socket\n * connects to `ws://.../threads/:thread_id/stream/events`.\n */\nexport class ProtocolWebSocketTransportAdapter implements TransportAdapter {\n  readonly threadId: string;\n\n  private readonly queue = new AsyncQueue<Message>();\n\n  private readonly apiUrl: string;\n\n  private readonly defaultHeaders?: Record<string, HeaderValue>;\n\n  private readonly onRequest?: ProtocolRequestHook;\n\n  private readonly webSocketFactory: (url: string) => WebSocket;\n\n  private readonly streamUrl: string;\n\n  private readonly pending = new Map<number, PendingResponse>();\n\n  private socket: WebSocket | null = null;\n\n  private closed = false;\n\n  private intentionalClose = false;\n\n  constructor(options: ProtocolWebSocketTransportOptions) {\n    this.apiUrl = options.apiUrl;\n    this.threadId = options.threadId;\n    this.defaultHeaders = options.defaultHeaders;\n    this.onRequest = options.onRequest;\n    this.webSocketFactory =\n      options.webSocketFactory ?? ((url) => new WebSocket(url));\n    this.streamUrl =\n      options.paths?.stream ?? `/threads/${this.threadId}/stream/events`;\n  }\n\n  async open(): Promise<void> {\n    if (this.socket != null) return;\n    this.assertBrowserSafeTransportConfig();\n\n    const wsUrl = toWebSocketUrl(\n      toAbsoluteUrl(this.apiUrl, this.streamUrl).toString()\n    );\n    const socket = this.webSocketFactory(wsUrl);\n    this.socket = socket;\n    this.closed = false;\n    this.intentionalClose = false;\n\n    socket.addEventListener(\"message\", this.handleMessage);\n    socket.addEventListener(\"close\", this.handleClose);\n    socket.addEventListener(\"error\", this.handleSocketError);\n\n    await new Promise<void>((resolve, reject) => {\n      const onOpen = () => {\n        cleanup();\n        resolve();\n      };\n      const onError = () => {\n        cleanup();\n        reject(new Error(\"Failed to open protocol WebSocket.\"));\n      };\n      const cleanup = () => {\n        socket.removeEventListener(\"open\", onOpen);\n        socket.removeEventListener(\"error\", onError);\n      };\n      socket.addEventListener(\"open\", onOpen, { once: true });\n      socket.addEventListener(\"error\", onError, { once: true });\n    });\n  }\n\n  async send(\n    command: Command\n  ): Promise<CommandResponse | ErrorResponse | void> {\n    return await this.sendCommand(command);\n  }\n\n  events(): AsyncIterable<Message> {\n    const queue = this.queue;\n    return {\n      [Symbol.asyncIterator]: () => ({\n        next: async () => await queue.shift(),\n        return: async () => {\n          queue.close();\n          return { done: true, value: undefined };\n        },\n      }),\n    };\n  }\n\n  async close(): Promise<void> {\n    if (this.closed) {\n      return;\n    }\n\n    this.closed = true;\n    this.intentionalClose = true;\n\n    for (const { reject } of this.pending.values()) {\n      reject(new Error(\"Protocol WebSocket connection closed.\"));\n    }\n    this.pending.clear();\n    this.queue.close();\n\n    const socket = this.socket;\n    this.socket = null;\n    if (!socket) {\n      return;\n    }\n\n    await new Promise<void>((resolve) => {\n      if (socket.readyState === WebSocket.CLOSED) {\n        resolve();\n        return;\n      }\n\n      const onClose = () => {\n        socket.removeEventListener(\"close\", onClose);\n        resolve();\n      };\n\n      socket.addEventListener(\"close\", onClose, { once: true });\n      if (\n        socket.readyState === WebSocket.OPEN ||\n        socket.readyState === WebSocket.CONNECTING\n      ) {\n        socket.close();\n      } else {\n        resolve();\n      }\n    });\n  }\n\n  private assertBrowserSafeTransportConfig(): void {\n    if (hasHeaders(this.defaultHeaders) || this.onRequest != null) {\n      throw new Error(\n        \"Browser WebSocket protocol transport does not support defaultHeaders or onRequest hooks. Supply a custom protocolWebSocketFactory if you need custom WebSocket setup.\"\n      );\n    }\n  }\n\n  private async sendCommand(\n    command: Command\n  ): Promise<CommandResponse | ErrorResponse> {\n    const socket = this.socket;\n    if (socket == null || socket.readyState !== WebSocket.OPEN) {\n      throw new Error(\"Protocol WebSocket is not open.\");\n    }\n\n    return await new Promise<CommandResponse | ErrorResponse>(\n      (resolve, reject) => {\n        this.pending.set(command.id, { resolve, reject });\n\n        try {\n          socket.send(JSON.stringify(command));\n        } catch (error) {\n          this.pending.delete(command.id);\n          reject(toError(error));\n        }\n      }\n    );\n  }\n\n  private readonly handleMessage = (event: MessageEvent): void => {\n    let payload: unknown;\n    try {\n      payload = JSON.parse(String(event.data));\n    } catch {\n      return;\n    }\n\n    if (\n      isRecord(payload) &&\n      typeof payload.id === \"number\" &&\n      (payload.type === \"success\" || payload.type === \"error\")\n    ) {\n      const pending = this.pending.get(payload.id);\n      if (pending) {\n        this.pending.delete(payload.id);\n        pending.resolve(payload as CommandResponse | ErrorResponse);\n      }\n      return;\n    }\n\n    if (isRecord(payload) && payload.type === \"event\") {\n      this.queue.push(payload as Message);\n    }\n  };\n\n  private readonly handleClose = (): void => {\n    this.socket = null;\n\n    if (this.intentionalClose || this.closed) {\n      this.queue.close();\n      return;\n    }\n\n    const error = new Error(\"Protocol WebSocket closed unexpectedly.\");\n    for (const { reject } of this.pending.values()) {\n      reject(error);\n    }\n    this.pending.clear();\n    this.queue.close(error);\n  };\n\n  private readonly handleSocketError = (): void => {\n    if (this.closed || this.intentionalClose) {\n      return;\n    }\n\n    const error = new Error(\"Protocol WebSocket encountered an error.\");\n    for (const { reject } of this.pending.values()) {\n      reject(error);\n    }\n    this.pending.clear();\n    this.queue.close(error);\n  };\n}\n"],"mappings":";;;;;;;;AA4BA,IAAa,oCAAb,MAA2E;CACzE;CAEA,QAAyB,IAAIA,cAAAA,YAAqB;CAElD;CAEA;CAEA;CAEA;CAEA;CAEA,0BAA2B,IAAI,KAA8B;CAE7D,SAAmC;CAEnC,SAAiB;CAEjB,mBAA2B;CAE3B,YAAY,SAA4C;AACtD,OAAK,SAAS,QAAQ;AACtB,OAAK,WAAW,QAAQ;AACxB,OAAK,iBAAiB,QAAQ;AAC9B,OAAK,YAAY,QAAQ;AACzB,OAAK,mBACH,QAAQ,sBAAsB,QAAQ,IAAI,UAAU,IAAI;AAC1D,OAAK,YACH,QAAQ,OAAO,UAAU,YAAY,KAAK,SAAS;;CAGvD,MAAM,OAAsB;AAC1B,MAAI,KAAK,UAAU,KAAM;AACzB,OAAK,kCAAkC;EAEvC,MAAM,QAAQC,cAAAA,eACZC,cAAAA,cAAc,KAAK,QAAQ,KAAK,UAAU,CAAC,UAAU,CACtD;EACD,MAAM,SAAS,KAAK,iBAAiB,MAAM;AAC3C,OAAK,SAAS;AACd,OAAK,SAAS;AACd,OAAK,mBAAmB;AAExB,SAAO,iBAAiB,WAAW,KAAK,cAAc;AACtD,SAAO,iBAAiB,SAAS,KAAK,YAAY;AAClD,SAAO,iBAAiB,SAAS,KAAK,kBAAkB;AAExD,QAAM,IAAI,SAAe,SAAS,WAAW;GAC3C,MAAM,eAAe;AACnB,aAAS;AACT,aAAS;;GAEX,MAAM,gBAAgB;AACpB,aAAS;AACT,2BAAO,IAAI,MAAM,qCAAqC,CAAC;;GAEzD,MAAM,gBAAgB;AACpB,WAAO,oBAAoB,QAAQ,OAAO;AAC1C,WAAO,oBAAoB,SAAS,QAAQ;;AAE9C,UAAO,iBAAiB,QAAQ,QAAQ,EAAE,MAAM,MAAM,CAAC;AACvD,UAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;IACzD;;CAGJ,MAAM,KACJ,SACiD;AACjD,SAAO,MAAM,KAAK,YAAY,QAAQ;;CAGxC,SAAiC;EAC/B,MAAM,QAAQ,KAAK;AACnB,SAAO,GACJ,OAAO,uBAAuB;GAC7B,MAAM,YAAY,MAAM,MAAM,OAAO;GACrC,QAAQ,YAAY;AAClB,UAAM,OAAO;AACb,WAAO;KAAE,MAAM;KAAM,OAAO,KAAA;KAAW;;GAE1C,GACF;;CAGH,MAAM,QAAuB;AAC3B,MAAI,KAAK,OACP;AAGF,OAAK,SAAS;AACd,OAAK,mBAAmB;AAExB,OAAK,MAAM,EAAE,YAAY,KAAK,QAAQ,QAAQ,CAC5C,wBAAO,IAAI,MAAM,wCAAwC,CAAC;AAE5D,OAAK,QAAQ,OAAO;AACpB,OAAK,MAAM,OAAO;EAElB,MAAM,SAAS,KAAK;AACpB,OAAK,SAAS;AACd,MAAI,CAAC,OACH;AAGF,QAAM,IAAI,SAAe,YAAY;AACnC,OAAI,OAAO,eAAe,UAAU,QAAQ;AAC1C,aAAS;AACT;;GAGF,MAAM,gBAAgB;AACpB,WAAO,oBAAoB,SAAS,QAAQ;AAC5C,aAAS;;AAGX,UAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AACzD,OACE,OAAO,eAAe,UAAU,QAChC,OAAO,eAAe,UAAU,WAEhC,QAAO,OAAO;OAEd,UAAS;IAEX;;CAGJ,mCAAiD;AAC/C,MAAIC,cAAAA,WAAW,KAAK,eAAe,IAAI,KAAK,aAAa,KACvD,OAAM,IAAI,MACR,wKACD;;CAIL,MAAc,YACZ,SAC0C;EAC1C,MAAM,SAAS,KAAK;AACpB,MAAI,UAAU,QAAQ,OAAO,eAAe,UAAU,KACpD,OAAM,IAAI,MAAM,kCAAkC;AAGpD,SAAO,MAAM,IAAI,SACd,SAAS,WAAW;AACnB,QAAK,QAAQ,IAAI,QAAQ,IAAI;IAAE;IAAS;IAAQ,CAAC;AAEjD,OAAI;AACF,WAAO,KAAK,KAAK,UAAU,QAAQ,CAAC;YAC7B,OAAO;AACd,SAAK,QAAQ,OAAO,QAAQ,GAAG;AAC/B,WAAOC,cAAAA,QAAQ,MAAM,CAAC;;IAG3B;;CAGH,iBAAkC,UAA8B;EAC9D,IAAI;AACJ,MAAI;AACF,aAAU,KAAK,MAAM,OAAO,MAAM,KAAK,CAAC;UAClC;AACN;;AAGF,MACEC,cAAAA,SAAS,QAAQ,IACjB,OAAO,QAAQ,OAAO,aACrB,QAAQ,SAAS,aAAa,QAAQ,SAAS,UAChD;GACA,MAAM,UAAU,KAAK,QAAQ,IAAI,QAAQ,GAAG;AAC5C,OAAI,SAAS;AACX,SAAK,QAAQ,OAAO,QAAQ,GAAG;AAC/B,YAAQ,QAAQ,QAA2C;;AAE7D;;AAGF,MAAIA,cAAAA,SAAS,QAAQ,IAAI,QAAQ,SAAS,QACxC,MAAK,MAAM,KAAK,QAAmB;;CAIvC,oBAA2C;AACzC,OAAK,SAAS;AAEd,MAAI,KAAK,oBAAoB,KAAK,QAAQ;AACxC,QAAK,MAAM,OAAO;AAClB;;EAGF,MAAM,wBAAQ,IAAI,MAAM,0CAA0C;AAClE,OAAK,MAAM,EAAE,YAAY,KAAK,QAAQ,QAAQ,CAC5C,QAAO,MAAM;AAEf,OAAK,QAAQ,OAAO;AACpB,OAAK,MAAM,MAAM,MAAM;;CAGzB,0BAAiD;AAC/C,MAAI,KAAK,UAAU,KAAK,iBACtB;EAGF,MAAM,wBAAQ,IAAI,MAAM,2CAA2C;AACnE,OAAK,MAAM,EAAE,YAAY,KAAK,QAAQ,QAAQ,CAC5C,QAAO,MAAM;AAEf,OAAK,QAAQ,OAAO;AACpB,OAAK,MAAM,MAAM,MAAM"}
+{"version":3,"file":"websocket.cjs","names":["AsyncQueue","resolveProtocolPath","#detachSocket","toWebSocketUrl","toAbsoluteUrl","#attachSocket","hasHeaders","toError","isRecord","#handleUnexpectedDisconnect","#scheduleReconnect","#runReconnectLoop","MaxWebSocketReconnectAttemptsError"],"sources":["../../../../src/client/stream/transport/websocket.ts"],"sourcesContent":["import { AsyncQueue } from \"./queue.js\";\nimport type {\n  Message,\n  Command,\n  CommandResponse,\n  ErrorResponse,\n} from \"@langchain/protocol\";\n\nimport {\n  toAbsoluteUrl,\n  toWebSocketUrl,\n  isRecord,\n  hasHeaders,\n  toError,\n  resolveProtocolPath,\n} from \"./utils.js\";\nimport type {\n  HeaderValue,\n  ProtocolRequestHook,\n  PendingResponse,\n  ProtocolTransportPaths,\n  ProtocolWebSocketTransportOptions,\n} from \"./types.js\";\nimport type { TransportAdapter } from \"../transport.js\";\nimport { MaxWebSocketReconnectAttemptsError } from \"../error.js\";\n\nconst WEB_SOCKET_CONNECTING = 0;\nconst WEB_SOCKET_OPEN = 1;\nconst WEB_SOCKET_CLOSED = 3;\n\n/**\n * Reconnect tuning for {@link ProtocolWebSocketTransportAdapter}. A subset\n * of {@link ProtocolWebSocketTransportOptions}.\n */\nexport interface WebSocketReconnectOptions {\n  /**\n   * Maximum reconnection attempts after an unexpected disconnect.\n   * Defaults to 5.\n   */\n  maxReconnectAttempts?: number;\n\n  /**\n   * Invoked before each reconnect attempt (after backoff).\n   */\n  onReconnect?: (options: { attempt: number; cause: unknown }) => void;\n}\n\n/**\n * Exponential backoff with jitter for WebSocket reconnect. Mirrors\n * {@link streamWithRetry} in `utils/stream.ts` (capped at 5s + 1s jitter).\n */\nexport function webSocketReconnectDelayMs(attempt: number): number {\n  const baseDelay = Math.min(1000 * 2 ** (attempt - 1), 5000);\n  const jitter = Math.random() * 1000;\n  return baseDelay + jitter;\n}\n\n/**\n * Transport adapter that speaks the thread-centric protocol over a\n * bidirectional WebSocket. Bound to a `threadId` at construction or later\n * via {@link setThreadId} — the socket connects to\n * `ws://.../threads/:thread_id/stream/events`.\n *\n * On unexpected disconnect the adapter reconnects with exponential\n * backoff (see {@link ProtocolWebSocketTransportOptions.maxReconnectAttempts}).\n * The server replays buffered events on the new socket; the SDK\n * deduplicates by `event_id`. {@link ProtocolWebSocketTransportOptions.onReconnected}\n * runs after each successful reconnect so `ThreadStream` can re-issue\n * `subscription.subscribe` commands.\n */\nexport class ProtocolWebSocketTransportAdapter implements TransportAdapter {\n  threadId: string;\n\n  private readonly queue = new AsyncQueue<Message>();\n\n  private readonly apiUrl: string;\n\n  private readonly defaultHeaders?: Record<string, HeaderValue>;\n\n  private readonly onRequest?: ProtocolRequestHook;\n\n  private readonly webSocketFactory: (url: string) => WebSocket;\n\n  private readonly paths?: Pick<ProtocolTransportPaths, \"stream\">;\n\n  private readonly maxReconnectAttempts: number;\n\n  private readonly onReconnect?: ProtocolWebSocketTransportOptions[\"onReconnect\"];\n\n  private readonly reconnectDelayMs: (attempt: number) => number;\n\n  private onReconnected?: () => void | Promise<void>;\n\n  private readonly pending = new Map<number, PendingResponse>();\n\n  private socket: WebSocket | null = null;\n\n  private closed = false;\n\n  private intentionalClose = false;\n\n  private reconnectInFlight: Promise<void> | null = null;\n\n  constructor(options: ProtocolWebSocketTransportOptions) {\n    this.apiUrl = options.apiUrl;\n    this.threadId = options.threadId ?? \"\";\n    this.defaultHeaders = options.defaultHeaders;\n    this.onRequest = options.onRequest;\n    this.webSocketFactory =\n      options.webSocketFactory ?? ((url) => new WebSocket(url));\n    this.paths = options.paths;\n    this.maxReconnectAttempts = options.maxReconnectAttempts ?? 5;\n    this.onReconnect = options.onReconnect;\n    this.onReconnected = options.onReconnected;\n    this.reconnectDelayMs =\n      options.reconnectDelayMs ?? webSocketReconnectDelayMs;\n  }\n\n  /** {@inheritDoc TransportAdapter.setThreadId} */\n  setThreadId(threadId: string): void {\n    if (threadId === this.threadId) {\n      return;\n    }\n    if (\n      this.reconnectInFlight != null ||\n      (this.socket != null && this.socket.readyState !== WEB_SOCKET_CLOSED)\n    ) {\n      throw new Error(\n        \"Protocol WebSocket transport cannot be rebound to a different thread while the socket is open. Close the current stream and create a new WebSocket transport for the new thread.\"\n      );\n    }\n    this.threadId = threadId;\n  }\n\n  /**\n   * Socket URL derives from the currently-bound thread so a single adapter\n   * can follow {@link setThreadId} re-binds; the next {@link open} connects\n   * to the new thread. A fixed `paths.stream` string overrides the default.\n   */\n  private get streamUrl(): string {\n    return resolveProtocolPath(\n      this.paths?.stream,\n      this.threadId,\n      (id) => `/threads/${id}/stream/events`\n    );\n  }\n\n  /**\n   * Register a callback invoked after each successful reconnect. Used\n   * by {@link ThreadStream} to re-send active `subscription.subscribe`\n   * commands.\n   */\n  setOnReconnected(handler: () => void | Promise<void>): void {\n    this.onReconnected = handler;\n  }\n\n  async open(): Promise<void> {\n    if (this.closed) {\n      throw new Error(\"Protocol WebSocket transport is closed.\");\n    }\n    if (this.socket?.readyState === WEB_SOCKET_OPEN) {\n      return;\n    }\n    if (this.socket != null) {\n      this.#detachSocket(this.socket);\n      this.socket = null;\n    }\n\n    this.assertBrowserSafeTransportConfig();\n\n    const wsUrl = toWebSocketUrl(\n      toAbsoluteUrl(this.apiUrl, this.streamUrl).toString()\n    );\n    const socket = this.webSocketFactory(wsUrl);\n    this.socket = socket;\n    this.intentionalClose = false;\n\n    this.#attachSocket(socket);\n\n    await new Promise<void>((resolve, reject) => {\n      const onOpen = () => {\n        cleanup();\n        resolve();\n      };\n      const onError = () => {\n        cleanup();\n        reject(new Error(\"Failed to open protocol WebSocket.\"));\n      };\n      const cleanup = () => {\n        socket.removeEventListener(\"open\", onOpen);\n        socket.removeEventListener(\"error\", onError);\n      };\n      socket.addEventListener(\"open\", onOpen, { once: true });\n      socket.addEventListener(\"error\", onError, { once: true });\n    });\n  }\n\n  async send(\n    command: Command\n  ): Promise<CommandResponse | ErrorResponse | void> {\n    return await this.sendCommand(command);\n  }\n\n  events(): AsyncIterable<Message> {\n    const queue = this.queue;\n    return {\n      [Symbol.asyncIterator]: () => ({\n        next: async () => await queue.shift(),\n        return: async () => {\n          queue.close();\n          return { done: true, value: undefined };\n        },\n      }),\n    };\n  }\n\n  async close(): Promise<void> {\n    if (this.closed) {\n      return;\n    }\n\n    this.closed = true;\n    this.intentionalClose = true;\n\n    for (const { reject } of this.pending.values()) {\n      reject(new Error(\"Protocol WebSocket connection closed.\"));\n    }\n    this.pending.clear();\n    this.queue.close();\n\n    const socket = this.socket;\n    this.socket = null;\n    if (!socket) {\n      return;\n    }\n\n    this.#detachSocket(socket);\n\n    await new Promise<void>((resolve) => {\n      if (socket.readyState === WEB_SOCKET_CLOSED) {\n        resolve();\n        return;\n      }\n\n      const onClose = () => {\n        socket.removeEventListener(\"close\", onClose);\n        resolve();\n      };\n\n      socket.addEventListener(\"close\", onClose, { once: true });\n      if (\n        socket.readyState === WEB_SOCKET_OPEN ||\n        socket.readyState === WEB_SOCKET_CONNECTING\n      ) {\n        socket.close();\n      } else {\n        resolve();\n      }\n    });\n  }\n\n  private assertBrowserSafeTransportConfig(): void {\n    if (hasHeaders(this.defaultHeaders) || this.onRequest != null) {\n      throw new Error(\n        \"Browser WebSocket protocol transport does not support defaultHeaders or onRequest hooks. Supply a custom protocolWebSocketFactory if you need custom WebSocket setup.\"\n      );\n    }\n  }\n\n  private async sendCommand(\n    command: Command\n  ): Promise<CommandResponse | ErrorResponse> {\n    // Wait for an in-flight reconnect only when the socket is not usable.\n    // After `open()` succeeds, `#runReconnectLoop` may still be awaiting\n    // `onReconnected` (e.g. ThreadStream resubscribe). Those callbacks call\n    // `sendCommand` and must not await the same `reconnectInFlight` promise.\n    let socket = this.socket;\n    if (\n      this.reconnectInFlight != null &&\n      (socket == null || socket.readyState !== WEB_SOCKET_OPEN)\n    ) {\n      await this.reconnectInFlight.catch(() => undefined);\n      socket = this.socket;\n    }\n\n    if (socket == null || socket.readyState !== WEB_SOCKET_OPEN) {\n      throw new Error(\"Protocol WebSocket is not open.\");\n    }\n\n    return await new Promise<CommandResponse | ErrorResponse>(\n      (resolve, reject) => {\n        this.pending.set(command.id, { resolve, reject });\n\n        try {\n          socket.send(JSON.stringify(command));\n        } catch (error) {\n          this.pending.delete(command.id);\n          reject(toError(error));\n        }\n      }\n    );\n  }\n\n  #attachSocket(socket: WebSocket): void {\n    socket.addEventListener(\"message\", this.handleMessage);\n    socket.addEventListener(\"close\", this.handleClose);\n    socket.addEventListener(\"error\", this.handleSocketError);\n  }\n\n  #detachSocket(socket: WebSocket): void {\n    socket.removeEventListener(\"message\", this.handleMessage);\n    socket.removeEventListener(\"close\", this.handleClose);\n    socket.removeEventListener(\"error\", this.handleSocketError);\n  }\n\n  private readonly handleMessage = (event: MessageEvent): void => {\n    let payload: unknown;\n    try {\n      payload = JSON.parse(String(event.data));\n    } catch {\n      return;\n    }\n\n    if (\n      isRecord(payload) &&\n      typeof payload.id === \"number\" &&\n      (payload.type === \"success\" || payload.type === \"error\")\n    ) {\n      const pending = this.pending.get(payload.id);\n      if (pending) {\n        this.pending.delete(payload.id);\n        pending.resolve(payload as CommandResponse | ErrorResponse);\n      }\n      return;\n    }\n\n    if (isRecord(payload) && payload.type === \"event\") {\n      this.queue.push(payload as Message);\n    }\n  };\n\n  private readonly handleClose = (): void => {\n    const socket = this.socket;\n    if (socket != null) {\n      this.#detachSocket(socket);\n    }\n    this.socket = null;\n\n    if (this.intentionalClose || this.closed) {\n      this.queue.close();\n      return;\n    }\n\n    this.#handleUnexpectedDisconnect(\n      new Error(\"Protocol WebSocket closed unexpectedly.\")\n    );\n  };\n\n  private readonly handleSocketError = (): void => {\n    if (this.closed || this.intentionalClose) {\n      return;\n    }\n\n    this.#handleUnexpectedDisconnect(\n      new Error(\"Protocol WebSocket encountered an error.\")\n    );\n  };\n\n  #handleUnexpectedDisconnect(cause: unknown): void {\n    const error = toError(cause);\n    for (const { reject } of this.pending.values()) {\n      reject(error);\n    }\n    this.pending.clear();\n\n    if (this.maxReconnectAttempts <= 0) {\n      this.queue.close(error);\n      return;\n    }\n\n    this.#scheduleReconnect(cause);\n  }\n\n  #scheduleReconnect(cause: unknown): void {\n    if (this.closed || this.intentionalClose) {\n      return;\n    }\n    if (this.reconnectInFlight != null) {\n      return;\n    }\n\n    this.reconnectInFlight = this.#runReconnectLoop(cause).finally(() => {\n      this.reconnectInFlight = null;\n    });\n  }\n\n  async #runReconnectLoop(initialCause: unknown): Promise<void> {\n    let lastError: unknown = initialCause;\n\n    for (let attempt = 1; attempt <= this.maxReconnectAttempts; attempt += 1) {\n      if (this.closed || this.intentionalClose) {\n        return;\n      }\n\n      this.onReconnect?.({ attempt, cause: lastError });\n\n      const delay = this.reconnectDelayMs(attempt);\n      if (delay > 0) {\n        await new Promise<void>((resolve) => {\n          setTimeout(resolve, delay);\n        });\n      }\n\n      if (this.closed || this.intentionalClose) {\n        return;\n      }\n\n      try {\n        await this.open();\n        if (this.onReconnected) {\n          await this.onReconnected();\n        }\n        return;\n      } catch (error) {\n        lastError = error;\n      }\n    }\n\n    this.queue.close(\n      new MaxWebSocketReconnectAttemptsError(\n        this.maxReconnectAttempts,\n        lastError\n      )\n    );\n  }\n}\n"],"mappings":";;;;AA0BA,MAAM,wBAAwB;AAC9B,MAAM,kBAAkB;AACxB,MAAM,oBAAoB;;;;;AAuB1B,SAAgB,0BAA0B,SAAyB;AAGjE,QAFkB,KAAK,IAAI,MAAO,MAAM,UAAU,IAAI,IAAK,GAC5C,KAAK,QAAQ,GAAG;;;;;;;;;;;;;;;AAiBjC,IAAa,oCAAb,MAA2E;CACzE;CAEA,QAAyB,IAAIA,cAAAA,YAAqB;CAElD;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA,0BAA2B,IAAI,KAA8B;CAE7D,SAAmC;CAEnC,SAAiB;CAEjB,mBAA2B;CAE3B,oBAAkD;CAElD,YAAY,SAA4C;AACtD,OAAK,SAAS,QAAQ;AACtB,OAAK,WAAW,QAAQ,YAAY;AACpC,OAAK,iBAAiB,QAAQ;AAC9B,OAAK,YAAY,QAAQ;AACzB,OAAK,mBACH,QAAQ,sBAAsB,QAAQ,IAAI,UAAU,IAAI;AAC1D,OAAK,QAAQ,QAAQ;AACrB,OAAK,uBAAuB,QAAQ,wBAAwB;AAC5D,OAAK,cAAc,QAAQ;AAC3B,OAAK,gBAAgB,QAAQ;AAC7B,OAAK,mBACH,QAAQ,oBAAoB;;;CAIhC,YAAY,UAAwB;AAClC,MAAI,aAAa,KAAK,SACpB;AAEF,MACE,KAAK,qBAAqB,QACzB,KAAK,UAAU,QAAQ,KAAK,OAAO,eAAe,kBAEnD,OAAM,IAAI,MACR,mLACD;AAEH,OAAK,WAAW;;;;;;;CAQlB,IAAY,YAAoB;AAC9B,SAAOC,cAAAA,oBACL,KAAK,OAAO,QACZ,KAAK,WACJ,OAAO,YAAY,GAAG,gBACxB;;;;;;;CAQH,iBAAiB,SAA2C;AAC1D,OAAK,gBAAgB;;CAGvB,MAAM,OAAsB;AAC1B,MAAI,KAAK,OACP,OAAM,IAAI,MAAM,0CAA0C;AAE5D,MAAI,KAAK,QAAQ,eAAe,gBAC9B;AAEF,MAAI,KAAK,UAAU,MAAM;AACvB,SAAA,aAAmB,KAAK,OAAO;AAC/B,QAAK,SAAS;;AAGhB,OAAK,kCAAkC;EAEvC,MAAM,QAAQE,cAAAA,eACZC,cAAAA,cAAc,KAAK,QAAQ,KAAK,UAAU,CAAC,UAAU,CACtD;EACD,MAAM,SAAS,KAAK,iBAAiB,MAAM;AAC3C,OAAK,SAAS;AACd,OAAK,mBAAmB;AAExB,QAAA,aAAmB,OAAO;AAE1B,QAAM,IAAI,SAAe,SAAS,WAAW;GAC3C,MAAM,eAAe;AACnB,aAAS;AACT,aAAS;;GAEX,MAAM,gBAAgB;AACpB,aAAS;AACT,2BAAO,IAAI,MAAM,qCAAqC,CAAC;;GAEzD,MAAM,gBAAgB;AACpB,WAAO,oBAAoB,QAAQ,OAAO;AAC1C,WAAO,oBAAoB,SAAS,QAAQ;;AAE9C,UAAO,iBAAiB,QAAQ,QAAQ,EAAE,MAAM,MAAM,CAAC;AACvD,UAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;IACzD;;CAGJ,MAAM,KACJ,SACiD;AACjD,SAAO,MAAM,KAAK,YAAY,QAAQ;;CAGxC,SAAiC;EAC/B,MAAM,QAAQ,KAAK;AACnB,SAAO,GACJ,OAAO,uBAAuB;GAC7B,MAAM,YAAY,MAAM,MAAM,OAAO;GACrC,QAAQ,YAAY;AAClB,UAAM,OAAO;AACb,WAAO;KAAE,MAAM;KAAM,OAAO,KAAA;KAAW;;GAE1C,GACF;;CAGH,MAAM,QAAuB;AAC3B,MAAI,KAAK,OACP;AAGF,OAAK,SAAS;AACd,OAAK,mBAAmB;AAExB,OAAK,MAAM,EAAE,YAAY,KAAK,QAAQ,QAAQ,CAC5C,wBAAO,IAAI,MAAM,wCAAwC,CAAC;AAE5D,OAAK,QAAQ,OAAO;AACpB,OAAK,MAAM,OAAO;EAElB,MAAM,SAAS,KAAK;AACpB,OAAK,SAAS;AACd,MAAI,CAAC,OACH;AAGF,QAAA,aAAmB,OAAO;AAE1B,QAAM,IAAI,SAAe,YAAY;AACnC,OAAI,OAAO,eAAe,mBAAmB;AAC3C,aAAS;AACT;;GAGF,MAAM,gBAAgB;AACpB,WAAO,oBAAoB,SAAS,QAAQ;AAC5C,aAAS;;AAGX,UAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AACzD,OACE,OAAO,eAAe,mBACtB,OAAO,eAAe,sBAEtB,QAAO,OAAO;OAEd,UAAS;IAEX;;CAGJ,mCAAiD;AAC/C,MAAIE,cAAAA,WAAW,KAAK,eAAe,IAAI,KAAK,aAAa,KACvD,OAAM,IAAI,MACR,wKACD;;CAIL,MAAc,YACZ,SAC0C;EAK1C,IAAI,SAAS,KAAK;AAClB,MACE,KAAK,qBAAqB,SACzB,UAAU,QAAQ,OAAO,eAAe,kBACzC;AACA,SAAM,KAAK,kBAAkB,YAAY,KAAA,EAAU;AACnD,YAAS,KAAK;;AAGhB,MAAI,UAAU,QAAQ,OAAO,eAAe,gBAC1C,OAAM,IAAI,MAAM,kCAAkC;AAGpD,SAAO,MAAM,IAAI,SACd,SAAS,WAAW;AACnB,QAAK,QAAQ,IAAI,QAAQ,IAAI;IAAE;IAAS;IAAQ,CAAC;AAEjD,OAAI;AACF,WAAO,KAAK,KAAK,UAAU,QAAQ,CAAC;YAC7B,OAAO;AACd,SAAK,QAAQ,OAAO,QAAQ,GAAG;AAC/B,WAAOC,cAAAA,QAAQ,MAAM,CAAC;;IAG3B;;CAGH,cAAc,QAAyB;AACrC,SAAO,iBAAiB,WAAW,KAAK,cAAc;AACtD,SAAO,iBAAiB,SAAS,KAAK,YAAY;AAClD,SAAO,iBAAiB,SAAS,KAAK,kBAAkB;;CAG1D,cAAc,QAAyB;AACrC,SAAO,oBAAoB,WAAW,KAAK,cAAc;AACzD,SAAO,oBAAoB,SAAS,KAAK,YAAY;AACrD,SAAO,oBAAoB,SAAS,KAAK,kBAAkB;;CAG7D,iBAAkC,UAA8B;EAC9D,IAAI;AACJ,MAAI;AACF,aAAU,KAAK,MAAM,OAAO,MAAM,KAAK,CAAC;UAClC;AACN;;AAGF,MACEC,cAAAA,SAAS,QAAQ,IACjB,OAAO,QAAQ,OAAO,aACrB,QAAQ,SAAS,aAAa,QAAQ,SAAS,UAChD;GACA,MAAM,UAAU,KAAK,QAAQ,IAAI,QAAQ,GAAG;AAC5C,OAAI,SAAS;AACX,SAAK,QAAQ,OAAO,QAAQ,GAAG;AAC/B,YAAQ,QAAQ,QAA2C;;AAE7D;;AAGF,MAAIA,cAAAA,SAAS,QAAQ,IAAI,QAAQ,SAAS,QACxC,MAAK,MAAM,KAAK,QAAmB;;CAIvC,oBAA2C;EACzC,MAAM,SAAS,KAAK;AACpB,MAAI,UAAU,KACZ,OAAA,aAAmB,OAAO;AAE5B,OAAK,SAAS;AAEd,MAAI,KAAK,oBAAoB,KAAK,QAAQ;AACxC,QAAK,MAAM,OAAO;AAClB;;AAGF,QAAA,2CACE,IAAI,MAAM,0CAA0C,CACrD;;CAGH,0BAAiD;AAC/C,MAAI,KAAK,UAAU,KAAK,iBACtB;AAGF,QAAA,2CACE,IAAI,MAAM,2CAA2C,CACtD;;CAGH,4BAA4B,OAAsB;EAChD,MAAM,QAAQD,cAAAA,QAAQ,MAAM;AAC5B,OAAK,MAAM,EAAE,YAAY,KAAK,QAAQ,QAAQ,CAC5C,QAAO,MAAM;AAEf,OAAK,QAAQ,OAAO;AAEpB,MAAI,KAAK,wBAAwB,GAAG;AAClC,QAAK,MAAM,MAAM,MAAM;AACvB;;AAGF,QAAA,kBAAwB,MAAM;;CAGhC,mBAAmB,OAAsB;AACvC,MAAI,KAAK,UAAU,KAAK,iBACtB;AAEF,MAAI,KAAK,qBAAqB,KAC5B;AAGF,OAAK,oBAAoB,MAAA,iBAAuB,MAAM,CAAC,cAAc;AACnE,QAAK,oBAAoB;IACzB;;CAGJ,OAAA,iBAAwB,cAAsC;EAC5D,IAAI,YAAqB;AAEzB,OAAK,IAAI,UAAU,GAAG,WAAW,KAAK,sBAAsB,WAAW,GAAG;AACxE,OAAI,KAAK,UAAU,KAAK,iBACtB;AAGF,QAAK,cAAc;IAAE;IAAS,OAAO;IAAW,CAAC;GAEjD,MAAM,QAAQ,KAAK,iBAAiB,QAAQ;AAC5C,OAAI,QAAQ,EACV,OAAM,IAAI,SAAe,YAAY;AACnC,eAAW,SAAS,MAAM;KAC1B;AAGJ,OAAI,KAAK,UAAU,KAAK,iBACtB;AAGF,OAAI;AACF,UAAM,KAAK,MAAM;AACjB,QAAI,KAAK,cACP,OAAM,KAAK,eAAe;AAE5B;YACO,OAAO;AACd,gBAAY;;;AAIhB,OAAK,MAAM,MACT,IAAIK,cAAAA,mCACF,KAAK,sBACL,UACD,CACF"}

package/dist/client/stream/transport/websocket.d.cts

@@ -5,22 +5,50 @@ import { Command, CommandResponse, ErrorResponse, Message } from "@langchain/pro
 //#region src/client/stream/transport/websocket.d.ts
 /**
  * Transport adapter that speaks the thread-centric protocol over a
- * bidirectional WebSocket. Bound to a specific `threadId` — the socket
- * connects to `ws://.../threads/:thread_id/stream/events`.
+ * bidirectional WebSocket. Bound to a `threadId` at construction or later
+ * via {@link setThreadId} — the socket connects to
+ * `ws://.../threads/:thread_id/stream/events`.
+ *
+ * On unexpected disconnect the adapter reconnects with exponential
+ * backoff (see {@link ProtocolWebSocketTransportOptions.maxReconnectAttempts}).
+ * The server replays buffered events on the new socket; the SDK
+ * deduplicates by `event_id`. {@link ProtocolWebSocketTransportOptions.onReconnected}
+ * runs after each successful reconnect so `ThreadStream` can re-issue
+ * `subscription.subscribe` commands.
  */
 declare class ProtocolWebSocketTransportAdapter implements TransportAdapter {
-  readonly threadId: string;
+  #private;
+  threadId: string;
   private readonly queue;
   private readonly apiUrl;
   private readonly defaultHeaders?;
   private readonly onRequest?;
   private readonly webSocketFactory;
-  private readonly streamUrl;
+  private readonly paths?;
+  private readonly maxReconnectAttempts;
+  private readonly onReconnect?;
+  private readonly reconnectDelayMs;
+  private onReconnected?;
   private readonly pending;
   private socket;
   private closed;
   private intentionalClose;
+  private reconnectInFlight;
   constructor(options: ProtocolWebSocketTransportOptions);
+  /** {@inheritDoc TransportAdapter.setThreadId} */
+  setThreadId(threadId: string): void;
+  /**
+   * Socket URL derives from the currently-bound thread so a single adapter
+   * can follow {@link setThreadId} re-binds; the next {@link open} connects
+   * to the new thread. A fixed `paths.stream` string overrides the default.
+   */
+  private get streamUrl();
+  /**
+   * Register a callback invoked after each successful reconnect. Used
+   * by {@link ThreadStream} to re-send active `subscription.subscribe`
+   * commands.
+   */
+  setOnReconnected(handler: () => void | Promise<void>): void;
   open(): Promise<void>;
   send(command: Command): Promise<CommandResponse | ErrorResponse | void>;
   events(): AsyncIterable<Message>;

package/dist/client/stream/transport/websocket.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"websocket.d.cts","names":[],"sources":["../../../../src/client/stream/transport/websocket.ts"],"mappings":";;;;;;;AA4BA;;;cAAa,iCAAA,YAA6C,gBAAA;EAAA,SAC/C,QAAA;EAAA,iBAEQ,KAAA;EAAA,iBAEA,MAAA;EAAA,iBAEA,cAAA;EAAA,iBAEA,SAAA;EAAA,iBAEA,gBAAA;EAAA,iBAEA,SAAA;EAAA,iBAEA,OAAA;EAAA,QAET,MAAA;EAAA,QAEA,MAAA;EAAA,QAEA,gBAAA;EAER,WAAA,CAAY,OAAA,EAAS,iCAAA;EAWf,IAAA,CAAA,GAAQ,OAAA;EAkCR,IAAA,CACJ,OAAA,EAAS,OAAA,GACR,OAAA,CAAQ,eAAA,GAAkB,aAAA;EAI7B,MAAA,CAAA,GAAU,aAAA,CAAc,OAAA;EAalB,KAAA,CAAA,GAAS,OAAA;EAAA,QA2CP,gCAAA;EAAA,QAQM,WAAA;EAAA,iBAsBG,aAAA;EAAA,iBA0BA,WAAA;EAAA,iBAgBA,iBAAA;AAAA"}
+{"version":3,"file":"websocket.d.cts","names":[],"sources":["../../../../src/client/stream/transport/websocket.ts"],"mappings":";;;;;;;;;;;;;;;;;;cAsEa,iCAAA,YAA6C,gBAAA;EAAA;EACxD,QAAA;EAAA,iBAEiB,KAAA;EAAA,iBAEA,MAAA;EAAA,iBAEA,cAAA;EAAA,iBAEA,SAAA;EAAA,iBAEA,gBAAA;EAAA,iBAEA,KAAA;EAAA,iBAEA,oBAAA;EAAA,iBAEA,WAAA;EAAA,iBAEA,gBAAA;EAAA,QAET,aAAA;EAAA,iBAES,OAAA;EAAA,QAET,MAAA;EAAA,QAEA,MAAA;EAAA,QAEA,gBAAA;EAAA,QAEA,iBAAA;EAER,WAAA,CAAY,OAAA,EAAS,iCAAA;EAoGG;EApFxB,WAAA,CAAY,QAAA;EAiGG;;;;;EAAA,YA7EH,SAAA,CAAA;EA2NsB;;;;;EA9MlC,gBAAA,CAAiB,OAAA,eAAsB,OAAA;EAIjC,IAAA,CAAA,GAAQ,OAAA;EAyCR,IAAA,CACJ,OAAA,EAAS,OAAA,GACR,OAAA,CAAQ,eAAA,GAAkB,aAAA;EAI7B,MAAA,CAAA,GAAU,aAAA,CAAc,OAAA;EAalB,KAAA,CAAA,GAAS,OAAA;EAAA,QA6CP,gCAAA;EAAA,QAQM,WAAA;EAAA,iBA8CG,aAAA;EAAA,iBA0BA,WAAA;EAAA,iBAiBA,iBAAA;AAAA"}

package/dist/client/stream/transport/websocket.d.ts

@@ -5,22 +5,50 @@ import { Command, CommandResponse, ErrorResponse, Message } from "@langchain/pro
 //#region src/client/stream/transport/websocket.d.ts
 /**
  * Transport adapter that speaks the thread-centric protocol over a
- * bidirectional WebSocket. Bound to a specific `threadId` — the socket
- * connects to `ws://.../threads/:thread_id/stream/events`.
+ * bidirectional WebSocket. Bound to a `threadId` at construction or later
+ * via {@link setThreadId} — the socket connects to
+ * `ws://.../threads/:thread_id/stream/events`.
+ *
+ * On unexpected disconnect the adapter reconnects with exponential
+ * backoff (see {@link ProtocolWebSocketTransportOptions.maxReconnectAttempts}).
+ * The server replays buffered events on the new socket; the SDK
+ * deduplicates by `event_id`. {@link ProtocolWebSocketTransportOptions.onReconnected}
+ * runs after each successful reconnect so `ThreadStream` can re-issue
+ * `subscription.subscribe` commands.
  */
 declare class ProtocolWebSocketTransportAdapter implements TransportAdapter {
-  readonly threadId: string;
+  #private;
+  threadId: string;
   private readonly queue;
   private readonly apiUrl;
   private readonly defaultHeaders?;
   private readonly onRequest?;
   private readonly webSocketFactory;
-  private readonly streamUrl;
+  private readonly paths?;
+  private readonly maxReconnectAttempts;
+  private readonly onReconnect?;
+  private readonly reconnectDelayMs;
+  private onReconnected?;
   private readonly pending;
   private socket;
   private closed;
   private intentionalClose;
+  private reconnectInFlight;
   constructor(options: ProtocolWebSocketTransportOptions);
+  /** {@inheritDoc TransportAdapter.setThreadId} */
+  setThreadId(threadId: string): void;
+  /**
+   * Socket URL derives from the currently-bound thread so a single adapter
+   * can follow {@link setThreadId} re-binds; the next {@link open} connects
+   * to the new thread. A fixed `paths.stream` string overrides the default.
+   */
+  private get streamUrl();
+  /**
+   * Register a callback invoked after each successful reconnect. Used
+   * by {@link ThreadStream} to re-send active `subscription.subscribe`
+   * commands.
+   */
+  setOnReconnected(handler: () => void | Promise<void>): void;
   open(): Promise<void>;
   send(command: Command): Promise<CommandResponse | ErrorResponse | void>;
   events(): AsyncIterable<Message>;

package/dist/client/stream/transport/websocket.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"websocket.d.ts","names":[],"sources":["../../../../src/client/stream/transport/websocket.ts"],"mappings":";;;;;;;AA4BA;;;cAAa,iCAAA,YAA6C,gBAAA;EAAA,SAC/C,QAAA;EAAA,iBAEQ,KAAA;EAAA,iBAEA,MAAA;EAAA,iBAEA,cAAA;EAAA,iBAEA,SAAA;EAAA,iBAEA,gBAAA;EAAA,iBAEA,SAAA;EAAA,iBAEA,OAAA;EAAA,QAET,MAAA;EAAA,QAEA,MAAA;EAAA,QAEA,gBAAA;EAER,WAAA,CAAY,OAAA,EAAS,iCAAA;EAWf,IAAA,CAAA,GAAQ,OAAA;EAkCR,IAAA,CACJ,OAAA,EAAS,OAAA,GACR,OAAA,CAAQ,eAAA,GAAkB,aAAA;EAI7B,MAAA,CAAA,GAAU,aAAA,CAAc,OAAA;EAalB,KAAA,CAAA,GAAS,OAAA;EAAA,QA2CP,gCAAA;EAAA,QAQM,WAAA;EAAA,iBAsBG,aAAA;EAAA,iBA0BA,WAAA;EAAA,iBAgBA,iBAAA;AAAA"}
+{"version":3,"file":"websocket.d.ts","names":[],"sources":["../../../../src/client/stream/transport/websocket.ts"],"mappings":";;;;;;;;;;;;;;;;;;cAsEa,iCAAA,YAA6C,gBAAA;EAAA;EACxD,QAAA;EAAA,iBAEiB,KAAA;EAAA,iBAEA,MAAA;EAAA,iBAEA,cAAA;EAAA,iBAEA,SAAA;EAAA,iBAEA,gBAAA;EAAA,iBAEA,KAAA;EAAA,iBAEA,oBAAA;EAAA,iBAEA,WAAA;EAAA,iBAEA,gBAAA;EAAA,QAET,aAAA;EAAA,iBAES,OAAA;EAAA,QAET,MAAA;EAAA,QAEA,MAAA;EAAA,QAEA,gBAAA;EAAA,QAEA,iBAAA;EAER,WAAA,CAAY,OAAA,EAAS,iCAAA;EAoGG;EApFxB,WAAA,CAAY,QAAA;EAiGG;;;;;EAAA,YA7EH,SAAA,CAAA;EA2NsB;;;;;EA9MlC,gBAAA,CAAiB,OAAA,eAAsB,OAAA;EAIjC,IAAA,CAAA,GAAQ,OAAA;EAyCR,IAAA,CACJ,OAAA,EAAS,OAAA,GACR,OAAA,CAAQ,eAAA,GAAkB,aAAA;EAI7B,MAAA,CAAA,GAAU,aAAA,CAAc,OAAA;EAalB,KAAA,CAAA,GAAS,OAAA;EAAA,QA6CP,gCAAA;EAAA,QAQM,WAAA;EAAA,iBA8CG,aAAA;EAAA,iBA0BA,WAAA;EAAA,iBAiBA,iBAAA;AAAA"}

package/dist/client/stream/transport/websocket.js

@@ -1,10 +1,29 @@
+import { MaxWebSocketReconnectAttemptsError } from "../error.js";
 import { AsyncQueue } from "./queue.js";
-import { hasHeaders, isRecord, toAbsoluteUrl, toError, toWebSocketUrl } from "./utils.js";
+import { hasHeaders, isRecord, resolveProtocolPath, toAbsoluteUrl, toError, toWebSocketUrl } from "./utils.js";
 //#region src/client/stream/transport/websocket.ts
+const WEB_SOCKET_CONNECTING = 0;
+const WEB_SOCKET_OPEN = 1;
+const WEB_SOCKET_CLOSED = 3;
+/**
+* Exponential backoff with jitter for WebSocket reconnect. Mirrors
+* {@link streamWithRetry} in `utils/stream.ts` (capped at 5s + 1s jitter).
+*/
+function webSocketReconnectDelayMs(attempt) {
+	return Math.min(1e3 * 2 ** (attempt - 1), 5e3) + Math.random() * 1e3;
+}
 /**
 * Transport adapter that speaks the thread-centric protocol over a
-* bidirectional WebSocket. Bound to a specific `threadId` — the socket
-* connects to `ws://.../threads/:thread_id/stream/events`.
+* bidirectional WebSocket. Bound to a `threadId` at construction or later
+* via {@link setThreadId} — the socket connects to
+* `ws://.../threads/:thread_id/stream/events`.
+*
+* On unexpected disconnect the adapter reconnects with exponential
+* backoff (see {@link ProtocolWebSocketTransportOptions.maxReconnectAttempts}).
+* The server replays buffered events on the new socket; the SDK
+* deduplicates by `event_id`. {@link ProtocolWebSocketTransportOptions.onReconnected}
+* runs after each successful reconnect so `ThreadStream` can re-issue
+* `subscription.subscribe` commands.
 */
 var ProtocolWebSocketTransportAdapter = class {
 	threadId;
@@ -13,30 +32,63 @@ var ProtocolWebSocketTransportAdapter = class {
 	defaultHeaders;
 	onRequest;
 	webSocketFactory;
-	streamUrl;
+	paths;
+	maxReconnectAttempts;
+	onReconnect;
+	reconnectDelayMs;
+	onReconnected;
 	pending = /* @__PURE__ */ new Map();
 	socket = null;
 	closed = false;
 	intentionalClose = false;
+	reconnectInFlight = null;
 	constructor(options) {
 		this.apiUrl = options.apiUrl;
-		this.threadId = options.threadId;
+		this.threadId = options.threadId ?? "";
 		this.defaultHeaders = options.defaultHeaders;
 		this.onRequest = options.onRequest;
 		this.webSocketFactory = options.webSocketFactory ?? ((url) => new WebSocket(url));
-		this.streamUrl = options.paths?.stream ?? `/threads/${this.threadId}/stream/events`;
+		this.paths = options.paths;
+		this.maxReconnectAttempts = options.maxReconnectAttempts ?? 5;
+		this.onReconnect = options.onReconnect;
+		this.onReconnected = options.onReconnected;
+		this.reconnectDelayMs = options.reconnectDelayMs ?? webSocketReconnectDelayMs;
+	}
+	/** {@inheritDoc TransportAdapter.setThreadId} */
+	setThreadId(threadId) {
+		if (threadId === this.threadId) return;
+		if (this.reconnectInFlight != null || this.socket != null && this.socket.readyState !== WEB_SOCKET_CLOSED) throw new Error("Protocol WebSocket transport cannot be rebound to a different thread while the socket is open. Close the current stream and create a new WebSocket transport for the new thread.");
+		this.threadId = threadId;
+	}
+	/**
+	* Socket URL derives from the currently-bound thread so a single adapter
+	* can follow {@link setThreadId} re-binds; the next {@link open} connects
+	* to the new thread. A fixed `paths.stream` string overrides the default.
+	*/
+	get streamUrl() {
+		return resolveProtocolPath(this.paths?.stream, this.threadId, (id) => `/threads/${id}/stream/events`);
+	}
+	/**
+	* Register a callback invoked after each successful reconnect. Used
+	* by {@link ThreadStream} to re-send active `subscription.subscribe`
+	* commands.
+	*/
+	setOnReconnected(handler) {
+		this.onReconnected = handler;
 	}
 	async open() {
-		if (this.socket != null) return;
+		if (this.closed) throw new Error("Protocol WebSocket transport is closed.");
+		if (this.socket?.readyState === WEB_SOCKET_OPEN) return;
+		if (this.socket != null) {
+			this.#detachSocket(this.socket);
+			this.socket = null;
+		}
 		this.assertBrowserSafeTransportConfig();
 		const wsUrl = toWebSocketUrl(toAbsoluteUrl(this.apiUrl, this.streamUrl).toString());
 		const socket = this.webSocketFactory(wsUrl);
 		this.socket = socket;
-		this.closed = false;
 		this.intentionalClose = false;
-		socket.addEventListener("message", this.handleMessage);
-		socket.addEventListener("close", this.handleClose);
-		socket.addEventListener("error", this.handleSocketError);
+		this.#attachSocket(socket);
 		await new Promise((resolve, reject) => {
 			const onOpen = () => {
 				cleanup();
@@ -80,8 +132,9 @@ var ProtocolWebSocketTransportAdapter = class {
 		const socket = this.socket;
 		this.socket = null;
 		if (!socket) return;
+		this.#detachSocket(socket);
 		await new Promise((resolve) => {
-			if (socket.readyState === WebSocket.CLOSED) {
+			if (socket.readyState === WEB_SOCKET_CLOSED) {
 				resolve();
 				return;
 			}
@@ -90,7 +143,7 @@ var ProtocolWebSocketTransportAdapter = class {
 				resolve();
 			};
 			socket.addEventListener("close", onClose, { once: true });
-			if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) socket.close();
+			if (socket.readyState === WEB_SOCKET_OPEN || socket.readyState === WEB_SOCKET_CONNECTING) socket.close();
 			else resolve();
 		});
 	}
@@ -98,8 +151,12 @@ var ProtocolWebSocketTransportAdapter = class {
 		if (hasHeaders(this.defaultHeaders) || this.onRequest != null) throw new Error("Browser WebSocket protocol transport does not support defaultHeaders or onRequest hooks. Supply a custom protocolWebSocketFactory if you need custom WebSocket setup.");
 	}
 	async sendCommand(command) {
-		const socket = this.socket;
-		if (socket == null || socket.readyState !== WebSocket.OPEN) throw new Error("Protocol WebSocket is not open.");
+		let socket = this.socket;
+		if (this.reconnectInFlight != null && (socket == null || socket.readyState !== WEB_SOCKET_OPEN)) {
+			await this.reconnectInFlight.catch(() => void 0);
+			socket = this.socket;
+		}
+		if (socket == null || socket.readyState !== WEB_SOCKET_OPEN) throw new Error("Protocol WebSocket is not open.");
 		return await new Promise((resolve, reject) => {
 			this.pending.set(command.id, {
 				resolve,
@@ -113,6 +170,16 @@ var ProtocolWebSocketTransportAdapter = class {
 			}
 		});
 	}
+	#attachSocket(socket) {
+		socket.addEventListener("message", this.handleMessage);
+		socket.addEventListener("close", this.handleClose);
+		socket.addEventListener("error", this.handleSocketError);
+	}
+	#detachSocket(socket) {
+		socket.removeEventListener("message", this.handleMessage);
+		socket.removeEventListener("close", this.handleClose);
+		socket.removeEventListener("error", this.handleSocketError);
+	}
 	handleMessage = (event) => {
 		let payload;
 		try {
@@ -131,25 +198,61 @@ var ProtocolWebSocketTransportAdapter = class {
 		if (isRecord(payload) && payload.type === "event") this.queue.push(payload);
 	};
 	handleClose = () => {
+		const socket = this.socket;
+		if (socket != null) this.#detachSocket(socket);
 		this.socket = null;
 		if (this.intentionalClose || this.closed) {
 			this.queue.close();
 			return;
 		}
-		const error = /* @__PURE__ */ new Error("Protocol WebSocket closed unexpectedly.");
-		for (const { reject } of this.pending.values()) reject(error);
-		this.pending.clear();
-		this.queue.close(error);
+		this.#handleUnexpectedDisconnect(/* @__PURE__ */ new Error("Protocol WebSocket closed unexpectedly."));
 	};
 	handleSocketError = () => {
 		if (this.closed || this.intentionalClose) return;
-		const error = /* @__PURE__ */ new Error("Protocol WebSocket encountered an error.");
+		this.#handleUnexpectedDisconnect(/* @__PURE__ */ new Error("Protocol WebSocket encountered an error."));
+	};
+	#handleUnexpectedDisconnect(cause) {
+		const error = toError(cause);
 		for (const { reject } of this.pending.values()) reject(error);
 		this.pending.clear();
-		this.queue.close(error);
-	};
+		if (this.maxReconnectAttempts <= 0) {
+			this.queue.close(error);
+			return;
+		}
+		this.#scheduleReconnect(cause);
+	}
+	#scheduleReconnect(cause) {
+		if (this.closed || this.intentionalClose) return;
+		if (this.reconnectInFlight != null) return;
+		this.reconnectInFlight = this.#runReconnectLoop(cause).finally(() => {
+			this.reconnectInFlight = null;
+		});
+	}
+	async #runReconnectLoop(initialCause) {
+		let lastError = initialCause;
+		for (let attempt = 1; attempt <= this.maxReconnectAttempts; attempt += 1) {
+			if (this.closed || this.intentionalClose) return;
+			this.onReconnect?.({
+				attempt,
+				cause: lastError
+			});
+			const delay = this.reconnectDelayMs(attempt);
+			if (delay > 0) await new Promise((resolve) => {
+				setTimeout(resolve, delay);
+			});
+			if (this.closed || this.intentionalClose) return;
+			try {
+				await this.open();
+				if (this.onReconnected) await this.onReconnected();
+				return;
+			} catch (error) {
+				lastError = error;
+			}
+		}
+		this.queue.close(new MaxWebSocketReconnectAttemptsError(this.maxReconnectAttempts, lastError));
+	}
 };
 //#endregion
-export { ProtocolWebSocketTransportAdapter };
+export { ProtocolWebSocketTransportAdapter, webSocketReconnectDelayMs };
 
 //# sourceMappingURL=websocket.js.map