@langchain/langgraph 1.3.6 → 1.4.0

package/dist/pregel/stream.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream.js","names":[],"sources":["../../src/pregel/stream.ts"],"sourcesContent":["import { IterableReadableStream } from \"@langchain/core/utils/stream\";\nimport type { RunnableConfig } from \"@langchain/core/runnables\";\nimport { BaseCallbackHandler } from \"@langchain/core/callbacks/base\";\nimport { Serialized } from \"@langchain/core/load/serializable\";\nimport { isCheckpointEnvelope } from \"../stream/convert.js\";\nimport type { Checkpoint } from \"../stream/types.js\";\nimport type { StreamMode, StreamOutputMap } from \"./types.js\";\nimport { TAG_HIDDEN } from \"../constants.js\";\n\n/**\n * Optional chunk-level metadata carried alongside the payload. Used by\n * `streamEvents(..., { version: \"v3\" })` to emit a companion `checkpoints`\n * protocol event adjacent to the `values` event for the same superstep, so\n * clients can build branching / time-travel UIs without subscribing to a\n * full-state `checkpoints` stream.\n *\n * Companion checkpoint envelopes are emitted as separate\n * ``[namespace, \"checkpoints\", envelope]`` chunks (see\n * ``PregelLoop._emitValuesWithCheckpointMeta``).\n */\nexport interface StreamChunkMeta {\n  /**\n   * Lightweight checkpoint envelope for the superstep that produced the\n   * paired `values` chunk. Shape matches the canonical {@link Checkpoint}\n   * generated from `protocol.cddl`.\n   */\n  checkpoint?: Checkpoint;\n}\n\n// [namespace, streamMode, payload]\nexport type StreamChunk = [string[], StreamMode, unknown];\n\ntype StreamCheckpointsOutput<StreamValues> = StreamOutputMap<\n  \"checkpoints\",\n  false,\n  StreamValues,\n  unknown,\n  string,\n  unknown,\n  unknown,\n  undefined\n>;\n\ntype AnyStreamOutput = StreamOutputMap<\n  StreamMode[],\n  true,\n  unknown,\n  unknown,\n  string,\n  unknown,\n  unknown,\n  undefined\n>;\n\ntype ToolRunInfo = {\n  ns: string[];\n  toolCallId?: string;\n  toolName: string;\n  input: unknown;\n};\n\n/**\n * A wrapper around an IterableReadableStream that allows for aborting the stream when\n * {@link cancel} is called.\n */\nexport class IterableReadableStreamWithAbortSignal<\n  T,\n> extends IterableReadableStream<T> {\n  protected _abortController: AbortController;\n\n  protected _innerReader: ReadableStreamDefaultReader<T>;\n\n  /**\n   * @param readableStream - The stream to wrap.\n   * @param abortController - The abort controller to use. Optional. One will be created if not provided.\n   */\n  constructor(\n    readableStream: ReadableStream<T>,\n    abortController?: AbortController\n  ) {\n    const reader = readableStream.getReader();\n    const ac = abortController ?? new AbortController();\n    super({\n      start(controller: ReadableStreamDefaultController<T>) {\n        return pump();\n        function pump(): Promise<T | undefined> {\n          return reader.read().then(({ done, value }) => {\n            // When no more data needs to be consumed, close the stream\n            if (done) {\n              controller.close();\n              return;\n            }\n            // Enqueue the next data chunk into our target stream\n            controller.enqueue(value);\n            return pump();\n          });\n        }\n      },\n    });\n    this._abortController = ac;\n    this._innerReader = reader;\n  }\n\n  /**\n   * Aborts the stream, abandoning any pending operations in progress. Calling this triggers an\n   * {@link AbortSignal} that is propagated to the tasks that are producing the data for this stream.\n   * @param reason - The reason for aborting the stream. Optional.\n   */\n  override async cancel(reason?: unknown) {\n    this._abortController.abort(reason);\n    this._innerReader.releaseLock();\n  }\n\n  /**\n   * The {@link AbortSignal} for the stream. Aborted when {@link cancel} is called.\n   */\n  get signal() {\n    return this._abortController.signal;\n  }\n}\n\nexport class IterableReadableWritableStream extends IterableReadableStream<StreamChunk> {\n  modes: Set<StreamMode>;\n\n  private controller: ReadableStreamDefaultController;\n\n  private passthroughFn?: (chunk: StreamChunk) => void;\n\n  private _closed: boolean = false;\n\n  get closed() {\n    return this._closed;\n  }\n\n  constructor(params: {\n    passthroughFn?: (chunk: StreamChunk) => void;\n    modes: Set<StreamMode>;\n  }) {\n    let streamControllerPromiseResolver: (\n      controller: ReadableStreamDefaultController\n    ) => void;\n    const streamControllerPromise: Promise<ReadableStreamDefaultController> =\n      new Promise<ReadableStreamDefaultController>((resolve) => {\n        streamControllerPromiseResolver = resolve;\n      });\n\n    super({\n      start: (controller) => {\n        streamControllerPromiseResolver!(controller);\n      },\n    });\n\n    // .start() will always be called before the stream can be interacted\n    // with anyway\n    void streamControllerPromise.then((controller) => {\n      this.controller = controller;\n    });\n\n    this.passthroughFn = params.passthroughFn;\n    this.modes = params.modes;\n  }\n\n  push(chunk: StreamChunk) {\n    // Prevent pushing to a closed stream to avoid race condition errors\n    if (this._closed || !this.controller) {\n      // Silently drop chunks when stream is closed - this is expected behavior\n      // when async operations try to push after stream termination\n      return;\n    }\n\n    // Forward chunk to passthrough function if provided\n    this.passthroughFn?.(chunk);\n\n    // Attempt to enqueue the chunk to the underlying stream\n    this.controller.enqueue(chunk);\n  }\n\n  close() {\n    try {\n      this.controller.close();\n    } catch {\n      // pass\n    } finally {\n      this._closed = true;\n    }\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  error(e: any) {\n    try {\n      this.controller?.error(e);\n    } finally {\n      // Mark the stream as closed so any late `push()` calls from in-flight\n      // parallel tasks are dropped instead of throwing on an errored controller.\n      this._closed = true;\n    }\n  }\n}\n\n/**\n * A callback handler that implements stream_mode=tools.\n * Emits on_tool_start, on_tool_event, on_tool_end, on_tool_error events.\n */\nexport class StreamToolsHandler extends BaseCallbackHandler {\n  name = \"StreamToolsHandler\";\n\n  streamFn: (streamChunk: StreamChunk) => void;\n\n  runs: Record<string, ToolRunInfo | undefined> = {};\n\n  constructor(streamFn: (streamChunk: StreamChunk) => void) {\n    super();\n    this.streamFn = streamFn;\n  }\n\n  handleToolStart(\n    _tool: Serialized,\n    input: string,\n    runId: string,\n    _parentRunId?: string,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    runName?: string,\n    toolCallId?: string\n  ) {\n    if (!metadata || (tags && tags.includes(TAG_HIDDEN))) return;\n\n    const ns = (metadata.langgraph_checkpoint_ns as string)?.split(\"|\") ?? [];\n    const info: ToolRunInfo = {\n      ns,\n      toolCallId,\n      toolName: runName ?? \"unknown\",\n      input,\n    };\n    this.runs[runId] = info;\n\n    this.streamFn([\n      ns,\n      \"tools\",\n      {\n        event: \"on_tool_start\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        input,\n      },\n    ]);\n  }\n\n  handleToolEvent(chunk: unknown, runId: string) {\n    const info = this.runs[runId];\n    if (!info) return;\n\n    this.streamFn([\n      info.ns,\n      \"tools\",\n      {\n        event: \"on_tool_event\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        data: chunk,\n      },\n    ]);\n  }\n\n  handleToolEnd(output: unknown, runId: string) {\n    const info = this.runs[runId];\n    delete this.runs[runId];\n    if (!info) return;\n\n    this.streamFn([\n      info.ns,\n      \"tools\",\n      {\n        event: \"on_tool_end\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        output,\n      },\n    ]);\n  }\n\n  handleToolError(err: unknown, runId: string) {\n    const info = this.runs[runId];\n    delete this.runs[runId];\n    if (!info) return;\n\n    this.streamFn([\n      info.ns,\n      \"tools\",\n      {\n        event: \"on_tool_error\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        error: err,\n      },\n    ]);\n  }\n}\n\nfunction _stringifyAsDict(obj: unknown) {\n  return JSON.stringify(obj, function (key: string | number, value: unknown) {\n    const rawValue = this[key];\n    if (\n      rawValue != null &&\n      typeof rawValue === \"object\" &&\n      \"toDict\" in rawValue &&\n      typeof rawValue.toDict === \"function\"\n    ) {\n      const { type, data } = rawValue.toDict();\n      return { ...data, type };\n    }\n\n    return value;\n  });\n}\n\nfunction _serializeError(error: unknown) {\n  // eslint-disable-next-line no-instanceof/no-instanceof\n  if (error instanceof Error) {\n    return { error: error.name, message: error.message };\n  }\n  return { error: \"Error\", message: JSON.stringify(error) };\n}\n\nfunction _isRunnableConfig(\n  config: unknown\n): config is RunnableConfig & { configurable: Record<string, unknown> } {\n  if (typeof config !== \"object\" || config == null) return false;\n  return (\n    \"configurable\" in config &&\n    typeof config.configurable === \"object\" &&\n    config.configurable != null\n  );\n}\n\nfunction _extractCheckpointFromConfig(\n  config: RunnableConfig | null | undefined\n) {\n  if (!_isRunnableConfig(config) || !config.configurable.thread_id) {\n    return null;\n  }\n\n  return {\n    thread_id: config.configurable.thread_id,\n    checkpoint_ns: config.configurable.checkpoint_ns || \"\",\n    checkpoint_id: config.configurable.checkpoint_id || null,\n    checkpoint_map: config.configurable.checkpoint_map || null,\n  };\n}\n\nfunction _serializeConfig(config: unknown) {\n  if (_isRunnableConfig(config)) {\n    const configurable = Object.fromEntries(\n      Object.entries(config.configurable).filter(\n        ([key]) => !key.startsWith(\"__\")\n      )\n    );\n\n    const newConfig = { ...config, configurable };\n    delete newConfig.callbacks;\n    return newConfig;\n  }\n\n  return config;\n}\n\nfunction _serializeCheckpoint(payload: StreamCheckpointsOutput<unknown>) {\n  const result: Record<string, unknown> = {\n    ...payload,\n    checkpoint: _extractCheckpointFromConfig(payload.config),\n    parent_checkpoint: _extractCheckpointFromConfig(payload.parentConfig),\n\n    config: _serializeConfig(payload.config),\n    parent_config: _serializeConfig(payload.parentConfig),\n\n    tasks: payload.tasks.map((task) => {\n      if (_isRunnableConfig(task.state)) {\n        const checkpoint = _extractCheckpointFromConfig(task.state);\n        if (checkpoint != null) {\n          const cloneTask: Record<string, unknown> = { ...task, checkpoint };\n          delete cloneTask.state;\n          return cloneTask;\n        }\n      }\n\n      return task;\n    }),\n  };\n\n  delete result.parentConfig;\n  return result;\n}\n\nexport function toEventStream(stream: AsyncGenerator) {\n  const encoder = new TextEncoder();\n  return new ReadableStream<Uint8Array>({\n    async start(controller) {\n      const enqueueChunk = (sse: {\n        id?: string;\n        event: string;\n        data: unknown;\n      }) => {\n        controller.enqueue(\n          encoder.encode(\n            `event: ${sse.event}\\ndata: ${_stringifyAsDict(sse.data)}\\n\\n`\n          )\n        );\n      };\n\n      try {\n        for await (const payload of stream) {\n          const [ns, mode, chunk] = payload as AnyStreamOutput;\n\n          let data: unknown = chunk;\n          if (mode === \"debug\") {\n            const debugChunk = chunk;\n\n            if (debugChunk.type === \"checkpoint\") {\n              data = {\n                ...debugChunk,\n                payload: _serializeCheckpoint(debugChunk.payload),\n              };\n            }\n          }\n\n          if (mode === \"checkpoints\") {\n            data = _serializeCheckpoint(chunk);\n          }\n\n          const event = ns?.length ? `${mode}|${ns.join(\"|\")}` : mode;\n          enqueueChunk({ event, data });\n        }\n      } catch (error) {\n        enqueueChunk({ event: \"error\", data: _serializeError(error) });\n      }\n\n      controller.close();\n    },\n  });\n}\n\n/** Multiplex subgraph stream chunks into the parent pregel stream. */\nexport function createDuplexStream(\n  ...streams: IterableReadableWritableStream[]\n) {\n  return new IterableReadableWritableStream({\n    passthroughFn: (value: StreamChunk) => {\n      const isEnvelope =\n        value[1] === \"checkpoints\" && isCheckpointEnvelope(value[2]);\n      for (const stream of streams) {\n        if (stream.modes.has(value[1]) || isEnvelope) {\n          stream.push(value);\n        }\n      }\n    },\n    modes: new Set(streams.flatMap((s) => Array.from(s.modes))),\n  });\n}\n"],"mappings":";;;;;;;;;AAiEA,IAAa,wCAAb,cAEU,uBAA0B;CAClC;CAEA;;;;;CAMA,YACE,gBACA,iBACA;EACA,MAAM,SAAS,eAAe,WAAW;EACzC,MAAM,KAAK,mBAAmB,IAAI,iBAAiB;AACnD,QAAM,EACJ,MAAM,YAAgD;AACpD,UAAO,MAAM;GACb,SAAS,OAA+B;AACtC,WAAO,OAAO,MAAM,CAAC,MAAM,EAAE,MAAM,YAAY;AAE7C,SAAI,MAAM;AACR,iBAAW,OAAO;AAClB;;AAGF,gBAAW,QAAQ,MAAM;AACzB,YAAO,MAAM;MACb;;KAGP,CAAC;AACF,OAAK,mBAAmB;AACxB,OAAK,eAAe;;;;;;;CAQtB,MAAe,OAAO,QAAkB;AACtC,OAAK,iBAAiB,MAAM,OAAO;AACnC,OAAK,aAAa,aAAa;;;;;CAMjC,IAAI,SAAS;AACX,SAAO,KAAK,iBAAiB;;;AAIjC,IAAa,iCAAb,cAAoD,uBAAoC;CACtF;CAEA;CAEA;CAEA,UAA2B;CAE3B,IAAI,SAAS;AACX,SAAO,KAAK;;CAGd,YAAY,QAGT;EACD,IAAI;EAGJ,MAAM,0BACJ,IAAI,SAA0C,YAAY;AACxD,qCAAkC;IAClC;AAEJ,QAAM,EACJ,QAAQ,eAAe;AACrB,mCAAiC,WAAW;KAE/C,CAAC;AAIG,0BAAwB,MAAM,eAAe;AAChD,QAAK,aAAa;IAClB;AAEF,OAAK,gBAAgB,OAAO;AAC5B,OAAK,QAAQ,OAAO;;CAGtB,KAAK,OAAoB;AAEvB,MAAI,KAAK,WAAW,CAAC,KAAK,WAGxB;AAIF,OAAK,gBAAgB,MAAM;AAG3B,OAAK,WAAW,QAAQ,MAAM;;CAGhC,QAAQ;AACN,MAAI;AACF,QAAK,WAAW,OAAO;UACjB,WAEE;AACR,QAAK,UAAU;;;CAKnB,MAAM,GAAQ;AACZ,MAAI;AACF,QAAK,YAAY,MAAM,EAAE;YACjB;AAGR,QAAK,UAAU;;;;;;;;AASrB,IAAa,qBAAb,cAAwC,oBAAoB;CAC1D,OAAO;CAEP;CAEA,OAAgD,EAAE;CAElD,YAAY,UAA8C;AACxD,SAAO;AACP,OAAK,WAAW;;CAGlB,gBACE,OACA,OACA,OACA,cACA,MACA,UACA,SACA,YACA;AACA,MAAI,CAAC,YAAa,QAAQ,KAAK,SAAA,mBAAoB,CAAG;EAEtD,MAAM,KAAM,SAAS,yBAAoC,MAAM,IAAI,IAAI,EAAE;EACzE,MAAM,OAAoB;GACxB;GACA;GACA,UAAU,WAAW;GACrB;GACD;AACD,OAAK,KAAK,SAAS;AAEnB,OAAK,SAAS;GACZ;GACA;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX;IACD;GACF,CAAC;;CAGJ,gBAAgB,OAAgB,OAAe;EAC7C,MAAM,OAAO,KAAK,KAAK;AACvB,MAAI,CAAC,KAAM;AAEX,OAAK,SAAS;GACZ,KAAK;GACL;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX,MAAM;IACP;GACF,CAAC;;CAGJ,cAAc,QAAiB,OAAe;EAC5C,MAAM,OAAO,KAAK,KAAK;AACvB,SAAO,KAAK,KAAK;AACjB,MAAI,CAAC,KAAM;AAEX,OAAK,SAAS;GACZ,KAAK;GACL;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX;IACD;GACF,CAAC;;CAGJ,gBAAgB,KAAc,OAAe;EAC3C,MAAM,OAAO,KAAK,KAAK;AACvB,SAAO,KAAK,KAAK;AACjB,MAAI,CAAC,KAAM;AAEX,OAAK,SAAS;GACZ,KAAK;GACL;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX,OAAO;IACR;GACF,CAAC;;;AAIN,SAAS,iBAAiB,KAAc;AACtC,QAAO,KAAK,UAAU,KAAK,SAAU,KAAsB,OAAgB;EACzE,MAAM,WAAW,KAAK;AACtB,MACE,YAAY,QACZ,OAAO,aAAa,YACpB,YAAY,YACZ,OAAO,SAAS,WAAW,YAC3B;GACA,MAAM,EAAE,MAAM,SAAS,SAAS,QAAQ;AACxC,UAAO;IAAE,GAAG;IAAM;IAAM;;AAG1B,SAAO;GACP;;AAGJ,SAAS,gBAAgB,OAAgB;AAEvC,KAAI,iBAAiB,MACnB,QAAO;EAAE,OAAO,MAAM;EAAM,SAAS,MAAM;EAAS;AAEtD,QAAO;EAAE,OAAO;EAAS,SAAS,KAAK,UAAU,MAAM;EAAE;;AAG3D,SAAS,kBACP,QACsE;AACtE,KAAI,OAAO,WAAW,YAAY,UAAU,KAAM,QAAO;AACzD,QACE,kBAAkB,UAClB,OAAO,OAAO,iBAAiB,YAC/B,OAAO,gBAAgB;;AAI3B,SAAS,6BACP,QACA;AACA,KAAI,CAAC,kBAAkB,OAAO,IAAI,CAAC,OAAO,aAAa,UACrD,QAAO;AAGT,QAAO;EACL,WAAW,OAAO,aAAa;EAC/B,eAAe,OAAO,aAAa,iBAAiB;EACpD,eAAe,OAAO,aAAa,iBAAiB;EACpD,gBAAgB,OAAO,aAAa,kBAAkB;EACvD;;AAGH,SAAS,iBAAiB,QAAiB;AACzC,KAAI,kBAAkB,OAAO,EAAE;EAC7B,MAAM,eAAe,OAAO,YAC1B,OAAO,QAAQ,OAAO,aAAa,CAAC,QACjC,CAAC,SAAS,CAAC,IAAI,WAAW,KAAK,CACjC,CACF;EAED,MAAM,YAAY;GAAE,GAAG;GAAQ;GAAc;AAC7C,SAAO,UAAU;AACjB,SAAO;;AAGT,QAAO;;AAGT,SAAS,qBAAqB,SAA2C;CACvE,MAAM,SAAkC;EACtC,GAAG;EACH,YAAY,6BAA6B,QAAQ,OAAO;EACxD,mBAAmB,6BAA6B,QAAQ,aAAa;EAErE,QAAQ,iBAAiB,QAAQ,OAAO;EACxC,eAAe,iBAAiB,QAAQ,aAAa;EAErD,OAAO,QAAQ,MAAM,KAAK,SAAS;AACjC,OAAI,kBAAkB,KAAK,MAAM,EAAE;IACjC,MAAM,aAAa,6BAA6B,KAAK,MAAM;AAC3D,QAAI,cAAc,MAAM;KACtB,MAAM,YAAqC;MAAE,GAAG;MAAM;MAAY;AAClE,YAAO,UAAU;AACjB,YAAO;;;AAIX,UAAO;IACP;EACH;AAED,QAAO,OAAO;AACd,QAAO;;AAGT,SAAgB,cAAc,QAAwB;CACpD,MAAM,UAAU,IAAI,aAAa;AACjC,QAAO,IAAI,eAA2B,EACpC,MAAM,MAAM,YAAY;EACtB,MAAM,gBAAgB,QAIhB;AACJ,cAAW,QACT,QAAQ,OACN,UAAU,IAAI,MAAM,UAAU,iBAAiB,IAAI,KAAK,CAAC,MAC1D,CACF;;AAGH,MAAI;AACF,cAAW,MAAM,WAAW,QAAQ;IAClC,MAAM,CAAC,IAAI,MAAM,SAAS;IAE1B,IAAI,OAAgB;AACpB,QAAI,SAAS,SAAS;KACpB,MAAM,aAAa;AAEnB,SAAI,WAAW,SAAS,aACtB,QAAO;MACL,GAAG;MACH,SAAS,qBAAqB,WAAW,QAAQ;MAClD;;AAIL,QAAI,SAAS,cACX,QAAO,qBAAqB,MAAM;AAIpC,iBAAa;KAAE,OADD,IAAI,SAAS,GAAG,KAAK,GAAG,GAAG,KAAK,IAAI,KAAK;KACjC;KAAM,CAAC;;WAExB,OAAO;AACd,gBAAa;IAAE,OAAO;IAAS,MAAM,gBAAgB,MAAM;IAAE,CAAC;;AAGhE,aAAW,OAAO;IAErB,CAAC;;;AAIJ,SAAgB,mBACd,GAAG,SACH;AACA,QAAO,IAAI,+BAA+B;EACxC,gBAAgB,UAAuB;GACrC,MAAM,aACJ,MAAM,OAAO,iBAAiB,qBAAqB,MAAM,GAAG;AAC9D,QAAK,MAAM,UAAU,QACnB,KAAI,OAAO,MAAM,IAAI,MAAM,GAAG,IAAI,WAChC,QAAO,KAAK,MAAM;;EAIxB,OAAO,IAAI,IAAI,QAAQ,SAAS,MAAM,MAAM,KAAK,EAAE,MAAM,CAAC,CAAC;EAC5D,CAAC"}
+{"version":3,"file":"stream.js","names":[],"sources":["../../src/pregel/stream.ts"],"sourcesContent":["import { IterableReadableStream } from \"@langchain/core/utils/stream\";\nimport type { RunnableConfig } from \"@langchain/core/runnables\";\nimport { BaseCallbackHandler } from \"@langchain/core/callbacks/base\";\nimport { Serialized } from \"@langchain/core/load/serializable\";\nimport { isCheckpointEnvelope } from \"../stream/convert.js\";\nimport type { Checkpoint } from \"../stream/types.js\";\nimport type { StreamMode, StreamOutputMap } from \"./types.js\";\nimport { TAG_HIDDEN } from \"../constants.js\";\n\n/**\n * Optional chunk-level metadata carried alongside the payload. Used by\n * `streamEvents(..., { version: \"v3\" })` to emit a companion `checkpoints`\n * protocol event adjacent to the `values` event for the same superstep, so\n * clients can build branching / time-travel UIs without subscribing to a\n * full-state `checkpoints` stream.\n *\n * Companion checkpoint envelopes are emitted as separate\n * ``[namespace, \"checkpoints\", envelope]`` chunks (see\n * ``PregelLoop._emitValuesWithCheckpointMeta``).\n */\nexport interface StreamChunkMeta {\n  /**\n   * Lightweight checkpoint envelope for the superstep that produced the\n   * paired `values` chunk. Shape matches the canonical {@link Checkpoint}\n   * generated from `protocol.cddl`.\n   */\n  checkpoint?: Checkpoint;\n}\n\n// [namespace, streamMode, payload]\nexport type StreamChunk = [string[], StreamMode, unknown];\n\ntype StreamCheckpointsOutput<StreamValues> = StreamOutputMap<\n  \"checkpoints\",\n  false,\n  StreamValues,\n  unknown,\n  string,\n  unknown,\n  unknown,\n  undefined\n>;\n\ntype AnyStreamOutput = StreamOutputMap<\n  StreamMode[],\n  true,\n  unknown,\n  unknown,\n  string,\n  unknown,\n  unknown,\n  undefined\n>;\n\ntype ToolRunInfo = {\n  ns: string[];\n  toolCallId?: string;\n  toolName: string;\n  input: unknown;\n};\n\n/**\n * A wrapper around an IterableReadableStream that allows for aborting the stream when\n * {@link cancel} is called.\n */\nexport class IterableReadableStreamWithAbortSignal<\n  T,\n> extends IterableReadableStream<T> {\n  protected _abortController: AbortController;\n\n  protected _innerReader: ReadableStreamDefaultReader<T>;\n\n  /**\n   * @param readableStream - The stream to wrap.\n   * @param abortController - The abort controller to use. Optional. One will be created if not provided.\n   */\n  constructor(\n    readableStream: ReadableStream<T>,\n    abortController?: AbortController\n  ) {\n    const reader = readableStream.getReader();\n    const ac = abortController ?? new AbortController();\n    super({\n      start(controller: ReadableStreamDefaultController<T>) {\n        return pump();\n        function pump(): Promise<T | undefined> {\n          return reader.read().then(({ done, value }) => {\n            // When no more data needs to be consumed, close the stream\n            if (done) {\n              controller.close();\n              return;\n            }\n            // Enqueue the next data chunk into our target stream\n            controller.enqueue(value);\n            return pump();\n          });\n        }\n      },\n    });\n    this._abortController = ac;\n    this._innerReader = reader;\n  }\n\n  /**\n   * Aborts the stream, abandoning any pending operations in progress. Calling this triggers an\n   * {@link AbortSignal} that is propagated to the tasks that are producing the data for this stream.\n   * @param reason - The reason for aborting the stream. Optional.\n   */\n  override async cancel(reason?: unknown) {\n    this._abortController.abort(reason);\n    this._innerReader.releaseLock();\n  }\n\n  /**\n   * The {@link AbortSignal} for the stream. Aborted when {@link cancel} is called.\n   */\n  get signal() {\n    return this._abortController.signal;\n  }\n}\n\nexport class IterableReadableWritableStream extends IterableReadableStream<StreamChunk> {\n  modes: Set<StreamMode>;\n\n  private controller: ReadableStreamDefaultController;\n\n  private passthroughFn?: (chunk: StreamChunk) => void;\n\n  private _closed: boolean = false;\n\n  get closed() {\n    return this._closed;\n  }\n\n  constructor(params: {\n    passthroughFn?: (chunk: StreamChunk) => void;\n    modes: Set<StreamMode>;\n  }) {\n    let streamControllerPromiseResolver: (\n      controller: ReadableStreamDefaultController\n    ) => void;\n    const streamControllerPromise: Promise<ReadableStreamDefaultController> =\n      new Promise<ReadableStreamDefaultController>((resolve) => {\n        streamControllerPromiseResolver = resolve;\n      });\n\n    super({\n      start: (controller) => {\n        streamControllerPromiseResolver!(controller);\n      },\n    });\n\n    // .start() will always be called before the stream can be interacted\n    // with anyway\n    void streamControllerPromise.then((controller) => {\n      this.controller = controller;\n    });\n\n    this.passthroughFn = params.passthroughFn;\n    this.modes = params.modes;\n  }\n\n  push(chunk: StreamChunk) {\n    // Prevent pushing to a closed stream to avoid race condition errors\n    if (this._closed || !this.controller) {\n      // Silently drop chunks when stream is closed - this is expected behavior\n      // when async operations try to push after stream termination\n      return;\n    }\n\n    // Forward chunk to passthrough function if provided\n    this.passthroughFn?.(chunk);\n\n    // Attempt to enqueue the chunk to the underlying stream\n    this.controller.enqueue(chunk);\n  }\n\n  close() {\n    try {\n      this.controller.close();\n    } catch {\n      // pass\n    } finally {\n      this._closed = true;\n    }\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  error(e: any) {\n    try {\n      this.controller?.error(e);\n    } finally {\n      // Mark the stream as closed so any late `push()` calls from in-flight\n      // parallel tasks are dropped instead of throwing on an errored controller.\n      this._closed = true;\n    }\n  }\n}\n\n/**\n * A callback handler that implements stream_mode=tools.\n * Emits on_tool_start, on_tool_event, on_tool_end, on_tool_error events.\n */\nexport class StreamToolsHandler extends BaseCallbackHandler {\n  name = \"StreamToolsHandler\";\n\n  /** Ensure tool lifecycle callbacks run before tool.invoke returns/errors. */\n  awaitHandlers = true;\n\n  streamFn: (streamChunk: StreamChunk) => void;\n\n  runs: Record<string, ToolRunInfo | undefined> = {};\n\n  constructor(streamFn: (streamChunk: StreamChunk) => void) {\n    super();\n    this.streamFn = streamFn;\n  }\n\n  handleToolStart(\n    _tool: Serialized,\n    input: string,\n    runId: string,\n    _parentRunId?: string,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    runName?: string,\n    toolCallId?: string\n  ) {\n    if (!metadata || (tags && tags.includes(TAG_HIDDEN))) return;\n\n    const ns = (metadata.langgraph_checkpoint_ns as string)?.split(\"|\") ?? [];\n    const info: ToolRunInfo = {\n      ns,\n      toolCallId,\n      toolName: runName ?? \"unknown\",\n      input,\n    };\n    this.runs[runId] = info;\n\n    this.streamFn([\n      ns,\n      \"tools\",\n      {\n        event: \"on_tool_start\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        input,\n      },\n    ]);\n  }\n\n  handleToolEvent(chunk: unknown, runId: string) {\n    const info = this.runs[runId];\n    if (!info) return;\n\n    this.streamFn([\n      info.ns,\n      \"tools\",\n      {\n        event: \"on_tool_event\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        data: chunk,\n      },\n    ]);\n  }\n\n  handleToolEnd(output: unknown, runId: string) {\n    const info = this.runs[runId];\n    delete this.runs[runId];\n    if (!info) return;\n\n    this.streamFn([\n      info.ns,\n      \"tools\",\n      {\n        event: \"on_tool_end\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        output,\n      },\n    ]);\n  }\n\n  handleToolError(err: unknown, runId: string) {\n    const info = this.runs[runId];\n    delete this.runs[runId];\n    if (!info) return;\n\n    this.streamFn([\n      info.ns,\n      \"tools\",\n      {\n        event: \"on_tool_error\",\n        toolCallId: info.toolCallId,\n        name: info.toolName,\n        error: err,\n      },\n    ]);\n  }\n}\n\nfunction _stringifyAsDict(obj: unknown) {\n  return JSON.stringify(obj, function (key: string | number, value: unknown) {\n    const rawValue = this[key];\n    if (\n      rawValue != null &&\n      typeof rawValue === \"object\" &&\n      \"toDict\" in rawValue &&\n      typeof rawValue.toDict === \"function\"\n    ) {\n      const { type, data } = rawValue.toDict();\n      return { ...data, type };\n    }\n\n    return value;\n  });\n}\n\nfunction _serializeError(error: unknown) {\n  // eslint-disable-next-line no-instanceof/no-instanceof\n  if (error instanceof Error) {\n    return { error: error.name, message: error.message };\n  }\n  return { error: \"Error\", message: JSON.stringify(error) };\n}\n\nfunction _isRunnableConfig(\n  config: unknown\n): config is RunnableConfig & { configurable: Record<string, unknown> } {\n  if (typeof config !== \"object\" || config == null) return false;\n  return (\n    \"configurable\" in config &&\n    typeof config.configurable === \"object\" &&\n    config.configurable != null\n  );\n}\n\nfunction _extractCheckpointFromConfig(\n  config: RunnableConfig | null | undefined\n) {\n  if (!_isRunnableConfig(config) || !config.configurable.thread_id) {\n    return null;\n  }\n\n  return {\n    thread_id: config.configurable.thread_id,\n    checkpoint_ns: config.configurable.checkpoint_ns || \"\",\n    checkpoint_id: config.configurable.checkpoint_id || null,\n    checkpoint_map: config.configurable.checkpoint_map || null,\n  };\n}\n\nfunction _serializeConfig(config: unknown) {\n  if (_isRunnableConfig(config)) {\n    const configurable = Object.fromEntries(\n      Object.entries(config.configurable).filter(\n        ([key]) => !key.startsWith(\"__\")\n      )\n    );\n\n    const newConfig = { ...config, configurable };\n    delete newConfig.callbacks;\n    return newConfig;\n  }\n\n  return config;\n}\n\nfunction _serializeCheckpoint(payload: StreamCheckpointsOutput<unknown>) {\n  const result: Record<string, unknown> = {\n    ...payload,\n    checkpoint: _extractCheckpointFromConfig(payload.config),\n    parent_checkpoint: _extractCheckpointFromConfig(payload.parentConfig),\n\n    config: _serializeConfig(payload.config),\n    parent_config: _serializeConfig(payload.parentConfig),\n\n    tasks: payload.tasks.map((task) => {\n      if (_isRunnableConfig(task.state)) {\n        const checkpoint = _extractCheckpointFromConfig(task.state);\n        if (checkpoint != null) {\n          const cloneTask: Record<string, unknown> = { ...task, checkpoint };\n          delete cloneTask.state;\n          return cloneTask;\n        }\n      }\n\n      return task;\n    }),\n  };\n\n  delete result.parentConfig;\n  return result;\n}\n\nexport function toEventStream(stream: AsyncGenerator) {\n  const encoder = new TextEncoder();\n  return new ReadableStream<Uint8Array>({\n    async start(controller) {\n      const enqueueChunk = (sse: {\n        id?: string;\n        event: string;\n        data: unknown;\n      }) => {\n        controller.enqueue(\n          encoder.encode(\n            `event: ${sse.event}\\ndata: ${_stringifyAsDict(sse.data)}\\n\\n`\n          )\n        );\n      };\n\n      try {\n        for await (const payload of stream) {\n          const [ns, mode, chunk] = payload as AnyStreamOutput;\n\n          let data: unknown = chunk;\n          if (mode === \"debug\") {\n            const debugChunk = chunk;\n\n            if (debugChunk.type === \"checkpoint\") {\n              data = {\n                ...debugChunk,\n                payload: _serializeCheckpoint(debugChunk.payload),\n              };\n            }\n          }\n\n          if (mode === \"checkpoints\") {\n            data = _serializeCheckpoint(chunk);\n          }\n\n          const event = ns?.length ? `${mode}|${ns.join(\"|\")}` : mode;\n          enqueueChunk({ event, data });\n        }\n      } catch (error) {\n        enqueueChunk({ event: \"error\", data: _serializeError(error) });\n      }\n\n      controller.close();\n    },\n  });\n}\n\n/** Multiplex subgraph stream chunks into the parent pregel stream. */\nexport function createDuplexStream(\n  ...streams: IterableReadableWritableStream[]\n) {\n  return new IterableReadableWritableStream({\n    passthroughFn: (value: StreamChunk) => {\n      const isEnvelope =\n        value[1] === \"checkpoints\" && isCheckpointEnvelope(value[2]);\n      for (const stream of streams) {\n        if (stream.modes.has(value[1]) || isEnvelope) {\n          stream.push(value);\n        }\n      }\n    },\n    modes: new Set(streams.flatMap((s) => Array.from(s.modes))),\n  });\n}\n"],"mappings":";;;;;;;;;AAiEA,IAAa,wCAAb,cAEU,uBAA0B;CAClC;CAEA;;;;;CAMA,YACE,gBACA,iBACA;EACA,MAAM,SAAS,eAAe,WAAW;EACzC,MAAM,KAAK,mBAAmB,IAAI,iBAAiB;AACnD,QAAM,EACJ,MAAM,YAAgD;AACpD,UAAO,MAAM;GACb,SAAS,OAA+B;AACtC,WAAO,OAAO,MAAM,CAAC,MAAM,EAAE,MAAM,YAAY;AAE7C,SAAI,MAAM;AACR,iBAAW,OAAO;AAClB;;AAGF,gBAAW,QAAQ,MAAM;AACzB,YAAO,MAAM;MACb;;KAGP,CAAC;AACF,OAAK,mBAAmB;AACxB,OAAK,eAAe;;;;;;;CAQtB,MAAe,OAAO,QAAkB;AACtC,OAAK,iBAAiB,MAAM,OAAO;AACnC,OAAK,aAAa,aAAa;;;;;CAMjC,IAAI,SAAS;AACX,SAAO,KAAK,iBAAiB;;;AAIjC,IAAa,iCAAb,cAAoD,uBAAoC;CACtF;CAEA;CAEA;CAEA,UAA2B;CAE3B,IAAI,SAAS;AACX,SAAO,KAAK;;CAGd,YAAY,QAGT;EACD,IAAI;EAGJ,MAAM,0BACJ,IAAI,SAA0C,YAAY;AACxD,qCAAkC;IAClC;AAEJ,QAAM,EACJ,QAAQ,eAAe;AACrB,mCAAiC,WAAW;KAE/C,CAAC;AAIG,0BAAwB,MAAM,eAAe;AAChD,QAAK,aAAa;IAClB;AAEF,OAAK,gBAAgB,OAAO;AAC5B,OAAK,QAAQ,OAAO;;CAGtB,KAAK,OAAoB;AAEvB,MAAI,KAAK,WAAW,CAAC,KAAK,WAGxB;AAIF,OAAK,gBAAgB,MAAM;AAG3B,OAAK,WAAW,QAAQ,MAAM;;CAGhC,QAAQ;AACN,MAAI;AACF,QAAK,WAAW,OAAO;UACjB,WAEE;AACR,QAAK,UAAU;;;CAKnB,MAAM,GAAQ;AACZ,MAAI;AACF,QAAK,YAAY,MAAM,EAAE;YACjB;AAGR,QAAK,UAAU;;;;;;;;AASrB,IAAa,qBAAb,cAAwC,oBAAoB;CAC1D,OAAO;;CAGP,gBAAgB;CAEhB;CAEA,OAAgD,EAAE;CAElD,YAAY,UAA8C;AACxD,SAAO;AACP,OAAK,WAAW;;CAGlB,gBACE,OACA,OACA,OACA,cACA,MACA,UACA,SACA,YACA;AACA,MAAI,CAAC,YAAa,QAAQ,KAAK,SAAA,mBAAoB,CAAG;EAEtD,MAAM,KAAM,SAAS,yBAAoC,MAAM,IAAI,IAAI,EAAE;EACzE,MAAM,OAAoB;GACxB;GACA;GACA,UAAU,WAAW;GACrB;GACD;AACD,OAAK,KAAK,SAAS;AAEnB,OAAK,SAAS;GACZ;GACA;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX;IACD;GACF,CAAC;;CAGJ,gBAAgB,OAAgB,OAAe;EAC7C,MAAM,OAAO,KAAK,KAAK;AACvB,MAAI,CAAC,KAAM;AAEX,OAAK,SAAS;GACZ,KAAK;GACL;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX,MAAM;IACP;GACF,CAAC;;CAGJ,cAAc,QAAiB,OAAe;EAC5C,MAAM,OAAO,KAAK,KAAK;AACvB,SAAO,KAAK,KAAK;AACjB,MAAI,CAAC,KAAM;AAEX,OAAK,SAAS;GACZ,KAAK;GACL;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX;IACD;GACF,CAAC;;CAGJ,gBAAgB,KAAc,OAAe;EAC3C,MAAM,OAAO,KAAK,KAAK;AACvB,SAAO,KAAK,KAAK;AACjB,MAAI,CAAC,KAAM;AAEX,OAAK,SAAS;GACZ,KAAK;GACL;GACA;IACE,OAAO;IACP,YAAY,KAAK;IACjB,MAAM,KAAK;IACX,OAAO;IACR;GACF,CAAC;;;AAIN,SAAS,iBAAiB,KAAc;AACtC,QAAO,KAAK,UAAU,KAAK,SAAU,KAAsB,OAAgB;EACzE,MAAM,WAAW,KAAK;AACtB,MACE,YAAY,QACZ,OAAO,aAAa,YACpB,YAAY,YACZ,OAAO,SAAS,WAAW,YAC3B;GACA,MAAM,EAAE,MAAM,SAAS,SAAS,QAAQ;AACxC,UAAO;IAAE,GAAG;IAAM;IAAM;;AAG1B,SAAO;GACP;;AAGJ,SAAS,gBAAgB,OAAgB;AAEvC,KAAI,iBAAiB,MACnB,QAAO;EAAE,OAAO,MAAM;EAAM,SAAS,MAAM;EAAS;AAEtD,QAAO;EAAE,OAAO;EAAS,SAAS,KAAK,UAAU,MAAM;EAAE;;AAG3D,SAAS,kBACP,QACsE;AACtE,KAAI,OAAO,WAAW,YAAY,UAAU,KAAM,QAAO;AACzD,QACE,kBAAkB,UAClB,OAAO,OAAO,iBAAiB,YAC/B,OAAO,gBAAgB;;AAI3B,SAAS,6BACP,QACA;AACA,KAAI,CAAC,kBAAkB,OAAO,IAAI,CAAC,OAAO,aAAa,UACrD,QAAO;AAGT,QAAO;EACL,WAAW,OAAO,aAAa;EAC/B,eAAe,OAAO,aAAa,iBAAiB;EACpD,eAAe,OAAO,aAAa,iBAAiB;EACpD,gBAAgB,OAAO,aAAa,kBAAkB;EACvD;;AAGH,SAAS,iBAAiB,QAAiB;AACzC,KAAI,kBAAkB,OAAO,EAAE;EAC7B,MAAM,eAAe,OAAO,YAC1B,OAAO,QAAQ,OAAO,aAAa,CAAC,QACjC,CAAC,SAAS,CAAC,IAAI,WAAW,KAAK,CACjC,CACF;EAED,MAAM,YAAY;GAAE,GAAG;GAAQ;GAAc;AAC7C,SAAO,UAAU;AACjB,SAAO;;AAGT,QAAO;;AAGT,SAAS,qBAAqB,SAA2C;CACvE,MAAM,SAAkC;EACtC,GAAG;EACH,YAAY,6BAA6B,QAAQ,OAAO;EACxD,mBAAmB,6BAA6B,QAAQ,aAAa;EAErE,QAAQ,iBAAiB,QAAQ,OAAO;EACxC,eAAe,iBAAiB,QAAQ,aAAa;EAErD,OAAO,QAAQ,MAAM,KAAK,SAAS;AACjC,OAAI,kBAAkB,KAAK,MAAM,EAAE;IACjC,MAAM,aAAa,6BAA6B,KAAK,MAAM;AAC3D,QAAI,cAAc,MAAM;KACtB,MAAM,YAAqC;MAAE,GAAG;MAAM;MAAY;AAClE,YAAO,UAAU;AACjB,YAAO;;;AAIX,UAAO;IACP;EACH;AAED,QAAO,OAAO;AACd,QAAO;;AAGT,SAAgB,cAAc,QAAwB;CACpD,MAAM,UAAU,IAAI,aAAa;AACjC,QAAO,IAAI,eAA2B,EACpC,MAAM,MAAM,YAAY;EACtB,MAAM,gBAAgB,QAIhB;AACJ,cAAW,QACT,QAAQ,OACN,UAAU,IAAI,MAAM,UAAU,iBAAiB,IAAI,KAAK,CAAC,MAC1D,CACF;;AAGH,MAAI;AACF,cAAW,MAAM,WAAW,QAAQ;IAClC,MAAM,CAAC,IAAI,MAAM,SAAS;IAE1B,IAAI,OAAgB;AACpB,QAAI,SAAS,SAAS;KACpB,MAAM,aAAa;AAEnB,SAAI,WAAW,SAAS,aACtB,QAAO;MACL,GAAG;MACH,SAAS,qBAAqB,WAAW,QAAQ;MAClD;;AAIL,QAAI,SAAS,cACX,QAAO,qBAAqB,MAAM;AAIpC,iBAAa;KAAE,OADD,IAAI,SAAS,GAAG,KAAK,GAAG,GAAG,KAAK,IAAI,KAAK;KACjC;KAAM,CAAC;;WAExB,OAAO;AACd,gBAAa;IAAE,OAAO;IAAS,MAAM,gBAAgB,MAAM;IAAE,CAAC;;AAGhE,aAAW,OAAO;IAErB,CAAC;;;AAIJ,SAAgB,mBACd,GAAG,SACH;AACA,QAAO,IAAI,+BAA+B;EACxC,gBAAgB,UAAuB;GACrC,MAAM,aACJ,MAAM,OAAO,iBAAiB,qBAAqB,MAAM,GAAG;AAC9D,QAAK,MAAM,UAAU,QACnB,KAAI,OAAO,MAAM,IAAI,MAAM,GAAG,IAAI,WAChC,QAAO,KAAK,MAAM;;EAIxB,OAAO,IAAI,IAAI,QAAQ,SAAS,MAAM,MAAM,KAAK,EAAE,MAAM,CAAC,CAAC;EAC5D,CAAC"}

package/dist/pregel/timeout.cjs

@@ -0,0 +1,216 @@
+const require_constants = require("../constants.cjs");
+const require_errors = require("../errors.cjs");
+const require_index = require("./utils/index.cjs");
+let _langchain_core_callbacks_base = require("@langchain/core/callbacks/base");
+//#region src/pregel/timeout.ts
+/**
+* Tracks the live progress state of a single timed node attempt.
+*
+* The scope guards observable-progress channels (writes, child-task
+* scheduling, the custom stream writer, callbacks) so that:
+*
+* 1. `idleTimeout` is refreshed whenever the node makes progress, and
+* 2. once the attempt is `close()`d after a timeout fires, late writes/calls
+*    from the still-running background task are dropped (Python parity:
+*    buffered writes from the failed attempt must not leak into the
+*    checkpoint).
+*
+* @internal
+*/
+var TimedAttemptScope = class {
+	active = true;
+	lastProgress = Date.now();
+	refreshOn;
+	constructor(refreshOn) {
+		this.refreshOn = refreshOn;
+	}
+	/** Record progress now. Always honored (used by `runtime.heartbeat()`). */
+	touch() {
+		this.lastProgress = Date.now();
+	}
+	/**
+	* Record progress for an automatic signal (write/call/stream/callback).
+	* No-op when `refreshOn === "heartbeat"`, where only explicit heartbeats
+	* count as progress.
+	*/
+	autoTouch() {
+		if (this.refreshOn === "auto") this.lastProgress = Date.now();
+	}
+	close() {
+		this.active = false;
+	}
+};
+/**
+* Callback handler that refreshes a {@link TimedAttemptScope} on any LangChain
+* callback event emitted under the node's run. Because it is attached via
+* `config.callbacks`, it only observes events from runs descended from this
+* node's attempt, not from sibling nodes.
+*
+* @internal
+*/
+var IdleProgressCallbackHandler = class extends _langchain_core_callbacks_base.BaseCallbackHandler {
+	name = "IdleProgressCallbackHandler";
+	awaitHandlers = false;
+	#scope;
+	constructor(scope) {
+		super();
+		this.#scope = scope;
+	}
+	#touch = () => {
+		this.#scope.autoTouch();
+	};
+	handleLLMStart = this.#touch;
+	handleChatModelStart = this.#touch;
+	handleLLMNewToken = this.#touch;
+	handleLLMEnd = this.#touch;
+	handleLLMError = this.#touch;
+	handleChainStart = this.#touch;
+	handleChainEnd = this.#touch;
+	handleChainError = this.#touch;
+	handleToolStart = this.#touch;
+	handleToolEnd = this.#touch;
+	handleToolError = this.#touch;
+	handleText = this.#touch;
+	handleRetrieverStart = this.#touch;
+	handleRetrieverEnd = this.#touch;
+	handleRetrieverError = this.#touch;
+	handleCustomEvent = this.#touch;
+};
+/**
+* Wrap the node attempt config so observable-progress signals refresh the idle
+* clock and are dropped once the scope is closed. Also injects
+* {@link LangGraphRunnableConfig.heartbeat}.
+*/
+function wrapConfig(config, scope, policy, taskName) {
+	const configurable = config.configurable ?? {};
+	const patch = {};
+	const send = configurable[require_constants.CONFIG_KEY_SEND];
+	if (typeof send === "function") patch[require_constants.CONFIG_KEY_SEND] = (writes) => {
+		if (!scope.active) return void 0;
+		if (writes && writes.length) scope.autoTouch();
+		return send(writes);
+	};
+	const callFn = configurable[require_constants.CONFIG_KEY_CALL];
+	if (typeof callFn === "function") patch[require_constants.CONFIG_KEY_CALL] = (...args) => {
+		if (!scope.active) throw new Error(`Node "${taskName}" attempt was cancelled after its timeout fired`);
+		scope.autoTouch();
+		return callFn(...args);
+	};
+	const wrapped = { ...Object.keys(patch).length > 0 ? require_index.patchConfigurable(config, patch) : config };
+	wrapped.heartbeat = () => {
+		if (policy.idleTimeout !== void 0) scope.touch();
+	};
+	if (typeof wrapped.writer === "function") {
+		const writer = wrapped.writer;
+		wrapped.writer = ((chunk) => {
+			if (!scope.active) return void 0;
+			scope.autoTouch();
+			return writer(chunk);
+		});
+	}
+	if ((policy.refreshOn ?? "auto") === "auto" && policy.idleTimeout !== void 0) {
+		const handler = new IdleProgressCallbackHandler(scope);
+		const cb = wrapped.callbacks;
+		if (cb === void 0) wrapped.callbacks = [handler];
+		else if (Array.isArray(cb)) wrapped.callbacks = [...cb, handler];
+		else {
+			const copied = cb.copy();
+			copied.addHandler(handler, true);
+			wrapped.callbacks = copied;
+		}
+	}
+	return wrapped;
+}
+/**
+* Run a single node attempt under a {@link TimeoutPolicy}.
+*
+* Races the node invocation against per-attempt run/idle watchdogs. On
+* successful completion (or node error), returns/rethrows normally. When a
+* watchdog fires first, the scope is closed, the task's buffered writes are
+* dropped, the attempt's {@link AbortSignal} is aborted, and a
+* {@link NodeTimeoutError} is thrown.
+*
+* @internal
+*/
+async function runAttemptWithTimeout(task, config, policy, invoke) {
+	const scope = new TimedAttemptScope(policy.refreshOn ?? "auto");
+	const timeoutController = new AbortController();
+	const { signal: composedSignal, dispose } = require_index.combineAbortSignals(config.signal, timeoutController.signal);
+	const scopedConfig = wrapConfig({
+		...config,
+		signal: composedSignal
+	}, scope, policy, String(task.name));
+	const start = Date.now();
+	const nodeOutcome = invoke(scopedConfig).then((value) => ({
+		type: "ok",
+		value
+	}), (error) => ({
+		type: "err",
+		error
+	}));
+	let runTimer;
+	let idleTimer;
+	const clearTimers = () => {
+		if (runTimer !== void 0) clearTimeout(runTimer);
+		if (idleTimer !== void 0) clearTimeout(idleTimer);
+	};
+	const watchdog = new Promise((resolve) => {
+		if (policy.runTimeout !== void 0) runTimer = setTimeout(() => resolve({
+			type: "timeout",
+			kind: "run"
+		}), policy.runTimeout);
+		if (policy.idleTimeout !== void 0) {
+			const idleMs = policy.idleTimeout;
+			const checkIdle = () => {
+				const remaining = scope.lastProgress + idleMs - Date.now();
+				if (remaining <= 0) resolve({
+					type: "timeout",
+					kind: "idle"
+				});
+				else idleTimer = setTimeout(checkIdle, remaining);
+			};
+			idleTimer = setTimeout(checkIdle, idleMs);
+		}
+	});
+	let outcome;
+	try {
+		outcome = await Promise.race([nodeOutcome, watchdog]);
+	} finally {
+		clearTimers();
+	}
+	if (outcome.type !== "timeout") {
+		const now = Date.now();
+		if (policy.runTimeout !== void 0 && now - start >= policy.runTimeout) outcome = {
+			type: "timeout",
+			kind: "run"
+		};
+		else if (policy.idleTimeout !== void 0 && now - scope.lastProgress >= policy.idleTimeout) outcome = {
+			type: "timeout",
+			kind: "idle"
+		};
+	}
+	if (outcome.type === "ok") {
+		dispose?.();
+		return outcome.value;
+	}
+	if (outcome.type === "err") {
+		dispose?.();
+		throw outcome.error;
+	}
+	const elapsed = Date.now() - start;
+	scope.close();
+	task.writes.splice(0, task.writes.length);
+	timeoutController.abort();
+	dispose?.();
+	throw new require_errors.NodeTimeoutError({
+		node: String(task.name),
+		elapsed,
+		kind: outcome.kind,
+		runTimeout: policy.runTimeout,
+		idleTimeout: policy.idleTimeout
+	});
+}
+//#endregion
+exports.runAttemptWithTimeout = runAttemptWithTimeout;
+
+//# sourceMappingURL=timeout.cjs.map

package/dist/pregel/timeout.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeout.cjs","names":["BaseCallbackHandler","#scope","#touch","CONFIG_KEY_SEND","CONFIG_KEY_CALL","patchConfigurable","combineAbortSignals","NodeTimeoutError"],"sources":["../../src/pregel/timeout.ts"],"sourcesContent":["import { BaseCallbackHandler } from \"@langchain/core/callbacks/base\";\nimport { PendingWrite } from \"@langchain/langgraph-checkpoint\";\nimport { CONFIG_KEY_CALL, CONFIG_KEY_SEND } from \"../constants.js\";\nimport { NodeTimeoutError } from \"../errors.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\nimport { PregelExecutableTask } from \"./types.js\";\nimport {\n  combineAbortSignals,\n  patchConfigurable,\n  TimeoutPolicy,\n} from \"./utils/index.js\";\n\n/**\n * Tracks the live progress state of a single timed node attempt.\n *\n * The scope guards observable-progress channels (writes, child-task\n * scheduling, the custom stream writer, callbacks) so that:\n *\n * 1. `idleTimeout` is refreshed whenever the node makes progress, and\n * 2. once the attempt is `close()`d after a timeout fires, late writes/calls\n *    from the still-running background task are dropped (Python parity:\n *    buffered writes from the failed attempt must not leak into the\n *    checkpoint).\n *\n * @internal\n */\nclass TimedAttemptScope {\n  active = true;\n\n  lastProgress = Date.now();\n\n  private refreshOn: \"auto\" | \"heartbeat\";\n\n  constructor(refreshOn: \"auto\" | \"heartbeat\") {\n    this.refreshOn = refreshOn;\n  }\n\n  /** Record progress now. Always honored (used by `runtime.heartbeat()`). */\n  touch(): void {\n    this.lastProgress = Date.now();\n  }\n\n  /**\n   * Record progress for an automatic signal (write/call/stream/callback).\n   * No-op when `refreshOn === \"heartbeat\"`, where only explicit heartbeats\n   * count as progress.\n   */\n  autoTouch(): void {\n    if (this.refreshOn === \"auto\") {\n      this.lastProgress = Date.now();\n    }\n  }\n\n  close(): void {\n    this.active = false;\n  }\n}\n\n/**\n * Callback handler that refreshes a {@link TimedAttemptScope} on any LangChain\n * callback event emitted under the node's run. Because it is attached via\n * `config.callbacks`, it only observes events from runs descended from this\n * node's attempt, not from sibling nodes.\n *\n * @internal\n */\nclass IdleProgressCallbackHandler extends BaseCallbackHandler {\n  name = \"IdleProgressCallbackHandler\";\n\n  awaitHandlers = false;\n\n  #scope: TimedAttemptScope;\n\n  constructor(scope: TimedAttemptScope) {\n    super();\n    this.#scope = scope;\n  }\n\n  #touch = () => {\n    this.#scope.autoTouch();\n  };\n\n  handleLLMStart = this.#touch;\n\n  handleChatModelStart = this.#touch;\n\n  handleLLMNewToken = this.#touch;\n\n  handleLLMEnd = this.#touch;\n\n  handleLLMError = this.#touch;\n\n  handleChainStart = this.#touch;\n\n  handleChainEnd = this.#touch;\n\n  handleChainError = this.#touch;\n\n  handleToolStart = this.#touch;\n\n  handleToolEnd = this.#touch;\n\n  handleToolError = this.#touch;\n\n  handleText = this.#touch;\n\n  handleRetrieverStart = this.#touch;\n\n  handleRetrieverEnd = this.#touch;\n\n  handleRetrieverError = this.#touch;\n\n  handleCustomEvent = this.#touch;\n}\n\n/**\n * Wrap the node attempt config so observable-progress signals refresh the idle\n * clock and are dropped once the scope is closed. Also injects\n * {@link LangGraphRunnableConfig.heartbeat}.\n */\nfunction wrapConfig(\n  config: LangGraphRunnableConfig,\n  scope: TimedAttemptScope,\n  policy: TimeoutPolicy,\n  taskName: string\n): LangGraphRunnableConfig {\n  const configurable = config.configurable ?? {};\n  const patch: Record<string, unknown> = {};\n\n  const send = configurable[CONFIG_KEY_SEND];\n  if (typeof send === \"function\") {\n    patch[CONFIG_KEY_SEND] = (writes: PendingWrite[]) => {\n      if (!scope.active) return undefined;\n      if (writes && writes.length) scope.autoTouch();\n      return send(writes);\n    };\n  }\n\n  const callFn = configurable[CONFIG_KEY_CALL];\n  if (typeof callFn === \"function\") {\n    patch[CONFIG_KEY_CALL] = (...args: unknown[]) => {\n      if (!scope.active) {\n        throw new Error(\n          `Node \"${taskName}\" attempt was cancelled after its timeout fired`\n        );\n      }\n      scope.autoTouch();\n      return (callFn as (...a: unknown[]) => unknown)(...args);\n    };\n  }\n\n  const out: LangGraphRunnableConfig =\n    Object.keys(patch).length > 0 ? patchConfigurable(config, patch) : config;\n  const wrapped: LangGraphRunnableConfig = { ...out };\n\n  // `heartbeat` always resets the idle clock, even under `refreshOn:\n  // \"heartbeat\"`. It is a no-op when no idle timeout is configured.\n  wrapped.heartbeat = () => {\n    if (policy.idleTimeout !== undefined) scope.touch();\n  };\n\n  if (typeof wrapped.writer === \"function\") {\n    const writer = wrapped.writer;\n    wrapped.writer = ((chunk: unknown) => {\n      if (!scope.active) return undefined;\n      scope.autoTouch();\n      return (writer as (c: unknown) => unknown)(chunk);\n    }) as typeof wrapped.writer;\n  }\n\n  if (\n    (policy.refreshOn ?? \"auto\") === \"auto\" &&\n    policy.idleTimeout !== undefined\n  ) {\n    const handler = new IdleProgressCallbackHandler(scope);\n    const cb = wrapped.callbacks;\n    if (cb === undefined) {\n      wrapped.callbacks = [handler];\n    } else if (Array.isArray(cb)) {\n      wrapped.callbacks = [...cb, handler];\n    } else {\n      const copied = cb.copy();\n      copied.addHandler(handler, true);\n      wrapped.callbacks = copied;\n    }\n  }\n\n  return wrapped;\n}\n\ntype AttemptOutcome<T> =\n  | { type: \"ok\"; value: T }\n  | { type: \"err\"; error: unknown }\n  | { type: \"timeout\"; kind: \"run\" | \"idle\" };\n\n/**\n * Run a single node attempt under a {@link TimeoutPolicy}.\n *\n * Races the node invocation against per-attempt run/idle watchdogs. On\n * successful completion (or node error), returns/rethrows normally. When a\n * watchdog fires first, the scope is closed, the task's buffered writes are\n * dropped, the attempt's {@link AbortSignal} is aborted, and a\n * {@link NodeTimeoutError} is thrown.\n *\n * @internal\n */\nexport async function runAttemptWithTimeout<T>(\n  task: PregelExecutableTask<string, string>,\n  config: LangGraphRunnableConfig,\n  policy: TimeoutPolicy,\n  invoke: (scopedConfig: LangGraphRunnableConfig) => Promise<T>\n): Promise<T> {\n  const refreshOn = policy.refreshOn ?? \"auto\";\n  const scope = new TimedAttemptScope(refreshOn);\n\n  const timeoutController = new AbortController();\n  const { signal: composedSignal, dispose } = combineAbortSignals(\n    config.signal,\n    timeoutController.signal\n  );\n\n  const scopedConfig = wrapConfig(\n    { ...config, signal: composedSignal },\n    scope,\n    policy,\n    String(task.name)\n  );\n\n  const start = Date.now();\n  const bg = invoke(scopedConfig);\n\n  // Normalize the node result into an outcome. Catching the rejection here\n  // (rather than via a separate `bg.catch`) keeps `bg` handled even when a\n  // watchdog wins the race below, so its late settlement can never surface as\n  // an unhandled rejection.\n  const nodeOutcome: Promise<AttemptOutcome<T>> = bg.then(\n    (value) => ({ type: \"ok\", value }),\n    (error) => ({ type: \"err\", error })\n  );\n\n  let runTimer: ReturnType<typeof setTimeout> | undefined;\n  let idleTimer: ReturnType<typeof setTimeout> | undefined;\n  const clearTimers = () => {\n    if (runTimer !== undefined) clearTimeout(runTimer);\n    if (idleTimer !== undefined) clearTimeout(idleTimer);\n  };\n\n  // The watchdog never rejects: it either resolves with a timeout outcome or\n  // stays pending forever (and is dropped) when the node wins the race. The\n  // idle timer re-arms itself off `scope.lastProgress`, so it can't be a single\n  // fixed `setTimeout`.\n  const watchdog = new Promise<AttemptOutcome<T>>((resolve) => {\n    if (policy.runTimeout !== undefined) {\n      runTimer = setTimeout(\n        () => resolve({ type: \"timeout\", kind: \"run\" }),\n        policy.runTimeout\n      );\n    }\n\n    if (policy.idleTimeout !== undefined) {\n      const idleMs = policy.idleTimeout;\n      const checkIdle = () => {\n        const remaining = scope.lastProgress + idleMs - Date.now();\n        if (remaining <= 0) {\n          resolve({ type: \"timeout\", kind: \"idle\" });\n        } else {\n          idleTimer = setTimeout(checkIdle, remaining);\n        }\n      };\n      idleTimer = setTimeout(checkIdle, idleMs);\n    }\n  });\n\n  let outcome: AttemptOutcome<T>;\n  try {\n    outcome = await Promise.race([nodeOutcome, watchdog]);\n  } finally {\n    clearTimers();\n  }\n\n  // Watchdog timers are macrotasks and cannot fire while a synchronous\n  // (CPU-bound) node blocks the event loop. Such a node — or one with a long\n  // synchronous prefix — can therefore settle the race as \"ok\"/\"err\" before an\n  // already-expired timer ever runs, bypassing the documented hard wall-clock\n  // cap. Re-check the budget against wall-clock time here so the caps hold for\n  // synchronous nodes too. (For genuinely async work the watchdog already\n  // wins the race, so this only catches what timers structurally cannot.)\n  if (outcome.type !== \"timeout\") {\n    const now = Date.now();\n    if (policy.runTimeout !== undefined && now - start >= policy.runTimeout) {\n      outcome = { type: \"timeout\", kind: \"run\" };\n    } else if (\n      policy.idleTimeout !== undefined &&\n      now - scope.lastProgress >= policy.idleTimeout\n    ) {\n      outcome = { type: \"timeout\", kind: \"idle\" };\n    }\n  }\n\n  if (outcome.type === \"ok\") {\n    dispose?.();\n    return outcome.value;\n  }\n  if (outcome.type === \"err\") {\n    dispose?.();\n    throw outcome.error;\n  }\n\n  // A watchdog fired: close the scope (drop late writes/calls), discard the\n  // attempt's buffered writes, abort the node, and surface a NodeTimeoutError.\n  const elapsed = Date.now() - start;\n  scope.close();\n  task.writes.splice(0, task.writes.length);\n  // Abort BEFORE disposing the combined signal so the abort actually\n  // propagates to the node's signal (dispose removes the relay listeners).\n  timeoutController.abort();\n  dispose?.();\n\n  throw new NodeTimeoutError({\n    node: String(task.name),\n    elapsed,\n    kind: outcome.kind,\n    runTimeout: policy.runTimeout,\n    idleTimeout: policy.idleTimeout,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AA0BA,IAAM,oBAAN,MAAwB;CACtB,SAAS;CAET,eAAe,KAAK,KAAK;CAEzB;CAEA,YAAY,WAAiC;AAC3C,OAAK,YAAY;;;CAInB,QAAc;AACZ,OAAK,eAAe,KAAK,KAAK;;;;;;;CAQhC,YAAkB;AAChB,MAAI,KAAK,cAAc,OACrB,MAAK,eAAe,KAAK,KAAK;;CAIlC,QAAc;AACZ,OAAK,SAAS;;;;;;;;;;;AAYlB,IAAM,8BAAN,cAA0CA,+BAAAA,oBAAoB;CAC5D,OAAO;CAEP,gBAAgB;CAEhB;CAEA,YAAY,OAA0B;AACpC,SAAO;AACP,QAAA,QAAc;;CAGhB,eAAe;AACb,QAAA,MAAY,WAAW;;CAGzB,iBAAiB,MAAA;CAEjB,uBAAuB,MAAA;CAEvB,oBAAoB,MAAA;CAEpB,eAAe,MAAA;CAEf,iBAAiB,MAAA;CAEjB,mBAAmB,MAAA;CAEnB,iBAAiB,MAAA;CAEjB,mBAAmB,MAAA;CAEnB,kBAAkB,MAAA;CAElB,gBAAgB,MAAA;CAEhB,kBAAkB,MAAA;CAElB,aAAa,MAAA;CAEb,uBAAuB,MAAA;CAEvB,qBAAqB,MAAA;CAErB,uBAAuB,MAAA;CAEvB,oBAAoB,MAAA;;;;;;;AAQtB,SAAS,WACP,QACA,OACA,QACA,UACyB;CACzB,MAAM,eAAe,OAAO,gBAAgB,EAAE;CAC9C,MAAM,QAAiC,EAAE;CAEzC,MAAM,OAAO,aAAaG,kBAAAA;AAC1B,KAAI,OAAO,SAAS,WAClB,OAAMA,kBAAAA,oBAAoB,WAA2B;AACnD,MAAI,CAAC,MAAM,OAAQ,QAAO,KAAA;AAC1B,MAAI,UAAU,OAAO,OAAQ,OAAM,WAAW;AAC9C,SAAO,KAAK,OAAO;;CAIvB,MAAM,SAAS,aAAaC,kBAAAA;AAC5B,KAAI,OAAO,WAAW,WACpB,OAAMA,kBAAAA,oBAAoB,GAAG,SAAoB;AAC/C,MAAI,CAAC,MAAM,OACT,OAAM,IAAI,MACR,SAAS,SAAS,iDACnB;AAEH,QAAM,WAAW;AACjB,SAAQ,OAAwC,GAAG,KAAK;;CAM5D,MAAM,UAAmC,EAAE,GADzC,OAAO,KAAK,MAAM,CAAC,SAAS,IAAIC,cAAAA,kBAAkB,QAAQ,MAAM,GAAG,QAClB;AAInD,SAAQ,kBAAkB;AACxB,MAAI,OAAO,gBAAgB,KAAA,EAAW,OAAM,OAAO;;AAGrD,KAAI,OAAO,QAAQ,WAAW,YAAY;EACxC,MAAM,SAAS,QAAQ;AACvB,UAAQ,WAAW,UAAmB;AACpC,OAAI,CAAC,MAAM,OAAQ,QAAO,KAAA;AAC1B,SAAM,WAAW;AACjB,UAAQ,OAAmC,MAAM;;;AAIrD,MACG,OAAO,aAAa,YAAY,UACjC,OAAO,gBAAgB,KAAA,GACvB;EACA,MAAM,UAAU,IAAI,4BAA4B,MAAM;EACtD,MAAM,KAAK,QAAQ;AACnB,MAAI,OAAO,KAAA,EACT,SAAQ,YAAY,CAAC,QAAQ;WACpB,MAAM,QAAQ,GAAG,CAC1B,SAAQ,YAAY,CAAC,GAAG,IAAI,QAAQ;OAC/B;GACL,MAAM,SAAS,GAAG,MAAM;AACxB,UAAO,WAAW,SAAS,KAAK;AAChC,WAAQ,YAAY;;;AAIxB,QAAO;;;;;;;;;;;;;AAmBT,eAAsB,sBACpB,MACA,QACA,QACA,QACY;CAEZ,MAAM,QAAQ,IAAI,kBADA,OAAO,aAAa,OACQ;CAE9C,MAAM,oBAAoB,IAAI,iBAAiB;CAC/C,MAAM,EAAE,QAAQ,gBAAgB,YAAYC,cAAAA,oBAC1C,OAAO,QACP,kBAAkB,OACnB;CAED,MAAM,eAAe,WACnB;EAAE,GAAG;EAAQ,QAAQ;EAAgB,EACrC,OACA,QACA,OAAO,KAAK,KAAK,CAClB;CAED,MAAM,QAAQ,KAAK,KAAK;CAOxB,MAAM,cANK,OAAO,aAAa,CAMoB,MAChD,WAAW;EAAE,MAAM;EAAM;EAAO,IAChC,WAAW;EAAE,MAAM;EAAO;EAAO,EACnC;CAED,IAAI;CACJ,IAAI;CACJ,MAAM,oBAAoB;AACxB,MAAI,aAAa,KAAA,EAAW,cAAa,SAAS;AAClD,MAAI,cAAc,KAAA,EAAW,cAAa,UAAU;;CAOtD,MAAM,WAAW,IAAI,SAA4B,YAAY;AAC3D,MAAI,OAAO,eAAe,KAAA,EACxB,YAAW,iBACH,QAAQ;GAAE,MAAM;GAAW,MAAM;GAAO,CAAC,EAC/C,OAAO,WACR;AAGH,MAAI,OAAO,gBAAgB,KAAA,GAAW;GACpC,MAAM,SAAS,OAAO;GACtB,MAAM,kBAAkB;IACtB,MAAM,YAAY,MAAM,eAAe,SAAS,KAAK,KAAK;AAC1D,QAAI,aAAa,EACf,SAAQ;KAAE,MAAM;KAAW,MAAM;KAAQ,CAAC;QAE1C,aAAY,WAAW,WAAW,UAAU;;AAGhD,eAAY,WAAW,WAAW,OAAO;;GAE3C;CAEF,IAAI;AACJ,KAAI;AACF,YAAU,MAAM,QAAQ,KAAK,CAAC,aAAa,SAAS,CAAC;WAC7C;AACR,eAAa;;AAUf,KAAI,QAAQ,SAAS,WAAW;EAC9B,MAAM,MAAM,KAAK,KAAK;AACtB,MAAI,OAAO,eAAe,KAAA,KAAa,MAAM,SAAS,OAAO,WAC3D,WAAU;GAAE,MAAM;GAAW,MAAM;GAAO;WAE1C,OAAO,gBAAgB,KAAA,KACvB,MAAM,MAAM,gBAAgB,OAAO,YAEnC,WAAU;GAAE,MAAM;GAAW,MAAM;GAAQ;;AAI/C,KAAI,QAAQ,SAAS,MAAM;AACzB,aAAW;AACX,SAAO,QAAQ;;AAEjB,KAAI,QAAQ,SAAS,OAAO;AAC1B,aAAW;AACX,QAAM,QAAQ;;CAKhB,MAAM,UAAU,KAAK,KAAK,GAAG;AAC7B,OAAM,OAAO;AACb,MAAK,OAAO,OAAO,GAAG,KAAK,OAAO,OAAO;AAGzC,mBAAkB,OAAO;AACzB,YAAW;AAEX,OAAM,IAAIC,eAAAA,iBAAiB;EACzB,MAAM,OAAO,KAAK,KAAK;EACvB;EACA,MAAM,QAAQ;EACd,YAAY,OAAO;EACnB,aAAa,OAAO;EACrB,CAAC"}

package/dist/pregel/timeout.js

@@ -0,0 +1,216 @@
+import { CONFIG_KEY_CALL, CONFIG_KEY_SEND } from "../constants.js";
+import { NodeTimeoutError } from "../errors.js";
+import { combineAbortSignals, patchConfigurable } from "./utils/index.js";
+import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
+//#region src/pregel/timeout.ts
+/**
+* Tracks the live progress state of a single timed node attempt.
+*
+* The scope guards observable-progress channels (writes, child-task
+* scheduling, the custom stream writer, callbacks) so that:
+*
+* 1. `idleTimeout` is refreshed whenever the node makes progress, and
+* 2. once the attempt is `close()`d after a timeout fires, late writes/calls
+*    from the still-running background task are dropped (Python parity:
+*    buffered writes from the failed attempt must not leak into the
+*    checkpoint).
+*
+* @internal
+*/
+var TimedAttemptScope = class {
+	active = true;
+	lastProgress = Date.now();
+	refreshOn;
+	constructor(refreshOn) {
+		this.refreshOn = refreshOn;
+	}
+	/** Record progress now. Always honored (used by `runtime.heartbeat()`). */
+	touch() {
+		this.lastProgress = Date.now();
+	}
+	/**
+	* Record progress for an automatic signal (write/call/stream/callback).
+	* No-op when `refreshOn === "heartbeat"`, where only explicit heartbeats
+	* count as progress.
+	*/
+	autoTouch() {
+		if (this.refreshOn === "auto") this.lastProgress = Date.now();
+	}
+	close() {
+		this.active = false;
+	}
+};
+/**
+* Callback handler that refreshes a {@link TimedAttemptScope} on any LangChain
+* callback event emitted under the node's run. Because it is attached via
+* `config.callbacks`, it only observes events from runs descended from this
+* node's attempt, not from sibling nodes.
+*
+* @internal
+*/
+var IdleProgressCallbackHandler = class extends BaseCallbackHandler {
+	name = "IdleProgressCallbackHandler";
+	awaitHandlers = false;
+	#scope;
+	constructor(scope) {
+		super();
+		this.#scope = scope;
+	}
+	#touch = () => {
+		this.#scope.autoTouch();
+	};
+	handleLLMStart = this.#touch;
+	handleChatModelStart = this.#touch;
+	handleLLMNewToken = this.#touch;
+	handleLLMEnd = this.#touch;
+	handleLLMError = this.#touch;
+	handleChainStart = this.#touch;
+	handleChainEnd = this.#touch;
+	handleChainError = this.#touch;
+	handleToolStart = this.#touch;
+	handleToolEnd = this.#touch;
+	handleToolError = this.#touch;
+	handleText = this.#touch;
+	handleRetrieverStart = this.#touch;
+	handleRetrieverEnd = this.#touch;
+	handleRetrieverError = this.#touch;
+	handleCustomEvent = this.#touch;
+};
+/**
+* Wrap the node attempt config so observable-progress signals refresh the idle
+* clock and are dropped once the scope is closed. Also injects
+* {@link LangGraphRunnableConfig.heartbeat}.
+*/
+function wrapConfig(config, scope, policy, taskName) {
+	const configurable = config.configurable ?? {};
+	const patch = {};
+	const send = configurable[CONFIG_KEY_SEND];
+	if (typeof send === "function") patch[CONFIG_KEY_SEND] = (writes) => {
+		if (!scope.active) return void 0;
+		if (writes && writes.length) scope.autoTouch();
+		return send(writes);
+	};
+	const callFn = configurable[CONFIG_KEY_CALL];
+	if (typeof callFn === "function") patch[CONFIG_KEY_CALL] = (...args) => {
+		if (!scope.active) throw new Error(`Node "${taskName}" attempt was cancelled after its timeout fired`);
+		scope.autoTouch();
+		return callFn(...args);
+	};
+	const wrapped = { ...Object.keys(patch).length > 0 ? patchConfigurable(config, patch) : config };
+	wrapped.heartbeat = () => {
+		if (policy.idleTimeout !== void 0) scope.touch();
+	};
+	if (typeof wrapped.writer === "function") {
+		const writer = wrapped.writer;
+		wrapped.writer = ((chunk) => {
+			if (!scope.active) return void 0;
+			scope.autoTouch();
+			return writer(chunk);
+		});
+	}
+	if ((policy.refreshOn ?? "auto") === "auto" && policy.idleTimeout !== void 0) {
+		const handler = new IdleProgressCallbackHandler(scope);
+		const cb = wrapped.callbacks;
+		if (cb === void 0) wrapped.callbacks = [handler];
+		else if (Array.isArray(cb)) wrapped.callbacks = [...cb, handler];
+		else {
+			const copied = cb.copy();
+			copied.addHandler(handler, true);
+			wrapped.callbacks = copied;
+		}
+	}
+	return wrapped;
+}
+/**
+* Run a single node attempt under a {@link TimeoutPolicy}.
+*
+* Races the node invocation against per-attempt run/idle watchdogs. On
+* successful completion (or node error), returns/rethrows normally. When a
+* watchdog fires first, the scope is closed, the task's buffered writes are
+* dropped, the attempt's {@link AbortSignal} is aborted, and a
+* {@link NodeTimeoutError} is thrown.
+*
+* @internal
+*/
+async function runAttemptWithTimeout(task, config, policy, invoke) {
+	const scope = new TimedAttemptScope(policy.refreshOn ?? "auto");
+	const timeoutController = new AbortController();
+	const { signal: composedSignal, dispose } = combineAbortSignals(config.signal, timeoutController.signal);
+	const scopedConfig = wrapConfig({
+		...config,
+		signal: composedSignal
+	}, scope, policy, String(task.name));
+	const start = Date.now();
+	const nodeOutcome = invoke(scopedConfig).then((value) => ({
+		type: "ok",
+		value
+	}), (error) => ({
+		type: "err",
+		error
+	}));
+	let runTimer;
+	let idleTimer;
+	const clearTimers = () => {
+		if (runTimer !== void 0) clearTimeout(runTimer);
+		if (idleTimer !== void 0) clearTimeout(idleTimer);
+	};
+	const watchdog = new Promise((resolve) => {
+		if (policy.runTimeout !== void 0) runTimer = setTimeout(() => resolve({
+			type: "timeout",
+			kind: "run"
+		}), policy.runTimeout);
+		if (policy.idleTimeout !== void 0) {
+			const idleMs = policy.idleTimeout;
+			const checkIdle = () => {
+				const remaining = scope.lastProgress + idleMs - Date.now();
+				if (remaining <= 0) resolve({
+					type: "timeout",
+					kind: "idle"
+				});
+				else idleTimer = setTimeout(checkIdle, remaining);
+			};
+			idleTimer = setTimeout(checkIdle, idleMs);
+		}
+	});
+	let outcome;
+	try {
+		outcome = await Promise.race([nodeOutcome, watchdog]);
+	} finally {
+		clearTimers();
+	}
+	if (outcome.type !== "timeout") {
+		const now = Date.now();
+		if (policy.runTimeout !== void 0 && now - start >= policy.runTimeout) outcome = {
+			type: "timeout",
+			kind: "run"
+		};
+		else if (policy.idleTimeout !== void 0 && now - scope.lastProgress >= policy.idleTimeout) outcome = {
+			type: "timeout",
+			kind: "idle"
+		};
+	}
+	if (outcome.type === "ok") {
+		dispose?.();
+		return outcome.value;
+	}
+	if (outcome.type === "err") {
+		dispose?.();
+		throw outcome.error;
+	}
+	const elapsed = Date.now() - start;
+	scope.close();
+	task.writes.splice(0, task.writes.length);
+	timeoutController.abort();
+	dispose?.();
+	throw new NodeTimeoutError({
+		node: String(task.name),
+		elapsed,
+		kind: outcome.kind,
+		runTimeout: policy.runTimeout,
+		idleTimeout: policy.idleTimeout
+	});
+}
+//#endregion
+export { runAttemptWithTimeout };
+
+//# sourceMappingURL=timeout.js.map

package/dist/pregel/timeout.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"timeout.js","names":["#scope","#touch"],"sources":["../../src/pregel/timeout.ts"],"sourcesContent":["import { BaseCallbackHandler } from \"@langchain/core/callbacks/base\";\nimport { PendingWrite } from \"@langchain/langgraph-checkpoint\";\nimport { CONFIG_KEY_CALL, CONFIG_KEY_SEND } from \"../constants.js\";\nimport { NodeTimeoutError } from \"../errors.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\nimport { PregelExecutableTask } from \"./types.js\";\nimport {\n  combineAbortSignals,\n  patchConfigurable,\n  TimeoutPolicy,\n} from \"./utils/index.js\";\n\n/**\n * Tracks the live progress state of a single timed node attempt.\n *\n * The scope guards observable-progress channels (writes, child-task\n * scheduling, the custom stream writer, callbacks) so that:\n *\n * 1. `idleTimeout` is refreshed whenever the node makes progress, and\n * 2. once the attempt is `close()`d after a timeout fires, late writes/calls\n *    from the still-running background task are dropped (Python parity:\n *    buffered writes from the failed attempt must not leak into the\n *    checkpoint).\n *\n * @internal\n */\nclass TimedAttemptScope {\n  active = true;\n\n  lastProgress = Date.now();\n\n  private refreshOn: \"auto\" | \"heartbeat\";\n\n  constructor(refreshOn: \"auto\" | \"heartbeat\") {\n    this.refreshOn = refreshOn;\n  }\n\n  /** Record progress now. Always honored (used by `runtime.heartbeat()`). */\n  touch(): void {\n    this.lastProgress = Date.now();\n  }\n\n  /**\n   * Record progress for an automatic signal (write/call/stream/callback).\n   * No-op when `refreshOn === \"heartbeat\"`, where only explicit heartbeats\n   * count as progress.\n   */\n  autoTouch(): void {\n    if (this.refreshOn === \"auto\") {\n      this.lastProgress = Date.now();\n    }\n  }\n\n  close(): void {\n    this.active = false;\n  }\n}\n\n/**\n * Callback handler that refreshes a {@link TimedAttemptScope} on any LangChain\n * callback event emitted under the node's run. Because it is attached via\n * `config.callbacks`, it only observes events from runs descended from this\n * node's attempt, not from sibling nodes.\n *\n * @internal\n */\nclass IdleProgressCallbackHandler extends BaseCallbackHandler {\n  name = \"IdleProgressCallbackHandler\";\n\n  awaitHandlers = false;\n\n  #scope: TimedAttemptScope;\n\n  constructor(scope: TimedAttemptScope) {\n    super();\n    this.#scope = scope;\n  }\n\n  #touch = () => {\n    this.#scope.autoTouch();\n  };\n\n  handleLLMStart = this.#touch;\n\n  handleChatModelStart = this.#touch;\n\n  handleLLMNewToken = this.#touch;\n\n  handleLLMEnd = this.#touch;\n\n  handleLLMError = this.#touch;\n\n  handleChainStart = this.#touch;\n\n  handleChainEnd = this.#touch;\n\n  handleChainError = this.#touch;\n\n  handleToolStart = this.#touch;\n\n  handleToolEnd = this.#touch;\n\n  handleToolError = this.#touch;\n\n  handleText = this.#touch;\n\n  handleRetrieverStart = this.#touch;\n\n  handleRetrieverEnd = this.#touch;\n\n  handleRetrieverError = this.#touch;\n\n  handleCustomEvent = this.#touch;\n}\n\n/**\n * Wrap the node attempt config so observable-progress signals refresh the idle\n * clock and are dropped once the scope is closed. Also injects\n * {@link LangGraphRunnableConfig.heartbeat}.\n */\nfunction wrapConfig(\n  config: LangGraphRunnableConfig,\n  scope: TimedAttemptScope,\n  policy: TimeoutPolicy,\n  taskName: string\n): LangGraphRunnableConfig {\n  const configurable = config.configurable ?? {};\n  const patch: Record<string, unknown> = {};\n\n  const send = configurable[CONFIG_KEY_SEND];\n  if (typeof send === \"function\") {\n    patch[CONFIG_KEY_SEND] = (writes: PendingWrite[]) => {\n      if (!scope.active) return undefined;\n      if (writes && writes.length) scope.autoTouch();\n      return send(writes);\n    };\n  }\n\n  const callFn = configurable[CONFIG_KEY_CALL];\n  if (typeof callFn === \"function\") {\n    patch[CONFIG_KEY_CALL] = (...args: unknown[]) => {\n      if (!scope.active) {\n        throw new Error(\n          `Node \"${taskName}\" attempt was cancelled after its timeout fired`\n        );\n      }\n      scope.autoTouch();\n      return (callFn as (...a: unknown[]) => unknown)(...args);\n    };\n  }\n\n  const out: LangGraphRunnableConfig =\n    Object.keys(patch).length > 0 ? patchConfigurable(config, patch) : config;\n  const wrapped: LangGraphRunnableConfig = { ...out };\n\n  // `heartbeat` always resets the idle clock, even under `refreshOn:\n  // \"heartbeat\"`. It is a no-op when no idle timeout is configured.\n  wrapped.heartbeat = () => {\n    if (policy.idleTimeout !== undefined) scope.touch();\n  };\n\n  if (typeof wrapped.writer === \"function\") {\n    const writer = wrapped.writer;\n    wrapped.writer = ((chunk: unknown) => {\n      if (!scope.active) return undefined;\n      scope.autoTouch();\n      return (writer as (c: unknown) => unknown)(chunk);\n    }) as typeof wrapped.writer;\n  }\n\n  if (\n    (policy.refreshOn ?? \"auto\") === \"auto\" &&\n    policy.idleTimeout !== undefined\n  ) {\n    const handler = new IdleProgressCallbackHandler(scope);\n    const cb = wrapped.callbacks;\n    if (cb === undefined) {\n      wrapped.callbacks = [handler];\n    } else if (Array.isArray(cb)) {\n      wrapped.callbacks = [...cb, handler];\n    } else {\n      const copied = cb.copy();\n      copied.addHandler(handler, true);\n      wrapped.callbacks = copied;\n    }\n  }\n\n  return wrapped;\n}\n\ntype AttemptOutcome<T> =\n  | { type: \"ok\"; value: T }\n  | { type: \"err\"; error: unknown }\n  | { type: \"timeout\"; kind: \"run\" | \"idle\" };\n\n/**\n * Run a single node attempt under a {@link TimeoutPolicy}.\n *\n * Races the node invocation against per-attempt run/idle watchdogs. On\n * successful completion (or node error), returns/rethrows normally. When a\n * watchdog fires first, the scope is closed, the task's buffered writes are\n * dropped, the attempt's {@link AbortSignal} is aborted, and a\n * {@link NodeTimeoutError} is thrown.\n *\n * @internal\n */\nexport async function runAttemptWithTimeout<T>(\n  task: PregelExecutableTask<string, string>,\n  config: LangGraphRunnableConfig,\n  policy: TimeoutPolicy,\n  invoke: (scopedConfig: LangGraphRunnableConfig) => Promise<T>\n): Promise<T> {\n  const refreshOn = policy.refreshOn ?? \"auto\";\n  const scope = new TimedAttemptScope(refreshOn);\n\n  const timeoutController = new AbortController();\n  const { signal: composedSignal, dispose } = combineAbortSignals(\n    config.signal,\n    timeoutController.signal\n  );\n\n  const scopedConfig = wrapConfig(\n    { ...config, signal: composedSignal },\n    scope,\n    policy,\n    String(task.name)\n  );\n\n  const start = Date.now();\n  const bg = invoke(scopedConfig);\n\n  // Normalize the node result into an outcome. Catching the rejection here\n  // (rather than via a separate `bg.catch`) keeps `bg` handled even when a\n  // watchdog wins the race below, so its late settlement can never surface as\n  // an unhandled rejection.\n  const nodeOutcome: Promise<AttemptOutcome<T>> = bg.then(\n    (value) => ({ type: \"ok\", value }),\n    (error) => ({ type: \"err\", error })\n  );\n\n  let runTimer: ReturnType<typeof setTimeout> | undefined;\n  let idleTimer: ReturnType<typeof setTimeout> | undefined;\n  const clearTimers = () => {\n    if (runTimer !== undefined) clearTimeout(runTimer);\n    if (idleTimer !== undefined) clearTimeout(idleTimer);\n  };\n\n  // The watchdog never rejects: it either resolves with a timeout outcome or\n  // stays pending forever (and is dropped) when the node wins the race. The\n  // idle timer re-arms itself off `scope.lastProgress`, so it can't be a single\n  // fixed `setTimeout`.\n  const watchdog = new Promise<AttemptOutcome<T>>((resolve) => {\n    if (policy.runTimeout !== undefined) {\n      runTimer = setTimeout(\n        () => resolve({ type: \"timeout\", kind: \"run\" }),\n        policy.runTimeout\n      );\n    }\n\n    if (policy.idleTimeout !== undefined) {\n      const idleMs = policy.idleTimeout;\n      const checkIdle = () => {\n        const remaining = scope.lastProgress + idleMs - Date.now();\n        if (remaining <= 0) {\n          resolve({ type: \"timeout\", kind: \"idle\" });\n        } else {\n          idleTimer = setTimeout(checkIdle, remaining);\n        }\n      };\n      idleTimer = setTimeout(checkIdle, idleMs);\n    }\n  });\n\n  let outcome: AttemptOutcome<T>;\n  try {\n    outcome = await Promise.race([nodeOutcome, watchdog]);\n  } finally {\n    clearTimers();\n  }\n\n  // Watchdog timers are macrotasks and cannot fire while a synchronous\n  // (CPU-bound) node blocks the event loop. Such a node — or one with a long\n  // synchronous prefix — can therefore settle the race as \"ok\"/\"err\" before an\n  // already-expired timer ever runs, bypassing the documented hard wall-clock\n  // cap. Re-check the budget against wall-clock time here so the caps hold for\n  // synchronous nodes too. (For genuinely async work the watchdog already\n  // wins the race, so this only catches what timers structurally cannot.)\n  if (outcome.type !== \"timeout\") {\n    const now = Date.now();\n    if (policy.runTimeout !== undefined && now - start >= policy.runTimeout) {\n      outcome = { type: \"timeout\", kind: \"run\" };\n    } else if (\n      policy.idleTimeout !== undefined &&\n      now - scope.lastProgress >= policy.idleTimeout\n    ) {\n      outcome = { type: \"timeout\", kind: \"idle\" };\n    }\n  }\n\n  if (outcome.type === \"ok\") {\n    dispose?.();\n    return outcome.value;\n  }\n  if (outcome.type === \"err\") {\n    dispose?.();\n    throw outcome.error;\n  }\n\n  // A watchdog fired: close the scope (drop late writes/calls), discard the\n  // attempt's buffered writes, abort the node, and surface a NodeTimeoutError.\n  const elapsed = Date.now() - start;\n  scope.close();\n  task.writes.splice(0, task.writes.length);\n  // Abort BEFORE disposing the combined signal so the abort actually\n  // propagates to the node's signal (dispose removes the relay listeners).\n  timeoutController.abort();\n  dispose?.();\n\n  throw new NodeTimeoutError({\n    node: String(task.name),\n    elapsed,\n    kind: outcome.kind,\n    runTimeout: policy.runTimeout,\n    idleTimeout: policy.idleTimeout,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AA0BA,IAAM,oBAAN,MAAwB;CACtB,SAAS;CAET,eAAe,KAAK,KAAK;CAEzB;CAEA,YAAY,WAAiC;AAC3C,OAAK,YAAY;;;CAInB,QAAc;AACZ,OAAK,eAAe,KAAK,KAAK;;;;;;;CAQhC,YAAkB;AAChB,MAAI,KAAK,cAAc,OACrB,MAAK,eAAe,KAAK,KAAK;;CAIlC,QAAc;AACZ,OAAK,SAAS;;;;;;;;;;;AAYlB,IAAM,8BAAN,cAA0C,oBAAoB;CAC5D,OAAO;CAEP,gBAAgB;CAEhB;CAEA,YAAY,OAA0B;AACpC,SAAO;AACP,QAAA,QAAc;;CAGhB,eAAe;AACb,QAAA,MAAY,WAAW;;CAGzB,iBAAiB,MAAA;CAEjB,uBAAuB,MAAA;CAEvB,oBAAoB,MAAA;CAEpB,eAAe,MAAA;CAEf,iBAAiB,MAAA;CAEjB,mBAAmB,MAAA;CAEnB,iBAAiB,MAAA;CAEjB,mBAAmB,MAAA;CAEnB,kBAAkB,MAAA;CAElB,gBAAgB,MAAA;CAEhB,kBAAkB,MAAA;CAElB,aAAa,MAAA;CAEb,uBAAuB,MAAA;CAEvB,qBAAqB,MAAA;CAErB,uBAAuB,MAAA;CAEvB,oBAAoB,MAAA;;;;;;;AAQtB,SAAS,WACP,QACA,OACA,QACA,UACyB;CACzB,MAAM,eAAe,OAAO,gBAAgB,EAAE;CAC9C,MAAM,QAAiC,EAAE;CAEzC,MAAM,OAAO,aAAa;AAC1B,KAAI,OAAO,SAAS,WAClB,OAAM,oBAAoB,WAA2B;AACnD,MAAI,CAAC,MAAM,OAAQ,QAAO,KAAA;AAC1B,MAAI,UAAU,OAAO,OAAQ,OAAM,WAAW;AAC9C,SAAO,KAAK,OAAO;;CAIvB,MAAM,SAAS,aAAa;AAC5B,KAAI,OAAO,WAAW,WACpB,OAAM,oBAAoB,GAAG,SAAoB;AAC/C,MAAI,CAAC,MAAM,OACT,OAAM,IAAI,MACR,SAAS,SAAS,iDACnB;AAEH,QAAM,WAAW;AACjB,SAAQ,OAAwC,GAAG,KAAK;;CAM5D,MAAM,UAAmC,EAAE,GADzC,OAAO,KAAK,MAAM,CAAC,SAAS,IAAI,kBAAkB,QAAQ,MAAM,GAAG,QAClB;AAInD,SAAQ,kBAAkB;AACxB,MAAI,OAAO,gBAAgB,KAAA,EAAW,OAAM,OAAO;;AAGrD,KAAI,OAAO,QAAQ,WAAW,YAAY;EACxC,MAAM,SAAS,QAAQ;AACvB,UAAQ,WAAW,UAAmB;AACpC,OAAI,CAAC,MAAM,OAAQ,QAAO,KAAA;AAC1B,SAAM,WAAW;AACjB,UAAQ,OAAmC,MAAM;;;AAIrD,MACG,OAAO,aAAa,YAAY,UACjC,OAAO,gBAAgB,KAAA,GACvB;EACA,MAAM,UAAU,IAAI,4BAA4B,MAAM;EACtD,MAAM,KAAK,QAAQ;AACnB,MAAI,OAAO,KAAA,EACT,SAAQ,YAAY,CAAC,QAAQ;WACpB,MAAM,QAAQ,GAAG,CAC1B,SAAQ,YAAY,CAAC,GAAG,IAAI,QAAQ;OAC/B;GACL,MAAM,SAAS,GAAG,MAAM;AACxB,UAAO,WAAW,SAAS,KAAK;AAChC,WAAQ,YAAY;;;AAIxB,QAAO;;;;;;;;;;;;;AAmBT,eAAsB,sBACpB,MACA,QACA,QACA,QACY;CAEZ,MAAM,QAAQ,IAAI,kBADA,OAAO,aAAa,OACQ;CAE9C,MAAM,oBAAoB,IAAI,iBAAiB;CAC/C,MAAM,EAAE,QAAQ,gBAAgB,YAAY,oBAC1C,OAAO,QACP,kBAAkB,OACnB;CAED,MAAM,eAAe,WACnB;EAAE,GAAG;EAAQ,QAAQ;EAAgB,EACrC,OACA,QACA,OAAO,KAAK,KAAK,CAClB;CAED,MAAM,QAAQ,KAAK,KAAK;CAOxB,MAAM,cANK,OAAO,aAAa,CAMoB,MAChD,WAAW;EAAE,MAAM;EAAM;EAAO,IAChC,WAAW;EAAE,MAAM;EAAO;EAAO,EACnC;CAED,IAAI;CACJ,IAAI;CACJ,MAAM,oBAAoB;AACxB,MAAI,aAAa,KAAA,EAAW,cAAa,SAAS;AAClD,MAAI,cAAc,KAAA,EAAW,cAAa,UAAU;;CAOtD,MAAM,WAAW,IAAI,SAA4B,YAAY;AAC3D,MAAI,OAAO,eAAe,KAAA,EACxB,YAAW,iBACH,QAAQ;GAAE,MAAM;GAAW,MAAM;GAAO,CAAC,EAC/C,OAAO,WACR;AAGH,MAAI,OAAO,gBAAgB,KAAA,GAAW;GACpC,MAAM,SAAS,OAAO;GACtB,MAAM,kBAAkB;IACtB,MAAM,YAAY,MAAM,eAAe,SAAS,KAAK,KAAK;AAC1D,QAAI,aAAa,EACf,SAAQ;KAAE,MAAM;KAAW,MAAM;KAAQ,CAAC;QAE1C,aAAY,WAAW,WAAW,UAAU;;AAGhD,eAAY,WAAW,WAAW,OAAO;;GAE3C;CAEF,IAAI;AACJ,KAAI;AACF,YAAU,MAAM,QAAQ,KAAK,CAAC,aAAa,SAAS,CAAC;WAC7C;AACR,eAAa;;AAUf,KAAI,QAAQ,SAAS,WAAW;EAC9B,MAAM,MAAM,KAAK,KAAK;AACtB,MAAI,OAAO,eAAe,KAAA,KAAa,MAAM,SAAS,OAAO,WAC3D,WAAU;GAAE,MAAM;GAAW,MAAM;GAAO;WAE1C,OAAO,gBAAgB,KAAA,KACvB,MAAM,MAAM,gBAAgB,OAAO,YAEnC,WAAU;GAAE,MAAM;GAAW,MAAM;GAAQ;;AAI/C,KAAI,QAAQ,SAAS,MAAM;AACzB,aAAW;AACX,SAAO,QAAQ;;AAEjB,KAAI,QAAQ,SAAS,OAAO;AAC1B,aAAW;AACX,QAAM,QAAQ;;CAKhB,MAAM,UAAU,KAAK,KAAK,GAAG;AAC7B,OAAM,OAAO;AACb,MAAK,OAAO,OAAO,GAAG,KAAK,OAAO,OAAO;AAGzC,mBAAkB,OAAO;AACzB,YAAW;AAEX,OAAM,IAAI,iBAAiB;EACzB,MAAM,OAAO,KAAK,KAAK;EACvB;EACA,MAAM,QAAQ;EACd,YAAY,OAAO;EACnB,aAAa,OAAO;EACrB,CAAC"}

package/dist/pregel/types.cjs

@@ -5,14 +5,16 @@ var Call = class {
 	input;
 	retry;
 	cache;
+	timeout;
 	callbacks;
 	__lg_type = "call";
-	constructor({ func, name, input, retry, cache, callbacks }) {
+	constructor({ func, name, input, retry, cache, timeout, callbacks }) {
 		this.func = func;
 		this.name = name;
 		this.input = input;
 		this.retry = retry;
 		this.cache = cache;
+		this.timeout = timeout;
 		this.callbacks = callbacks;
 	}
 };