@langchain/langgraph-sdk 1.10.3-rc.1 → 1.11.0

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

@@ -1 +1 @@
-{"version":3,"file":"websocket.cjs","names":["AsyncQueue","reconnectDelayMs","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\";\nimport {\n  DEFAULT_MAX_RECONNECT_ATTEMPTS,\n  reconnectDelayMs,\n} from \"../../../utils/reconnect.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 {@link DEFAULT_MAX_RECONNECT_ATTEMPTS}.\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 * 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 =\n      options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT_ATTEMPTS;\n    this.onReconnect = options.onReconnect;\n    this.onReconnected = options.onReconnected;\n    this.reconnectDelayMs = options.reconnectDelayMs ?? reconnectDelayMs;\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":";;;;;AA8BA,MAAM,wBAAwB;AAC9B,MAAM,kBAAkB;AACxB,MAAM,oBAAoB;;;;;;;;;;;;;;AAgC1B,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,uBACH,QAAQ,wBAAA;AACV,OAAK,cAAc,QAAQ;AAC3B,OAAK,gBAAgB,QAAQ;AAC7B,OAAK,mBAAmB,QAAQ,oBAAoBC,kBAAAA;;;CAItD,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"}
+{"version":3,"file":"websocket.cjs","names":["AsyncQueue","reconnectDelayMs","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\";\nimport {\n  DEFAULT_MAX_RECONNECT_ATTEMPTS,\n  reconnectDelayMs,\n  type ReconnectInfo,\n} from \"../../../utils/reconnect.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 {@link DEFAULT_MAX_RECONNECT_ATTEMPTS}.\n   */\n  maxReconnectAttempts?: number;\n\n  /** Invoked before each reconnect attempt with its scheduled delay. */\n  onReconnect?: (options: ReconnectInfo) => void;\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 onConnected?: ProtocolWebSocketTransportOptions[\"onConnected\"];\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 =\n      options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT_ATTEMPTS;\n    this.onReconnect = options.onReconnect;\n    this.onConnected = options.onConnected;\n    this.onReconnected = options.onReconnected;\n    this.reconnectDelayMs = options.reconnectDelayMs ?? reconnectDelayMs;\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    const reconnecting = this.reconnectInFlight != null;\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    if (!reconnecting) {\n      await this.onConnected?.({ kind: \"initial\", attempt: 0 });\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 = Promise.resolve()\n      .then(() => this.#runReconnectLoop(cause))\n      .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      const delay = this.reconnectDelayMs(attempt);\n      this.onReconnect?.({ attempt, cause: lastError, delayMs: delay });\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        await this.onConnected?.({ kind: \"reconnected\", attempt });\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":";;;;;AA+BA,MAAM,wBAAwB;AAC9B,MAAM,kBAAkB;AACxB,MAAM,oBAAoB;;;;;;;;;;;;;;AA8B1B,IAAa,oCAAb,MAA2E;CACzE;CAEA,QAAyB,IAAIA,cAAAA,YAAqB;CAElD;CAEA;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,uBACH,QAAQ,wBAAA;AACV,OAAK,cAAc,QAAQ;AAC3B,OAAK,cAAc,QAAQ;AAC3B,OAAK,gBAAgB,QAAQ;AAC7B,OAAK,mBAAmB,QAAQ,oBAAoBC,kBAAAA;;;CAItD,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;EAC1B,MAAM,eAAe,KAAK,qBAAqB;AAC/C,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;AAEF,MAAI,CAAC,aACH,OAAM,KAAK,cAAc;GAAE,MAAM;GAAW,SAAS;GAAG,CAAC;;CAI7D,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,QAAQ,SAAS,CACvC,WAAW,MAAA,iBAAuB,MAAM,CAAC,CACzC,cAAc;AACb,QAAK,oBAAoB;IACzB;;CAGN,OAAA,iBAAwB,cAAsC;EAC5D,IAAI,YAAqB;AAEzB,OAAK,IAAI,UAAU,GAAG,WAAW,KAAK,sBAAsB,WAAW,GAAG;AACxE,OAAI,KAAK,UAAU,KAAK,iBACtB;GAGF,MAAM,QAAQ,KAAK,iBAAiB,QAAQ;AAC5C,QAAK,cAAc;IAAE;IAAS,OAAO;IAAW,SAAS;IAAO,CAAC;AACjE,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,UAAM,KAAK,cAAc;KAAE,MAAM;KAAe;KAAS,CAAC;AAC1D;YACO,OAAO;AACd,gBAAY;;;AAIhB,OAAK,MAAM,MACT,IAAIK,cAAAA,mCACF,KAAK,sBACL,UACD,CACF"}

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

@@ -27,6 +27,7 @@ declare class ProtocolWebSocketTransportAdapter implements TransportAdapter {
   private readonly paths?;
   private readonly maxReconnectAttempts;
   private readonly onReconnect?;
+  private readonly onConnected?;
   private readonly reconnectDelayMs;
   private onReconnected?;
   private readonly pending;

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":";;;;;;;;;;;;;;;;;;cAgEa,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;EAgGlB;EAhFH,WAAA,CAAY,QAAA;EAgFiB;;;;;EAAA,YA5DjB,SAAA,CAAA;EA0HJ;;;;;EA7GR,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"}
+{"version":3,"file":"websocket.d.cts","names":[],"sources":["../../../../src/client/stream/transport/websocket.ts"],"mappings":";;;;;;;;;;;;;;;;;;cA+Da,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,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;EAsDP;EArCd,WAAA,CAAY,QAAA;EAoFD;;;;;EAAA,YAhEC,SAAA,CAAA;EAqEF;;;;;EAxDV,gBAAA,CAAiB,OAAA,eAAsB,OAAA;EAIjC,IAAA,CAAA,GAAQ,OAAA;EA8CR,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

@@ -27,6 +27,7 @@ declare class ProtocolWebSocketTransportAdapter implements TransportAdapter {
   private readonly paths?;
   private readonly maxReconnectAttempts;
   private readonly onReconnect?;
+  private readonly onConnected?;
   private readonly reconnectDelayMs;
   private onReconnected?;
   private readonly pending;

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":";;;;;;;;;;;;;;;;;;cAgEa,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;EAgGlB;EAhFH,WAAA,CAAY,QAAA;EAgFiB;;;;;EAAA,YA5DjB,SAAA,CAAA;EA0HJ;;;;;EA7GR,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"}
+{"version":3,"file":"websocket.d.ts","names":[],"sources":["../../../../src/client/stream/transport/websocket.ts"],"mappings":";;;;;;;;;;;;;;;;;;cA+Da,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,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;EAsDP;EArCd,WAAA,CAAY,QAAA;EAoFD;;;;;EAAA,YAhEC,SAAA,CAAA;EAqEF;;;;;EAxDV,gBAAA,CAAiB,OAAA,eAAsB,OAAA;EAIjC,IAAA,CAAA,GAAQ,OAAA;EA8CR,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

@@ -29,6 +29,7 @@ var ProtocolWebSocketTransportAdapter = class {
 	paths;
 	maxReconnectAttempts;
 	onReconnect;
+	onConnected;
 	reconnectDelayMs;
 	onReconnected;
 	pending = /* @__PURE__ */ new Map();
@@ -45,6 +46,7 @@ var ProtocolWebSocketTransportAdapter = class {
 		this.paths = options.paths;
 		this.maxReconnectAttempts = options.maxReconnectAttempts ?? 5;
 		this.onReconnect = options.onReconnect;
+		this.onConnected = options.onConnected;
 		this.onReconnected = options.onReconnected;
 		this.reconnectDelayMs = options.reconnectDelayMs ?? reconnectDelayMs;
 	}
@@ -71,6 +73,7 @@ var ProtocolWebSocketTransportAdapter = class {
 		this.onReconnected = handler;
 	}
 	async open() {
+		const reconnecting = this.reconnectInFlight != null;
 		if (this.closed) throw new Error("Protocol WebSocket transport is closed.");
 		if (this.socket?.readyState === WEB_SOCKET_OPEN) return;
 		if (this.socket != null) {
@@ -99,6 +102,10 @@ var ProtocolWebSocketTransportAdapter = class {
 			socket.addEventListener("open", onOpen, { once: true });
 			socket.addEventListener("error", onError, { once: true });
 		});
+		if (!reconnecting) await this.onConnected?.({
+			kind: "initial",
+			attempt: 0
+		});
 	}
 	async send(command) {
 		return await this.sendCommand(command);
@@ -218,7 +225,7 @@ var ProtocolWebSocketTransportAdapter = class {
 	#scheduleReconnect(cause) {
 		if (this.closed || this.intentionalClose) return;
 		if (this.reconnectInFlight != null) return;
-		this.reconnectInFlight = this.#runReconnectLoop(cause).finally(() => {
+		this.reconnectInFlight = Promise.resolve().then(() => this.#runReconnectLoop(cause)).finally(() => {
 			this.reconnectInFlight = null;
 		});
 	}
@@ -226,11 +233,12 @@ var ProtocolWebSocketTransportAdapter = class {
 		let lastError = initialCause;
 		for (let attempt = 1; attempt <= this.maxReconnectAttempts; attempt += 1) {
 			if (this.closed || this.intentionalClose) return;
+			const delay = this.reconnectDelayMs(attempt);
 			this.onReconnect?.({
 				attempt,
-				cause: lastError
+				cause: lastError,
+				delayMs: delay
 			});
-			const delay = this.reconnectDelayMs(attempt);
 			if (delay > 0) await new Promise((resolve) => {
 				setTimeout(resolve, delay);
 			});
@@ -238,6 +246,10 @@ var ProtocolWebSocketTransportAdapter = class {
 			try {
 				await this.open();
 				if (this.onReconnected) await this.onReconnected();
+				await this.onConnected?.({
+					kind: "reconnected",
+					attempt
+				});
 				return;
 			} catch (error) {
 				lastError = error;

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

@@ -1 +1 @@
-{"version":3,"file":"websocket.js","names":["#detachSocket","#attachSocket","#handleUnexpectedDisconnect","#scheduleReconnect","#runReconnectLoop"],"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\";\nimport {\n  DEFAULT_MAX_RECONNECT_ATTEMPTS,\n  reconnectDelayMs,\n} from \"../../../utils/reconnect.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 {@link DEFAULT_MAX_RECONNECT_ATTEMPTS}.\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 * 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 =\n      options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT_ATTEMPTS;\n    this.onReconnect = options.onReconnect;\n    this.onReconnected = options.onReconnected;\n    this.reconnectDelayMs = options.reconnectDelayMs ?? reconnectDelayMs;\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":";;;;;AA8BA,MAAM,wBAAwB;AAC9B,MAAM,kBAAkB;AACxB,MAAM,oBAAoB;;;;;;;;;;;;;;AAgC1B,IAAa,oCAAb,MAA2E;CACzE;CAEA,QAAyB,IAAI,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,uBACH,QAAQ,wBAAA;AACV,OAAK,cAAc,QAAQ;AAC3B,OAAK,gBAAgB,QAAQ;AAC7B,OAAK,mBAAmB,QAAQ,oBAAoB;;;CAItD,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,SAAO,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,QAAQ,eACZ,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,MAAI,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,WAAO,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,MACE,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,MAAI,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,QAAQ,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,IAAI,mCACF,KAAK,sBACL,UACD,CACF"}
+{"version":3,"file":"websocket.js","names":["#detachSocket","#attachSocket","#handleUnexpectedDisconnect","#scheduleReconnect","#runReconnectLoop"],"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\";\nimport {\n  DEFAULT_MAX_RECONNECT_ATTEMPTS,\n  reconnectDelayMs,\n  type ReconnectInfo,\n} from \"../../../utils/reconnect.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 {@link DEFAULT_MAX_RECONNECT_ATTEMPTS}.\n   */\n  maxReconnectAttempts?: number;\n\n  /** Invoked before each reconnect attempt with its scheduled delay. */\n  onReconnect?: (options: ReconnectInfo) => void;\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 onConnected?: ProtocolWebSocketTransportOptions[\"onConnected\"];\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 =\n      options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT_ATTEMPTS;\n    this.onReconnect = options.onReconnect;\n    this.onConnected = options.onConnected;\n    this.onReconnected = options.onReconnected;\n    this.reconnectDelayMs = options.reconnectDelayMs ?? reconnectDelayMs;\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    const reconnecting = this.reconnectInFlight != null;\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    if (!reconnecting) {\n      await this.onConnected?.({ kind: \"initial\", attempt: 0 });\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 = Promise.resolve()\n      .then(() => this.#runReconnectLoop(cause))\n      .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      const delay = this.reconnectDelayMs(attempt);\n      this.onReconnect?.({ attempt, cause: lastError, delayMs: delay });\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        await this.onConnected?.({ kind: \"reconnected\", attempt });\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":";;;;;AA+BA,MAAM,wBAAwB;AAC9B,MAAM,kBAAkB;AACxB,MAAM,oBAAoB;;;;;;;;;;;;;;AA8B1B,IAAa,oCAAb,MAA2E;CACzE;CAEA,QAAyB,IAAI,YAAqB;CAElD;CAEA;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,uBACH,QAAQ,wBAAA;AACV,OAAK,cAAc,QAAQ;AAC3B,OAAK,cAAc,QAAQ;AAC3B,OAAK,gBAAgB,QAAQ;AAC7B,OAAK,mBAAmB,QAAQ,oBAAoB;;;CAItD,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,SAAO,oBACL,KAAK,OAAO,QACZ,KAAK,WACJ,OAAO,YAAY,GAAG,gBACxB;;;;;;;CAQH,iBAAiB,SAA2C;AAC1D,OAAK,gBAAgB;;CAGvB,MAAM,OAAsB;EAC1B,MAAM,eAAe,KAAK,qBAAqB;AAC/C,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,QAAQ,eACZ,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;AAEF,MAAI,CAAC,aACH,OAAM,KAAK,cAAc;GAAE,MAAM;GAAW,SAAS;GAAG,CAAC;;CAI7D,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,MAAI,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,WAAO,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,MACE,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,MAAI,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,QAAQ,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,QAAQ,SAAS,CACvC,WAAW,MAAA,iBAAuB,MAAM,CAAC,CACzC,cAAc;AACb,QAAK,oBAAoB;IACzB;;CAGN,OAAA,iBAAwB,cAAsC;EAC5D,IAAI,YAAqB;AAEzB,OAAK,IAAI,UAAU,GAAG,WAAW,KAAK,sBAAsB,WAAW,GAAG;AACxE,OAAI,KAAK,UAAU,KAAK,iBACtB;GAGF,MAAM,QAAQ,KAAK,iBAAiB,QAAQ;AAC5C,QAAK,cAAc;IAAE;IAAS,OAAO;IAAW,SAAS;IAAO,CAAC;AACjE,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,UAAM,KAAK,cAAc;KAAE,MAAM;KAAe;KAAS,CAAC;AAC1D;YACO,OAAO;AACd,gBAAY;;;AAIhB,OAAK,MAAM,MACT,IAAI,mCACF,KAAK,sBACL,UACD,CACF"}

package/dist/client/stream/types.d.cts

@@ -1,6 +1,7 @@
 import { IdleReconnectMode } from "../../utils/stream.cjs";
 import { AssembledMessage } from "./messages.cjs";
 import { AgentServerAdapter } from "./transport.cjs";
+import { ConnectedInfo, ReconnectInfo } from "../../utils/reconnect.cjs";
 import { AgentResult, Channel, Event, InputInjectParams, InputRespondParams, ListCheckpointsParams, ListCheckpointsResult, RunResult, RunStartParams, StateForkParams, StateForkResult, StateGetParams, StateGetResult, SubscribeParams } from "@langchain/protocol";
 
 //#region src/client/stream/types.d.ts
@@ -123,13 +124,10 @@ interface ThreadStreamOptions {
    * Defaults to {@link reconnectDelayMs}.
    */
   reconnectDelayMs?: (attempt: number) => number;
-  /**
-   * Built-in transports only: invoked before each reconnect attempt.
-   */
-  onReconnect?: (options: {
-    attempt: number;
-    cause: unknown;
-  }) => void;
+  /** Built-in transports only: invoked before each reconnect attempt. */
+  onReconnect?: (options: ReconnectInfo) => void;
+  /** Built-in transports only: invoked after every usable connection. */
+  onConnected?: (options: ConnectedInfo) => void | Promise<void>;
 }
 interface SessionOrderingState {
   lastSeenSeq?: number;

package/dist/client/stream/types.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/client/stream/types.ts"],"mappings":";;;;;;UAkCiB,sBAAA,SAA+B,cAAA;EAC9C,QAAA;EACA,iBAAA;AAAA;AAAA,KAGU,gBAAA,GAAmB,IAAA,CAAK,eAAA;AAAA,KAExB,oBAAA;EACV,MAAA;EACA,OAAA;EACA,QAAA;EACA,KAAA;EACA,MAAA;EACA,SAAA;EACA,KAAA;EACA,KAAA;EACA,WAAA;EACA,KAAA;AAAA;AAAA,KAGU,eAAA,kBAAiC,OAAA,IAC3C,QAAA,eAAuB,oBAAA,GACnB,OAAA,CAAQ,KAAA;EAAS,MAAA,EAAQ,oBAAA,CAAqB,QAAA;AAAA,KAC9C,QAAA,8BACE,OAAA,CAAQ,KAAA;EAAS,MAAA;AAAA;AAAA,KAGb,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;;;;;;;KAQN,eAAA,kBAAiC,OAAA,IAC3C,QAAA,wCAAgD,eAAA,CAAgB,QAAA;AAAA,KAEtD,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;;;;;;;;;KAUN,yBAAA;;;;;;;;;KAUA,qBAAA,GACR,yBAAA,GACA,kBAAA;;;;UAKa,mBAAA;EA7CmC;;;;;EAmDlD,WAAA;EAjD6B;AAG/B;;;;;;;;;;;;EA4DE,SAAA,GAAY,qBAAA;EA3Da;AAQ3B;;;EAwDE,iBAAA;EAvDA;;;;;;;EA+DA,KAAA,UAAe,KAAA;EA/DiC;;;;AAElD;;EAoEE,gBAAA,IAAoB,GAAA,aAAgB,SAAA;EApEkB;;;;;EA0EtD,oBAAA;EA1EsD;;;;;AAWxD;;;;;AAUA;;;;EAoEE,mBAAA,GAAsB,iBAAA;EA7DP;;;;EAkEf,gBAAA,IAAoB,OAAA;EA1BgB;;;EA8BpC,WAAA,IAAe,OAAA;IAAW,OAAA;IAAiB,KAAA;EAAA;AAAA;AAAA,UAG5B,oBAAA;EACf,WAAA;EACA,qBAAA;EACA,WAAA;AAAA;AAAA,UAGe,iBAAA,UACN,KAAA,UACD,aAAA,CAAc,MAAA;EAAA,SACb,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,mBAAA,SAA4B,aAAA,CAAc,gBAAA;EAAA,SAChD,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,WAAA;EACf,OAAA,CAAQ,MAAA,EAAQ,kBAAA,GAAqB,OAAA;EACrC,MAAA,CAAO,MAAA,EAAQ,iBAAA,GAAoB,OAAA;AAAA;AAAA,UAGpB,WAAA;EACf,GAAA,CAAI,MAAA,EAAQ,cAAA,GAAiB,OAAA,CAAQ,cAAA;EACrC,eAAA,CACE,MAAA,EAAQ,qBAAA,GACP,OAAA,CAAQ,qBAAA;EACX,IAAA,CAAK,MAAA,EAAQ,eAAA,GAAkB,OAAA,CAAQ,eAAA;AAAA;;AAxBzC;;UA8BiB,aAAA;EACf,GAAA;IA7BsB;;;;;IAmCpB,KAAA,CACE,MAAA,EAAQ,IAAA,CAAK,sBAAA,oBACZ,OAAA,CAAQ,SAAA;EAAA;EAEb,KAAA;IACE,OAAA,CAAQ,MAAA;MAAW,MAAA;IAAA,IAAoB,OAAA,CAAQ,WAAA;EAAA;EAEjD,KAAA,EAAO,WAAA;EACP,KAAA,EAAO,WAAA;AAAA;;;AArCT;;;;;;;;UAkDiB,gBAAA;EACf,WAAA;EACA,OAAA,EAAS,QAAA;EAnDA;EAqDT,SAAA;AAAA;;;;;AAhDF;;;;;;;;;;;;;;UAqEiB,eAAA,sBACP,aAAA,CAAc,CAAA,GAAI,WAAA,CAAY,CAAA;;;;AAjExC;;;;;;;;;;KAgFY,eAAA,MACV,CAAA,SAAU,WAAA,YAAuB,CAAA,GAAI,CAAA,SAAU,aAAA,YAAyB,CAAA,GAAI,CAAA;;;;;;;;;;;;;;;;KAiBlE,gBAAA,qBACU,MAAA,oBAA0B,MAAA,4CAEzB,WAAA,GAAc,eAAA,CACjC,eAAA,CAAgB,WAAA,CAAY,CAAA;EAAA,UAGpB,IAAA,WAAe,eAAA;AAAA"}
+{"version":3,"file":"types.d.cts","names":[],"sources":["../../../src/client/stream/types.ts"],"mappings":";;;;;;;UAoCiB,sBAAA,SAA+B,cAAA;EAC9C,QAAA;EACA,iBAAA;AAAA;AAAA,KAGU,gBAAA,GAAmB,IAAA,CAAK,eAAA;AAAA,KAExB,oBAAA;EACV,MAAA;EACA,OAAA;EACA,QAAA;EACA,KAAA;EACA,MAAA;EACA,SAAA;EACA,KAAA;EACA,KAAA;EACA,WAAA;EACA,KAAA;AAAA;AAAA,KAGU,eAAA,kBAAiC,OAAA,IAC3C,QAAA,eAAuB,oBAAA,GACnB,OAAA,CAAQ,KAAA;EAAS,MAAA,EAAQ,oBAAA,CAAqB,QAAA;AAAA,KAC9C,QAAA,8BACE,OAAA,CAAQ,KAAA;EAAS,MAAA;AAAA;AAAA,KAGb,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;;;;;;;KAQN,eAAA,kBAAiC,OAAA,IAC3C,QAAA,wCAAgD,eAAA,CAAgB,QAAA;AAAA,KAEtD,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;AApBlB;;;;;;;;AAAA,KA8BY,yBAAA;;;;;;;;;KAUA,qBAAA,GACR,yBAAA,GACA,kBAAA;;;;UAKa,mBAAA;EA7Cc;;;;;EAmD7B,WAAA;EAjD6B;;AAG/B;;;;;;;;;;;EA4DE,SAAA,GAAY,qBAAA;EA3Da;;AAQ3B;;EAwDE,iBAAA;EAxD2C;;;;;;;EAgE3C,KAAA,UAAe,KAAA;EA/Df;;;;;AAEF;EAoEE,gBAAA,IAAoB,GAAA,aAAgB,SAAA;EApEV;;;;;EA0E1B,oBAAA;EA1E2B;;;;;;AAW7B;;;;;AAUA;;;EAoEE,mBAAA,GAAsB,iBAAA;EAlEF;AAKtB;;;EAkEE,gBAAA,IAAoB,OAAA;EAjCL;EAmCf,WAAA,IAAe,OAAA,EAAS,aAAA;EAPF;EAStB,WAAA,IAAe,OAAA,EAAS,aAAA,YAAyB,OAAA;AAAA;AAAA,UAGlC,oBAAA;EACf,WAAA;EACA,qBAAA;EACA,WAAA;AAAA;AAAA,UAGe,iBAAA,UACN,KAAA,UACD,aAAA,CAAc,MAAA;EAAA,SACb,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,mBAAA,SAA4B,aAAA,CAAc,gBAAA;EAAA,SAChD,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,WAAA;EACf,OAAA,CAAQ,MAAA,EAAQ,kBAAA,GAAqB,OAAA;EACrC,MAAA,CAAO,MAAA,EAAQ,iBAAA,GAAoB,OAAA;AAAA;AAAA,UAGpB,WAAA;EACf,GAAA,CAAI,MAAA,EAAQ,cAAA,GAAiB,OAAA,CAAQ,cAAA;EACrC,eAAA,CACE,MAAA,EAAQ,qBAAA,GACP,OAAA,CAAQ,qBAAA;EACX,IAAA,CAAK,MAAA,EAAQ,eAAA,GAAkB,OAAA,CAAQ,eAAA;AAAA;;;AA9BzC;UAoCiB,aAAA;EACf,GAAA;IArCmC;;;;;IA2CjC,KAAA,CACE,MAAA,EAAQ,IAAA,CAAK,sBAAA,oBACZ,OAAA,CAAQ,SAAA;EAAA;EAEb,KAAA;IACE,OAAA,CAAQ,MAAA;MAAW,MAAA;IAAA,IAAoB,OAAA,CAAQ,WAAA;EAAA;EAEjD,KAAA,EAAO,WAAA;EACP,KAAA,EAAO,WAAA;AAAA;;;;;;;;;;;UAaQ,gBAAA;EACf,WAAA;EACA,OAAA,EAAS,QAAA;EApD0B;EAsDnC,SAAA;AAAA;;;;;;;;;;;;;;;AAhDF;;;;UAqEiB,eAAA,sBACP,aAAA,CAAc,CAAA,GAAI,WAAA,CAAY,CAAA;;;;;;;;;;;;;;KAe5B,eAAA,MACV,CAAA,SAAU,WAAA,YAAuB,CAAA,GAAI,CAAA,SAAU,aAAA,YAAyB,CAAA,GAAI,CAAA;;;;;;;;;;;;;;;;KAiBlE,gBAAA,qBACU,MAAA,oBAA0B,MAAA,4CAEzB,WAAA,GAAc,eAAA,CACjC,eAAA,CAAgB,WAAA,CAAY,CAAA;EAAA,UAGpB,IAAA,WAAe,eAAA;AAAA"}

package/dist/client/stream/types.d.ts

@@ -1,6 +1,7 @@
 import { IdleReconnectMode } from "../../utils/stream.js";
 import { AssembledMessage } from "./messages.js";
 import { AgentServerAdapter } from "./transport.js";
+import { ConnectedInfo, ReconnectInfo } from "../../utils/reconnect.js";
 import { AgentResult, Channel, Event, InputInjectParams, InputRespondParams, ListCheckpointsParams, ListCheckpointsResult, RunResult, RunStartParams, StateForkParams, StateForkResult, StateGetParams, StateGetResult, SubscribeParams } from "@langchain/protocol";
 
 //#region src/client/stream/types.d.ts
@@ -123,13 +124,10 @@ interface ThreadStreamOptions {
    * Defaults to {@link reconnectDelayMs}.
    */
   reconnectDelayMs?: (attempt: number) => number;
-  /**
-   * Built-in transports only: invoked before each reconnect attempt.
-   */
-  onReconnect?: (options: {
-    attempt: number;
-    cause: unknown;
-  }) => void;
+  /** Built-in transports only: invoked before each reconnect attempt. */
+  onReconnect?: (options: ReconnectInfo) => void;
+  /** Built-in transports only: invoked after every usable connection. */
+  onConnected?: (options: ConnectedInfo) => void | Promise<void>;
 }
 interface SessionOrderingState {
   lastSeenSeq?: number;

package/dist/client/stream/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/client/stream/types.ts"],"mappings":";;;;;;UAkCiB,sBAAA,SAA+B,cAAA;EAC9C,QAAA;EACA,iBAAA;AAAA;AAAA,KAGU,gBAAA,GAAmB,IAAA,CAAK,eAAA;AAAA,KAExB,oBAAA;EACV,MAAA;EACA,OAAA;EACA,QAAA;EACA,KAAA;EACA,MAAA;EACA,SAAA;EACA,KAAA;EACA,KAAA;EACA,WAAA;EACA,KAAA;AAAA;AAAA,KAGU,eAAA,kBAAiC,OAAA,IAC3C,QAAA,eAAuB,oBAAA,GACnB,OAAA,CAAQ,KAAA;EAAS,MAAA,EAAQ,oBAAA,CAAqB,QAAA;AAAA,KAC9C,QAAA,8BACE,OAAA,CAAQ,KAAA;EAAS,MAAA;AAAA;AAAA,KAGb,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;;;;;;;KAQN,eAAA,kBAAiC,OAAA,IAC3C,QAAA,wCAAgD,eAAA,CAAgB,QAAA;AAAA,KAEtD,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;;;;;;;;;KAUN,yBAAA;;;;;;;;;KAUA,qBAAA,GACR,yBAAA,GACA,kBAAA;;;;UAKa,mBAAA;EA7CmC;;;;;EAmDlD,WAAA;EAjD6B;AAG/B;;;;;;;;;;;;EA4DE,SAAA,GAAY,qBAAA;EA3Da;AAQ3B;;;EAwDE,iBAAA;EAvDA;;;;;;;EA+DA,KAAA,UAAe,KAAA;EA/DiC;;;;AAElD;;EAoEE,gBAAA,IAAoB,GAAA,aAAgB,SAAA;EApEkB;;;;;EA0EtD,oBAAA;EA1EsD;;;;;AAWxD;;;;;AAUA;;;;EAoEE,mBAAA,GAAsB,iBAAA;EA7DP;;;;EAkEf,gBAAA,IAAoB,OAAA;EA1BgB;;;EA8BpC,WAAA,IAAe,OAAA;IAAW,OAAA;IAAiB,KAAA;EAAA;AAAA;AAAA,UAG5B,oBAAA;EACf,WAAA;EACA,qBAAA;EACA,WAAA;AAAA;AAAA,UAGe,iBAAA,UACN,KAAA,UACD,aAAA,CAAc,MAAA;EAAA,SACb,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,mBAAA,SAA4B,aAAA,CAAc,gBAAA;EAAA,SAChD,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,WAAA;EACf,OAAA,CAAQ,MAAA,EAAQ,kBAAA,GAAqB,OAAA;EACrC,MAAA,CAAO,MAAA,EAAQ,iBAAA,GAAoB,OAAA;AAAA;AAAA,UAGpB,WAAA;EACf,GAAA,CAAI,MAAA,EAAQ,cAAA,GAAiB,OAAA,CAAQ,cAAA;EACrC,eAAA,CACE,MAAA,EAAQ,qBAAA,GACP,OAAA,CAAQ,qBAAA;EACX,IAAA,CAAK,MAAA,EAAQ,eAAA,GAAkB,OAAA,CAAQ,eAAA;AAAA;;AAxBzC;;UA8BiB,aAAA;EACf,GAAA;IA7BsB;;;;;IAmCpB,KAAA,CACE,MAAA,EAAQ,IAAA,CAAK,sBAAA,oBACZ,OAAA,CAAQ,SAAA;EAAA;EAEb,KAAA;IACE,OAAA,CAAQ,MAAA;MAAW,MAAA;IAAA,IAAoB,OAAA,CAAQ,WAAA;EAAA;EAEjD,KAAA,EAAO,WAAA;EACP,KAAA,EAAO,WAAA;AAAA;;;AArCT;;;;;;;;UAkDiB,gBAAA;EACf,WAAA;EACA,OAAA,EAAS,QAAA;EAnDA;EAqDT,SAAA;AAAA;;;;;AAhDF;;;;;;;;;;;;;;UAqEiB,eAAA,sBACP,aAAA,CAAc,CAAA,GAAI,WAAA,CAAY,CAAA;;;;AAjExC;;;;;;;;;;KAgFY,eAAA,MACV,CAAA,SAAU,WAAA,YAAuB,CAAA,GAAI,CAAA,SAAU,aAAA,YAAyB,CAAA,GAAI,CAAA;;;;;;;;;;;;;;;;KAiBlE,gBAAA,qBACU,MAAA,oBAA0B,MAAA,4CAEzB,WAAA,GAAc,eAAA,CACjC,eAAA,CAAgB,WAAA,CAAY,CAAA;EAAA,UAGpB,IAAA,WAAe,eAAA;AAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/client/stream/types.ts"],"mappings":";;;;;;;UAoCiB,sBAAA,SAA+B,cAAA;EAC9C,QAAA;EACA,iBAAA;AAAA;AAAA,KAGU,gBAAA,GAAmB,IAAA,CAAK,eAAA;AAAA,KAExB,oBAAA;EACV,MAAA;EACA,OAAA;EACA,QAAA;EACA,KAAA;EACA,MAAA;EACA,SAAA;EACA,KAAA;EACA,KAAA;EACA,WAAA;EACA,KAAA;AAAA;AAAA,KAGU,eAAA,kBAAiC,OAAA,IAC3C,QAAA,eAAuB,oBAAA,GACnB,OAAA,CAAQ,KAAA;EAAS,MAAA,EAAQ,oBAAA,CAAqB,QAAA;AAAA,KAC9C,QAAA,8BACE,OAAA,CAAQ,KAAA;EAAS,MAAA;AAAA;AAAA,KAGb,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;;;;;;;KAQN,eAAA,kBAAiC,OAAA,IAC3C,QAAA,wCAAgD,eAAA,CAAgB,QAAA;AAAA,KAEtD,gBAAA,4BAA4C,OAAA,MACtD,eAAA,CAAgB,SAAA;AApBlB;;;;;;;;AAAA,KA8BY,yBAAA;;;;;;;;;KAUA,qBAAA,GACR,yBAAA,GACA,kBAAA;;;;UAKa,mBAAA;EA7Cc;;;;;EAmD7B,WAAA;EAjD6B;;AAG/B;;;;;;;;;;;EA4DE,SAAA,GAAY,qBAAA;EA3Da;;AAQ3B;;EAwDE,iBAAA;EAxD2C;;;;;;;EAgE3C,KAAA,UAAe,KAAA;EA/Df;;;;;AAEF;EAoEE,gBAAA,IAAoB,GAAA,aAAgB,SAAA;EApEV;;;;;EA0E1B,oBAAA;EA1E2B;;;;;;AAW7B;;;;;AAUA;;;EAoEE,mBAAA,GAAsB,iBAAA;EAlEF;AAKtB;;;EAkEE,gBAAA,IAAoB,OAAA;EAjCL;EAmCf,WAAA,IAAe,OAAA,EAAS,aAAA;EAPF;EAStB,WAAA,IAAe,OAAA,EAAS,aAAA,YAAyB,OAAA;AAAA;AAAA,UAGlC,oBAAA;EACf,WAAA;EACA,qBAAA;EACA,WAAA;AAAA;AAAA,UAGe,iBAAA,UACN,KAAA,UACD,aAAA,CAAc,MAAA;EAAA,SACb,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,mBAAA,SAA4B,aAAA,CAAc,gBAAA;EAAA,SAChD,cAAA;EAAA,SACA,MAAA,EAAQ,eAAA;EACjB,WAAA,IAAe,OAAA;AAAA;AAAA,UAGA,WAAA;EACf,OAAA,CAAQ,MAAA,EAAQ,kBAAA,GAAqB,OAAA;EACrC,MAAA,CAAO,MAAA,EAAQ,iBAAA,GAAoB,OAAA;AAAA;AAAA,UAGpB,WAAA;EACf,GAAA,CAAI,MAAA,EAAQ,cAAA,GAAiB,OAAA,CAAQ,cAAA;EACrC,eAAA,CACE,MAAA,EAAQ,qBAAA,GACP,OAAA,CAAQ,qBAAA;EACX,IAAA,CAAK,MAAA,EAAQ,eAAA,GAAkB,OAAA,CAAQ,eAAA;AAAA;;;AA9BzC;UAoCiB,aAAA;EACf,GAAA;IArCmC;;;;;IA2CjC,KAAA,CACE,MAAA,EAAQ,IAAA,CAAK,sBAAA,oBACZ,OAAA,CAAQ,SAAA;EAAA;EAEb,KAAA;IACE,OAAA,CAAQ,MAAA;MAAW,MAAA;IAAA,IAAoB,OAAA,CAAQ,WAAA;EAAA;EAEjD,KAAA,EAAO,WAAA;EACP,KAAA,EAAO,WAAA;AAAA;;;;;;;;;;;UAaQ,gBAAA;EACf,WAAA;EACA,OAAA,EAAS,QAAA;EApD0B;EAsDnC,SAAA;AAAA;;;;;;;;;;;;;;;AAhDF;;;;UAqEiB,eAAA,sBACP,aAAA,CAAc,CAAA,GAAI,WAAA,CAAY,CAAA;;;;;;;;;;;;;;KAe5B,eAAA,MACV,CAAA,SAAU,WAAA,YAAuB,CAAA,GAAI,CAAA,SAAU,aAAA,YAAyB,CAAA,GAAI,CAAA;;;;;;;;;;;;;;;;KAiBlE,gBAAA,qBACU,MAAA,oBAA0B,MAAA,4CAEzB,WAAA,GAAc,eAAA,CACjC,eAAA,CAAgB,WAAA,CAAY,CAAA;EAAA,UAGpB,IAAA,WAAe,eAAA;AAAA"}

package/dist/client/threads/index.cjs

@@ -271,7 +271,8 @@ var ThreadsClient = class extends require_base.BaseClient {
 				onRequest: this.onRequest,
 				maxReconnectAttempts,
 				reconnectDelayMs: options.reconnectDelayMs,
-				onReconnect: options.onReconnect
+				onReconnect: options.onReconnect,
+				onConnected: options.onConnected
 			};
 			transport = transportKind === "websocket" ? new require_websocket.ProtocolWebSocketTransportAdapter({
 				...commonOpts,

package/dist/client/threads/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["BaseClient","ProtocolWebSocketTransportAdapter","ProtocolSseTransportAdapter","ThreadStream"],"sources":["../../../src/client/threads/index.ts"],"sourcesContent":["import { v7 as uuidv7 } from \"@langchain/core/utils/uuid\";\n\nimport {\n  Checkpoint,\n  Config,\n  DefaultValues,\n  Metadata,\n  SortOrder,\n  Thread,\n  ThreadSelectField,\n  ThreadSortBy,\n  ThreadState,\n  ThreadStatus,\n  ThreadValuesFilter,\n} from \"../../schema.js\";\nimport type { Command, OnConflictBehavior, StreamEvent } from \"../../types.js\";\nimport type { ThreadStreamMode } from \"../../types.stream.js\";\nimport { BaseClient } from \"../base.js\";\nimport { ThreadStream } from \"../stream/index.js\";\nimport type {\n  ThreadStreamOptions,\n  ThreadStreamTransportKind,\n} from \"../stream/types.js\";\nimport { DEFAULT_MAX_RECONNECT_ATTEMPTS } from \"../../utils/reconnect.js\";\nimport { ProtocolSseTransportAdapter } from \"../stream/transport/http.js\";\nimport { ProtocolWebSocketTransportAdapter } from \"../stream/transport/websocket.js\";\nimport type { TransportAdapter } from \"../stream/transport.js\";\n\nexport class ThreadsClient<\n  TStateType = DefaultValues,\n  TUpdateType = TStateType,\n> extends BaseClient {\n  /**\n   * Get a thread by ID.\n   *\n   * @param threadId ID of the thread.\n   * @returns The thread.\n   */\n  async get<ValuesType = TStateType>(\n    threadId: string,\n    options?: { signal?: AbortSignal; include?: string[] }\n  ): Promise<Thread<ValuesType>> {\n    return this.fetch<Thread<ValuesType>>(`/threads/${threadId}`, {\n      params: {\n        include: options?.include ?? undefined,\n      },\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Create a new thread.\n   *\n   * @param payload Payload for creating a thread.\n   * @returns The created thread.\n   */\n  async create(payload?: {\n    metadata?: Metadata;\n    threadId?: string;\n    ifExists?: OnConflictBehavior;\n    graphId?: string;\n    supersteps?: Array<{\n      updates: Array<{ values: unknown; command?: Command; asNode: string }>;\n    }>;\n    ttl?: number | { ttl: number; strategy?: \"delete\" };\n    signal?: AbortSignal;\n  }): Promise<Thread<TStateType>> {\n    const ttlPayload =\n      typeof payload?.ttl === \"number\"\n        ? { ttl: payload.ttl, strategy: \"delete\" as const }\n        : payload?.ttl;\n\n    return this.fetch<Thread<TStateType>>(`/threads`, {\n      method: \"POST\",\n      json: {\n        metadata: {\n          ...payload?.metadata,\n          graph_id: payload?.graphId,\n        },\n        thread_id: payload?.threadId,\n        if_exists: payload?.ifExists,\n        supersteps: payload?.supersteps?.map((s) => ({\n          updates: s.updates.map((u) => ({\n            values: u.values,\n            command: u.command,\n            as_node: u.asNode,\n          })),\n        })),\n        ttl: ttlPayload,\n      },\n      signal: payload?.signal,\n    });\n  }\n\n  /**\n   * Copy an existing thread\n   * @param threadId ID of the thread to be copied\n   * @returns Newly copied thread\n   */\n  async copy(\n    threadId: string,\n    options?: { signal?: AbortSignal }\n  ): Promise<Thread<TStateType>> {\n    return this.fetch<Thread<TStateType>>(`/threads/${threadId}/copy`, {\n      method: \"POST\",\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Update a thread.\n   *\n   * @param threadId ID of the thread.\n   * @param payload Payload for updating the thread.\n   * @returns The updated thread.\n   */\n  async update(\n    threadId: string,\n    payload?: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal?: false;\n      signal?: AbortSignal;\n    }\n  ): Promise<Thread>;\n  async update(\n    threadId: string,\n    payload: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal: true;\n      signal?: AbortSignal;\n    }\n  ): Promise<void>;\n  async update(\n    threadId: string,\n    payload: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal: boolean;\n      signal?: AbortSignal;\n    }\n  ): Promise<Thread | void>;\n  async update(\n    threadId: string,\n    payload?: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal?: boolean;\n      signal?: AbortSignal;\n    }\n  ): Promise<Thread | void> {\n    const ttlPayload =\n      typeof payload?.ttl === \"number\"\n        ? { ttl: payload.ttl, strategy: \"delete\" as const }\n        : payload?.ttl;\n\n    return this.fetch<Thread | void>(`/threads/${threadId}`, {\n      method: \"PATCH\",\n      headers: payload?.returnMinimal\n        ? { Prefer: \"return=minimal\" }\n        : undefined,\n      json: { metadata: payload?.metadata, ttl: ttlPayload },\n      signal: payload?.signal,\n    });\n  }\n\n  /**\n   * Delete a thread.\n   *\n   * @param threadId ID of the thread.\n   */\n  async delete(\n    threadId: string,\n    options?: { signal?: AbortSignal }\n  ): Promise<void> {\n    return this.fetch<void>(`/threads/${threadId}`, {\n      method: \"DELETE\",\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Prune threads by ID. The 'delete' strategy removes threads entirely.\n   * The 'keep_latest' strategy prunes old checkpoints but keeps threads\n   * and their latest state.\n   *\n   * @param threadIds List of thread IDs to prune.\n   * @param options Additional options for pruning.\n   * @param options.strategy The prune strategy. Defaults to 'delete'.\n   * @param options.signal Signal to abort the request.\n   * @returns An object containing `pruned_count`.\n   */\n  async prune(\n    threadIds: string[],\n    options?: {\n      strategy?: \"delete\" | \"keep_latest\";\n      signal?: AbortSignal;\n    }\n  ): Promise<{ pruned_count: number }> {\n    return this.fetch<{ pruned_count: number }>(\"/threads/prune\", {\n      method: \"POST\",\n      json: {\n        thread_ids: threadIds,\n        strategy: options?.strategy ?? \"delete\",\n      },\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * List threads\n   *\n   * @param query Query options\n   * @returns List of threads\n   */\n  async search<ValuesType = TStateType>(query?: {\n    metadata?: Metadata;\n    ids?: string[];\n    limit?: number;\n    offset?: number;\n    status?: ThreadStatus;\n    sortBy?: ThreadSortBy;\n    sortOrder?: SortOrder;\n    select?: ThreadSelectField[];\n    values?: ThreadValuesFilter;\n    extract?: Record<string, string>;\n    signal?: AbortSignal;\n  }): Promise<Thread<ValuesType>[]> {\n    return this.fetch<Thread<ValuesType>[]>(\"/threads/search\", {\n      method: \"POST\",\n      json: {\n        metadata: query?.metadata ?? undefined,\n        ids: query?.ids ?? undefined,\n        limit: query?.limit ?? 10,\n        offset: query?.offset ?? 0,\n        status: query?.status,\n        sort_by: query?.sortBy,\n        sort_order: query?.sortOrder,\n        select: query?.select ?? undefined,\n        values: query?.values ?? undefined,\n        extract: query?.extract ?? undefined,\n      },\n      signal: query?.signal,\n    });\n  }\n\n  /**\n   * Count threads matching filters.\n   *\n   * @param query.metadata Thread metadata to filter on.\n   * @param query.values State values to filter on.\n   * @param query.status Thread status to filter on.\n   * @returns Number of threads matching the criteria.\n   */\n  async count<ValuesType = TStateType>(query?: {\n    metadata?: Metadata;\n    values?: ValuesType;\n    status?: ThreadStatus;\n    signal?: AbortSignal;\n  }): Promise<number> {\n    return this.fetch<number>(`/threads/count`, {\n      method: \"POST\",\n      json: {\n        metadata: query?.metadata ?? undefined,\n        values: query?.values ?? undefined,\n        status: query?.status ?? undefined,\n      },\n      signal: query?.signal,\n    });\n  }\n\n  /**\n   * Get state for a thread.\n   *\n   * @param threadId ID of the thread.\n   * @returns Thread state.\n   */\n  async getState<ValuesType = TStateType>(\n    threadId: string,\n    checkpoint?: Checkpoint | string,\n    options?: { subgraphs?: boolean; signal?: AbortSignal }\n  ): Promise<ThreadState<ValuesType>> {\n    if (checkpoint != null) {\n      if (typeof checkpoint !== \"string\") {\n        return this.fetch<ThreadState<ValuesType>>(\n          `/threads/${threadId}/state/checkpoint`,\n          {\n            method: \"POST\",\n            json: { checkpoint, subgraphs: options?.subgraphs },\n            signal: options?.signal,\n          }\n        );\n      }\n\n      // deprecated\n      return this.fetch<ThreadState<ValuesType>>(\n        `/threads/${threadId}/state/${checkpoint}`,\n        { params: { subgraphs: options?.subgraphs }, signal: options?.signal }\n      );\n    }\n\n    return this.fetch<ThreadState<ValuesType>>(`/threads/${threadId}/state`, {\n      params: { subgraphs: options?.subgraphs },\n      signal: options?.signal,\n      // Coalesce concurrent identical reads (e.g. two controllers\n      // hydrating the same thread on reconnect). Skipped automatically\n      // when a caller supplies its own `signal`.\n      dedupe: true,\n    });\n  }\n\n  /**\n   * Add state to a thread.\n   *\n   * @param threadId The ID of the thread.\n   * @returns\n   */\n  async updateState<ValuesType = TUpdateType>(\n    threadId: string,\n    options: {\n      values: ValuesType;\n      checkpoint?: Checkpoint;\n      checkpointId?: string;\n      asNode?: string;\n      signal?: AbortSignal;\n    }\n  ): Promise<Pick<Config, \"configurable\">> {\n    return this.fetch<Pick<Config, \"configurable\">>(\n      `/threads/${threadId}/state`,\n      {\n        method: \"POST\",\n        json: {\n          values: options.values,\n          checkpoint: options.checkpoint,\n          checkpoint_id: options.checkpointId,\n          as_node: options?.asNode,\n        },\n        signal: options?.signal,\n      }\n    );\n  }\n\n  /**\n   * Patch the metadata of a thread.\n   *\n   * @param threadIdOrConfig Thread ID or config to patch the state of.\n   * @param metadata Metadata to patch the state with.\n   */\n  async patchState(\n    threadIdOrConfig: string | Config,\n    metadata: Metadata,\n    options?: { signal?: AbortSignal }\n  ): Promise<void> {\n    let threadId: string;\n\n    if (typeof threadIdOrConfig !== \"string\") {\n      if (typeof threadIdOrConfig.configurable?.thread_id !== \"string\") {\n        throw new Error(\n          \"Thread ID is required when updating state with a config.\"\n        );\n      }\n      threadId = threadIdOrConfig.configurable.thread_id;\n    } else {\n      threadId = threadIdOrConfig;\n    }\n\n    return this.fetch<void>(`/threads/${threadId}/state`, {\n      method: \"PATCH\",\n      json: { metadata },\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Get all past states for a thread.\n   *\n   * @param threadId ID of the thread.\n   * @param options Additional options.\n   * @returns List of thread states.\n   */\n  async getHistory<ValuesType = TStateType>(\n    threadId: string,\n    options?: {\n      limit?: number;\n      before?: Config;\n      checkpoint?: Partial<Omit<Checkpoint, \"thread_id\">>;\n      metadata?: Metadata;\n      signal?: AbortSignal;\n    }\n  ): Promise<ThreadState<ValuesType>[]> {\n    return this.fetch<ThreadState<ValuesType>[]>(\n      `/threads/${threadId}/history`,\n      {\n        method: \"POST\",\n        json: {\n          limit: options?.limit ?? 10,\n          before: options?.before,\n          metadata: options?.metadata,\n          checkpoint: options?.checkpoint,\n        },\n        signal: options?.signal,\n        // `getHistory` is a read despite using POST — coalesce the\n        // hydrate-time discovery seed when two controllers reconnect to\n        // the same thread concurrently. Skipped when `signal` is set.\n        dedupe: true,\n      }\n    );\n  }\n\n  async *joinStream(\n    threadId: string,\n    options?: {\n      lastEventId?: string;\n      streamMode?: ThreadStreamMode | ThreadStreamMode[];\n      signal?: AbortSignal;\n    }\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  ): AsyncGenerator<{ id?: string; event: StreamEvent; data: any }> {\n    yield* this.streamWithRetry({\n      endpoint: `/threads/${threadId}/stream`,\n      method: \"GET\",\n      signal: options?.signal,\n      headers: options?.lastEventId\n        ? { \"Last-Event-ID\": options.lastEventId }\n        : undefined,\n      params: options?.streamMode\n        ? { stream_mode: options.streamMode }\n        : undefined,\n    });\n  }\n\n  /**\n   * Open a protocol stream over the thread-centric v2 protocol.\n   *\n   * Returns a {@link ThreadStream} with lazy getters\n   * (`.messages`, `.values`, `.toolCalls`, `.subgraphs`, `.subagents`,\n   * `.output`) and `thread.run.start({ input, ... })` for starting runs.\n   * Mirrors the in-process `graph.streamEvents(..., { version: \"v3\" })` API.\n   *\n   * The thread is bound to `options.assistantId` for its lifetime.\n   * The wire transport defaults to SSE; pass `transport: \"websocket\"`\n   * in options (or configure `streamProtocol: \"v2-websocket\"` on the\n   * client) to use a WebSocket instead.\n   *\n   * @example New thread (UUID generated client-side)\n   * ```ts\n   * const thread = client.threads.stream({ assistantId: \"my-agent\" });\n   * ```\n   *\n   * @example Attach to an existing thread\n   * ```ts\n   * const thread = client.threads.stream(threadId, { assistantId: \"my-agent\" });\n   * ```\n   *\n   * @example WebSocket transport\n   * ```ts\n   * const thread = client.threads.stream({\n   *   assistantId: \"my-agent\",\n   *   transport: \"websocket\",\n   * });\n   * ```\n   */\n  stream<TExtensions extends Record<string, unknown> = Record<string, unknown>>(\n    options: ThreadStreamOptions\n  ): ThreadStream<TExtensions>;\n  stream<TExtensions extends Record<string, unknown> = Record<string, unknown>>(\n    threadId: string,\n    options: ThreadStreamOptions\n  ): ThreadStream<TExtensions>;\n  stream<TExtensions extends Record<string, unknown> = Record<string, unknown>>(\n    threadIdOrOptions: string | ThreadStreamOptions,\n    maybeOptions?: ThreadStreamOptions\n  ): ThreadStream<TExtensions> {\n    const { threadId, options } =\n      typeof threadIdOrOptions === \"string\"\n        ? {\n            threadId: threadIdOrOptions,\n            options: maybeOptions as ThreadStreamOptions,\n          }\n        : // A nullish first arg is the generate-thread-id form, not the\n          // options-only overload. Without this branch, a runtime\n          // `stream(null, opts)` binds `null` to `options` and the\n          // `options.fetch` read below throws. Use the second arg as options.\n          // (Cast: the declared type excludes null, but callers can reach\n          // here at runtime, e.g. an in-flight reconnect after the id clears.)\n          (threadIdOrOptions as ThreadStreamOptions | null | undefined) == null\n          ? {\n              threadId: uuidv7(),\n              options: maybeOptions as ThreadStreamOptions,\n            }\n          : { threadId: uuidv7(), options: threadIdOrOptions };\n\n    // `transport` accepts either a preset string (`\"sse\"` / `\"websocket\"`)\n    // or a custom {@link AgentServerAdapter}. A custom adapter replaces\n    // the built-in factories entirely — this is the seam that lets users\n    // point `useStream` at any agent server (including the thin wrappers\n    // produced by `HttpAgentServerAdapter`).\n    // When callers supply `fetch`, use it verbatim (tests, auth shims). Otherwise\n    // route protocol HTTP and media URL fetches through AsyncCaller like REST.\n    const userFetch = options.fetch;\n    const protocolFetch =\n      userFetch ?? this.asyncCaller.fetch.bind(this.asyncCaller);\n\n    let transport: TransportAdapter;\n    if (options.transport != null && typeof options.transport !== \"string\") {\n      transport = options.transport;\n      // Bind (or re-bind) the caller-supplied adapter to this thread so a\n      // single instance can follow lazy creation (the first `submit()` on a\n      // `threadId: null` controller mints the id here) and thread switches.\n      // No-op for adapters that bake `threadId` at construction.\n      transport.setThreadId?.(threadId);\n    } else {\n      const transportKind: ThreadStreamTransportKind =\n        options.transport ??\n        (this.streamProtocol === \"v2-websocket\" ? \"websocket\" : \"sse\");\n      const maxReconnectAttempts =\n        options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT_ATTEMPTS;\n\n      /**\n       * Common options for both transports.\n       */\n      const commonOpts = {\n        apiUrl: this.apiUrl,\n        threadId,\n        defaultHeaders: this.defaultHeaders,\n        onRequest: this.onRequest,\n        maxReconnectAttempts,\n        reconnectDelayMs: options.reconnectDelayMs,\n        onReconnect: options.onReconnect,\n      };\n\n      transport =\n        transportKind === \"websocket\"\n          ? new ProtocolWebSocketTransportAdapter({\n              ...commonOpts,\n              webSocketFactory: options.webSocketFactory,\n            })\n          : new ProtocolSseTransportAdapter({\n              ...commonOpts,\n              idleReconnect: options.streamIdleReconnect,\n              fetch: userFetch,\n              asyncCaller: userFetch ? undefined : this.asyncCaller,\n            });\n    }\n\n    return new ThreadStream<TExtensions>(transport, {\n      ...options,\n      fetch: protocolFetch,\n    });\n  }\n}\n"],"mappings":";;;;;;;AA4BA,IAAa,gBAAb,cAGUA,aAAAA,WAAW;;;;;;;CAOnB,MAAM,IACJ,UACA,SAC6B;AAC7B,SAAO,KAAK,MAA0B,YAAY,YAAY;GAC5D,QAAQ,EACN,SAAS,SAAS,WAAW,KAAA,GAC9B;GACD,QAAQ,SAAS;GAClB,CAAC;;;;;;;;CASJ,MAAM,OAAO,SAUmB;EAC9B,MAAM,aACJ,OAAO,SAAS,QAAQ,WACpB;GAAE,KAAK,QAAQ;GAAK,UAAU;GAAmB,GACjD,SAAS;AAEf,SAAO,KAAK,MAA0B,YAAY;GAChD,QAAQ;GACR,MAAM;IACJ,UAAU;KACR,GAAG,SAAS;KACZ,UAAU,SAAS;KACpB;IACD,WAAW,SAAS;IACpB,WAAW,SAAS;IACpB,YAAY,SAAS,YAAY,KAAK,OAAO,EAC3C,SAAS,EAAE,QAAQ,KAAK,OAAO;KAC7B,QAAQ,EAAE;KACV,SAAS,EAAE;KACX,SAAS,EAAE;KACZ,EAAE,EACJ,EAAE;IACH,KAAK;IACN;GACD,QAAQ,SAAS;GAClB,CAAC;;;;;;;CAQJ,MAAM,KACJ,UACA,SAC6B;AAC7B,SAAO,KAAK,MAA0B,YAAY,SAAS,QAAQ;GACjE,QAAQ;GACR,QAAQ,SAAS;GAClB,CAAC;;CAqCJ,MAAM,OACJ,UACA,SAMwB;EACxB,MAAM,aACJ,OAAO,SAAS,QAAQ,WACpB;GAAE,KAAK,QAAQ;GAAK,UAAU;GAAmB,GACjD,SAAS;AAEf,SAAO,KAAK,MAAqB,YAAY,YAAY;GACvD,QAAQ;GACR,SAAS,SAAS,gBACd,EAAE,QAAQ,kBAAkB,GAC5B,KAAA;GACJ,MAAM;IAAE,UAAU,SAAS;IAAU,KAAK;IAAY;GACtD,QAAQ,SAAS;GAClB,CAAC;;;;;;;CAQJ,MAAM,OACJ,UACA,SACe;AACf,SAAO,KAAK,MAAY,YAAY,YAAY;GAC9C,QAAQ;GACR,QAAQ,SAAS;GAClB,CAAC;;;;;;;;;;;;;CAcJ,MAAM,MACJ,WACA,SAImC;AACnC,SAAO,KAAK,MAAgC,kBAAkB;GAC5D,QAAQ;GACR,MAAM;IACJ,YAAY;IACZ,UAAU,SAAS,YAAY;IAChC;GACD,QAAQ,SAAS;GAClB,CAAC;;;;;;;;CASJ,MAAM,OAAgC,OAYJ;AAChC,SAAO,KAAK,MAA4B,mBAAmB;GACzD,QAAQ;GACR,MAAM;IACJ,UAAU,OAAO,YAAY,KAAA;IAC7B,KAAK,OAAO,OAAO,KAAA;IACnB,OAAO,OAAO,SAAS;IACvB,QAAQ,OAAO,UAAU;IACzB,QAAQ,OAAO;IACf,SAAS,OAAO;IAChB,YAAY,OAAO;IACnB,QAAQ,OAAO,UAAU,KAAA;IACzB,QAAQ,OAAO,UAAU,KAAA;IACzB,SAAS,OAAO,WAAW,KAAA;IAC5B;GACD,QAAQ,OAAO;GAChB,CAAC;;;;;;;;;;CAWJ,MAAM,MAA+B,OAKjB;AAClB,SAAO,KAAK,MAAc,kBAAkB;GAC1C,QAAQ;GACR,MAAM;IACJ,UAAU,OAAO,YAAY,KAAA;IAC7B,QAAQ,OAAO,UAAU,KAAA;IACzB,QAAQ,OAAO,UAAU,KAAA;IAC1B;GACD,QAAQ,OAAO;GAChB,CAAC;;;;;;;;CASJ,MAAM,SACJ,UACA,YACA,SACkC;AAClC,MAAI,cAAc,MAAM;AACtB,OAAI,OAAO,eAAe,SACxB,QAAO,KAAK,MACV,YAAY,SAAS,oBACrB;IACE,QAAQ;IACR,MAAM;KAAE;KAAY,WAAW,SAAS;KAAW;IACnD,QAAQ,SAAS;IAClB,CACF;AAIH,UAAO,KAAK,MACV,YAAY,SAAS,SAAS,cAC9B;IAAE,QAAQ,EAAE,WAAW,SAAS,WAAW;IAAE,QAAQ,SAAS;IAAQ,CACvE;;AAGH,SAAO,KAAK,MAA+B,YAAY,SAAS,SAAS;GACvE,QAAQ,EAAE,WAAW,SAAS,WAAW;GACzC,QAAQ,SAAS;GAIjB,QAAQ;GACT,CAAC;;;;;;;;CASJ,MAAM,YACJ,UACA,SAOuC;AACvC,SAAO,KAAK,MACV,YAAY,SAAS,SACrB;GACE,QAAQ;GACR,MAAM;IACJ,QAAQ,QAAQ;IAChB,YAAY,QAAQ;IACpB,eAAe,QAAQ;IACvB,SAAS,SAAS;IACnB;GACD,QAAQ,SAAS;GAClB,CACF;;;;;;;;CASH,MAAM,WACJ,kBACA,UACA,SACe;EACf,IAAI;AAEJ,MAAI,OAAO,qBAAqB,UAAU;AACxC,OAAI,OAAO,iBAAiB,cAAc,cAAc,SACtD,OAAM,IAAI,MACR,2DACD;AAEH,cAAW,iBAAiB,aAAa;QAEzC,YAAW;AAGb,SAAO,KAAK,MAAY,YAAY,SAAS,SAAS;GACpD,QAAQ;GACR,MAAM,EAAE,UAAU;GAClB,QAAQ,SAAS;GAClB,CAAC;;;;;;;;;CAUJ,MAAM,WACJ,UACA,SAOoC;AACpC,SAAO,KAAK,MACV,YAAY,SAAS,WACrB;GACE,QAAQ;GACR,MAAM;IACJ,OAAO,SAAS,SAAS;IACzB,QAAQ,SAAS;IACjB,UAAU,SAAS;IACnB,YAAY,SAAS;IACtB;GACD,QAAQ,SAAS;GAIjB,QAAQ;GACT,CACF;;CAGH,OAAO,WACL,UACA,SAMgE;AAChE,SAAO,KAAK,gBAAgB;GAC1B,UAAU,YAAY,SAAS;GAC/B,QAAQ;GACR,QAAQ,SAAS;GACjB,SAAS,SAAS,cACd,EAAE,iBAAiB,QAAQ,aAAa,GACxC,KAAA;GACJ,QAAQ,SAAS,aACb,EAAE,aAAa,QAAQ,YAAY,GACnC,KAAA;GACL,CAAC;;CAyCJ,OACE,mBACA,cAC2B;EAC3B,MAAM,EAAE,UAAU,YAChB,OAAO,sBAAsB,WACzB;GACE,UAAU;GACV,SAAS;GACV,GAOA,qBAAgE,OAC/D;GACE,WAAA,GAAA,2BAAA,KAAkB;GAClB,SAAS;GACV,GACD;GAAE,WAAA,GAAA,2BAAA,KAAkB;GAAE,SAAS;GAAmB;EAS1D,MAAM,YAAY,QAAQ;EAC1B,MAAM,gBACJ,aAAa,KAAK,YAAY,MAAM,KAAK,KAAK,YAAY;EAE5D,IAAI;AACJ,MAAI,QAAQ,aAAa,QAAQ,OAAO,QAAQ,cAAc,UAAU;AACtE,eAAY,QAAQ;AAKpB,aAAU,cAAc,SAAS;SAC5B;GACL,MAAM,gBACJ,QAAQ,cACP,KAAK,mBAAmB,iBAAiB,cAAc;GAC1D,MAAM,uBACJ,QAAQ,wBAAA;;;;GAKV,MAAM,aAAa;IACjB,QAAQ,KAAK;IACb;IACA,gBAAgB,KAAK;IACrB,WAAW,KAAK;IAChB;IACA,kBAAkB,QAAQ;IAC1B,aAAa,QAAQ;IACtB;AAED,eACE,kBAAkB,cACd,IAAIC,kBAAAA,kCAAkC;IACpC,GAAG;IACH,kBAAkB,QAAQ;IAC3B,CAAC,GACF,IAAIC,aAAAA,4BAA4B;IAC9B,GAAG;IACH,eAAe,QAAQ;IACvB,OAAO;IACP,aAAa,YAAY,KAAA,IAAY,KAAK;IAC3C,CAAC;;AAGV,SAAO,IAAIC,cAAAA,aAA0B,WAAW;GAC9C,GAAG;GACH,OAAO;GACR,CAAC"}
+{"version":3,"file":"index.cjs","names":["BaseClient","ProtocolWebSocketTransportAdapter","ProtocolSseTransportAdapter","ThreadStream"],"sources":["../../../src/client/threads/index.ts"],"sourcesContent":["import { v7 as uuidv7 } from \"@langchain/core/utils/uuid\";\n\nimport {\n  Checkpoint,\n  Config,\n  DefaultValues,\n  Metadata,\n  SortOrder,\n  Thread,\n  ThreadSelectField,\n  ThreadSortBy,\n  ThreadState,\n  ThreadStatus,\n  ThreadValuesFilter,\n} from \"../../schema.js\";\nimport type { Command, OnConflictBehavior, StreamEvent } from \"../../types.js\";\nimport type { ThreadStreamMode } from \"../../types.stream.js\";\nimport { BaseClient } from \"../base.js\";\nimport { ThreadStream } from \"../stream/index.js\";\nimport type {\n  ThreadStreamOptions,\n  ThreadStreamTransportKind,\n} from \"../stream/types.js\";\nimport { DEFAULT_MAX_RECONNECT_ATTEMPTS } from \"../../utils/reconnect.js\";\nimport { ProtocolSseTransportAdapter } from \"../stream/transport/http.js\";\nimport { ProtocolWebSocketTransportAdapter } from \"../stream/transport/websocket.js\";\nimport type { TransportAdapter } from \"../stream/transport.js\";\n\nexport class ThreadsClient<\n  TStateType = DefaultValues,\n  TUpdateType = TStateType,\n> extends BaseClient {\n  /**\n   * Get a thread by ID.\n   *\n   * @param threadId ID of the thread.\n   * @returns The thread.\n   */\n  async get<ValuesType = TStateType>(\n    threadId: string,\n    options?: { signal?: AbortSignal; include?: string[] }\n  ): Promise<Thread<ValuesType>> {\n    return this.fetch<Thread<ValuesType>>(`/threads/${threadId}`, {\n      params: {\n        include: options?.include ?? undefined,\n      },\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Create a new thread.\n   *\n   * @param payload Payload for creating a thread.\n   * @returns The created thread.\n   */\n  async create(payload?: {\n    metadata?: Metadata;\n    threadId?: string;\n    ifExists?: OnConflictBehavior;\n    graphId?: string;\n    supersteps?: Array<{\n      updates: Array<{ values: unknown; command?: Command; asNode: string }>;\n    }>;\n    ttl?: number | { ttl: number; strategy?: \"delete\" };\n    signal?: AbortSignal;\n  }): Promise<Thread<TStateType>> {\n    const ttlPayload =\n      typeof payload?.ttl === \"number\"\n        ? { ttl: payload.ttl, strategy: \"delete\" as const }\n        : payload?.ttl;\n\n    return this.fetch<Thread<TStateType>>(`/threads`, {\n      method: \"POST\",\n      json: {\n        metadata: {\n          ...payload?.metadata,\n          graph_id: payload?.graphId,\n        },\n        thread_id: payload?.threadId,\n        if_exists: payload?.ifExists,\n        supersteps: payload?.supersteps?.map((s) => ({\n          updates: s.updates.map((u) => ({\n            values: u.values,\n            command: u.command,\n            as_node: u.asNode,\n          })),\n        })),\n        ttl: ttlPayload,\n      },\n      signal: payload?.signal,\n    });\n  }\n\n  /**\n   * Copy an existing thread\n   * @param threadId ID of the thread to be copied\n   * @returns Newly copied thread\n   */\n  async copy(\n    threadId: string,\n    options?: { signal?: AbortSignal }\n  ): Promise<Thread<TStateType>> {\n    return this.fetch<Thread<TStateType>>(`/threads/${threadId}/copy`, {\n      method: \"POST\",\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Update a thread.\n   *\n   * @param threadId ID of the thread.\n   * @param payload Payload for updating the thread.\n   * @returns The updated thread.\n   */\n  async update(\n    threadId: string,\n    payload?: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal?: false;\n      signal?: AbortSignal;\n    }\n  ): Promise<Thread>;\n  async update(\n    threadId: string,\n    payload: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal: true;\n      signal?: AbortSignal;\n    }\n  ): Promise<void>;\n  async update(\n    threadId: string,\n    payload: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal: boolean;\n      signal?: AbortSignal;\n    }\n  ): Promise<Thread | void>;\n  async update(\n    threadId: string,\n    payload?: {\n      metadata?: Metadata;\n      ttl?: number | { ttl: number; strategy?: \"delete\" };\n      returnMinimal?: boolean;\n      signal?: AbortSignal;\n    }\n  ): Promise<Thread | void> {\n    const ttlPayload =\n      typeof payload?.ttl === \"number\"\n        ? { ttl: payload.ttl, strategy: \"delete\" as const }\n        : payload?.ttl;\n\n    return this.fetch<Thread | void>(`/threads/${threadId}`, {\n      method: \"PATCH\",\n      headers: payload?.returnMinimal\n        ? { Prefer: \"return=minimal\" }\n        : undefined,\n      json: { metadata: payload?.metadata, ttl: ttlPayload },\n      signal: payload?.signal,\n    });\n  }\n\n  /**\n   * Delete a thread.\n   *\n   * @param threadId ID of the thread.\n   */\n  async delete(\n    threadId: string,\n    options?: { signal?: AbortSignal }\n  ): Promise<void> {\n    return this.fetch<void>(`/threads/${threadId}`, {\n      method: \"DELETE\",\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Prune threads by ID. The 'delete' strategy removes threads entirely.\n   * The 'keep_latest' strategy prunes old checkpoints but keeps threads\n   * and their latest state.\n   *\n   * @param threadIds List of thread IDs to prune.\n   * @param options Additional options for pruning.\n   * @param options.strategy The prune strategy. Defaults to 'delete'.\n   * @param options.signal Signal to abort the request.\n   * @returns An object containing `pruned_count`.\n   */\n  async prune(\n    threadIds: string[],\n    options?: {\n      strategy?: \"delete\" | \"keep_latest\";\n      signal?: AbortSignal;\n    }\n  ): Promise<{ pruned_count: number }> {\n    return this.fetch<{ pruned_count: number }>(\"/threads/prune\", {\n      method: \"POST\",\n      json: {\n        thread_ids: threadIds,\n        strategy: options?.strategy ?? \"delete\",\n      },\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * List threads\n   *\n   * @param query Query options\n   * @returns List of threads\n   */\n  async search<ValuesType = TStateType>(query?: {\n    metadata?: Metadata;\n    ids?: string[];\n    limit?: number;\n    offset?: number;\n    status?: ThreadStatus;\n    sortBy?: ThreadSortBy;\n    sortOrder?: SortOrder;\n    select?: ThreadSelectField[];\n    values?: ThreadValuesFilter;\n    extract?: Record<string, string>;\n    signal?: AbortSignal;\n  }): Promise<Thread<ValuesType>[]> {\n    return this.fetch<Thread<ValuesType>[]>(\"/threads/search\", {\n      method: \"POST\",\n      json: {\n        metadata: query?.metadata ?? undefined,\n        ids: query?.ids ?? undefined,\n        limit: query?.limit ?? 10,\n        offset: query?.offset ?? 0,\n        status: query?.status,\n        sort_by: query?.sortBy,\n        sort_order: query?.sortOrder,\n        select: query?.select ?? undefined,\n        values: query?.values ?? undefined,\n        extract: query?.extract ?? undefined,\n      },\n      signal: query?.signal,\n    });\n  }\n\n  /**\n   * Count threads matching filters.\n   *\n   * @param query.metadata Thread metadata to filter on.\n   * @param query.values State values to filter on.\n   * @param query.status Thread status to filter on.\n   * @returns Number of threads matching the criteria.\n   */\n  async count<ValuesType = TStateType>(query?: {\n    metadata?: Metadata;\n    values?: ValuesType;\n    status?: ThreadStatus;\n    signal?: AbortSignal;\n  }): Promise<number> {\n    return this.fetch<number>(`/threads/count`, {\n      method: \"POST\",\n      json: {\n        metadata: query?.metadata ?? undefined,\n        values: query?.values ?? undefined,\n        status: query?.status ?? undefined,\n      },\n      signal: query?.signal,\n    });\n  }\n\n  /**\n   * Get state for a thread.\n   *\n   * @param threadId ID of the thread.\n   * @returns Thread state.\n   */\n  async getState<ValuesType = TStateType>(\n    threadId: string,\n    checkpoint?: Checkpoint | string,\n    options?: { subgraphs?: boolean; signal?: AbortSignal }\n  ): Promise<ThreadState<ValuesType>> {\n    if (checkpoint != null) {\n      if (typeof checkpoint !== \"string\") {\n        return this.fetch<ThreadState<ValuesType>>(\n          `/threads/${threadId}/state/checkpoint`,\n          {\n            method: \"POST\",\n            json: { checkpoint, subgraphs: options?.subgraphs },\n            signal: options?.signal,\n          }\n        );\n      }\n\n      // deprecated\n      return this.fetch<ThreadState<ValuesType>>(\n        `/threads/${threadId}/state/${checkpoint}`,\n        { params: { subgraphs: options?.subgraphs }, signal: options?.signal }\n      );\n    }\n\n    return this.fetch<ThreadState<ValuesType>>(`/threads/${threadId}/state`, {\n      params: { subgraphs: options?.subgraphs },\n      signal: options?.signal,\n      // Coalesce concurrent identical reads (e.g. two controllers\n      // hydrating the same thread on reconnect). Skipped automatically\n      // when a caller supplies its own `signal`.\n      dedupe: true,\n    });\n  }\n\n  /**\n   * Add state to a thread.\n   *\n   * @param threadId The ID of the thread.\n   * @returns\n   */\n  async updateState<ValuesType = TUpdateType>(\n    threadId: string,\n    options: {\n      values: ValuesType;\n      checkpoint?: Checkpoint;\n      checkpointId?: string;\n      asNode?: string;\n      signal?: AbortSignal;\n    }\n  ): Promise<Pick<Config, \"configurable\">> {\n    return this.fetch<Pick<Config, \"configurable\">>(\n      `/threads/${threadId}/state`,\n      {\n        method: \"POST\",\n        json: {\n          values: options.values,\n          checkpoint: options.checkpoint,\n          checkpoint_id: options.checkpointId,\n          as_node: options?.asNode,\n        },\n        signal: options?.signal,\n      }\n    );\n  }\n\n  /**\n   * Patch the metadata of a thread.\n   *\n   * @param threadIdOrConfig Thread ID or config to patch the state of.\n   * @param metadata Metadata to patch the state with.\n   */\n  async patchState(\n    threadIdOrConfig: string | Config,\n    metadata: Metadata,\n    options?: { signal?: AbortSignal }\n  ): Promise<void> {\n    let threadId: string;\n\n    if (typeof threadIdOrConfig !== \"string\") {\n      if (typeof threadIdOrConfig.configurable?.thread_id !== \"string\") {\n        throw new Error(\n          \"Thread ID is required when updating state with a config.\"\n        );\n      }\n      threadId = threadIdOrConfig.configurable.thread_id;\n    } else {\n      threadId = threadIdOrConfig;\n    }\n\n    return this.fetch<void>(`/threads/${threadId}/state`, {\n      method: \"PATCH\",\n      json: { metadata },\n      signal: options?.signal,\n    });\n  }\n\n  /**\n   * Get all past states for a thread.\n   *\n   * @param threadId ID of the thread.\n   * @param options Additional options.\n   * @returns List of thread states.\n   */\n  async getHistory<ValuesType = TStateType>(\n    threadId: string,\n    options?: {\n      limit?: number;\n      before?: Config;\n      checkpoint?: Partial<Omit<Checkpoint, \"thread_id\">>;\n      metadata?: Metadata;\n      signal?: AbortSignal;\n    }\n  ): Promise<ThreadState<ValuesType>[]> {\n    return this.fetch<ThreadState<ValuesType>[]>(\n      `/threads/${threadId}/history`,\n      {\n        method: \"POST\",\n        json: {\n          limit: options?.limit ?? 10,\n          before: options?.before,\n          metadata: options?.metadata,\n          checkpoint: options?.checkpoint,\n        },\n        signal: options?.signal,\n        // `getHistory` is a read despite using POST — coalesce the\n        // hydrate-time discovery seed when two controllers reconnect to\n        // the same thread concurrently. Skipped when `signal` is set.\n        dedupe: true,\n      }\n    );\n  }\n\n  async *joinStream(\n    threadId: string,\n    options?: {\n      lastEventId?: string;\n      streamMode?: ThreadStreamMode | ThreadStreamMode[];\n      signal?: AbortSignal;\n    }\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  ): AsyncGenerator<{ id?: string; event: StreamEvent; data: any }> {\n    yield* this.streamWithRetry({\n      endpoint: `/threads/${threadId}/stream`,\n      method: \"GET\",\n      signal: options?.signal,\n      headers: options?.lastEventId\n        ? { \"Last-Event-ID\": options.lastEventId }\n        : undefined,\n      params: options?.streamMode\n        ? { stream_mode: options.streamMode }\n        : undefined,\n    });\n  }\n\n  /**\n   * Open a protocol stream over the thread-centric v2 protocol.\n   *\n   * Returns a {@link ThreadStream} with lazy getters\n   * (`.messages`, `.values`, `.toolCalls`, `.subgraphs`, `.subagents`,\n   * `.output`) and `thread.run.start({ input, ... })` for starting runs.\n   * Mirrors the in-process `graph.streamEvents(..., { version: \"v3\" })` API.\n   *\n   * The thread is bound to `options.assistantId` for its lifetime.\n   * The wire transport defaults to SSE; pass `transport: \"websocket\"`\n   * in options (or configure `streamProtocol: \"v2-websocket\"` on the\n   * client) to use a WebSocket instead.\n   *\n   * @example New thread (UUID generated client-side)\n   * ```ts\n   * const thread = client.threads.stream({ assistantId: \"my-agent\" });\n   * ```\n   *\n   * @example Attach to an existing thread\n   * ```ts\n   * const thread = client.threads.stream(threadId, { assistantId: \"my-agent\" });\n   * ```\n   *\n   * @example WebSocket transport\n   * ```ts\n   * const thread = client.threads.stream({\n   *   assistantId: \"my-agent\",\n   *   transport: \"websocket\",\n   * });\n   * ```\n   */\n  stream<TExtensions extends Record<string, unknown> = Record<string, unknown>>(\n    options: ThreadStreamOptions\n  ): ThreadStream<TExtensions>;\n  stream<TExtensions extends Record<string, unknown> = Record<string, unknown>>(\n    threadId: string,\n    options: ThreadStreamOptions\n  ): ThreadStream<TExtensions>;\n  stream<TExtensions extends Record<string, unknown> = Record<string, unknown>>(\n    threadIdOrOptions: string | ThreadStreamOptions,\n    maybeOptions?: ThreadStreamOptions\n  ): ThreadStream<TExtensions> {\n    const { threadId, options } =\n      typeof threadIdOrOptions === \"string\"\n        ? {\n            threadId: threadIdOrOptions,\n            options: maybeOptions as ThreadStreamOptions,\n          }\n        : // A nullish first arg is the generate-thread-id form, not the\n          // options-only overload. Without this branch, a runtime\n          // `stream(null, opts)` binds `null` to `options` and the\n          // `options.fetch` read below throws. Use the second arg as options.\n          // (Cast: the declared type excludes null, but callers can reach\n          // here at runtime, e.g. an in-flight reconnect after the id clears.)\n          (threadIdOrOptions as ThreadStreamOptions | null | undefined) == null\n          ? {\n              threadId: uuidv7(),\n              options: maybeOptions as ThreadStreamOptions,\n            }\n          : { threadId: uuidv7(), options: threadIdOrOptions };\n\n    // `transport` accepts either a preset string (`\"sse\"` / `\"websocket\"`)\n    // or a custom {@link AgentServerAdapter}. A custom adapter replaces\n    // the built-in factories entirely — this is the seam that lets users\n    // point `useStream` at any agent server (including the thin wrappers\n    // produced by `HttpAgentServerAdapter`).\n    // When callers supply `fetch`, use it verbatim (tests, auth shims). Otherwise\n    // route protocol HTTP and media URL fetches through AsyncCaller like REST.\n    const userFetch = options.fetch;\n    const protocolFetch =\n      userFetch ?? this.asyncCaller.fetch.bind(this.asyncCaller);\n\n    let transport: TransportAdapter;\n    if (options.transport != null && typeof options.transport !== \"string\") {\n      transport = options.transport;\n      // Bind (or re-bind) the caller-supplied adapter to this thread so a\n      // single instance can follow lazy creation (the first `submit()` on a\n      // `threadId: null` controller mints the id here) and thread switches.\n      // No-op for adapters that bake `threadId` at construction.\n      transport.setThreadId?.(threadId);\n    } else {\n      const transportKind: ThreadStreamTransportKind =\n        options.transport ??\n        (this.streamProtocol === \"v2-websocket\" ? \"websocket\" : \"sse\");\n      const maxReconnectAttempts =\n        options.maxReconnectAttempts ?? DEFAULT_MAX_RECONNECT_ATTEMPTS;\n\n      /**\n       * Common options for both transports.\n       */\n      const commonOpts = {\n        apiUrl: this.apiUrl,\n        threadId,\n        defaultHeaders: this.defaultHeaders,\n        onRequest: this.onRequest,\n        maxReconnectAttempts,\n        reconnectDelayMs: options.reconnectDelayMs,\n        onReconnect: options.onReconnect,\n        onConnected: options.onConnected,\n      };\n\n      transport =\n        transportKind === \"websocket\"\n          ? new ProtocolWebSocketTransportAdapter({\n              ...commonOpts,\n              webSocketFactory: options.webSocketFactory,\n            })\n          : new ProtocolSseTransportAdapter({\n              ...commonOpts,\n              idleReconnect: options.streamIdleReconnect,\n              fetch: userFetch,\n              asyncCaller: userFetch ? undefined : this.asyncCaller,\n            });\n    }\n\n    return new ThreadStream<TExtensions>(transport, {\n      ...options,\n      fetch: protocolFetch,\n    });\n  }\n}\n"],"mappings":";;;;;;;AA4BA,IAAa,gBAAb,cAGUA,aAAAA,WAAW;;;;;;;CAOnB,MAAM,IACJ,UACA,SAC6B;AAC7B,SAAO,KAAK,MAA0B,YAAY,YAAY;GAC5D,QAAQ,EACN,SAAS,SAAS,WAAW,KAAA,GAC9B;GACD,QAAQ,SAAS;GAClB,CAAC;;;;;;;;CASJ,MAAM,OAAO,SAUmB;EAC9B,MAAM,aACJ,OAAO,SAAS,QAAQ,WACpB;GAAE,KAAK,QAAQ;GAAK,UAAU;GAAmB,GACjD,SAAS;AAEf,SAAO,KAAK,MAA0B,YAAY;GAChD,QAAQ;GACR,MAAM;IACJ,UAAU;KACR,GAAG,SAAS;KACZ,UAAU,SAAS;KACpB;IACD,WAAW,SAAS;IACpB,WAAW,SAAS;IACpB,YAAY,SAAS,YAAY,KAAK,OAAO,EAC3C,SAAS,EAAE,QAAQ,KAAK,OAAO;KAC7B,QAAQ,EAAE;KACV,SAAS,EAAE;KACX,SAAS,EAAE;KACZ,EAAE,EACJ,EAAE;IACH,KAAK;IACN;GACD,QAAQ,SAAS;GAClB,CAAC;;;;;;;CAQJ,MAAM,KACJ,UACA,SAC6B;AAC7B,SAAO,KAAK,MAA0B,YAAY,SAAS,QAAQ;GACjE,QAAQ;GACR,QAAQ,SAAS;GAClB,CAAC;;CAqCJ,MAAM,OACJ,UACA,SAMwB;EACxB,MAAM,aACJ,OAAO,SAAS,QAAQ,WACpB;GAAE,KAAK,QAAQ;GAAK,UAAU;GAAmB,GACjD,SAAS;AAEf,SAAO,KAAK,MAAqB,YAAY,YAAY;GACvD,QAAQ;GACR,SAAS,SAAS,gBACd,EAAE,QAAQ,kBAAkB,GAC5B,KAAA;GACJ,MAAM;IAAE,UAAU,SAAS;IAAU,KAAK;IAAY;GACtD,QAAQ,SAAS;GAClB,CAAC;;;;;;;CAQJ,MAAM,OACJ,UACA,SACe;AACf,SAAO,KAAK,MAAY,YAAY,YAAY;GAC9C,QAAQ;GACR,QAAQ,SAAS;GAClB,CAAC;;;;;;;;;;;;;CAcJ,MAAM,MACJ,WACA,SAImC;AACnC,SAAO,KAAK,MAAgC,kBAAkB;GAC5D,QAAQ;GACR,MAAM;IACJ,YAAY;IACZ,UAAU,SAAS,YAAY;IAChC;GACD,QAAQ,SAAS;GAClB,CAAC;;;;;;;;CASJ,MAAM,OAAgC,OAYJ;AAChC,SAAO,KAAK,MAA4B,mBAAmB;GACzD,QAAQ;GACR,MAAM;IACJ,UAAU,OAAO,YAAY,KAAA;IAC7B,KAAK,OAAO,OAAO,KAAA;IACnB,OAAO,OAAO,SAAS;IACvB,QAAQ,OAAO,UAAU;IACzB,QAAQ,OAAO;IACf,SAAS,OAAO;IAChB,YAAY,OAAO;IACnB,QAAQ,OAAO,UAAU,KAAA;IACzB,QAAQ,OAAO,UAAU,KAAA;IACzB,SAAS,OAAO,WAAW,KAAA;IAC5B;GACD,QAAQ,OAAO;GAChB,CAAC;;;;;;;;;;CAWJ,MAAM,MAA+B,OAKjB;AAClB,SAAO,KAAK,MAAc,kBAAkB;GAC1C,QAAQ;GACR,MAAM;IACJ,UAAU,OAAO,YAAY,KAAA;IAC7B,QAAQ,OAAO,UAAU,KAAA;IACzB,QAAQ,OAAO,UAAU,KAAA;IAC1B;GACD,QAAQ,OAAO;GAChB,CAAC;;;;;;;;CASJ,MAAM,SACJ,UACA,YACA,SACkC;AAClC,MAAI,cAAc,MAAM;AACtB,OAAI,OAAO,eAAe,SACxB,QAAO,KAAK,MACV,YAAY,SAAS,oBACrB;IACE,QAAQ;IACR,MAAM;KAAE;KAAY,WAAW,SAAS;KAAW;IACnD,QAAQ,SAAS;IAClB,CACF;AAIH,UAAO,KAAK,MACV,YAAY,SAAS,SAAS,cAC9B;IAAE,QAAQ,EAAE,WAAW,SAAS,WAAW;IAAE,QAAQ,SAAS;IAAQ,CACvE;;AAGH,SAAO,KAAK,MAA+B,YAAY,SAAS,SAAS;GACvE,QAAQ,EAAE,WAAW,SAAS,WAAW;GACzC,QAAQ,SAAS;GAIjB,QAAQ;GACT,CAAC;;;;;;;;CASJ,MAAM,YACJ,UACA,SAOuC;AACvC,SAAO,KAAK,MACV,YAAY,SAAS,SACrB;GACE,QAAQ;GACR,MAAM;IACJ,QAAQ,QAAQ;IAChB,YAAY,QAAQ;IACpB,eAAe,QAAQ;IACvB,SAAS,SAAS;IACnB;GACD,QAAQ,SAAS;GAClB,CACF;;;;;;;;CASH,MAAM,WACJ,kBACA,UACA,SACe;EACf,IAAI;AAEJ,MAAI,OAAO,qBAAqB,UAAU;AACxC,OAAI,OAAO,iBAAiB,cAAc,cAAc,SACtD,OAAM,IAAI,MACR,2DACD;AAEH,cAAW,iBAAiB,aAAa;QAEzC,YAAW;AAGb,SAAO,KAAK,MAAY,YAAY,SAAS,SAAS;GACpD,QAAQ;GACR,MAAM,EAAE,UAAU;GAClB,QAAQ,SAAS;GAClB,CAAC;;;;;;;;;CAUJ,MAAM,WACJ,UACA,SAOoC;AACpC,SAAO,KAAK,MACV,YAAY,SAAS,WACrB;GACE,QAAQ;GACR,MAAM;IACJ,OAAO,SAAS,SAAS;IACzB,QAAQ,SAAS;IACjB,UAAU,SAAS;IACnB,YAAY,SAAS;IACtB;GACD,QAAQ,SAAS;GAIjB,QAAQ;GACT,CACF;;CAGH,OAAO,WACL,UACA,SAMgE;AAChE,SAAO,KAAK,gBAAgB;GAC1B,UAAU,YAAY,SAAS;GAC/B,QAAQ;GACR,QAAQ,SAAS;GACjB,SAAS,SAAS,cACd,EAAE,iBAAiB,QAAQ,aAAa,GACxC,KAAA;GACJ,QAAQ,SAAS,aACb,EAAE,aAAa,QAAQ,YAAY,GACnC,KAAA;GACL,CAAC;;CAyCJ,OACE,mBACA,cAC2B;EAC3B,MAAM,EAAE,UAAU,YAChB,OAAO,sBAAsB,WACzB;GACE,UAAU;GACV,SAAS;GACV,GAOA,qBAAgE,OAC/D;GACE,WAAA,GAAA,2BAAA,KAAkB;GAClB,SAAS;GACV,GACD;GAAE,WAAA,GAAA,2BAAA,KAAkB;GAAE,SAAS;GAAmB;EAS1D,MAAM,YAAY,QAAQ;EAC1B,MAAM,gBACJ,aAAa,KAAK,YAAY,MAAM,KAAK,KAAK,YAAY;EAE5D,IAAI;AACJ,MAAI,QAAQ,aAAa,QAAQ,OAAO,QAAQ,cAAc,UAAU;AACtE,eAAY,QAAQ;AAKpB,aAAU,cAAc,SAAS;SAC5B;GACL,MAAM,gBACJ,QAAQ,cACP,KAAK,mBAAmB,iBAAiB,cAAc;GAC1D,MAAM,uBACJ,QAAQ,wBAAA;;;;GAKV,MAAM,aAAa;IACjB,QAAQ,KAAK;IACb;IACA,gBAAgB,KAAK;IACrB,WAAW,KAAK;IAChB;IACA,kBAAkB,QAAQ;IAC1B,aAAa,QAAQ;IACrB,aAAa,QAAQ;IACtB;AAED,eACE,kBAAkB,cACd,IAAIC,kBAAAA,kCAAkC;IACpC,GAAG;IACH,kBAAkB,QAAQ;IAC3B,CAAC,GACF,IAAIC,aAAAA,4BAA4B;IAC9B,GAAG;IACH,eAAe,QAAQ;IACvB,OAAO;IACP,aAAa,YAAY,KAAA,IAAY,KAAK;IAC3C,CAAC;;AAGV,SAAO,IAAIC,cAAAA,aAA0B,WAAW;GAC9C,GAAG;GACH,OAAO;GACR,CAAC"}

package/dist/client/threads/index.js

@@ -270,7 +270,8 @@ var ThreadsClient = class extends BaseClient {
 				onRequest: this.onRequest,
 				maxReconnectAttempts,
 				reconnectDelayMs: options.reconnectDelayMs,
-				onReconnect: options.onReconnect
+				onReconnect: options.onReconnect,
+				onConnected: options.onConnected
 			};
 			transport = transportKind === "websocket" ? new ProtocolWebSocketTransportAdapter({
 				...commonOpts,