@langchain/langgraph 1.0.7 → 1.0.12

package/dist/pregel/remote.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"remote.js","names":["updatedStreamModes: StreamMode[]","tasks: PregelTaskDescription[]","namespace: string[]","nodesMap: Record<string, DrawableNode>","DrawableGraph"],"sources":["../../src/pregel/remote.ts"],"sourcesContent":["import {\n  Client,\n  type Checkpoint,\n  type ThreadState,\n} from \"@langchain/langgraph-sdk\";\nimport {\n  Graph as DrawableGraph,\n  Node as DrawableNode,\n} from \"@langchain/core/runnables/graph\";\nimport {\n  mergeConfigs,\n  Runnable,\n  RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport {\n  All,\n  CheckpointListOptions,\n  CheckpointMetadata,\n} from \"@langchain/langgraph-checkpoint\";\nimport { StreamEvent } from \"@langchain/core/tracers/log_stream\";\nimport { IterableReadableStream } from \"@langchain/core/utils/stream\";\nimport { isBaseMessage } from \"@langchain/core/messages\";\n\nimport {\n  BaseChannel,\n  GraphInterrupt,\n  LangGraphRunnableConfig,\n  RemoteException,\n} from \"../web.js\";\nimport { StrRecord } from \"./algo.js\";\nimport { PregelInputType, PregelOptions, PregelOutputType } from \"./index.js\";\nimport { PregelNode } from \"./read.js\";\nimport {\n  PregelParams,\n  PregelInterface,\n  PregelTaskDescription,\n  StateSnapshot,\n  StreamMode,\n} from \"./types.js\";\nimport {\n  CHECKPOINT_NAMESPACE_SEPARATOR,\n  CONFIG_KEY_STREAM,\n  INTERRUPT,\n  isCommand,\n} from \"../constants.js\";\n\nexport type RemoteGraphParams = Omit<\n  PregelParams<StrRecord<string, PregelNode>, StrRecord<string, BaseChannel>>,\n  \"channels\" | \"nodes\" | \"inputChannels\" | \"outputChannels\"\n> & {\n  graphId: string;\n  client?: Client;\n  url?: string;\n  apiKey?: string;\n  headers?: Record<string, string>;\n  streamResumable?: boolean;\n};\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst _serializeInputs = (obj: any): any => {\n  if (obj === null || typeof obj !== \"object\") {\n    return obj;\n  }\n\n  if (Array.isArray(obj)) {\n    return obj.map(_serializeInputs);\n  }\n\n  // Handle BaseMessage instances by converting them to a serializable format\n  if (isBaseMessage(obj)) {\n    const dict = obj.toDict();\n    return {\n      ...dict.data,\n      role: obj.getType(),\n    };\n  }\n\n  return Object.fromEntries(\n    Object.entries(obj).map(([key, value]) => [key, _serializeInputs(value)])\n  );\n};\n\n/**\n * Return a tuple of the final list of stream modes sent to the\n * remote graph and a boolean flag indicating if only one stream mode was\n * originally requested and whether stream mode 'updates'\n * was present in the original list of stream modes.\n *\n * 'updates' mode is always added to the list of stream modes so that interrupts\n * can be detected in the remote graph.\n */\nconst getStreamModes = (\n  streamMode?: StreamMode | StreamMode[],\n  defaultStreamMode: StreamMode = \"updates\"\n) => {\n  const updatedStreamModes: StreamMode[] = [];\n  let reqUpdates = false;\n  let reqSingle = true;\n\n  if (\n    streamMode !== undefined &&\n    (typeof streamMode === \"string\" ||\n      (Array.isArray(streamMode) && streamMode.length > 0))\n  ) {\n    reqSingle = typeof streamMode === \"string\";\n    const mapped = Array.isArray(streamMode) ? streamMode : [streamMode];\n    updatedStreamModes.push(...mapped);\n  } else {\n    updatedStreamModes.push(defaultStreamMode);\n  }\n  if (updatedStreamModes.includes(\"updates\")) {\n    reqUpdates = true;\n  } else {\n    updatedStreamModes.push(\"updates\");\n  }\n  return {\n    updatedStreamModes,\n    reqUpdates,\n    reqSingle,\n  };\n};\n\n/**\n * The `RemoteGraph` class is a client implementation for calling remote\n * APIs that implement the LangGraph Server API specification.\n *\n * For example, the `RemoteGraph` class can be used to call APIs from deployments\n * on LangSmith Deployment.\n *\n * `RemoteGraph` behaves the same way as a `StateGraph` and can be used directly as\n * a node in another `StateGraph`.\n *\n * @example\n * ```ts\n * import { RemoteGraph } from \"@langchain/langgraph/remote\";\n *\n * // Can also pass a LangGraph SDK client instance directly\n * const remoteGraph = new RemoteGraph({\n *   graphId: process.env.LANGGRAPH_REMOTE_GRAPH_ID!,\n *   apiKey: process.env.LANGGRAPH_REMOTE_GRAPH_API_KEY,\n *   url: process.env.LANGGRAPH_REMOTE_GRAPH_API_URL,\n * });\n *\n * const input = {\n *   messages: [\n *     {\n *       role: \"human\",\n *       content: \"Hello world!\",\n *     },\n *   ],\n * };\n *\n * const config = {\n *   configurable: { thread_id: \"threadId1\" },\n * };\n *\n * await remoteGraph.invoke(input, config);\n * ```\n */\nexport class RemoteGraph<\n    Nn extends StrRecord<string, PregelNode> = StrRecord<string, PregelNode>,\n    Cc extends StrRecord<string, BaseChannel> = StrRecord<string, BaseChannel>,\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    ContextType extends Record<string, any> = StrRecord<string, any>\n  >\n  extends Runnable<\n    PregelInputType,\n    PregelOutputType,\n    PregelOptions<Nn, Cc, ContextType>\n  >\n  implements PregelInterface<Nn, Cc, ContextType>\n{\n  static lc_name() {\n    return \"RemoteGraph\";\n  }\n\n  lc_namespace = [\"langgraph\", \"pregel\"];\n\n  lg_is_pregel = true;\n\n  config?: RunnableConfig;\n\n  graphId: string;\n\n  protected client: Client;\n\n  protected interruptBefore?: Array<keyof Nn> | All;\n\n  protected interruptAfter?: Array<keyof Nn> | All;\n\n  protected streamResumable?: boolean;\n\n  constructor(params: RemoteGraphParams) {\n    super(params);\n\n    this.graphId = params.graphId;\n    this.client =\n      params.client ??\n      new Client({\n        apiUrl: params.url,\n        apiKey: params.apiKey,\n        defaultHeaders: params.headers,\n      });\n    this.config = params.config;\n    this.interruptBefore = params.interruptBefore;\n    this.interruptAfter = params.interruptAfter;\n    this.streamResumable = params.streamResumable;\n  }\n\n  // eslint-disable-next-line @typescript-eslint/ban-ts-comment\n  // @ts-ignore Remove ignore when we remove support for 0.2 versions of core\n  override withConfig(config: RunnableConfig): typeof this {\n    const mergedConfig = mergeConfigs(this.config, config);\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return new (this.constructor as any)({ ...this, config: mergedConfig });\n  }\n\n  protected _sanitizeConfig(config: RunnableConfig) {\n    const reservedConfigurableKeys = new Set([\n      \"callbacks\",\n      \"checkpoint_map\",\n      \"checkpoint_id\",\n      \"checkpoint_ns\",\n    ]);\n\n    const sanitizeObj = <T>(obj: T): T => {\n      try {\n        // This will only throw if we're trying to serialize a circular reference\n        // or trying to serialize a BigInt...\n        JSON.stringify(obj);\n        return obj;\n      } catch {\n        const seen = new WeakSet();\n        return JSON.parse(\n          JSON.stringify(obj, (_, value) => {\n            if (typeof value === \"object\" && value != null) {\n              if (seen.has(value)) return \"[Circular]\";\n              seen.add(value);\n            }\n\n            if (typeof value === \"bigint\") return value.toString();\n            return value;\n          })\n        );\n      }\n    };\n\n    // Remove non-JSON serializable fields from the config\n    const sanitizedConfig = sanitizeObj(config);\n\n    // Only include configurable keys that are not reserved and\n    // not starting with \"__pregel_\" prefix\n    const newConfigurable = Object.fromEntries(\n      Object.entries(sanitizedConfig.configurable ?? {}).filter(\n        ([k]) => !reservedConfigurableKeys.has(k) && !k.startsWith(\"__pregel_\")\n      )\n    );\n\n    return {\n      tags: sanitizedConfig.tags ?? [],\n      metadata: sanitizedConfig.metadata ?? {},\n      configurable: newConfigurable,\n      recursion_limit: sanitizedConfig.recursionLimit,\n    };\n  }\n\n  protected _getConfig(checkpoint: Record<string, unknown>): RunnableConfig {\n    return {\n      configurable: {\n        thread_id: checkpoint.thread_id,\n        checkpoint_ns: checkpoint.checkpoint_ns,\n        checkpoint_id: checkpoint.checkpoint_id,\n        checkpoint_map: checkpoint.checkpoint_map ?? {},\n      },\n    };\n  }\n\n  protected _getCheckpoint(config?: RunnableConfig): Checkpoint | undefined {\n    if (config?.configurable === undefined) {\n      return undefined;\n    }\n\n    const checkpointKeys = [\n      \"thread_id\",\n      \"checkpoint_ns\",\n      \"checkpoint_id\",\n      \"checkpoint_map\",\n    ] as const;\n\n    const checkpoint = Object.fromEntries(\n      checkpointKeys\n        .map((key) => [key, config.configurable![key]])\n        .filter(([_, value]) => value !== undefined)\n    );\n\n    return Object.keys(checkpoint).length > 0 ? checkpoint : undefined;\n  }\n\n  protected _createStateSnapshot(state: ThreadState): StateSnapshot {\n    const tasks: PregelTaskDescription[] = state.tasks.map((task) => {\n      return {\n        id: task.id,\n        name: task.name,\n        error: task.error ? { message: task.error } : undefined,\n        // TODO: remove in LangGraph.js 0.4\n        interrupts: task.interrupts.map(({ id, ...rest }) => ({\n          interrupt_id: id,\n          ...rest,\n        })),\n        // eslint-disable-next-line no-nested-ternary\n        state: task.state\n          ? this._createStateSnapshot(task.state)\n          : task.checkpoint\n          ? { configurable: task.checkpoint }\n          : undefined,\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        result: (task as any).result,\n      };\n    });\n\n    return {\n      values: state.values,\n      next: state.next ? [...state.next] : [],\n      config: {\n        configurable: {\n          thread_id: state.checkpoint.thread_id,\n          checkpoint_ns: state.checkpoint.checkpoint_ns,\n          checkpoint_id: state.checkpoint.checkpoint_id,\n          checkpoint_map: state.checkpoint.checkpoint_map ?? {},\n        },\n      },\n      metadata: state.metadata\n        ? (state.metadata as CheckpointMetadata)\n        : undefined,\n      createdAt: state.created_at ?? undefined,\n      parentConfig: state.parent_checkpoint\n        ? {\n            configurable: {\n              thread_id: state.parent_checkpoint.thread_id,\n              checkpoint_ns: state.parent_checkpoint.checkpoint_ns,\n              checkpoint_id: state.parent_checkpoint.checkpoint_id,\n              checkpoint_map: state.parent_checkpoint.checkpoint_map ?? {},\n            },\n          }\n        : undefined,\n      tasks,\n    };\n  }\n\n  override async invoke(\n    input: PregelInputType,\n    options?: Partial<PregelOptions<Nn, Cc, ContextType>>\n  ): Promise<PregelOutputType> {\n    let lastValue;\n    const stream = await this.stream(input, {\n      ...options,\n      streamMode: \"values\",\n    });\n    for await (const chunk of stream) {\n      lastValue = chunk;\n    }\n    return lastValue;\n  }\n\n  override streamEvents(\n    input: PregelInputType,\n    options: Partial<PregelOptions<Nn, Cc, ContextType>> & {\n      version: \"v1\" | \"v2\";\n    }\n  ): IterableReadableStream<StreamEvent>;\n\n  override streamEvents(\n    input: PregelInputType,\n    options: Partial<PregelOptions<Nn, Cc, ContextType>> & {\n      version: \"v1\" | \"v2\";\n      encoding: never;\n    }\n  ): IterableReadableStream<never>;\n\n  override streamEvents(\n    _input: PregelInputType,\n    _options: Partial<PregelOptions<Nn, Cc, ContextType>> & {\n      version: \"v1\" | \"v2\";\n      encoding?: never;\n    }\n  ): IterableReadableStream<StreamEvent> {\n    throw new Error(\"Not implemented.\");\n  }\n\n  override async *_streamIterator(\n    input: PregelInputType,\n    options?: Partial<PregelOptions<Nn, Cc, ContextType>>\n  ): AsyncGenerator<PregelOutputType> {\n    const mergedConfig = mergeConfigs(this.config, options);\n    const sanitizedConfig = this._sanitizeConfig(mergedConfig);\n\n    const streamProtocolInstance = options?.configurable?.[CONFIG_KEY_STREAM];\n\n    const streamSubgraphs =\n      options?.subgraphs ?? streamProtocolInstance !== undefined;\n\n    const interruptBefore = options?.interruptBefore ?? this.interruptBefore;\n    const interruptAfter = options?.interruptAfter ?? this.interruptAfter;\n\n    const { updatedStreamModes, reqSingle, reqUpdates } = getStreamModes(\n      options?.streamMode\n    );\n\n    const extendedStreamModes = [\n      ...new Set([\n        ...updatedStreamModes,\n        ...(streamProtocolInstance?.modes ?? new Set()),\n      ]),\n    ].map((mode) => {\n      if (mode === \"messages\") return \"messages-tuple\";\n      return mode;\n    });\n\n    let command;\n    let serializedInput;\n    if (isCommand(input)) {\n      // TODO: Remove cast when SDK type fix gets merged\n      command = input.toJSON() as Record<string, unknown>;\n      serializedInput = undefined;\n    } else {\n      serializedInput = _serializeInputs(input);\n    }\n\n    for await (const chunk of this.client.runs.stream(\n      sanitizedConfig.configurable.thread_id as string,\n      this.graphId,\n      {\n        command,\n        input: serializedInput,\n        config: sanitizedConfig,\n        streamMode: extendedStreamModes,\n        interruptBefore: interruptBefore as string[],\n        interruptAfter: interruptAfter as string[],\n        streamSubgraphs,\n        ifNotExists: \"create\",\n        signal: mergedConfig.signal,\n        streamResumable: this.streamResumable,\n      }\n    )) {\n      let mode;\n      let namespace: string[];\n      if (chunk.event.includes(CHECKPOINT_NAMESPACE_SEPARATOR)) {\n        const eventComponents = chunk.event.split(\n          CHECKPOINT_NAMESPACE_SEPARATOR\n        );\n        // eslint-disable-next-line prefer-destructuring\n        mode = eventComponents[0];\n        namespace = eventComponents.slice(1);\n      } else {\n        mode = chunk.event;\n        namespace = [];\n      }\n      const callerNamespace = options?.configurable?.checkpoint_ns;\n      if (typeof callerNamespace === \"string\") {\n        namespace = callerNamespace\n          .split(CHECKPOINT_NAMESPACE_SEPARATOR)\n          .concat(namespace);\n      }\n      if (\n        streamProtocolInstance !== undefined &&\n        streamProtocolInstance.modes?.has(chunk.event)\n      ) {\n        streamProtocolInstance.push([namespace, mode, chunk.data]);\n      }\n      if (chunk.event.startsWith(\"updates\")) {\n        if (\n          typeof chunk.data === \"object\" &&\n          chunk.data?.[INTERRUPT] !== undefined\n        ) {\n          throw new GraphInterrupt(chunk.data[INTERRUPT]);\n        }\n        if (!reqUpdates) {\n          continue;\n        }\n      } else if (chunk.event?.startsWith(\"error\")) {\n        throw new RemoteException(\n          typeof chunk.data === \"string\"\n            ? chunk.data\n            : JSON.stringify(chunk.data)\n        );\n      }\n      if (\n        !updatedStreamModes.includes(\n          chunk.event.split(CHECKPOINT_NAMESPACE_SEPARATOR)[0] as StreamMode\n        )\n      ) {\n        continue;\n      }\n      if (options?.subgraphs) {\n        if (reqSingle) {\n          yield [namespace, chunk.data];\n        } else {\n          yield [namespace, mode, chunk.data];\n        }\n      } else if (reqSingle) {\n        yield chunk.data;\n      } else {\n        yield [mode, chunk.data];\n      }\n    }\n  }\n\n  async updateState(\n    inputConfig: LangGraphRunnableConfig,\n    values: Record<string, unknown>,\n    asNode?: string\n  ): Promise<RunnableConfig> {\n    const mergedConfig = mergeConfigs(this.config, inputConfig);\n    const response = await this.client.threads.updateState(\n      mergedConfig.configurable?.thread_id,\n      { values, asNode, checkpoint: this._getCheckpoint(mergedConfig) }\n    );\n    // TODO: Fix SDK typing\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return this._getConfig((response as any).checkpoint);\n  }\n\n  async *getStateHistory(\n    config: RunnableConfig,\n    options?: CheckpointListOptions\n  ): AsyncIterableIterator<StateSnapshot> {\n    const mergedConfig = mergeConfigs(this.config, config);\n    const states = await this.client.threads.getHistory(\n      mergedConfig.configurable?.thread_id,\n      {\n        limit: options?.limit ?? 10,\n        // TODO: Fix type\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        before: this._getCheckpoint(options?.before) as any,\n        metadata: options?.filter,\n        checkpoint: this._getCheckpoint(mergedConfig),\n      }\n    );\n    for (const state of states) {\n      yield this._createStateSnapshot(state);\n    }\n  }\n\n  protected _getDrawableNodes(\n    nodes: Array<{\n      id: string | number;\n      name?: string;\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      data?: Record<string, any> | string;\n      metadata?: unknown;\n    }>\n  ): Record<string, DrawableNode> {\n    const nodesMap: Record<string, DrawableNode> = {};\n    for (const node of nodes) {\n      const nodeId = node.id;\n      nodesMap[nodeId] = {\n        id: nodeId.toString(),\n        name: typeof node.data === \"string\" ? node.data : node.data?.name ?? \"\",\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        data: (node.data as any) ?? {},\n        metadata:\n          typeof node.data !== \"string\" ? node.data?.metadata ?? {} : {},\n      };\n    }\n    return nodesMap;\n  }\n\n  async getState(\n    config: RunnableConfig,\n    options?: { subgraphs?: boolean }\n  ): Promise<StateSnapshot> {\n    const mergedConfig = mergeConfigs(this.config, config);\n\n    const state = await this.client.threads.getState(\n      mergedConfig.configurable?.thread_id,\n      this._getCheckpoint(mergedConfig),\n      options\n    );\n    return this._createStateSnapshot(state);\n  }\n\n  /** @deprecated Use getGraphAsync instead. The async method will become the default in the next minor release. */\n  override getGraph(\n    _?: RunnableConfig & { xray?: boolean | number }\n  ): DrawableGraph {\n    throw new Error(\n      `The synchronous \"getGraph\" is not supported for this graph. Call \"getGraphAsync\" instead.`\n    );\n  }\n\n  /**\n   * Returns a drawable representation of the computation graph.\n   */\n  async getGraphAsync(config?: RunnableConfig & { xray?: boolean | number }) {\n    const graph = await this.client.assistants.getGraph(this.graphId, {\n      xray: config?.xray,\n    });\n    return new DrawableGraph({\n      nodes: this._getDrawableNodes(graph.nodes),\n      edges: graph.edges,\n    });\n  }\n\n  /** @deprecated Use getSubgraphsAsync instead. The async method will become the default in the next minor release. */\n  getSubgraphs(): Generator<[string, PregelInterface<Nn, Cc, ContextType>]> {\n    throw new Error(\n      `The synchronous \"getSubgraphs\" method is not supported for this graph. Call \"getSubgraphsAsync\" instead.`\n    );\n  }\n\n  async *getSubgraphsAsync(\n    namespace?: string,\n    recurse = false\n  ): AsyncGenerator<[string, PregelInterface<Nn, Cc, ContextType>]> {\n    const subgraphs = await this.client.assistants.getSubgraphs(this.graphId, {\n      namespace,\n      recurse,\n    });\n\n    for (const [ns, graphSchema] of Object.entries(subgraphs)) {\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      const remoteSubgraph = new (this.constructor as any)({\n        ...this,\n        graphId: graphSchema.graph_id,\n      });\n      yield [ns, remoteSubgraph];\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AA2DA,MAAM,oBAAoB,QAAkB;AAC1C,KAAI,QAAQ,QAAQ,OAAO,QAAQ,SACjC,QAAO;AAGT,KAAI,MAAM,QAAQ,KAChB,QAAO,IAAI,IAAI;AAIjB,KAAI,cAAc,MAAM;EACtB,MAAM,OAAO,IAAI;AACjB,SAAO;GACL,GAAG,KAAK;GACR,MAAM,IAAI;;;AAId,QAAO,OAAO,YACZ,OAAO,QAAQ,KAAK,KAAK,CAAC,KAAK,WAAW,CAAC,KAAK,iBAAiB;;;;;;;;;;;AAarE,MAAM,kBACJ,YACA,oBAAgC,cAC7B;CACH,MAAMA,qBAAmC;CACzC,IAAI,aAAa;CACjB,IAAI,YAAY;AAEhB,KACE,eAAe,WACd,OAAO,eAAe,YACpB,MAAM,QAAQ,eAAe,WAAW,SAAS,IACpD;AACA,cAAY,OAAO,eAAe;EAClC,MAAM,SAAS,MAAM,QAAQ,cAAc,aAAa,CAAC;AACzD,qBAAmB,KAAK,GAAG;OAE3B,oBAAmB,KAAK;AAE1B,KAAI,mBAAmB,SAAS,WAC9B,cAAa;KAEb,oBAAmB,KAAK;AAE1B,QAAO;EACL;EACA;EACA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyCJ,IAAa,cAAb,cAMU,SAMV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,eAAe,CAAC,aAAa;CAE7B,eAAe;CAEf;CAEA;CAEA,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,YAAY,QAA2B;AACrC,QAAM;AAEN,OAAK,UAAU,OAAO;AACtB,OAAK,SACH,OAAO,UACP,IAAI,OAAO;GACT,QAAQ,OAAO;GACf,QAAQ,OAAO;GACf,gBAAgB,OAAO;;AAE3B,OAAK,SAAS,OAAO;AACrB,OAAK,kBAAkB,OAAO;AAC9B,OAAK,iBAAiB,OAAO;AAC7B,OAAK,kBAAkB,OAAO;;CAKhC,AAAS,WAAW,QAAqC;EACvD,MAAM,eAAe,aAAa,KAAK,QAAQ;AAE/C,SAAO,IAAK,KAAK,YAAoB;GAAE,GAAG;GAAM,QAAQ;;;CAG1D,AAAU,gBAAgB,QAAwB;EAChD,MAAM,2BAA2B,IAAI,IAAI;GACvC;GACA;GACA;GACA;;EAGF,MAAM,eAAkB,QAAc;AACpC,OAAI;AAGF,SAAK,UAAU;AACf,WAAO;WACD;IACN,MAAM,uBAAO,IAAI;AACjB,WAAO,KAAK,MACV,KAAK,UAAU,MAAM,GAAG,UAAU;AAChC,SAAI,OAAO,UAAU,YAAY,SAAS,MAAM;AAC9C,UAAI,KAAK,IAAI,OAAQ,QAAO;AAC5B,WAAK,IAAI;;AAGX,SAAI,OAAO,UAAU,SAAU,QAAO,MAAM;AAC5C,YAAO;;;;EAOf,MAAM,kBAAkB,YAAY;EAIpC,MAAM,kBAAkB,OAAO,YAC7B,OAAO,QAAQ,gBAAgB,gBAAgB,IAAI,QAChD,CAAC,OAAO,CAAC,yBAAyB,IAAI,MAAM,CAAC,EAAE,WAAW;AAI/D,SAAO;GACL,MAAM,gBAAgB,QAAQ;GAC9B,UAAU,gBAAgB,YAAY;GACtC,cAAc;GACd,iBAAiB,gBAAgB;;;CAIrC,AAAU,WAAW,YAAqD;AACxE,SAAO,EACL,cAAc;GACZ,WAAW,WAAW;GACtB,eAAe,WAAW;GAC1B,eAAe,WAAW;GAC1B,gBAAgB,WAAW,kBAAkB;;;CAKnD,AAAU,eAAe,QAAiD;AACxE,MAAI,QAAQ,iBAAiB,OAC3B,QAAO;EAGT,MAAM,iBAAiB;GACrB;GACA;GACA;GACA;;EAGF,MAAM,aAAa,OAAO,YACxB,eACG,KAAK,QAAQ,CAAC,KAAK,OAAO,aAAc,OACxC,QAAQ,CAAC,GAAG,WAAW,UAAU;AAGtC,SAAO,OAAO,KAAK,YAAY,SAAS,IAAI,aAAa;;CAG3D,AAAU,qBAAqB,OAAmC;EAChE,MAAMC,QAAiC,MAAM,MAAM,KAAK,SAAS;AAC/D,UAAO;IACL,IAAI,KAAK;IACT,MAAM,KAAK;IACX,OAAO,KAAK,QAAQ,EAAE,SAAS,KAAK,UAAU;IAE9C,YAAY,KAAK,WAAW,KAAK,EAAE,GAAI,GAAG,YAAY;KACpD,cAAc;KACd,GAAG;;IAGL,OAAO,KAAK,QACR,KAAK,qBAAqB,KAAK,SAC/B,KAAK,aACL,EAAE,cAAc,KAAK,eACrB;IAEJ,QAAS,KAAa;;;AAI1B,SAAO;GACL,QAAQ,MAAM;GACd,MAAM,MAAM,OAAO,CAAC,GAAG,MAAM,QAAQ;GACrC,QAAQ,EACN,cAAc;IACZ,WAAW,MAAM,WAAW;IAC5B,eAAe,MAAM,WAAW;IAChC,eAAe,MAAM,WAAW;IAChC,gBAAgB,MAAM,WAAW,kBAAkB;;GAGvD,UAAU,MAAM,WACX,MAAM,WACP;GACJ,WAAW,MAAM,cAAc;GAC/B,cAAc,MAAM,oBAChB,EACE,cAAc;IACZ,WAAW,MAAM,kBAAkB;IACnC,eAAe,MAAM,kBAAkB;IACvC,eAAe,MAAM,kBAAkB;IACvC,gBAAgB,MAAM,kBAAkB,kBAAkB;SAG9D;GACJ;;;CAIJ,MAAe,OACb,OACA,SAC2B;EAC3B,IAAI;EACJ,MAAM,SAAS,MAAM,KAAK,OAAO,OAAO;GACtC,GAAG;GACH,YAAY;;AAEd,aAAW,MAAM,SAAS,OACxB,aAAY;AAEd,SAAO;;CAkBT,AAAS,aACP,QACA,UAIqC;AACrC,QAAM,IAAI,MAAM;;CAGlB,OAAgB,gBACd,OACA,SACkC;EAClC,MAAM,eAAe,aAAa,KAAK,QAAQ;EAC/C,MAAM,kBAAkB,KAAK,gBAAgB;EAE7C,MAAM,yBAAyB,SAAS,eAAe;EAEvD,MAAM,kBACJ,SAAS,aAAa,2BAA2B;EAEnD,MAAM,kBAAkB,SAAS,mBAAmB,KAAK;EACzD,MAAM,iBAAiB,SAAS,kBAAkB,KAAK;EAEvD,MAAM,EAAE,oBAAoB,WAAW,eAAe,eACpD,SAAS;EAGX,MAAM,sBAAsB,CAC1B,GAAG,IAAI,IAAI,CACT,GAAG,oBACH,GAAI,wBAAwB,yBAAS,IAAI,SAE3C,KAAK,SAAS;AACd,OAAI,SAAS,WAAY,QAAO;AAChC,UAAO;;EAGT,IAAI;EACJ,IAAI;AACJ,MAAI,UAAU,QAAQ;AAEpB,aAAU,MAAM;AAChB,qBAAkB;QAElB,mBAAkB,iBAAiB;AAGrC,aAAW,MAAM,SAAS,KAAK,OAAO,KAAK,OACzC,gBAAgB,aAAa,WAC7B,KAAK,SACL;GACE;GACA,OAAO;GACP,QAAQ;GACR,YAAY;GACK;GACD;GAChB;GACA,aAAa;GACb,QAAQ,aAAa;GACrB,iBAAiB,KAAK;MAEvB;GACD,IAAI;GACJ,IAAIC;AACJ,OAAI,MAAM,MAAM,SAAS,iCAAiC;IACxD,MAAM,kBAAkB,MAAM,MAAM,MAClC;AAGF,WAAO,gBAAgB;AACvB,gBAAY,gBAAgB,MAAM;UAC7B;AACL,WAAO,MAAM;AACb,gBAAY;;GAEd,MAAM,kBAAkB,SAAS,cAAc;AAC/C,OAAI,OAAO,oBAAoB,SAC7B,aAAY,gBACT,MAAM,gCACN,OAAO;AAEZ,OACE,2BAA2B,UAC3B,uBAAuB,OAAO,IAAI,MAAM,OAExC,wBAAuB,KAAK;IAAC;IAAW;IAAM,MAAM;;AAEtD,OAAI,MAAM,MAAM,WAAW,YAAY;AACrC,QACE,OAAO,MAAM,SAAS,YACtB,MAAM,OAAO,eAAe,OAE5B,OAAM,IAAI,eAAe,MAAM,KAAK;AAEtC,QAAI,CAAC,WACH;cAEO,MAAM,OAAO,WAAW,SACjC,OAAM,IAAI,gBACR,OAAO,MAAM,SAAS,WAClB,MAAM,OACN,KAAK,UAAU,MAAM;AAG7B,OACE,CAAC,mBAAmB,SAClB,MAAM,MAAM,MAAM,gCAAgC,IAGpD;AAEF,OAAI,SAAS,UACX,KAAI,UACF,OAAM,CAAC,WAAW,MAAM;OAExB,OAAM;IAAC;IAAW;IAAM,MAAM;;YAEvB,UACT,OAAM,MAAM;OAEZ,OAAM,CAAC,MAAM,MAAM;;;CAKzB,MAAM,YACJ,aACA,QACA,QACyB;EACzB,MAAM,eAAe,aAAa,KAAK,QAAQ;EAC/C,MAAM,WAAW,MAAM,KAAK,OAAO,QAAQ,YACzC,aAAa,cAAc,WAC3B;GAAE;GAAQ;GAAQ,YAAY,KAAK,eAAe;;AAIpD,SAAO,KAAK,WAAY,SAAiB;;CAG3C,OAAO,gBACL,QACA,SACsC;EACtC,MAAM,eAAe,aAAa,KAAK,QAAQ;EAC/C,MAAM,SAAS,MAAM,KAAK,OAAO,QAAQ,WACvC,aAAa,cAAc,WAC3B;GACE,OAAO,SAAS,SAAS;GAGzB,QAAQ,KAAK,eAAe,SAAS;GACrC,UAAU,SAAS;GACnB,YAAY,KAAK,eAAe;;AAGpC,OAAK,MAAM,SAAS,OAClB,OAAM,KAAK,qBAAqB;;CAIpC,AAAU,kBACR,OAO8B;EAC9B,MAAMC,WAAyC;AAC/C,OAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,SAAS,KAAK;AACpB,YAAS,UAAU;IACjB,IAAI,OAAO;IACX,MAAM,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO,KAAK,MAAM,QAAQ;IAErE,MAAO,KAAK,QAAgB;IAC5B,UACE,OAAO,KAAK,SAAS,WAAW,KAAK,MAAM,YAAY,KAAK;;;AAGlE,SAAO;;CAGT,MAAM,SACJ,QACA,SACwB;EACxB,MAAM,eAAe,aAAa,KAAK,QAAQ;EAE/C,MAAM,QAAQ,MAAM,KAAK,OAAO,QAAQ,SACtC,aAAa,cAAc,WAC3B,KAAK,eAAe,eACpB;AAEF,SAAO,KAAK,qBAAqB;;;CAInC,AAAS,SACP,GACe;AACf,QAAM,IAAI,MACR;;;;;CAOJ,MAAM,cAAc,QAAuD;EACzE,MAAM,QAAQ,MAAM,KAAK,OAAO,WAAW,SAAS,KAAK,SAAS,EAChE,MAAM,QAAQ;AAEhB,SAAO,IAAIC,MAAc;GACvB,OAAO,KAAK,kBAAkB,MAAM;GACpC,OAAO,MAAM;;;;CAKjB,eAA0E;AACxE,QAAM,IAAI,MACR;;CAIJ,OAAO,kBACL,WACA,UAAU,OACsD;EAChE,MAAM,YAAY,MAAM,KAAK,OAAO,WAAW,aAAa,KAAK,SAAS;GACxE;GACA;;AAGF,OAAK,MAAM,CAAC,IAAI,gBAAgB,OAAO,QAAQ,YAAY;GAEzD,MAAM,iBAAiB,IAAK,KAAK,YAAoB;IACnD,GAAG;IACH,SAAS,YAAY;;AAEvB,SAAM,CAAC,IAAI"}

package/dist/pregel/retry.cjs

@@ -1,92 +0,0 @@
-const require_errors = require('../errors.cjs');
-const require_constants = require('../constants.cjs');
-const require_config = require('./utils/config.cjs');
-const require_index = require('./utils/index.cjs');
-
-//#region src/pregel/retry.ts
-const DEFAULT_INITIAL_INTERVAL = 500;
-const DEFAULT_BACKOFF_FACTOR = 2;
-const DEFAULT_MAX_INTERVAL = 128e3;
-const DEFAULT_MAX_RETRIES = 3;
-const DEFAULT_STATUS_NO_RETRY = [
-	400,
-	401,
-	402,
-	403,
-	404,
-	405,
-	406,
-	407,
-	409
-];
-const DEFAULT_RETRY_ON_HANDLER = (error) => {
-	if (error.message.startsWith("Cancel") || error.message.startsWith("AbortError") || error.name === "AbortError") return false;
-	if (error.name === "GraphValueError") return false;
-	if (error?.code === "ECONNABORTED") return false;
-	const status = error?.response?.status ?? error?.status;
-	if (status && DEFAULT_STATUS_NO_RETRY.includes(+status)) return false;
-	if (error?.error?.code === "insufficient_quota") return false;
-	return true;
-};
-async function _runWithRetry(pregelTask, retryPolicy, configurable, signal) {
-	const resolvedRetryPolicy = pregelTask.retry_policy ?? retryPolicy;
-	let interval = resolvedRetryPolicy !== void 0 ? resolvedRetryPolicy.initialInterval ?? DEFAULT_INITIAL_INTERVAL : 0;
-	let attempts = 0;
-	let error;
-	let result;
-	let { config } = pregelTask;
-	if (configurable) config = require_index.patchConfigurable(config, configurable);
-	config = {
-		...config,
-		signal
-	};
-	while (true) {
-		if (signal?.aborted) break;
-		pregelTask.writes.splice(0, pregelTask.writes.length);
-		error = void 0;
-		try {
-			result = await pregelTask.proc.invoke(pregelTask.input, config);
-			break;
-		} catch (e) {
-			error = e;
-			error.pregelTaskId = pregelTask.id;
-			if (require_errors.isParentCommand(error)) {
-				const ns = config?.configurable?.checkpoint_ns;
-				const cmd = error.command;
-				if (cmd.graph === ns) {
-					for (const writer of pregelTask.writers) await writer.invoke(cmd, config);
-					error = void 0;
-					break;
-				} else if (cmd.graph === require_constants.Command.PARENT) {
-					const parentNs = require_config.getParentCheckpointNamespace(ns);
-					error.command = new require_constants.Command({
-						...error.command,
-						graph: parentNs
-					});
-				}
-			}
-			if (require_errors.isGraphBubbleUp(error)) break;
-			if (resolvedRetryPolicy === void 0) break;
-			attempts += 1;
-			if (attempts >= (resolvedRetryPolicy.maxAttempts ?? DEFAULT_MAX_RETRIES)) break;
-			const retryOn = resolvedRetryPolicy.retryOn ?? DEFAULT_RETRY_ON_HANDLER;
-			if (!retryOn(error)) break;
-			interval = Math.min(resolvedRetryPolicy.maxInterval ?? DEFAULT_MAX_INTERVAL, interval * (resolvedRetryPolicy.backoffFactor ?? DEFAULT_BACKOFF_FACTOR));
-			const intervalWithJitter = resolvedRetryPolicy.jitter ? Math.floor(interval + Math.random() * 1e3) : interval;
-			await new Promise((resolve) => setTimeout(resolve, intervalWithJitter));
-			const errorName = error.name ?? error.constructor.unminifiable_name ?? error.constructor.name;
-			if (resolvedRetryPolicy?.logWarning ?? true) console.log(`Retrying task "${String(pregelTask.name)}" after ${interval.toFixed(2)}ms (attempt ${attempts}) after ${errorName}: ${error}`);
-			config = require_index.patchConfigurable(config, { [require_constants.CONFIG_KEY_RESUMING]: true });
-		}
-	}
-	return {
-		task: pregelTask,
-		result,
-		error,
-		signalAborted: signal?.aborted
-	};
-}
-
-//#endregion
-exports._runWithRetry = _runWithRetry;
-//# sourceMappingURL=retry.cjs.map

package/dist/pregel/retry.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"retry.cjs","names":["patchConfigurable","e: unknown","isParentCommand","ns: string","Command","getParentCheckpointNamespace","isGraphBubbleUp","CONFIG_KEY_RESUMING"],"sources":["../../src/pregel/retry.ts"],"sourcesContent":["import { Command, CONFIG_KEY_RESUMING } from \"../constants.js\";\nimport { isGraphBubbleUp, isParentCommand } from \"../errors.js\";\nimport { PregelExecutableTask } from \"./types.js\";\nimport { getParentCheckpointNamespace } from \"./utils/config.js\";\nimport { patchConfigurable, type RetryPolicy } from \"./utils/index.js\";\n\nexport const DEFAULT_INITIAL_INTERVAL = 500;\nexport const DEFAULT_BACKOFF_FACTOR = 2;\nexport const DEFAULT_MAX_INTERVAL = 128000;\nexport const DEFAULT_MAX_RETRIES = 3;\n\nconst DEFAULT_STATUS_NO_RETRY = [\n  400, // Bad Request\n  401, // Unauthorized\n  402, // Payment Required\n  403, // Forbidden\n  404, // Not Found\n  405, // Method Not Allowed\n  406, // Not Acceptable\n  407, // Proxy Authentication Required\n  409, // Conflict\n];\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst DEFAULT_RETRY_ON_HANDLER = (error: any) => {\n  if (\n    error.message.startsWith(\"Cancel\") ||\n    error.message.startsWith(\"AbortError\") ||\n    error.name === \"AbortError\"\n  ) {\n    return false;\n  }\n\n  // Thrown when interrupt is called without a checkpointer\n  if (error.name === \"GraphValueError\") {\n    return false;\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  if ((error as any)?.code === \"ECONNABORTED\") {\n    return false;\n  }\n\n  const status =\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    (error as any)?.response?.status ?? (error as any)?.status;\n  if (status && DEFAULT_STATUS_NO_RETRY.includes(+status)) {\n    return false;\n  }\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  if ((error as any)?.error?.code === \"insufficient_quota\") {\n    return false;\n  }\n  return true;\n};\n\nexport type SettledPregelTask = {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  task: PregelExecutableTask<any, any>;\n  error: Error;\n  signalAborted?: boolean;\n};\n\nexport async function _runWithRetry<\n  N extends PropertyKey,\n  C extends PropertyKey\n>(\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  pregelTask: PregelExecutableTask<N, C>,\n  retryPolicy?: RetryPolicy,\n  configurable?: Record<string, unknown>,\n  signal?: AbortSignal\n): Promise<{\n  task: PregelExecutableTask<N, C>;\n  result: unknown;\n  error: Error | undefined;\n  signalAborted?: boolean;\n}> {\n  const resolvedRetryPolicy = pregelTask.retry_policy ?? retryPolicy;\n  let interval =\n    resolvedRetryPolicy !== undefined\n      ? resolvedRetryPolicy.initialInterval ?? DEFAULT_INITIAL_INTERVAL\n      : 0;\n  let attempts = 0;\n  let error;\n  let result;\n\n  let { config } = pregelTask;\n  if (configurable) config = patchConfigurable(config, configurable);\n  config = { ...config, signal };\n\n  // eslint-disable-next-line no-constant-condition\n  while (true) {\n    if (signal?.aborted) {\n      // no need to throw here - we'll throw from the runner, instead.\n      // there's just no point in retrying if the user has requested an abort.\n      break;\n    }\n    // Clear any writes from previous attempts\n    pregelTask.writes.splice(0, pregelTask.writes.length);\n    error = undefined;\n    try {\n      result = await pregelTask.proc.invoke(pregelTask.input, config);\n      break;\n    } catch (e: unknown) {\n      error = e;\n      (error as { pregelTaskId: string }).pregelTaskId = pregelTask.id;\n      if (isParentCommand(error)) {\n        const ns: string = config?.configurable?.checkpoint_ns;\n        const cmd = error.command;\n        if (cmd.graph === ns) {\n          // this command is for the current graph, handle it\n          for (const writer of pregelTask.writers) {\n            await writer.invoke(cmd, config);\n          }\n          error = undefined;\n          break;\n        } else if (cmd.graph === Command.PARENT) {\n          // this command is for the parent graph, assign it to the parent\n          const parentNs = getParentCheckpointNamespace(ns);\n          error.command = new Command({\n            ...error.command,\n            graph: parentNs,\n          });\n        }\n      }\n      if (isGraphBubbleUp(error)) {\n        break;\n      }\n      if (resolvedRetryPolicy === undefined) {\n        break;\n      }\n      attempts += 1;\n      // check if we should give up\n      if (\n        attempts >= (resolvedRetryPolicy.maxAttempts ?? DEFAULT_MAX_RETRIES)\n      ) {\n        break;\n      }\n      const retryOn = resolvedRetryPolicy.retryOn ?? DEFAULT_RETRY_ON_HANDLER;\n      if (!retryOn(error)) {\n        break;\n      }\n      interval = Math.min(\n        resolvedRetryPolicy.maxInterval ?? DEFAULT_MAX_INTERVAL,\n        interval * (resolvedRetryPolicy.backoffFactor ?? DEFAULT_BACKOFF_FACTOR)\n      );\n      const intervalWithJitter = resolvedRetryPolicy.jitter\n        ? Math.floor(interval + Math.random() * 1000)\n        : interval;\n      // sleep before retrying\n      // eslint-disable-next-line no-promise-executor-return\n      await new Promise((resolve) => setTimeout(resolve, intervalWithJitter));\n      // log the retry\n      const errorName =\n        (error as Error).name ??\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        ((error as Error).constructor as any).unminifiable_name ??\n        (error as Error).constructor.name;\n      if (resolvedRetryPolicy?.logWarning ?? true) {\n        console.log(\n          `Retrying task \"${String(pregelTask.name)}\" after ${interval.toFixed(\n            2\n          )}ms (attempt ${attempts}) after ${errorName}: ${error}`\n        );\n      }\n\n      // signal subgraphs to resume (if available)\n      config = patchConfigurable(config, { [CONFIG_KEY_RESUMING]: true });\n    }\n  }\n  return {\n    task: pregelTask,\n    result,\n    error: error as Error | undefined,\n    signalAborted: signal?.aborted,\n  };\n}\n"],"mappings":";;;;;;AAMA,MAAa,2BAA2B;AACxC,MAAa,yBAAyB;AACtC,MAAa,uBAAuB;AACpC,MAAa,sBAAsB;AAEnC,MAAM,0BAA0B;CAC9B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;;AAIF,MAAM,4BAA4B,UAAe;AAC/C,KACE,MAAM,QAAQ,WAAW,aACzB,MAAM,QAAQ,WAAW,iBACzB,MAAM,SAAS,aAEf,QAAO;AAIT,KAAI,MAAM,SAAS,kBACjB,QAAO;AAIT,KAAK,OAAe,SAAS,eAC3B,QAAO;CAGT,MAAM,SAEH,OAAe,UAAU,UAAW,OAAe;AACtD,KAAI,UAAU,wBAAwB,SAAS,CAAC,QAC9C,QAAO;AAGT,KAAK,OAAe,OAAO,SAAS,qBAClC,QAAO;AAET,QAAO;;AAUT,eAAsB,cAKpB,YACA,aACA,cACA,QAMC;CACD,MAAM,sBAAsB,WAAW,gBAAgB;CACvD,IAAI,WACF,wBAAwB,SACpB,oBAAoB,mBAAmB,2BACvC;CACN,IAAI,WAAW;CACf,IAAI;CACJ,IAAI;CAEJ,IAAI,EAAE,WAAW;AACjB,KAAI,aAAc,UAASA,gCAAkB,QAAQ;AACrD,UAAS;EAAE,GAAG;EAAQ;;AAGtB,QAAO,MAAM;AACX,MAAI,QAAQ,QAGV;AAGF,aAAW,OAAO,OAAO,GAAG,WAAW,OAAO;AAC9C,UAAQ;AACR,MAAI;AACF,YAAS,MAAM,WAAW,KAAK,OAAO,WAAW,OAAO;AACxD;WACOC,GAAY;AACnB,WAAQ;AACR,GAAC,MAAmC,eAAe,WAAW;AAC9D,OAAIC,+BAAgB,QAAQ;IAC1B,MAAMC,KAAa,QAAQ,cAAc;IACzC,MAAM,MAAM,MAAM;AAClB,QAAI,IAAI,UAAU,IAAI;AAEpB,UAAK,MAAM,UAAU,WAAW,QAC9B,OAAM,OAAO,OAAO,KAAK;AAE3B,aAAQ;AACR;eACS,IAAI,UAAUC,0BAAQ,QAAQ;KAEvC,MAAM,WAAWC,4CAA6B;AAC9C,WAAM,UAAU,IAAID,0BAAQ;MAC1B,GAAG,MAAM;MACT,OAAO;;;;AAIb,OAAIE,+BAAgB,OAClB;AAEF,OAAI,wBAAwB,OAC1B;AAEF,eAAY;AAEZ,OACE,aAAa,oBAAoB,eAAe,qBAEhD;GAEF,MAAM,UAAU,oBAAoB,WAAW;AAC/C,OAAI,CAAC,QAAQ,OACX;AAEF,cAAW,KAAK,IACd,oBAAoB,eAAe,sBACnC,YAAY,oBAAoB,iBAAiB;GAEnD,MAAM,qBAAqB,oBAAoB,SAC3C,KAAK,MAAM,WAAW,KAAK,WAAW,OACtC;AAGJ,SAAM,IAAI,SAAS,YAAY,WAAW,SAAS;GAEnD,MAAM,YACH,MAAgB,QAEf,MAAgB,YAAoB,qBACrC,MAAgB,YAAY;AAC/B,OAAI,qBAAqB,cAAc,KACrC,SAAQ,IACN,kBAAkB,OAAO,WAAW,MAAM,UAAU,SAAS,QAC3D,GACA,cAAc,SAAS,UAAU,UAAU,IAAI;AAKrD,YAASN,gCAAkB,QAAQ,GAAGO,wCAAsB;;;AAGhE,QAAO;EACL,MAAM;EACN;EACO;EACP,eAAe,QAAQ"}

package/dist/pregel/retry.js

@@ -1,92 +0,0 @@
-import { isGraphBubbleUp, isParentCommand } from "../errors.js";
-import { CONFIG_KEY_RESUMING, Command } from "../constants.js";
-import { getParentCheckpointNamespace } from "./utils/config.js";
-import { patchConfigurable } from "./utils/index.js";
-
-//#region src/pregel/retry.ts
-const DEFAULT_INITIAL_INTERVAL = 500;
-const DEFAULT_BACKOFF_FACTOR = 2;
-const DEFAULT_MAX_INTERVAL = 128e3;
-const DEFAULT_MAX_RETRIES = 3;
-const DEFAULT_STATUS_NO_RETRY = [
-	400,
-	401,
-	402,
-	403,
-	404,
-	405,
-	406,
-	407,
-	409
-];
-const DEFAULT_RETRY_ON_HANDLER = (error) => {
-	if (error.message.startsWith("Cancel") || error.message.startsWith("AbortError") || error.name === "AbortError") return false;
-	if (error.name === "GraphValueError") return false;
-	if (error?.code === "ECONNABORTED") return false;
-	const status = error?.response?.status ?? error?.status;
-	if (status && DEFAULT_STATUS_NO_RETRY.includes(+status)) return false;
-	if (error?.error?.code === "insufficient_quota") return false;
-	return true;
-};
-async function _runWithRetry(pregelTask, retryPolicy, configurable, signal) {
-	const resolvedRetryPolicy = pregelTask.retry_policy ?? retryPolicy;
-	let interval = resolvedRetryPolicy !== void 0 ? resolvedRetryPolicy.initialInterval ?? DEFAULT_INITIAL_INTERVAL : 0;
-	let attempts = 0;
-	let error;
-	let result;
-	let { config } = pregelTask;
-	if (configurable) config = patchConfigurable(config, configurable);
-	config = {
-		...config,
-		signal
-	};
-	while (true) {
-		if (signal?.aborted) break;
-		pregelTask.writes.splice(0, pregelTask.writes.length);
-		error = void 0;
-		try {
-			result = await pregelTask.proc.invoke(pregelTask.input, config);
-			break;
-		} catch (e) {
-			error = e;
-			error.pregelTaskId = pregelTask.id;
-			if (isParentCommand(error)) {
-				const ns = config?.configurable?.checkpoint_ns;
-				const cmd = error.command;
-				if (cmd.graph === ns) {
-					for (const writer of pregelTask.writers) await writer.invoke(cmd, config);
-					error = void 0;
-					break;
-				} else if (cmd.graph === Command.PARENT) {
-					const parentNs = getParentCheckpointNamespace(ns);
-					error.command = new Command({
-						...error.command,
-						graph: parentNs
-					});
-				}
-			}
-			if (isGraphBubbleUp(error)) break;
-			if (resolvedRetryPolicy === void 0) break;
-			attempts += 1;
-			if (attempts >= (resolvedRetryPolicy.maxAttempts ?? DEFAULT_MAX_RETRIES)) break;
-			const retryOn = resolvedRetryPolicy.retryOn ?? DEFAULT_RETRY_ON_HANDLER;
-			if (!retryOn(error)) break;
-			interval = Math.min(resolvedRetryPolicy.maxInterval ?? DEFAULT_MAX_INTERVAL, interval * (resolvedRetryPolicy.backoffFactor ?? DEFAULT_BACKOFF_FACTOR));
-			const intervalWithJitter = resolvedRetryPolicy.jitter ? Math.floor(interval + Math.random() * 1e3) : interval;
-			await new Promise((resolve) => setTimeout(resolve, intervalWithJitter));
-			const errorName = error.name ?? error.constructor.unminifiable_name ?? error.constructor.name;
-			if (resolvedRetryPolicy?.logWarning ?? true) console.log(`Retrying task "${String(pregelTask.name)}" after ${interval.toFixed(2)}ms (attempt ${attempts}) after ${errorName}: ${error}`);
-			config = patchConfigurable(config, { [CONFIG_KEY_RESUMING]: true });
-		}
-	}
-	return {
-		task: pregelTask,
-		result,
-		error,
-		signalAborted: signal?.aborted
-	};
-}
-
-//#endregion
-export { _runWithRetry };
-//# sourceMappingURL=retry.js.map

package/dist/pregel/retry.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"retry.js","names":["e: unknown","ns: string"],"sources":["../../src/pregel/retry.ts"],"sourcesContent":["import { Command, CONFIG_KEY_RESUMING } from \"../constants.js\";\nimport { isGraphBubbleUp, isParentCommand } from \"../errors.js\";\nimport { PregelExecutableTask } from \"./types.js\";\nimport { getParentCheckpointNamespace } from \"./utils/config.js\";\nimport { patchConfigurable, type RetryPolicy } from \"./utils/index.js\";\n\nexport const DEFAULT_INITIAL_INTERVAL = 500;\nexport const DEFAULT_BACKOFF_FACTOR = 2;\nexport const DEFAULT_MAX_INTERVAL = 128000;\nexport const DEFAULT_MAX_RETRIES = 3;\n\nconst DEFAULT_STATUS_NO_RETRY = [\n  400, // Bad Request\n  401, // Unauthorized\n  402, // Payment Required\n  403, // Forbidden\n  404, // Not Found\n  405, // Method Not Allowed\n  406, // Not Acceptable\n  407, // Proxy Authentication Required\n  409, // Conflict\n];\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nconst DEFAULT_RETRY_ON_HANDLER = (error: any) => {\n  if (\n    error.message.startsWith(\"Cancel\") ||\n    error.message.startsWith(\"AbortError\") ||\n    error.name === \"AbortError\"\n  ) {\n    return false;\n  }\n\n  // Thrown when interrupt is called without a checkpointer\n  if (error.name === \"GraphValueError\") {\n    return false;\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  if ((error as any)?.code === \"ECONNABORTED\") {\n    return false;\n  }\n\n  const status =\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    (error as any)?.response?.status ?? (error as any)?.status;\n  if (status && DEFAULT_STATUS_NO_RETRY.includes(+status)) {\n    return false;\n  }\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  if ((error as any)?.error?.code === \"insufficient_quota\") {\n    return false;\n  }\n  return true;\n};\n\nexport type SettledPregelTask = {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  task: PregelExecutableTask<any, any>;\n  error: Error;\n  signalAborted?: boolean;\n};\n\nexport async function _runWithRetry<\n  N extends PropertyKey,\n  C extends PropertyKey\n>(\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  pregelTask: PregelExecutableTask<N, C>,\n  retryPolicy?: RetryPolicy,\n  configurable?: Record<string, unknown>,\n  signal?: AbortSignal\n): Promise<{\n  task: PregelExecutableTask<N, C>;\n  result: unknown;\n  error: Error | undefined;\n  signalAborted?: boolean;\n}> {\n  const resolvedRetryPolicy = pregelTask.retry_policy ?? retryPolicy;\n  let interval =\n    resolvedRetryPolicy !== undefined\n      ? resolvedRetryPolicy.initialInterval ?? DEFAULT_INITIAL_INTERVAL\n      : 0;\n  let attempts = 0;\n  let error;\n  let result;\n\n  let { config } = pregelTask;\n  if (configurable) config = patchConfigurable(config, configurable);\n  config = { ...config, signal };\n\n  // eslint-disable-next-line no-constant-condition\n  while (true) {\n    if (signal?.aborted) {\n      // no need to throw here - we'll throw from the runner, instead.\n      // there's just no point in retrying if the user has requested an abort.\n      break;\n    }\n    // Clear any writes from previous attempts\n    pregelTask.writes.splice(0, pregelTask.writes.length);\n    error = undefined;\n    try {\n      result = await pregelTask.proc.invoke(pregelTask.input, config);\n      break;\n    } catch (e: unknown) {\n      error = e;\n      (error as { pregelTaskId: string }).pregelTaskId = pregelTask.id;\n      if (isParentCommand(error)) {\n        const ns: string = config?.configurable?.checkpoint_ns;\n        const cmd = error.command;\n        if (cmd.graph === ns) {\n          // this command is for the current graph, handle it\n          for (const writer of pregelTask.writers) {\n            await writer.invoke(cmd, config);\n          }\n          error = undefined;\n          break;\n        } else if (cmd.graph === Command.PARENT) {\n          // this command is for the parent graph, assign it to the parent\n          const parentNs = getParentCheckpointNamespace(ns);\n          error.command = new Command({\n            ...error.command,\n            graph: parentNs,\n          });\n        }\n      }\n      if (isGraphBubbleUp(error)) {\n        break;\n      }\n      if (resolvedRetryPolicy === undefined) {\n        break;\n      }\n      attempts += 1;\n      // check if we should give up\n      if (\n        attempts >= (resolvedRetryPolicy.maxAttempts ?? DEFAULT_MAX_RETRIES)\n      ) {\n        break;\n      }\n      const retryOn = resolvedRetryPolicy.retryOn ?? DEFAULT_RETRY_ON_HANDLER;\n      if (!retryOn(error)) {\n        break;\n      }\n      interval = Math.min(\n        resolvedRetryPolicy.maxInterval ?? DEFAULT_MAX_INTERVAL,\n        interval * (resolvedRetryPolicy.backoffFactor ?? DEFAULT_BACKOFF_FACTOR)\n      );\n      const intervalWithJitter = resolvedRetryPolicy.jitter\n        ? Math.floor(interval + Math.random() * 1000)\n        : interval;\n      // sleep before retrying\n      // eslint-disable-next-line no-promise-executor-return\n      await new Promise((resolve) => setTimeout(resolve, intervalWithJitter));\n      // log the retry\n      const errorName =\n        (error as Error).name ??\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        ((error as Error).constructor as any).unminifiable_name ??\n        (error as Error).constructor.name;\n      if (resolvedRetryPolicy?.logWarning ?? true) {\n        console.log(\n          `Retrying task \"${String(pregelTask.name)}\" after ${interval.toFixed(\n            2\n          )}ms (attempt ${attempts}) after ${errorName}: ${error}`\n        );\n      }\n\n      // signal subgraphs to resume (if available)\n      config = patchConfigurable(config, { [CONFIG_KEY_RESUMING]: true });\n    }\n  }\n  return {\n    task: pregelTask,\n    result,\n    error: error as Error | undefined,\n    signalAborted: signal?.aborted,\n  };\n}\n"],"mappings":";;;;;;AAMA,MAAa,2BAA2B;AACxC,MAAa,yBAAyB;AACtC,MAAa,uBAAuB;AACpC,MAAa,sBAAsB;AAEnC,MAAM,0BAA0B;CAC9B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;;AAIF,MAAM,4BAA4B,UAAe;AAC/C,KACE,MAAM,QAAQ,WAAW,aACzB,MAAM,QAAQ,WAAW,iBACzB,MAAM,SAAS,aAEf,QAAO;AAIT,KAAI,MAAM,SAAS,kBACjB,QAAO;AAIT,KAAK,OAAe,SAAS,eAC3B,QAAO;CAGT,MAAM,SAEH,OAAe,UAAU,UAAW,OAAe;AACtD,KAAI,UAAU,wBAAwB,SAAS,CAAC,QAC9C,QAAO;AAGT,KAAK,OAAe,OAAO,SAAS,qBAClC,QAAO;AAET,QAAO;;AAUT,eAAsB,cAKpB,YACA,aACA,cACA,QAMC;CACD,MAAM,sBAAsB,WAAW,gBAAgB;CACvD,IAAI,WACF,wBAAwB,SACpB,oBAAoB,mBAAmB,2BACvC;CACN,IAAI,WAAW;CACf,IAAI;CACJ,IAAI;CAEJ,IAAI,EAAE,WAAW;AACjB,KAAI,aAAc,UAAS,kBAAkB,QAAQ;AACrD,UAAS;EAAE,GAAG;EAAQ;;AAGtB,QAAO,MAAM;AACX,MAAI,QAAQ,QAGV;AAGF,aAAW,OAAO,OAAO,GAAG,WAAW,OAAO;AAC9C,UAAQ;AACR,MAAI;AACF,YAAS,MAAM,WAAW,KAAK,OAAO,WAAW,OAAO;AACxD;WACOA,GAAY;AACnB,WAAQ;AACR,GAAC,MAAmC,eAAe,WAAW;AAC9D,OAAI,gBAAgB,QAAQ;IAC1B,MAAMC,KAAa,QAAQ,cAAc;IACzC,MAAM,MAAM,MAAM;AAClB,QAAI,IAAI,UAAU,IAAI;AAEpB,UAAK,MAAM,UAAU,WAAW,QAC9B,OAAM,OAAO,OAAO,KAAK;AAE3B,aAAQ;AACR;eACS,IAAI,UAAU,QAAQ,QAAQ;KAEvC,MAAM,WAAW,6BAA6B;AAC9C,WAAM,UAAU,IAAI,QAAQ;MAC1B,GAAG,MAAM;MACT,OAAO;;;;AAIb,OAAI,gBAAgB,OAClB;AAEF,OAAI,wBAAwB,OAC1B;AAEF,eAAY;AAEZ,OACE,aAAa,oBAAoB,eAAe,qBAEhD;GAEF,MAAM,UAAU,oBAAoB,WAAW;AAC/C,OAAI,CAAC,QAAQ,OACX;AAEF,cAAW,KAAK,IACd,oBAAoB,eAAe,sBACnC,YAAY,oBAAoB,iBAAiB;GAEnD,MAAM,qBAAqB,oBAAoB,SAC3C,KAAK,MAAM,WAAW,KAAK,WAAW,OACtC;AAGJ,SAAM,IAAI,SAAS,YAAY,WAAW,SAAS;GAEnD,MAAM,YACH,MAAgB,QAEf,MAAgB,YAAoB,qBACrC,MAAgB,YAAY;AAC/B,OAAI,qBAAqB,cAAc,KACrC,SAAQ,IACN,kBAAkB,OAAO,WAAW,MAAM,UAAU,SAAS,QAC3D,GACA,cAAc,SAAS,UAAU,UAAU,IAAI;AAKrD,YAAS,kBAAkB,QAAQ,GAAG,sBAAsB;;;AAGhE,QAAO;EACL,MAAM;EACN;EACO;EACP,eAAe,QAAQ"}

package/dist/pregel/runnable_types.d.cts

@@ -1,49 +0,0 @@
-import { BaseStore } from "@langchain/langgraph-checkpoint";
-import { RunnableConfig, RunnableInterface } from "@langchain/core/runnables";
-
-//#region src/pregel/runnable_types.d.ts
-type RunnableFunc$1<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = (input: RunInput,
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-options: CallOptions) => RunOutput | Promise<RunOutput>;
-type RunnableMapLike<RunInput, RunOutput> = { [K in keyof RunOutput]: RunnableLike$1<RunInput, RunOutput[K]> };
-type RunnableLike$1<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = RunnableInterface<RunInput, RunOutput, CallOptions> | RunnableFunc$1<RunInput, RunOutput, CallOptions> | RunnableMapLike<RunInput, RunOutput>;
-type IsEqual<T, U> = [T] extends [U] ? ([U] extends [T] ? true : false) : false;
-interface Runtime<ContextType = Record<string, unknown>, InterruptType = unknown, WriterType = unknown> {
-  configurable?: ContextType;
-  /** User provided context */
-  context?: ContextType;
-  /** Persistent key-value store */
-  store?: BaseStore;
-  /** Callback to send custom data chunks via the `custom` stream mode */
-  writer: IsEqual<WriterType, unknown> extends true ? (chunk: unknown) => void : WriterType;
-  /**
-   * Interrupts the execution of a graph node.
-   *
-   * This function can be used to pause execution of a node, and return the value of the `resume`
-   * input when the graph is re-invoked using `Command`.
-   * Multiple interrupts can be called within a single node, and each will be handled sequentially.
-   *
-   * When an interrupt is called:
-   * 1. If there's a `resume` value available (from a previous `Command`), it returns that value.
-   * 2. Otherwise, it throws a `GraphInterrupt` with the provided value
-   * 3. The graph can be resumed by passing a `Command` with a `resume` value
-   *
-   * Because the `interrupt` function propagates by throwing a special `GraphInterrupt` error,
-   * you should avoid using `try/catch` blocks around the `interrupt` function,
-   * or if you do, ensure that the `GraphInterrupt` error is thrown again within your `catch` block.
-   *
-   * @param value - The value to include in the interrupt.
-   * @returns The `resume` value provided when the graph is re-invoked with a Command.
-   */
-  interrupt: IsEqual<InterruptType, unknown> extends true ? (value: unknown) => unknown : InterruptType;
-  /** Abort signal to cancel the run. */
-  signal: AbortSignal;
-}
-interface LangGraphRunnableConfig<
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-ContextType extends Record<string, any> = Record<string, any>
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-> extends RunnableConfig<ContextType>, Partial<Runtime<ContextType, unknown, unknown>> {}
-//#endregion
-export { LangGraphRunnableConfig, RunnableLike$1 as RunnableLike, Runtime };
-//# sourceMappingURL=runnable_types.d.cts.map

package/dist/pregel/runnable_types.d.cts.map

@@ -1 +0,0 @@
-{"version":3,"file":"runnable_types.d.cts","names":["RunnableConfig","RunnableInterface","BaseStore","RunnableFunc","RunInput","CallOptions","RunOutput","Promise","RunnableMapLike","K","RunnableLike","IsEqual","T","U","Runtime","Record","ContextType","WriterType","InterruptType","AbortSignal","LangGraphRunnableConfig","Partial"],"sources":["../../src/pregel/runnable_types.d.ts"],"sourcesContent":["import { RunnableConfig, RunnableInterface } from \"@langchain/core/runnables\";\nimport { BaseStore } from \"@langchain/langgraph-checkpoint\";\ntype RunnableFunc<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = (input: RunInput, \n// eslint-disable-next-line @typescript-eslint/no-explicit-any\noptions: CallOptions) => RunOutput | Promise<RunOutput>;\ntype RunnableMapLike<RunInput, RunOutput> = {\n    [K in keyof RunOutput]: RunnableLike<RunInput, RunOutput[K]>;\n};\nexport type RunnableLike<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = RunnableInterface<RunInput, RunOutput, CallOptions> | RunnableFunc<RunInput, RunOutput, CallOptions> | RunnableMapLike<RunInput, RunOutput>;\ntype IsEqual<T, U> = [T] extends [U] ? ([U] extends [T] ? true : false) : false;\nexport interface Runtime<ContextType = Record<string, unknown>, InterruptType = unknown, WriterType = unknown> {\n    configurable?: ContextType;\n    /** User provided context */\n    context?: ContextType;\n    /** Persistent key-value store */\n    store?: BaseStore;\n    /** Callback to send custom data chunks via the `custom` stream mode */\n    writer: IsEqual<WriterType, unknown> extends true ? (chunk: unknown) => void : WriterType;\n    /**\n     * Interrupts the execution of a graph node.\n     *\n     * This function can be used to pause execution of a node, and return the value of the `resume`\n     * input when the graph is re-invoked using `Command`.\n     * Multiple interrupts can be called within a single node, and each will be handled sequentially.\n     *\n     * When an interrupt is called:\n     * 1. If there's a `resume` value available (from a previous `Command`), it returns that value.\n     * 2. Otherwise, it throws a `GraphInterrupt` with the provided value\n     * 3. The graph can be resumed by passing a `Command` with a `resume` value\n     *\n     * Because the `interrupt` function propagates by throwing a special `GraphInterrupt` error,\n     * you should avoid using `try/catch` blocks around the `interrupt` function,\n     * or if you do, ensure that the `GraphInterrupt` error is thrown again within your `catch` block.\n     *\n     * @param value - The value to include in the interrupt.\n     * @returns The `resume` value provided when the graph is re-invoked with a Command.\n     */\n    interrupt: IsEqual<InterruptType, unknown> extends true ? (value: unknown) => unknown : InterruptType;\n    /** Abort signal to cancel the run. */\n    signal: AbortSignal;\n}\nexport interface LangGraphRunnableConfig<\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nContextType extends Record<string, any> = Record<string, any>\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\n> extends RunnableConfig<ContextType>, Partial<Runtime<ContextType, unknown, unknown>> {\n}\nexport {};\n"],"mappings":";;;;KAEKG,wDAAsDH,iBAAiBA,0BAA0BI;;AAD1C,OACvDD,EAEIE,WAFJF,EAAY,GAEQG,SAFR,GAEoBC,OAFpB,CAE4BD,SAF5B,CAAA;KAGZE,eAHY,CAAA,QAAA,EAAA,SAAA,CAAA,GAAA,QAA0CR,MAI3CM,SAJ2CN,GAI/BU,cAJ+BV,CAIlBI,QAJkBJ,EAIRM,SAJQN,CAIES,CAJFT,CAAAA,CAAAA;AAA2CI,KAM1FM,cAN0FN,CAAAA,QAAAA,EAAAA,SAAAA,EAAAA,oBAMpCJ,cANoCI,GAMnBJ,cANmBI,CAAAA,GAMDH,iBANCG,CAMiBA,QANjBA,EAM2BE,SAN3BF,EAMsCC,WANtCD,CAAAA,GAMqDD,cANrDC,CAMkEA,QANlEA,EAM4EE,SAN5EF,EAMuFC,WANvFD,CAAAA,GAMsGI,eANtGJ,CAMsHA,QANtHA,EAMgIE,SANhIF,CAAAA;KAOjGO,OALIN,CAAAA,CAAAA,EAAAA,CAAAA,CAAAA,GAAAA,CAKaO,CALbP,CAAAA,SAAAA,CAKyBQ,CALzBR,CAAAA,GAAAA,CAAAA,CAKgCQ,CALhCR,CAAAA,SAAAA,CAK4CO,CAL5CP,CAAAA,GAAAA,IAAAA,GAAAA,KAAAA,CAAAA,GAAAA,KAAAA;AAAgBC,UAMRQ,OANQR,CAAAA,cAMcS,MANdT,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,gBAAAA,OAAAA,EAAAA,aAAAA,OAAAA,CAAAA,CAAAA;cAAoBA,CAAAA,EAO1BU,WAP0BV;;YAS/BU;EARTR;EAAe,KAAA,CAAA,EAURN,SAVQ;;QACqBE,EAW7BO,OAX6BP,CAWrBa,UAXqBb,EAAAA,OAAAA,CAAAA,SAAAA,IAAAA,GAAAA,CAAAA,KAAAA,EAAAA,OAAAA,EAAAA,GAAAA,IAAAA,GAW0Ca,UAX1Cb;;;;;AAEzC;;;;;;;;;;;;;;;aA6BeO,QAAQO,qEAAqEA;EA5BvFP;EAAO,MAAA,EA8BAQ,WA9BA;;AAAsBN,UAgCjBO,uBAhCiBP;;oBAkCdE,MAlCiCH,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAkCXG,MAlCWH,CAAAA,MAAAA,EAAAA,GAAAA;;CACrD,SAmCUZ,cAnCc,CAmCCgB,WAnCD,CAAA,EAmCeK,OAnCf,CAmCuBP,OAnCvB,CAmC+BE,WAnC/B,EAAA,OAAA,EAAA,OAAA,CAAA,CAAA,CAAA"}

package/dist/pregel/runnable_types.d.ts

@@ -1,49 +0,0 @@
-import { BaseStore } from "@langchain/langgraph-checkpoint";
-import { RunnableConfig, RunnableInterface } from "@langchain/core/runnables";
-
-//#region src/pregel/runnable_types.d.ts
-type RunnableFunc$1<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = (input: RunInput,
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-options: CallOptions) => RunOutput | Promise<RunOutput>;
-type RunnableMapLike<RunInput, RunOutput> = { [K in keyof RunOutput]: RunnableLike$1<RunInput, RunOutput[K]> };
-type RunnableLike$1<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = RunnableInterface<RunInput, RunOutput, CallOptions> | RunnableFunc$1<RunInput, RunOutput, CallOptions> | RunnableMapLike<RunInput, RunOutput>;
-type IsEqual<T, U> = [T] extends [U] ? ([U] extends [T] ? true : false) : false;
-interface Runtime<ContextType = Record<string, unknown>, InterruptType = unknown, WriterType = unknown> {
-  configurable?: ContextType;
-  /** User provided context */
-  context?: ContextType;
-  /** Persistent key-value store */
-  store?: BaseStore;
-  /** Callback to send custom data chunks via the `custom` stream mode */
-  writer: IsEqual<WriterType, unknown> extends true ? (chunk: unknown) => void : WriterType;
-  /**
-   * Interrupts the execution of a graph node.
-   *
-   * This function can be used to pause execution of a node, and return the value of the `resume`
-   * input when the graph is re-invoked using `Command`.
-   * Multiple interrupts can be called within a single node, and each will be handled sequentially.
-   *
-   * When an interrupt is called:
-   * 1. If there's a `resume` value available (from a previous `Command`), it returns that value.
-   * 2. Otherwise, it throws a `GraphInterrupt` with the provided value
-   * 3. The graph can be resumed by passing a `Command` with a `resume` value
-   *
-   * Because the `interrupt` function propagates by throwing a special `GraphInterrupt` error,
-   * you should avoid using `try/catch` blocks around the `interrupt` function,
-   * or if you do, ensure that the `GraphInterrupt` error is thrown again within your `catch` block.
-   *
-   * @param value - The value to include in the interrupt.
-   * @returns The `resume` value provided when the graph is re-invoked with a Command.
-   */
-  interrupt: IsEqual<InterruptType, unknown> extends true ? (value: unknown) => unknown : InterruptType;
-  /** Abort signal to cancel the run. */
-  signal: AbortSignal;
-}
-interface LangGraphRunnableConfig<
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-ContextType extends Record<string, any> = Record<string, any>
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-> extends RunnableConfig<ContextType>, Partial<Runtime<ContextType, unknown, unknown>> {}
-//#endregion
-export { LangGraphRunnableConfig, RunnableLike$1 as RunnableLike, Runtime };
-//# sourceMappingURL=runnable_types.d.ts.map

package/dist/pregel/runnable_types.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"runnable_types.d.ts","names":["RunnableConfig","RunnableInterface","BaseStore","RunnableFunc","RunInput","CallOptions","RunOutput","Promise","RunnableMapLike","K","RunnableLike","IsEqual","T","U","Runtime","Record","ContextType","WriterType","InterruptType","AbortSignal","LangGraphRunnableConfig","Partial"],"sources":["../../src/pregel/runnable_types.d.ts"],"sourcesContent":["import { RunnableConfig, RunnableInterface } from \"@langchain/core/runnables\";\nimport { BaseStore } from \"@langchain/langgraph-checkpoint\";\ntype RunnableFunc<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = (input: RunInput, \n// eslint-disable-next-line @typescript-eslint/no-explicit-any\noptions: CallOptions) => RunOutput | Promise<RunOutput>;\ntype RunnableMapLike<RunInput, RunOutput> = {\n    [K in keyof RunOutput]: RunnableLike<RunInput, RunOutput[K]>;\n};\nexport type RunnableLike<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = RunnableInterface<RunInput, RunOutput, CallOptions> | RunnableFunc<RunInput, RunOutput, CallOptions> | RunnableMapLike<RunInput, RunOutput>;\ntype IsEqual<T, U> = [T] extends [U] ? ([U] extends [T] ? true : false) : false;\nexport interface Runtime<ContextType = Record<string, unknown>, InterruptType = unknown, WriterType = unknown> {\n    configurable?: ContextType;\n    /** User provided context */\n    context?: ContextType;\n    /** Persistent key-value store */\n    store?: BaseStore;\n    /** Callback to send custom data chunks via the `custom` stream mode */\n    writer: IsEqual<WriterType, unknown> extends true ? (chunk: unknown) => void : WriterType;\n    /**\n     * Interrupts the execution of a graph node.\n     *\n     * This function can be used to pause execution of a node, and return the value of the `resume`\n     * input when the graph is re-invoked using `Command`.\n     * Multiple interrupts can be called within a single node, and each will be handled sequentially.\n     *\n     * When an interrupt is called:\n     * 1. If there's a `resume` value available (from a previous `Command`), it returns that value.\n     * 2. Otherwise, it throws a `GraphInterrupt` with the provided value\n     * 3. The graph can be resumed by passing a `Command` with a `resume` value\n     *\n     * Because the `interrupt` function propagates by throwing a special `GraphInterrupt` error,\n     * you should avoid using `try/catch` blocks around the `interrupt` function,\n     * or if you do, ensure that the `GraphInterrupt` error is thrown again within your `catch` block.\n     *\n     * @param value - The value to include in the interrupt.\n     * @returns The `resume` value provided when the graph is re-invoked with a Command.\n     */\n    interrupt: IsEqual<InterruptType, unknown> extends true ? (value: unknown) => unknown : InterruptType;\n    /** Abort signal to cancel the run. */\n    signal: AbortSignal;\n}\nexport interface LangGraphRunnableConfig<\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nContextType extends Record<string, any> = Record<string, any>\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\n> extends RunnableConfig<ContextType>, Partial<Runtime<ContextType, unknown, unknown>> {\n}\nexport {};\n"],"mappings":";;;;KAEKG,wDAAsDH,iBAAiBA,0BAA0BI;;AAD1C,OACvDD,EAEIE,WAFJF,EAAY,GAEQG,SAFR,GAEoBC,OAFpB,CAE4BD,SAF5B,CAAA;KAGZE,eAHY,CAAA,QAAA,EAAA,SAAA,CAAA,GAAA,QAA0CR,MAI3CM,SAJ2CN,GAI/BU,cAJ+BV,CAIlBI,QAJkBJ,EAIRM,SAJQN,CAIES,CAJFT,CAAAA,CAAAA;AAA2CI,KAM1FM,cAN0FN,CAAAA,QAAAA,EAAAA,SAAAA,EAAAA,oBAMpCJ,cANoCI,GAMnBJ,cANmBI,CAAAA,GAMDH,iBANCG,CAMiBA,QANjBA,EAM2BE,SAN3BF,EAMsCC,WANtCD,CAAAA,GAMqDD,cANrDC,CAMkEA,QANlEA,EAM4EE,SAN5EF,EAMuFC,WANvFD,CAAAA,GAMsGI,eANtGJ,CAMsHA,QANtHA,EAMgIE,SANhIF,CAAAA;KAOjGO,OALIN,CAAAA,CAAAA,EAAAA,CAAAA,CAAAA,GAAAA,CAKaO,CALbP,CAAAA,SAAAA,CAKyBQ,CALzBR,CAAAA,GAAAA,CAAAA,CAKgCQ,CALhCR,CAAAA,SAAAA,CAK4CO,CAL5CP,CAAAA,GAAAA,IAAAA,GAAAA,KAAAA,CAAAA,GAAAA,KAAAA;AAAgBC,UAMRQ,OANQR,CAAAA,cAMcS,MANdT,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,gBAAAA,OAAAA,EAAAA,aAAAA,OAAAA,CAAAA,CAAAA;cAAoBA,CAAAA,EAO1BU,WAP0BV;;YAS/BU;EARTR;EAAe,KAAA,CAAA,EAURN,SAVQ;;QACqBE,EAW7BO,OAX6BP,CAWrBa,UAXqBb,EAAAA,OAAAA,CAAAA,SAAAA,IAAAA,GAAAA,CAAAA,KAAAA,EAAAA,OAAAA,EAAAA,GAAAA,IAAAA,GAW0Ca,UAX1Cb;;;;;AAEzC;;;;;;;;;;;;;;;aA6BeO,QAAQO,qEAAqEA;EA5BvFP;EAAO,MAAA,EA8BAQ,WA9BA;;AAAsBN,UAgCjBO,uBAhCiBP;;oBAkCdE,MAlCiCH,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAkCXG,MAlCWH,CAAAA,MAAAA,EAAAA,GAAAA;;CACrD,SAmCUZ,cAnCc,CAmCCgB,WAnCD,CAAA,EAmCeK,OAnCf,CAmCuBP,OAnCvB,CAmC+BE,WAnC/B,EAAA,OAAA,EAAA,OAAA,CAAA,CAAA,CAAA"}

package/dist/pregel/runner.cjs

@@ -1,228 +0,0 @@
-const require_errors = require('../errors.cjs');
-const require_constants = require('../constants.cjs');
-const require_types = require('./types.cjs');
-const require_index = require('./utils/index.cjs');
-const require_retry = require('./retry.cjs');
-
-//#region src/pregel/runner.ts
-const PROMISE_ADDED_SYMBOL = Symbol.for("promiseAdded");
-function createPromiseBarrier() {
-	const barrier = {
-		next: () => void 0,
-		wait: Promise.resolve(PROMISE_ADDED_SYMBOL)
-	};
-	function waitHandler(resolve) {
-		barrier.next = () => {
-			barrier.wait = new Promise(waitHandler);
-			resolve(PROMISE_ADDED_SYMBOL);
-		};
-	}
-	barrier.wait = new Promise(waitHandler);
-	return barrier;
-}
-/**
-* Responsible for handling task execution on each tick of the {@link PregelLoop}.
-*/
-var PregelRunner = class {
-	nodeFinished;
-	loop;
-	/**
-	* Construct a new PregelRunner, which executes tasks from the provided PregelLoop.
-	* @param loop - The PregelLoop that produces tasks for this runner to execute.
-	*/
-	constructor({ loop, nodeFinished }) {
-		this.loop = loop;
-		this.nodeFinished = nodeFinished;
-	}
-	/**
-	* Execute tasks from the current step of the PregelLoop.
-	*
-	* Note: this method does NOT call {@link PregelLoop}#tick. That must be handled externally.
-	* @param options - Options for the execution.
-	*/
-	async tick(options = {}) {
-		const { timeout, retryPolicy, onStepWrite, maxConcurrency } = options;
-		const nodeErrors = /* @__PURE__ */ new Set();
-		let graphBubbleUp;
-		const exceptionSignalController = new AbortController();
-		const exceptionSignal = exceptionSignalController.signal;
-		const stepTimeoutSignal = timeout ? AbortSignal.timeout(timeout) : void 0;
-		const pendingTasks = Object.values(this.loop.tasks).filter((t) => t.writes.length === 0);
-		const { signals, disposeCombinedSignal } = this._initializeAbortSignals({
-			exceptionSignal,
-			stepTimeoutSignal,
-			signal: options.signal
-		});
-		const taskStream = this._executeTasksWithRetry(pendingTasks, {
-			signals,
-			retryPolicy,
-			maxConcurrency
-		});
-		for await (const { task, error, signalAborted } of taskStream) {
-			this._commit(task, error);
-			if (require_errors.isGraphInterrupt(error)) graphBubbleUp = error;
-			else if (require_errors.isGraphBubbleUp(error) && !require_errors.isGraphInterrupt(graphBubbleUp)) graphBubbleUp = error;
-			else if (error && (nodeErrors.size === 0 || !signalAborted)) {
-				exceptionSignalController.abort();
-				nodeErrors.add(error);
-			}
-		}
-		disposeCombinedSignal?.();
-		onStepWrite?.(this.loop.step, Object.values(this.loop.tasks).map((task) => task.writes).flat());
-		if (nodeErrors.size === 1) throw Array.from(nodeErrors)[0];
-		else if (nodeErrors.size > 1) throw new AggregateError(Array.from(nodeErrors), `Multiple errors occurred during superstep ${this.loop.step}. See the "errors" field of this exception for more details.`);
-		if (require_errors.isGraphInterrupt(graphBubbleUp)) throw graphBubbleUp;
-		if (require_errors.isGraphBubbleUp(graphBubbleUp) && this.loop.isNested) throw graphBubbleUp;
-	}
-	/**
-	* Initializes the current AbortSignals for the PregelRunner, handling the various ways that
-	* AbortSignals must be chained together so that the PregelLoop can be interrupted if necessary
-	* while still allowing nodes to gracefully exit.
-	*
-	* This method must only be called once per PregelRunner#tick. It has the side effect of updating
-	* the PregelLoop#config with the new AbortSignals so they may be propagated correctly to future
-	* ticks and subgraph calls.
-	*
-	* @param options - Options for the initialization.
-	* @returns The current abort signals.
-	* @internal
-	*/
-	_initializeAbortSignals({ exceptionSignal, stepTimeoutSignal, signal }) {
-		const previousSignals = this.loop.config.configurable?.[require_constants.CONFIG_KEY_ABORT_SIGNALS] ?? {};
-		const externalAbortSignal = previousSignals.externalAbortSignal ?? signal;
-		const timeoutAbortSignal = stepTimeoutSignal ?? previousSignals.timeoutAbortSignal;
-		const { signal: composedAbortSignal, dispose: disposeCombinedSignal } = require_index.combineAbortSignals(externalAbortSignal, timeoutAbortSignal, exceptionSignal);
-		const signals = {
-			externalAbortSignal,
-			timeoutAbortSignal,
-			composedAbortSignal
-		};
-		this.loop.config = require_index.patchConfigurable(this.loop.config, { [require_constants.CONFIG_KEY_ABORT_SIGNALS]: signals });
-		return {
-			signals,
-			disposeCombinedSignal
-		};
-	}
-	/**
-	* Concurrently executes tasks with the requested retry policy, yielding a {@link SettledPregelTask} for each task as it completes.
-	* @param tasks - The tasks to execute.
-	* @param options - Options for the execution.
-	*/
-	async *_executeTasksWithRetry(tasks, options) {
-		const { retryPolicy, maxConcurrency, signals } = options ?? {};
-		const barrier = createPromiseBarrier();
-		const executingTasksMap = {};
-		const thisCall = {
-			executingTasksMap,
-			barrier,
-			retryPolicy,
-			scheduleTask: async (task, writeIdx, call$1) => this.loop.acceptPush(task, writeIdx, call$1)
-		};
-		if (signals?.composedAbortSignal?.aborted) throw new Error("Abort");
-		let startedTasksCount = 0;
-		let listener;
-		const timeoutOrCancelSignal = require_index.combineAbortSignals(signals?.externalAbortSignal, signals?.timeoutAbortSignal);
-		const abortPromise = timeoutOrCancelSignal.signal ? new Promise((_resolve, reject) => {
-			listener = () => reject(/* @__PURE__ */ new Error("Abort"));
-			timeoutOrCancelSignal.signal?.addEventListener("abort", listener, { once: true });
-		}) : void 0;
-		while ((startedTasksCount === 0 || Object.keys(executingTasksMap).length > 0) && tasks.length) {
-			for (; Object.values(executingTasksMap).length < (maxConcurrency ?? tasks.length) && startedTasksCount < tasks.length; startedTasksCount += 1) {
-				const task = tasks[startedTasksCount];
-				executingTasksMap[task.id] = require_retry._runWithRetry(task, retryPolicy, { [require_constants.CONFIG_KEY_CALL]: call?.bind(thisCall, this, task) }, signals?.composedAbortSignal).catch((error) => {
-					return {
-						task,
-						error,
-						signalAborted: signals?.composedAbortSignal?.aborted
-					};
-				});
-			}
-			const settledTask = await Promise.race([
-				...Object.values(executingTasksMap),
-				...abortPromise ? [abortPromise] : [],
-				barrier.wait
-			]);
-			if (settledTask === PROMISE_ADDED_SYMBOL) continue;
-			yield settledTask;
-			if (listener != null) {
-				timeoutOrCancelSignal.signal?.removeEventListener("abort", listener);
-				timeoutOrCancelSignal.dispose?.();
-			}
-			delete executingTasksMap[settledTask.task.id];
-		}
-	}
-	/**
-	* Determines what writes to apply based on whether the task completed successfully, and what type of error occurred.
-	*
-	* Throws an error if the error is a {@link GraphBubbleUp} error and {@link PregelLoop}#isNested is true.
-	*
-	* @param task - The task to commit.
-	* @param error - The error that occurred, if any.
-	*/
-	_commit(task, error) {
-		if (error !== void 0) if (require_errors.isGraphInterrupt(error)) {
-			if (error.interrupts.length) {
-				const interrupts = error.interrupts.map((interrupt) => [require_constants.INTERRUPT, interrupt]);
-				const resumes = task.writes.filter((w) => w[0] === require_constants.RESUME);
-				if (resumes.length) interrupts.push(...resumes);
-				this.loop.putWrites(task.id, interrupts);
-			}
-		} else if (require_errors.isGraphBubbleUp(error) && task.writes.length) this.loop.putWrites(task.id, task.writes);
-		else this.loop.putWrites(task.id, [[require_constants.ERROR, {
-			message: error.message,
-			name: error.name
-		}]]);
-		else {
-			if (this.nodeFinished && (task.config?.tags == null || !task.config.tags.includes(require_constants.TAG_HIDDEN))) this.nodeFinished(String(task.name));
-			if (task.writes.length === 0) task.writes.push([require_constants.NO_WRITES, null]);
-			this.loop.putWrites(task.id, task.writes);
-		}
-	}
-};
-async function call(runner, task, func, name, input, options = {}) {
-	const scratchpad = task.config?.configurable?.[require_constants.CONFIG_KEY_SCRATCHPAD];
-	if (!scratchpad) throw new Error(`BUG: No scratchpad found on task ${task.name}__${task.id}`);
-	const cnt = scratchpad.callCounter;
-	scratchpad.callCounter += 1;
-	const wcall = new require_types.Call({
-		func,
-		name,
-		input,
-		cache: options.cache,
-		retry: options.retry,
-		callbacks: options.callbacks
-	});
-	const nextTask = await this.scheduleTask(task, cnt, wcall);
-	if (!nextTask) return void 0;
-	const existingPromise = this.executingTasksMap[nextTask.id];
-	if (existingPromise !== void 0) return existingPromise;
-	if (nextTask.writes.length > 0) {
-		const returns = nextTask.writes.filter(([c]) => c === require_constants.RETURN);
-		const errors = nextTask.writes.filter(([c]) => c === require_constants.ERROR);
-		if (returns.length > 0) {
-			if (returns.length === 1) return Promise.resolve(returns[0][1]);
-			throw new Error(`BUG: multiple returns found for task ${nextTask.name}__${nextTask.id}`);
-		}
-		if (errors.length > 0) {
-			if (errors.length === 1) {
-				const errorValue = errors[0][1];
-				const error = errorValue instanceof Error ? errorValue : new Error(String(errorValue));
-				return Promise.reject(error);
-			}
-			throw new Error(`BUG: multiple errors found for task ${nextTask.name}__${nextTask.id}`);
-		}
-		return void 0;
-	} else {
-		const prom = require_retry._runWithRetry(nextTask, options.retry, { [require_constants.CONFIG_KEY_CALL]: call.bind(this, runner, nextTask) });
-		this.executingTasksMap[nextTask.id] = prom;
-		this.barrier.next();
-		return prom.then(({ result, error }) => {
-			if (error) return Promise.reject(error);
-			return result;
-		});
-	}
-}
-
-//#endregion
-exports.PregelRunner = PregelRunner;
-//# sourceMappingURL=runner.cjs.map