@langchain/classic 1.0.23 → 1.0.24-dev-1773698445534

package/dist/chains/base.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"base.cjs","names":["BaseLangChain","CallbackManager","RUN_KEY"],"sources":["../../src/chains/base.ts"],"sourcesContent":["import { BaseMemory } from \"@langchain/core/memory\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { RUN_KEY } from \"@langchain/core/outputs\";\nimport {\n  CallbackManagerForChainRun,\n  CallbackManager,\n  Callbacks,\n  parseCallbackConfigArg,\n} from \"@langchain/core/callbacks/manager\";\nimport { ensureConfig, type RunnableConfig } from \"@langchain/core/runnables\";\nimport {\n  BaseLangChain,\n  BaseLangChainParams,\n} from \"@langchain/core/language_models/base\";\nimport { SerializedBaseChain } from \"./serde.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nexport interface ChainInputs extends BaseLangChainParams {\n  memory?: BaseMemory;\n\n  /**\n   * @deprecated Use `callbacks` instead\n   */\n  callbackManager?: CallbackManager;\n}\n\n/**\n * Base interface that all chains must implement.\n */\nexport abstract class BaseChain<\n  RunInput extends ChainValues = ChainValues,\n  RunOutput extends ChainValues = ChainValues,\n>\n  extends BaseLangChain<RunInput, RunOutput>\n  implements ChainInputs\n{\n  declare memory?: BaseMemory;\n\n  get lc_namespace(): string[] {\n    return [\"langchain\", \"chains\", this._chainType()];\n  }\n\n  constructor(\n    fields?: BaseMemory | ChainInputs,\n    /** @deprecated */\n    verbose?: boolean,\n    /** @deprecated */\n    callbacks?: Callbacks\n  ) {\n    if (\n      arguments.length === 1 &&\n      typeof fields === \"object\" &&\n      !(\"saveContext\" in fields)\n    ) {\n      // fields is not a BaseMemory\n      const { memory, callbackManager, ...rest } = fields;\n      super({ ...rest, callbacks: callbackManager ?? rest.callbacks });\n      this.memory = memory;\n    } else {\n      // fields is a BaseMemory\n      super({ verbose, callbacks });\n      this.memory = fields as BaseMemory;\n    }\n  }\n\n  /** @ignore */\n  _selectMemoryInputs(values: ChainValues): ChainValues {\n    const valuesForMemory = { ...values };\n    if (\"signal\" in valuesForMemory) {\n      delete valuesForMemory.signal;\n    }\n    if (\"timeout\" in valuesForMemory) {\n      delete valuesForMemory.timeout;\n    }\n    return valuesForMemory;\n  }\n\n  /**\n   * Invoke the chain with the provided input and returns the output.\n   * @param input Input values for the chain run.\n   * @param config Optional configuration for the Runnable.\n   * @returns Promise that resolves with the output of the chain run.\n   */\n  async invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput> {\n    const config = ensureConfig(options);\n    const fullValues = await this._formatValues(input);\n    const callbackManager_ = await CallbackManager.configure(\n      config?.callbacks,\n      this.callbacks,\n      config?.tags,\n      this.tags,\n      config?.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const runManager = await callbackManager_?.handleChainStart(\n      this.toJSON(),\n      fullValues,\n      undefined,\n      undefined,\n      undefined,\n      undefined,\n      config?.runName\n    );\n    let outputValues: RunOutput;\n    try {\n      if (fullValues.signal) {\n        let listener: (() => void) | undefined;\n        outputValues = (await Promise.race([\n          this._call(fullValues as RunInput, runManager, config),\n          new Promise<never>((_, reject) => {\n            listener = () => {\n              reject(new Error(\"AbortError\"));\n            };\n            fullValues.signal?.addEventListener(\"abort\", listener, {\n              once: true,\n            });\n          }),\n        ]).finally(() => {\n          if (fullValues.signal && listener) {\n            fullValues.signal.removeEventListener(\"abort\", listener);\n          }\n        })) as RunOutput;\n      } else {\n        outputValues = await this._call(\n          fullValues as RunInput,\n          runManager,\n          config\n        );\n      }\n    } catch (e) {\n      await runManager?.handleChainError(e);\n      throw e;\n    }\n    if (!(this.memory == null)) {\n      await this.memory.saveContext(\n        this._selectMemoryInputs(input),\n        outputValues\n      );\n    }\n    await runManager?.handleChainEnd(outputValues);\n    // add the runManager's currentRunId to the outputValues\n    Object.defineProperty(outputValues, RUN_KEY, {\n      value: runManager ? { runId: runManager?.runId } : undefined,\n      configurable: true,\n    });\n    return outputValues;\n  }\n\n  private _validateOutputs(outputs: Record<string, unknown>): void {\n    const missingKeys = this.outputKeys.filter((k) => !(k in outputs));\n    if (missingKeys.length) {\n      throw new Error(\n        `Missing output keys: ${missingKeys.join(\n          \", \"\n        )} from chain ${this._chainType()}`\n      );\n    }\n  }\n\n  async prepOutputs(\n    inputs: Record<string, unknown>,\n    outputs: Record<string, unknown>,\n    returnOnlyOutputs = false\n  ) {\n    this._validateOutputs(outputs);\n    if (this.memory) {\n      await this.memory.saveContext(inputs, outputs);\n    }\n    if (returnOnlyOutputs) {\n      return outputs;\n    }\n    return { ...inputs, ...outputs };\n  }\n\n  /**\n   * Run the core logic of this chain and return the output\n   */\n  abstract _call(\n    values: RunInput,\n    runManager?: CallbackManagerForChainRun,\n    config?: RunnableConfig\n  ): Promise<RunOutput>;\n\n  /**\n   * Return the string type key uniquely identifying this class of chain.\n   */\n  abstract _chainType(): string;\n\n  /**\n   * Return a json-like object representing this chain.\n   */\n  serialize(): SerializedBaseChain {\n    throw new Error(\"Method not implemented.\");\n  }\n\n  abstract get inputKeys(): string[];\n\n  abstract get outputKeys(): string[];\n\n  /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */\n  async run(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any,\n    config?: Callbacks | RunnableConfig\n  ): Promise<string> {\n    const inputKeys = this.inputKeys.filter(\n      (k) => !this.memory?.memoryKeys.includes(k)\n    );\n    const isKeylessInput = inputKeys.length <= 1;\n    if (!isKeylessInput) {\n      throw new Error(\n        `Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `\n      );\n    }\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const values = inputKeys.length ? { [inputKeys[0]]: input } : ({} as any);\n    const returnValues = await this.call(values, config);\n    const keys = Object.keys(returnValues);\n\n    if (keys.length === 1) {\n      return returnValues[keys[0]];\n    }\n    throw new Error(\n      \"return values have multiple keys, `run` only supported when one key currently\"\n    );\n  }\n\n  protected async _formatValues(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number }\n  ) {\n    const fullValues = { ...values } as typeof values;\n    if (fullValues.timeout && !fullValues.signal) {\n      fullValues.signal = AbortSignal.timeout(fullValues.timeout);\n      delete fullValues.timeout;\n    }\n    if (!(this.memory == null)) {\n      const newValues = await this.memory.loadMemoryVariables(\n        this._selectMemoryInputs(values)\n      );\n      for (const [key, value] of Object.entries(newValues)) {\n        fullValues[key] = value;\n      }\n    }\n    return fullValues;\n  }\n\n  /**\n   * @deprecated Use .invoke() instead. Will be removed in 0.2.0.\n   *\n   * Run the core logic of this chain and add to output if desired.\n   *\n   * Wraps _call and handles memory.\n   */\n  async call(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number },\n    config?: Callbacks | RunnableConfig,\n    /** @deprecated */\n    tags?: string[]\n  ): Promise<RunOutput> {\n    const parsedConfig = { tags, ...parseCallbackConfigArg(config) };\n    return this.invoke(values as RunInput, parsedConfig);\n  }\n\n  /**\n   * @deprecated Use .batch() instead. Will be removed in 0.2.0.\n   *\n   * Call the chain on all inputs in the list\n   */\n  async apply(\n    inputs: RunInput[],\n    config?: (Callbacks | RunnableConfig)[]\n  ): Promise<RunOutput[]> {\n    return Promise.all(\n      inputs.map(async (i, idx) => this.call(i, config?.[idx]))\n    );\n  }\n\n  /**\n   * Load a chain from a json-like object describing it.\n   */\n  static async deserialize(\n    data: SerializedBaseChain,\n    values: LoadValues = {}\n  ): Promise<BaseChain> {\n    switch (data._type) {\n      case \"llm_chain\": {\n        const { LLMChain } = await import(\"./llm_chain.js\");\n        return LLMChain.deserialize(data);\n      }\n      case \"sequential_chain\": {\n        const { SequentialChain } = await import(\"./sequential_chain.js\");\n        return SequentialChain.deserialize(data);\n      }\n      case \"simple_sequential_chain\": {\n        const { SimpleSequentialChain } = await import(\"./sequential_chain.js\");\n        return SimpleSequentialChain.deserialize(data);\n      }\n      case \"stuff_documents_chain\": {\n        const { StuffDocumentsChain } = await import(\"./combine_docs_chain.js\");\n        return StuffDocumentsChain.deserialize(data);\n      }\n      case \"map_reduce_documents_chain\": {\n        const { MapReduceDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return MapReduceDocumentsChain.deserialize(data);\n      }\n      case \"refine_documents_chain\": {\n        const { RefineDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return RefineDocumentsChain.deserialize(data);\n      }\n      case \"vector_db_qa\": {\n        const { VectorDBQAChain } = await import(\"./vector_db_qa.js\");\n        return VectorDBQAChain.deserialize(data, values);\n      }\n      case \"api_chain\": {\n        const { APIChain } = await import(\"./api/api_chain.js\");\n        return APIChain.deserialize(data);\n      }\n      default:\n        throw new Error(\n          `Invalid prompt type in config: ${\n            (data as SerializedBaseChain)._type\n          }`\n        );\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAsB,YAAtB,cAIUA,mDAEV;CAGE,IAAI,eAAyB;AAC3B,SAAO;GAAC;GAAa;GAAU,KAAK,YAAY;GAAC;;CAGnD,YACE,QAEA,SAEA,WACA;AACA,MACE,UAAU,WAAW,KACrB,OAAO,WAAW,YAClB,EAAE,iBAAiB,SACnB;GAEA,MAAM,EAAE,QAAQ,iBAAiB,GAAG,SAAS;AAC7C,SAAM;IAAE,GAAG;IAAM,WAAW,mBAAmB,KAAK;IAAW,CAAC;AAChE,QAAK,SAAS;SACT;AAEL,SAAM;IAAE;IAAS;IAAW,CAAC;AAC7B,QAAK,SAAS;;;;CAKlB,oBAAoB,QAAkC;EACpD,MAAM,kBAAkB,EAAE,GAAG,QAAQ;AACrC,MAAI,YAAY,gBACd,QAAO,gBAAgB;AAEzB,MAAI,aAAa,gBACf,QAAO,gBAAgB;AAEzB,SAAO;;;;;;;;CAST,MAAM,OAAO,OAAiB,SAA8C;EAC1E,MAAM,qDAAsB,QAAQ;EACpC,MAAM,aAAa,MAAM,KAAK,cAAc,MAAM;EAUlD,MAAM,aAAa,OATM,MAAMC,kDAAgB,UAC7C,QAAQ,WACR,KAAK,WACL,QAAQ,MACR,KAAK,MACL,QAAQ,UACR,KAAK,UACL,EAAE,SAAS,KAAK,SAAS,CAC1B,GAC0C,iBACzC,KAAK,QAAQ,EACb,YACA,QACA,QACA,QACA,QACA,QAAQ,QACT;EACD,IAAI;AACJ,MAAI;AACF,OAAI,WAAW,QAAQ;IACrB,IAAI;AACJ,mBAAgB,MAAM,QAAQ,KAAK,CACjC,KAAK,MAAM,YAAwB,YAAY,OAAO,EACtD,IAAI,SAAgB,GAAG,WAAW;AAChC,sBAAiB;AACf,6BAAO,IAAI,MAAM,aAAa,CAAC;;AAEjC,gBAAW,QAAQ,iBAAiB,SAAS,UAAU,EACrD,MAAM,MACP,CAAC;MACF,CACH,CAAC,CAAC,cAAc;AACf,SAAI,WAAW,UAAU,SACvB,YAAW,OAAO,oBAAoB,SAAS,SAAS;MAE1D;SAEF,gBAAe,MAAM,KAAK,MACxB,YACA,YACA,OACD;WAEI,GAAG;AACV,SAAM,YAAY,iBAAiB,EAAE;AACrC,SAAM;;AAER,MAAI,EAAE,KAAK,UAAU,MACnB,OAAM,KAAK,OAAO,YAChB,KAAK,oBAAoB,MAAM,EAC/B,aACD;AAEH,QAAM,YAAY,eAAe,aAAa;AAE9C,SAAO,eAAe,cAAcC,iCAAS;GAC3C,OAAO,aAAa,EAAE,OAAO,YAAY,OAAO,GAAG;GACnD,cAAc;GACf,CAAC;AACF,SAAO;;CAGT,AAAQ,iBAAiB,SAAwC;EAC/D,MAAM,cAAc,KAAK,WAAW,QAAQ,MAAM,EAAE,KAAK,SAAS;AAClE,MAAI,YAAY,OACd,OAAM,IAAI,MACR,wBAAwB,YAAY,KAClC,KACD,CAAC,cAAc,KAAK,YAAY,GAClC;;CAIL,MAAM,YACJ,QACA,SACA,oBAAoB,OACpB;AACA,OAAK,iBAAiB,QAAQ;AAC9B,MAAI,KAAK,OACP,OAAM,KAAK,OAAO,YAAY,QAAQ,QAAQ;AAEhD,MAAI,kBACF,QAAO;AAET,SAAO;GAAE,GAAG;GAAQ,GAAG;GAAS;;;;;CAoBlC,YAAiC;AAC/B,QAAM,IAAI,MAAM,0BAA0B;;;CAQ5C,MAAM,IAEJ,OACA,QACiB;EACjB,MAAM,YAAY,KAAK,UAAU,QAC9B,MAAM,CAAC,KAAK,QAAQ,WAAW,SAAS,EAAE,CAC5C;AAED,MAAI,EADmB,UAAU,UAAU,GAEzC,OAAM,IAAI,MACR,SAAS,KAAK,YAAY,CAAC,6CAC5B;EAGH,MAAM,SAAS,UAAU,SAAS,GAAG,UAAU,KAAK,OAAO,GAAI,EAAE;EACjE,MAAM,eAAe,MAAM,KAAK,KAAK,QAAQ,OAAO;EACpD,MAAM,OAAO,OAAO,KAAK,aAAa;AAEtC,MAAI,KAAK,WAAW,EAClB,QAAO,aAAa,KAAK;AAE3B,QAAM,IAAI,MACR,gFACD;;CAGH,MAAgB,cACd,QACA;EACA,MAAM,aAAa,EAAE,GAAG,QAAQ;AAChC,MAAI,WAAW,WAAW,CAAC,WAAW,QAAQ;AAC5C,cAAW,SAAS,YAAY,QAAQ,WAAW,QAAQ;AAC3D,UAAO,WAAW;;AAEpB,MAAI,EAAE,KAAK,UAAU,OAAO;GAC1B,MAAM,YAAY,MAAM,KAAK,OAAO,oBAClC,KAAK,oBAAoB,OAAO,CACjC;AACD,QAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,UAAU,CAClD,YAAW,OAAO;;AAGtB,SAAO;;;;;;;;;CAUT,MAAM,KACJ,QACA,QAEA,MACoB;EACpB,MAAM,eAAe;GAAE;GAAM,iEAA0B,OAAO;GAAE;AAChE,SAAO,KAAK,OAAO,QAAoB,aAAa;;;;;;;CAQtD,MAAM,MACJ,QACA,QACsB;AACtB,SAAO,QAAQ,IACb,OAAO,IAAI,OAAO,GAAG,QAAQ,KAAK,KAAK,GAAG,SAAS,KAAK,CAAC,CAC1D;;;;;CAMH,aAAa,YACX,MACA,SAAqB,EAAE,EACH;AACpB,UAAQ,KAAK,OAAb;GACE,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,2CAAM;AAC3B,WAAO,SAAS,YAAY,KAAK;;GAEnC,KAAK,oBAAoB;IACvB,MAAM,EAAE,oBAAoB,2CAAM;AAClC,WAAO,gBAAgB,YAAY,KAAK;;GAE1C,KAAK,2BAA2B;IAC9B,MAAM,EAAE,0BAA0B,2CAAM;AACxC,WAAO,sBAAsB,YAAY,KAAK;;GAEhD,KAAK,yBAAyB;IAC5B,MAAM,EAAE,wBAAwB,2CAAM;AACtC,WAAO,oBAAoB,YAAY,KAAK;;GAE9C,KAAK,8BAA8B;IACjC,MAAM,EAAE,4BACN,2CAAM;AACR,WAAO,wBAAwB,YAAY,KAAK;;GAElD,KAAK,0BAA0B;IAC7B,MAAM,EAAE,yBACN,2CAAM;AACR,WAAO,qBAAqB,YAAY,KAAK;;GAE/C,KAAK,gBAAgB;IACnB,MAAM,EAAE,oBAAoB,2CAAM;AAClC,WAAO,gBAAgB,YAAY,MAAM,OAAO;;GAElD,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,2CAAM;AAC3B,WAAO,SAAS,YAAY,KAAK;;GAEnC,QACE,OAAM,IAAI,MACR,kCACG,KAA6B,QAEjC"}
+{"version":3,"file":"base.cjs","names":["BaseLangChain","CallbackManager","RUN_KEY"],"sources":["../../src/chains/base.ts"],"sourcesContent":["import { BaseMemory } from \"@langchain/core/memory\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { RUN_KEY } from \"@langchain/core/outputs\";\nimport {\n  CallbackManagerForChainRun,\n  CallbackManager,\n  Callbacks,\n  parseCallbackConfigArg,\n} from \"@langchain/core/callbacks/manager\";\nimport { ensureConfig, type RunnableConfig } from \"@langchain/core/runnables\";\nimport {\n  BaseLangChain,\n  BaseLangChainParams,\n} from \"@langchain/core/language_models/base\";\nimport { SerializedBaseChain } from \"./serde.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nexport interface ChainInputs extends BaseLangChainParams {\n  memory?: BaseMemory;\n\n  /**\n   * @deprecated Use `callbacks` instead\n   */\n  callbackManager?: CallbackManager;\n}\n\n/**\n * Base interface that all chains must implement.\n */\nexport abstract class BaseChain<\n  RunInput extends ChainValues = ChainValues,\n  RunOutput extends ChainValues = ChainValues,\n>\n  extends BaseLangChain<RunInput, RunOutput>\n  implements ChainInputs\n{\n  declare memory?: BaseMemory;\n\n  get lc_namespace(): string[] {\n    return [\"langchain\", \"chains\", this._chainType()];\n  }\n\n  constructor(\n    fields?: BaseMemory | ChainInputs,\n    /** @deprecated */\n    verbose?: boolean,\n    /** @deprecated */\n    callbacks?: Callbacks\n  ) {\n    if (\n      arguments.length === 1 &&\n      typeof fields === \"object\" &&\n      !(\"saveContext\" in fields)\n    ) {\n      // fields is not a BaseMemory\n      const { memory, callbackManager, ...rest } = fields;\n      super({ ...rest, callbacks: callbackManager ?? rest.callbacks });\n      this.memory = memory;\n    } else {\n      // fields is a BaseMemory\n      super({ verbose, callbacks });\n      this.memory = fields as BaseMemory;\n    }\n  }\n\n  /** @ignore */\n  _selectMemoryInputs(values: ChainValues): ChainValues {\n    const valuesForMemory = { ...values };\n    if (\"signal\" in valuesForMemory) {\n      delete valuesForMemory.signal;\n    }\n    if (\"timeout\" in valuesForMemory) {\n      delete valuesForMemory.timeout;\n    }\n    return valuesForMemory;\n  }\n\n  /**\n   * Invoke the chain with the provided input and returns the output.\n   * @param input Input values for the chain run.\n   * @param config Optional configuration for the Runnable.\n   * @returns Promise that resolves with the output of the chain run.\n   */\n  async invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput> {\n    const config = ensureConfig(options);\n    const fullValues = await this._formatValues(input);\n    const callbackManager_ = await CallbackManager.configure(\n      config?.callbacks,\n      this.callbacks,\n      config?.tags,\n      this.tags,\n      config?.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const runManager = await callbackManager_?.handleChainStart(\n      this.toJSON(),\n      fullValues,\n      undefined,\n      undefined,\n      undefined,\n      undefined,\n      config?.runName\n    );\n    let outputValues: RunOutput;\n    try {\n      if (fullValues.signal) {\n        let listener: (() => void) | undefined;\n        outputValues = (await Promise.race([\n          this._call(fullValues as RunInput, runManager, config),\n          new Promise<never>((_, reject) => {\n            listener = () => {\n              reject(new Error(\"AbortError\"));\n            };\n            fullValues.signal?.addEventListener(\"abort\", listener, {\n              once: true,\n            });\n          }),\n        ]).finally(() => {\n          if (fullValues.signal && listener) {\n            fullValues.signal.removeEventListener(\"abort\", listener);\n          }\n        })) as RunOutput;\n      } else {\n        outputValues = await this._call(\n          fullValues as RunInput,\n          runManager,\n          config\n        );\n      }\n    } catch (e) {\n      await runManager?.handleChainError(e);\n      throw e;\n    }\n    if (!(this.memory == null)) {\n      await this.memory.saveContext(\n        this._selectMemoryInputs(input),\n        outputValues\n      );\n    }\n    await runManager?.handleChainEnd(outputValues);\n    // add the runManager's currentRunId to the outputValues\n    Object.defineProperty(outputValues, RUN_KEY, {\n      value: runManager ? { runId: runManager?.runId } : undefined,\n      configurable: true,\n    });\n    return outputValues;\n  }\n\n  private _validateOutputs(outputs: Record<string, unknown>): void {\n    const missingKeys = this.outputKeys.filter((k) => !(k in outputs));\n    if (missingKeys.length) {\n      throw new Error(\n        `Missing output keys: ${missingKeys.join(\n          \", \"\n        )} from chain ${this._chainType()}`\n      );\n    }\n  }\n\n  async prepOutputs(\n    inputs: Record<string, unknown>,\n    outputs: Record<string, unknown>,\n    returnOnlyOutputs = false\n  ) {\n    this._validateOutputs(outputs);\n    if (this.memory) {\n      await this.memory.saveContext(inputs, outputs);\n    }\n    if (returnOnlyOutputs) {\n      return outputs;\n    }\n    return { ...inputs, ...outputs };\n  }\n\n  /**\n   * Run the core logic of this chain and return the output\n   */\n  abstract _call(\n    values: RunInput,\n    runManager?: CallbackManagerForChainRun,\n    config?: RunnableConfig\n  ): Promise<RunOutput>;\n\n  /**\n   * Return the string type key uniquely identifying this class of chain.\n   */\n  abstract _chainType(): string;\n\n  /**\n   * Return a json-like object representing this chain.\n   */\n  serialize(): SerializedBaseChain {\n    throw new Error(\"Method not implemented.\");\n  }\n\n  abstract get inputKeys(): string[];\n\n  abstract get outputKeys(): string[];\n\n  /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */\n  async run(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any,\n    config?: Callbacks | RunnableConfig\n  ): Promise<string> {\n    const inputKeys = this.inputKeys.filter(\n      (k) => !this.memory?.memoryKeys.includes(k)\n    );\n    const isKeylessInput = inputKeys.length <= 1;\n    if (!isKeylessInput) {\n      throw new Error(\n        `Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `\n      );\n    }\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const values = inputKeys.length ? { [inputKeys[0]]: input } : ({} as any);\n    const returnValues = await this.call(values, config);\n    const keys = Object.keys(returnValues);\n\n    if (keys.length === 1) {\n      return returnValues[keys[0]];\n    }\n    throw new Error(\n      \"return values have multiple keys, `run` only supported when one key currently\"\n    );\n  }\n\n  protected async _formatValues(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number }\n  ) {\n    const fullValues = { ...values } as typeof values;\n    if (fullValues.timeout && !fullValues.signal) {\n      fullValues.signal = AbortSignal.timeout(fullValues.timeout);\n      delete fullValues.timeout;\n    }\n    if (!(this.memory == null)) {\n      const newValues = await this.memory.loadMemoryVariables(\n        this._selectMemoryInputs(values)\n      );\n      for (const [key, value] of Object.entries(newValues)) {\n        fullValues[key] = value;\n      }\n    }\n    return fullValues;\n  }\n\n  /**\n   * @deprecated Use .invoke() instead. Will be removed in 0.2.0.\n   *\n   * Run the core logic of this chain and add to output if desired.\n   *\n   * Wraps _call and handles memory.\n   */\n  async call(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number },\n    config?: Callbacks | RunnableConfig,\n    /** @deprecated */\n    tags?: string[]\n  ): Promise<RunOutput> {\n    const parsedConfig = { tags, ...parseCallbackConfigArg(config) };\n    return this.invoke(values as RunInput, parsedConfig);\n  }\n\n  /**\n   * @deprecated Use .batch() instead. Will be removed in 0.2.0.\n   *\n   * Call the chain on all inputs in the list\n   */\n  async apply(\n    inputs: RunInput[],\n    config?: (Callbacks | RunnableConfig)[]\n  ): Promise<RunOutput[]> {\n    return Promise.all(\n      inputs.map(async (i, idx) => this.call(i, config?.[idx]))\n    );\n  }\n\n  /**\n   * Load a chain from a json-like object describing it.\n   */\n  static async deserialize(\n    data: SerializedBaseChain,\n    values: LoadValues = {}\n  ): Promise<BaseChain> {\n    switch (data._type) {\n      case \"llm_chain\": {\n        const { LLMChain } = await import(\"./llm_chain.js\");\n        return LLMChain.deserialize(data);\n      }\n      case \"sequential_chain\": {\n        const { SequentialChain } = await import(\"./sequential_chain.js\");\n        return SequentialChain.deserialize(data);\n      }\n      case \"simple_sequential_chain\": {\n        const { SimpleSequentialChain } = await import(\"./sequential_chain.js\");\n        return SimpleSequentialChain.deserialize(data);\n      }\n      case \"stuff_documents_chain\": {\n        const { StuffDocumentsChain } = await import(\"./combine_docs_chain.js\");\n        return StuffDocumentsChain.deserialize(data);\n      }\n      case \"map_reduce_documents_chain\": {\n        const { MapReduceDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return MapReduceDocumentsChain.deserialize(data);\n      }\n      case \"refine_documents_chain\": {\n        const { RefineDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return RefineDocumentsChain.deserialize(data);\n      }\n      case \"vector_db_qa\": {\n        const { VectorDBQAChain } = await import(\"./vector_db_qa.js\");\n        return VectorDBQAChain.deserialize(data, values);\n      }\n      case \"api_chain\": {\n        const { APIChain } = await import(\"./api/api_chain.js\");\n        return APIChain.deserialize(data);\n      }\n      default:\n        throw new Error(\n          `Invalid prompt type in config: ${\n            (data as SerializedBaseChain)._type\n          }`\n        );\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AA+BA,IAAsB,YAAtB,cAIUA,qCAAAA,cAEV;CAGE,IAAI,eAAyB;AAC3B,SAAO;GAAC;GAAa;GAAU,KAAK,YAAY;GAAC;;CAGnD,YACE,QAEA,SAEA,WACA;AACA,MACE,UAAU,WAAW,KACrB,OAAO,WAAW,YAClB,EAAE,iBAAiB,SACnB;GAEA,MAAM,EAAE,QAAQ,iBAAiB,GAAG,SAAS;AAC7C,SAAM;IAAE,GAAG;IAAM,WAAW,mBAAmB,KAAK;IAAW,CAAC;AAChE,QAAK,SAAS;SACT;AAEL,SAAM;IAAE;IAAS;IAAW,CAAC;AAC7B,QAAK,SAAS;;;;CAKlB,oBAAoB,QAAkC;EACpD,MAAM,kBAAkB,EAAE,GAAG,QAAQ;AACrC,MAAI,YAAY,gBACd,QAAO,gBAAgB;AAEzB,MAAI,aAAa,gBACf,QAAO,gBAAgB;AAEzB,SAAO;;;;;;;;CAST,MAAM,OAAO,OAAiB,SAA8C;EAC1E,MAAM,UAAA,GAAA,0BAAA,cAAsB,QAAQ;EACpC,MAAM,aAAa,MAAM,KAAK,cAAc,MAAM;EAUlD,MAAM,aAAa,OATM,MAAMC,kCAAAA,gBAAgB,UAC7C,QAAQ,WACR,KAAK,WACL,QAAQ,MACR,KAAK,MACL,QAAQ,UACR,KAAK,UACL,EAAE,SAAS,KAAK,SAAS,CAC1B,GAC0C,iBACzC,KAAK,QAAQ,EACb,YACA,KAAA,GACA,KAAA,GACA,KAAA,GACA,KAAA,GACA,QAAQ,QACT;EACD,IAAI;AACJ,MAAI;AACF,OAAI,WAAW,QAAQ;IACrB,IAAI;AACJ,mBAAgB,MAAM,QAAQ,KAAK,CACjC,KAAK,MAAM,YAAwB,YAAY,OAAO,EACtD,IAAI,SAAgB,GAAG,WAAW;AAChC,sBAAiB;AACf,6BAAO,IAAI,MAAM,aAAa,CAAC;;AAEjC,gBAAW,QAAQ,iBAAiB,SAAS,UAAU,EACrD,MAAM,MACP,CAAC;MACF,CACH,CAAC,CAAC,cAAc;AACf,SAAI,WAAW,UAAU,SACvB,YAAW,OAAO,oBAAoB,SAAS,SAAS;MAE1D;SAEF,gBAAe,MAAM,KAAK,MACxB,YACA,YACA,OACD;WAEI,GAAG;AACV,SAAM,YAAY,iBAAiB,EAAE;AACrC,SAAM;;AAER,MAAI,EAAE,KAAK,UAAU,MACnB,OAAM,KAAK,OAAO,YAChB,KAAK,oBAAoB,MAAM,EAC/B,aACD;AAEH,QAAM,YAAY,eAAe,aAAa;AAE9C,SAAO,eAAe,cAAcC,wBAAAA,SAAS;GAC3C,OAAO,aAAa,EAAE,OAAO,YAAY,OAAO,GAAG,KAAA;GACnD,cAAc;GACf,CAAC;AACF,SAAO;;CAGT,iBAAyB,SAAwC;EAC/D,MAAM,cAAc,KAAK,WAAW,QAAQ,MAAM,EAAE,KAAK,SAAS;AAClE,MAAI,YAAY,OACd,OAAM,IAAI,MACR,wBAAwB,YAAY,KAClC,KACD,CAAC,cAAc,KAAK,YAAY,GAClC;;CAIL,MAAM,YACJ,QACA,SACA,oBAAoB,OACpB;AACA,OAAK,iBAAiB,QAAQ;AAC9B,MAAI,KAAK,OACP,OAAM,KAAK,OAAO,YAAY,QAAQ,QAAQ;AAEhD,MAAI,kBACF,QAAO;AAET,SAAO;GAAE,GAAG;GAAQ,GAAG;GAAS;;;;;CAoBlC,YAAiC;AAC/B,QAAM,IAAI,MAAM,0BAA0B;;;CAQ5C,MAAM,IAEJ,OACA,QACiB;EACjB,MAAM,YAAY,KAAK,UAAU,QAC9B,MAAM,CAAC,KAAK,QAAQ,WAAW,SAAS,EAAE,CAC5C;AAED,MAAI,EADmB,UAAU,UAAU,GAEzC,OAAM,IAAI,MACR,SAAS,KAAK,YAAY,CAAC,6CAC5B;EAGH,MAAM,SAAS,UAAU,SAAS,GAAG,UAAU,KAAK,OAAO,GAAI,EAAE;EACjE,MAAM,eAAe,MAAM,KAAK,KAAK,QAAQ,OAAO;EACpD,MAAM,OAAO,OAAO,KAAK,aAAa;AAEtC,MAAI,KAAK,WAAW,EAClB,QAAO,aAAa,KAAK;AAE3B,QAAM,IAAI,MACR,gFACD;;CAGH,MAAgB,cACd,QACA;EACA,MAAM,aAAa,EAAE,GAAG,QAAQ;AAChC,MAAI,WAAW,WAAW,CAAC,WAAW,QAAQ;AAC5C,cAAW,SAAS,YAAY,QAAQ,WAAW,QAAQ;AAC3D,UAAO,WAAW;;AAEpB,MAAI,EAAE,KAAK,UAAU,OAAO;GAC1B,MAAM,YAAY,MAAM,KAAK,OAAO,oBAClC,KAAK,oBAAoB,OAAO,CACjC;AACD,QAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,UAAU,CAClD,YAAW,OAAO;;AAGtB,SAAO;;;;;;;;;CAUT,MAAM,KACJ,QACA,QAEA,MACoB;EACpB,MAAM,eAAe;GAAE;GAAM,IAAA,GAAA,kCAAA,wBAA0B,OAAO;GAAE;AAChE,SAAO,KAAK,OAAO,QAAoB,aAAa;;;;;;;CAQtD,MAAM,MACJ,QACA,QACsB;AACtB,SAAO,QAAQ,IACb,OAAO,IAAI,OAAO,GAAG,QAAQ,KAAK,KAAK,GAAG,SAAS,KAAK,CAAC,CAC1D;;;;;CAMH,aAAa,YACX,MACA,SAAqB,EAAE,EACH;AACpB,UAAQ,KAAK,OAAb;GACE,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,kBAAA,CAAA;AAC3B,WAAO,SAAS,YAAY,KAAK;;GAEnC,KAAK,oBAAoB;IACvB,MAAM,EAAE,oBAAoB,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,yBAAA,CAAA;AAClC,WAAO,gBAAgB,YAAY,KAAK;;GAE1C,KAAK,2BAA2B;IAC9B,MAAM,EAAE,0BAA0B,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,yBAAA,CAAA;AACxC,WAAO,sBAAsB,YAAY,KAAK;;GAEhD,KAAK,yBAAyB;IAC5B,MAAM,EAAE,wBAAwB,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,2BAAA,CAAA;AACtC,WAAO,oBAAoB,YAAY,KAAK;;GAE9C,KAAK,8BAA8B;IACjC,MAAM,EAAE,4BACN,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,2BAAA,CAAA;AACR,WAAO,wBAAwB,YAAY,KAAK;;GAElD,KAAK,0BAA0B;IAC7B,MAAM,EAAE,yBACN,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,2BAAA,CAAA;AACR,WAAO,qBAAqB,YAAY,KAAK;;GAE/C,KAAK,gBAAgB;IACnB,MAAM,EAAE,oBAAoB,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,qBAAA,CAAA;AAClC,WAAO,gBAAgB,YAAY,MAAM,OAAO;;GAElD,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,MAAA,QAAA,SAAA,CAAA,WAAA,QAAM,sBAAA,CAAA;AAC3B,WAAO,SAAS,YAAY,KAAK;;GAEnC,QACE,OAAM,IAAI,MACR,kCACG,KAA6B,QAEjC"}

package/dist/chains/base.js

@@ -2,7 +2,6 @@ import { ensureConfig } from "@langchain/core/runnables";
 import { BaseLangChain } from "@langchain/core/language_models/base";
 import { RUN_KEY } from "@langchain/core/outputs";
 import { CallbackManager, parseCallbackConfigArg } from "@langchain/core/callbacks/manager";
-
 //#region src/chains/base.ts
 /**
 * Base interface that all chains must implement.
@@ -177,7 +176,7 @@ var BaseChain = class extends BaseLangChain {
 		}
 	}
 };
-
 //#endregion
 export { BaseChain };
+
 //# sourceMappingURL=base.js.map

package/dist/chains/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","names":[],"sources":["../../src/chains/base.ts"],"sourcesContent":["import { BaseMemory } from \"@langchain/core/memory\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { RUN_KEY } from \"@langchain/core/outputs\";\nimport {\n  CallbackManagerForChainRun,\n  CallbackManager,\n  Callbacks,\n  parseCallbackConfigArg,\n} from \"@langchain/core/callbacks/manager\";\nimport { ensureConfig, type RunnableConfig } from \"@langchain/core/runnables\";\nimport {\n  BaseLangChain,\n  BaseLangChainParams,\n} from \"@langchain/core/language_models/base\";\nimport { SerializedBaseChain } from \"./serde.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nexport interface ChainInputs extends BaseLangChainParams {\n  memory?: BaseMemory;\n\n  /**\n   * @deprecated Use `callbacks` instead\n   */\n  callbackManager?: CallbackManager;\n}\n\n/**\n * Base interface that all chains must implement.\n */\nexport abstract class BaseChain<\n  RunInput extends ChainValues = ChainValues,\n  RunOutput extends ChainValues = ChainValues,\n>\n  extends BaseLangChain<RunInput, RunOutput>\n  implements ChainInputs\n{\n  declare memory?: BaseMemory;\n\n  get lc_namespace(): string[] {\n    return [\"langchain\", \"chains\", this._chainType()];\n  }\n\n  constructor(\n    fields?: BaseMemory | ChainInputs,\n    /** @deprecated */\n    verbose?: boolean,\n    /** @deprecated */\n    callbacks?: Callbacks\n  ) {\n    if (\n      arguments.length === 1 &&\n      typeof fields === \"object\" &&\n      !(\"saveContext\" in fields)\n    ) {\n      // fields is not a BaseMemory\n      const { memory, callbackManager, ...rest } = fields;\n      super({ ...rest, callbacks: callbackManager ?? rest.callbacks });\n      this.memory = memory;\n    } else {\n      // fields is a BaseMemory\n      super({ verbose, callbacks });\n      this.memory = fields as BaseMemory;\n    }\n  }\n\n  /** @ignore */\n  _selectMemoryInputs(values: ChainValues): ChainValues {\n    const valuesForMemory = { ...values };\n    if (\"signal\" in valuesForMemory) {\n      delete valuesForMemory.signal;\n    }\n    if (\"timeout\" in valuesForMemory) {\n      delete valuesForMemory.timeout;\n    }\n    return valuesForMemory;\n  }\n\n  /**\n   * Invoke the chain with the provided input and returns the output.\n   * @param input Input values for the chain run.\n   * @param config Optional configuration for the Runnable.\n   * @returns Promise that resolves with the output of the chain run.\n   */\n  async invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput> {\n    const config = ensureConfig(options);\n    const fullValues = await this._formatValues(input);\n    const callbackManager_ = await CallbackManager.configure(\n      config?.callbacks,\n      this.callbacks,\n      config?.tags,\n      this.tags,\n      config?.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const runManager = await callbackManager_?.handleChainStart(\n      this.toJSON(),\n      fullValues,\n      undefined,\n      undefined,\n      undefined,\n      undefined,\n      config?.runName\n    );\n    let outputValues: RunOutput;\n    try {\n      if (fullValues.signal) {\n        let listener: (() => void) | undefined;\n        outputValues = (await Promise.race([\n          this._call(fullValues as RunInput, runManager, config),\n          new Promise<never>((_, reject) => {\n            listener = () => {\n              reject(new Error(\"AbortError\"));\n            };\n            fullValues.signal?.addEventListener(\"abort\", listener, {\n              once: true,\n            });\n          }),\n        ]).finally(() => {\n          if (fullValues.signal && listener) {\n            fullValues.signal.removeEventListener(\"abort\", listener);\n          }\n        })) as RunOutput;\n      } else {\n        outputValues = await this._call(\n          fullValues as RunInput,\n          runManager,\n          config\n        );\n      }\n    } catch (e) {\n      await runManager?.handleChainError(e);\n      throw e;\n    }\n    if (!(this.memory == null)) {\n      await this.memory.saveContext(\n        this._selectMemoryInputs(input),\n        outputValues\n      );\n    }\n    await runManager?.handleChainEnd(outputValues);\n    // add the runManager's currentRunId to the outputValues\n    Object.defineProperty(outputValues, RUN_KEY, {\n      value: runManager ? { runId: runManager?.runId } : undefined,\n      configurable: true,\n    });\n    return outputValues;\n  }\n\n  private _validateOutputs(outputs: Record<string, unknown>): void {\n    const missingKeys = this.outputKeys.filter((k) => !(k in outputs));\n    if (missingKeys.length) {\n      throw new Error(\n        `Missing output keys: ${missingKeys.join(\n          \", \"\n        )} from chain ${this._chainType()}`\n      );\n    }\n  }\n\n  async prepOutputs(\n    inputs: Record<string, unknown>,\n    outputs: Record<string, unknown>,\n    returnOnlyOutputs = false\n  ) {\n    this._validateOutputs(outputs);\n    if (this.memory) {\n      await this.memory.saveContext(inputs, outputs);\n    }\n    if (returnOnlyOutputs) {\n      return outputs;\n    }\n    return { ...inputs, ...outputs };\n  }\n\n  /**\n   * Run the core logic of this chain and return the output\n   */\n  abstract _call(\n    values: RunInput,\n    runManager?: CallbackManagerForChainRun,\n    config?: RunnableConfig\n  ): Promise<RunOutput>;\n\n  /**\n   * Return the string type key uniquely identifying this class of chain.\n   */\n  abstract _chainType(): string;\n\n  /**\n   * Return a json-like object representing this chain.\n   */\n  serialize(): SerializedBaseChain {\n    throw new Error(\"Method not implemented.\");\n  }\n\n  abstract get inputKeys(): string[];\n\n  abstract get outputKeys(): string[];\n\n  /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */\n  async run(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any,\n    config?: Callbacks | RunnableConfig\n  ): Promise<string> {\n    const inputKeys = this.inputKeys.filter(\n      (k) => !this.memory?.memoryKeys.includes(k)\n    );\n    const isKeylessInput = inputKeys.length <= 1;\n    if (!isKeylessInput) {\n      throw new Error(\n        `Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `\n      );\n    }\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const values = inputKeys.length ? { [inputKeys[0]]: input } : ({} as any);\n    const returnValues = await this.call(values, config);\n    const keys = Object.keys(returnValues);\n\n    if (keys.length === 1) {\n      return returnValues[keys[0]];\n    }\n    throw new Error(\n      \"return values have multiple keys, `run` only supported when one key currently\"\n    );\n  }\n\n  protected async _formatValues(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number }\n  ) {\n    const fullValues = { ...values } as typeof values;\n    if (fullValues.timeout && !fullValues.signal) {\n      fullValues.signal = AbortSignal.timeout(fullValues.timeout);\n      delete fullValues.timeout;\n    }\n    if (!(this.memory == null)) {\n      const newValues = await this.memory.loadMemoryVariables(\n        this._selectMemoryInputs(values)\n      );\n      for (const [key, value] of Object.entries(newValues)) {\n        fullValues[key] = value;\n      }\n    }\n    return fullValues;\n  }\n\n  /**\n   * @deprecated Use .invoke() instead. Will be removed in 0.2.0.\n   *\n   * Run the core logic of this chain and add to output if desired.\n   *\n   * Wraps _call and handles memory.\n   */\n  async call(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number },\n    config?: Callbacks | RunnableConfig,\n    /** @deprecated */\n    tags?: string[]\n  ): Promise<RunOutput> {\n    const parsedConfig = { tags, ...parseCallbackConfigArg(config) };\n    return this.invoke(values as RunInput, parsedConfig);\n  }\n\n  /**\n   * @deprecated Use .batch() instead. Will be removed in 0.2.0.\n   *\n   * Call the chain on all inputs in the list\n   */\n  async apply(\n    inputs: RunInput[],\n    config?: (Callbacks | RunnableConfig)[]\n  ): Promise<RunOutput[]> {\n    return Promise.all(\n      inputs.map(async (i, idx) => this.call(i, config?.[idx]))\n    );\n  }\n\n  /**\n   * Load a chain from a json-like object describing it.\n   */\n  static async deserialize(\n    data: SerializedBaseChain,\n    values: LoadValues = {}\n  ): Promise<BaseChain> {\n    switch (data._type) {\n      case \"llm_chain\": {\n        const { LLMChain } = await import(\"./llm_chain.js\");\n        return LLMChain.deserialize(data);\n      }\n      case \"sequential_chain\": {\n        const { SequentialChain } = await import(\"./sequential_chain.js\");\n        return SequentialChain.deserialize(data);\n      }\n      case \"simple_sequential_chain\": {\n        const { SimpleSequentialChain } = await import(\"./sequential_chain.js\");\n        return SimpleSequentialChain.deserialize(data);\n      }\n      case \"stuff_documents_chain\": {\n        const { StuffDocumentsChain } = await import(\"./combine_docs_chain.js\");\n        return StuffDocumentsChain.deserialize(data);\n      }\n      case \"map_reduce_documents_chain\": {\n        const { MapReduceDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return MapReduceDocumentsChain.deserialize(data);\n      }\n      case \"refine_documents_chain\": {\n        const { RefineDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return RefineDocumentsChain.deserialize(data);\n      }\n      case \"vector_db_qa\": {\n        const { VectorDBQAChain } = await import(\"./vector_db_qa.js\");\n        return VectorDBQAChain.deserialize(data, values);\n      }\n      case \"api_chain\": {\n        const { APIChain } = await import(\"./api/api_chain.js\");\n        return APIChain.deserialize(data);\n      }\n      default:\n        throw new Error(\n          `Invalid prompt type in config: ${\n            (data as SerializedBaseChain)._type\n          }`\n        );\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AA+BA,IAAsB,YAAtB,cAIU,cAEV;CAGE,IAAI,eAAyB;AAC3B,SAAO;GAAC;GAAa;GAAU,KAAK,YAAY;GAAC;;CAGnD,YACE,QAEA,SAEA,WACA;AACA,MACE,UAAU,WAAW,KACrB,OAAO,WAAW,YAClB,EAAE,iBAAiB,SACnB;GAEA,MAAM,EAAE,QAAQ,iBAAiB,GAAG,SAAS;AAC7C,SAAM;IAAE,GAAG;IAAM,WAAW,mBAAmB,KAAK;IAAW,CAAC;AAChE,QAAK,SAAS;SACT;AAEL,SAAM;IAAE;IAAS;IAAW,CAAC;AAC7B,QAAK,SAAS;;;;CAKlB,oBAAoB,QAAkC;EACpD,MAAM,kBAAkB,EAAE,GAAG,QAAQ;AACrC,MAAI,YAAY,gBACd,QAAO,gBAAgB;AAEzB,MAAI,aAAa,gBACf,QAAO,gBAAgB;AAEzB,SAAO;;;;;;;;CAST,MAAM,OAAO,OAAiB,SAA8C;EAC1E,MAAM,SAAS,aAAa,QAAQ;EACpC,MAAM,aAAa,MAAM,KAAK,cAAc,MAAM;EAUlD,MAAM,aAAa,OATM,MAAM,gBAAgB,UAC7C,QAAQ,WACR,KAAK,WACL,QAAQ,MACR,KAAK,MACL,QAAQ,UACR,KAAK,UACL,EAAE,SAAS,KAAK,SAAS,CAC1B,GAC0C,iBACzC,KAAK,QAAQ,EACb,YACA,QACA,QACA,QACA,QACA,QAAQ,QACT;EACD,IAAI;AACJ,MAAI;AACF,OAAI,WAAW,QAAQ;IACrB,IAAI;AACJ,mBAAgB,MAAM,QAAQ,KAAK,CACjC,KAAK,MAAM,YAAwB,YAAY,OAAO,EACtD,IAAI,SAAgB,GAAG,WAAW;AAChC,sBAAiB;AACf,6BAAO,IAAI,MAAM,aAAa,CAAC;;AAEjC,gBAAW,QAAQ,iBAAiB,SAAS,UAAU,EACrD,MAAM,MACP,CAAC;MACF,CACH,CAAC,CAAC,cAAc;AACf,SAAI,WAAW,UAAU,SACvB,YAAW,OAAO,oBAAoB,SAAS,SAAS;MAE1D;SAEF,gBAAe,MAAM,KAAK,MACxB,YACA,YACA,OACD;WAEI,GAAG;AACV,SAAM,YAAY,iBAAiB,EAAE;AACrC,SAAM;;AAER,MAAI,EAAE,KAAK,UAAU,MACnB,OAAM,KAAK,OAAO,YAChB,KAAK,oBAAoB,MAAM,EAC/B,aACD;AAEH,QAAM,YAAY,eAAe,aAAa;AAE9C,SAAO,eAAe,cAAc,SAAS;GAC3C,OAAO,aAAa,EAAE,OAAO,YAAY,OAAO,GAAG;GACnD,cAAc;GACf,CAAC;AACF,SAAO;;CAGT,AAAQ,iBAAiB,SAAwC;EAC/D,MAAM,cAAc,KAAK,WAAW,QAAQ,MAAM,EAAE,KAAK,SAAS;AAClE,MAAI,YAAY,OACd,OAAM,IAAI,MACR,wBAAwB,YAAY,KAClC,KACD,CAAC,cAAc,KAAK,YAAY,GAClC;;CAIL,MAAM,YACJ,QACA,SACA,oBAAoB,OACpB;AACA,OAAK,iBAAiB,QAAQ;AAC9B,MAAI,KAAK,OACP,OAAM,KAAK,OAAO,YAAY,QAAQ,QAAQ;AAEhD,MAAI,kBACF,QAAO;AAET,SAAO;GAAE,GAAG;GAAQ,GAAG;GAAS;;;;;CAoBlC,YAAiC;AAC/B,QAAM,IAAI,MAAM,0BAA0B;;;CAQ5C,MAAM,IAEJ,OACA,QACiB;EACjB,MAAM,YAAY,KAAK,UAAU,QAC9B,MAAM,CAAC,KAAK,QAAQ,WAAW,SAAS,EAAE,CAC5C;AAED,MAAI,EADmB,UAAU,UAAU,GAEzC,OAAM,IAAI,MACR,SAAS,KAAK,YAAY,CAAC,6CAC5B;EAGH,MAAM,SAAS,UAAU,SAAS,GAAG,UAAU,KAAK,OAAO,GAAI,EAAE;EACjE,MAAM,eAAe,MAAM,KAAK,KAAK,QAAQ,OAAO;EACpD,MAAM,OAAO,OAAO,KAAK,aAAa;AAEtC,MAAI,KAAK,WAAW,EAClB,QAAO,aAAa,KAAK;AAE3B,QAAM,IAAI,MACR,gFACD;;CAGH,MAAgB,cACd,QACA;EACA,MAAM,aAAa,EAAE,GAAG,QAAQ;AAChC,MAAI,WAAW,WAAW,CAAC,WAAW,QAAQ;AAC5C,cAAW,SAAS,YAAY,QAAQ,WAAW,QAAQ;AAC3D,UAAO,WAAW;;AAEpB,MAAI,EAAE,KAAK,UAAU,OAAO;GAC1B,MAAM,YAAY,MAAM,KAAK,OAAO,oBAClC,KAAK,oBAAoB,OAAO,CACjC;AACD,QAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,UAAU,CAClD,YAAW,OAAO;;AAGtB,SAAO;;;;;;;;;CAUT,MAAM,KACJ,QACA,QAEA,MACoB;EACpB,MAAM,eAAe;GAAE;GAAM,GAAG,uBAAuB,OAAO;GAAE;AAChE,SAAO,KAAK,OAAO,QAAoB,aAAa;;;;;;;CAQtD,MAAM,MACJ,QACA,QACsB;AACtB,SAAO,QAAQ,IACb,OAAO,IAAI,OAAO,GAAG,QAAQ,KAAK,KAAK,GAAG,SAAS,KAAK,CAAC,CAC1D;;;;;CAMH,aAAa,YACX,MACA,SAAqB,EAAE,EACH;AACpB,UAAQ,KAAK,OAAb;GACE,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,MAAM,OAAO;AAClC,WAAO,SAAS,YAAY,KAAK;;GAEnC,KAAK,oBAAoB;IACvB,MAAM,EAAE,oBAAoB,MAAM,OAAO;AACzC,WAAO,gBAAgB,YAAY,KAAK;;GAE1C,KAAK,2BAA2B;IAC9B,MAAM,EAAE,0BAA0B,MAAM,OAAO;AAC/C,WAAO,sBAAsB,YAAY,KAAK;;GAEhD,KAAK,yBAAyB;IAC5B,MAAM,EAAE,wBAAwB,MAAM,OAAO;AAC7C,WAAO,oBAAoB,YAAY,KAAK;;GAE9C,KAAK,8BAA8B;IACjC,MAAM,EAAE,4BACN,MAAM,OAAO;AACf,WAAO,wBAAwB,YAAY,KAAK;;GAElD,KAAK,0BAA0B;IAC7B,MAAM,EAAE,yBACN,MAAM,OAAO;AACf,WAAO,qBAAqB,YAAY,KAAK;;GAE/C,KAAK,gBAAgB;IACnB,MAAM,EAAE,oBAAoB,MAAM,OAAO;AACzC,WAAO,gBAAgB,YAAY,MAAM,OAAO;;GAElD,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,MAAM,OAAO;AAClC,WAAO,SAAS,YAAY,KAAK;;GAEnC,QACE,OAAM,IAAI,MACR,kCACG,KAA6B,QAEjC"}
+{"version":3,"file":"base.js","names":[],"sources":["../../src/chains/base.ts"],"sourcesContent":["import { BaseMemory } from \"@langchain/core/memory\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { RUN_KEY } from \"@langchain/core/outputs\";\nimport {\n  CallbackManagerForChainRun,\n  CallbackManager,\n  Callbacks,\n  parseCallbackConfigArg,\n} from \"@langchain/core/callbacks/manager\";\nimport { ensureConfig, type RunnableConfig } from \"@langchain/core/runnables\";\nimport {\n  BaseLangChain,\n  BaseLangChainParams,\n} from \"@langchain/core/language_models/base\";\nimport { SerializedBaseChain } from \"./serde.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nexport interface ChainInputs extends BaseLangChainParams {\n  memory?: BaseMemory;\n\n  /**\n   * @deprecated Use `callbacks` instead\n   */\n  callbackManager?: CallbackManager;\n}\n\n/**\n * Base interface that all chains must implement.\n */\nexport abstract class BaseChain<\n  RunInput extends ChainValues = ChainValues,\n  RunOutput extends ChainValues = ChainValues,\n>\n  extends BaseLangChain<RunInput, RunOutput>\n  implements ChainInputs\n{\n  declare memory?: BaseMemory;\n\n  get lc_namespace(): string[] {\n    return [\"langchain\", \"chains\", this._chainType()];\n  }\n\n  constructor(\n    fields?: BaseMemory | ChainInputs,\n    /** @deprecated */\n    verbose?: boolean,\n    /** @deprecated */\n    callbacks?: Callbacks\n  ) {\n    if (\n      arguments.length === 1 &&\n      typeof fields === \"object\" &&\n      !(\"saveContext\" in fields)\n    ) {\n      // fields is not a BaseMemory\n      const { memory, callbackManager, ...rest } = fields;\n      super({ ...rest, callbacks: callbackManager ?? rest.callbacks });\n      this.memory = memory;\n    } else {\n      // fields is a BaseMemory\n      super({ verbose, callbacks });\n      this.memory = fields as BaseMemory;\n    }\n  }\n\n  /** @ignore */\n  _selectMemoryInputs(values: ChainValues): ChainValues {\n    const valuesForMemory = { ...values };\n    if (\"signal\" in valuesForMemory) {\n      delete valuesForMemory.signal;\n    }\n    if (\"timeout\" in valuesForMemory) {\n      delete valuesForMemory.timeout;\n    }\n    return valuesForMemory;\n  }\n\n  /**\n   * Invoke the chain with the provided input and returns the output.\n   * @param input Input values for the chain run.\n   * @param config Optional configuration for the Runnable.\n   * @returns Promise that resolves with the output of the chain run.\n   */\n  async invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput> {\n    const config = ensureConfig(options);\n    const fullValues = await this._formatValues(input);\n    const callbackManager_ = await CallbackManager.configure(\n      config?.callbacks,\n      this.callbacks,\n      config?.tags,\n      this.tags,\n      config?.metadata,\n      this.metadata,\n      { verbose: this.verbose }\n    );\n    const runManager = await callbackManager_?.handleChainStart(\n      this.toJSON(),\n      fullValues,\n      undefined,\n      undefined,\n      undefined,\n      undefined,\n      config?.runName\n    );\n    let outputValues: RunOutput;\n    try {\n      if (fullValues.signal) {\n        let listener: (() => void) | undefined;\n        outputValues = (await Promise.race([\n          this._call(fullValues as RunInput, runManager, config),\n          new Promise<never>((_, reject) => {\n            listener = () => {\n              reject(new Error(\"AbortError\"));\n            };\n            fullValues.signal?.addEventListener(\"abort\", listener, {\n              once: true,\n            });\n          }),\n        ]).finally(() => {\n          if (fullValues.signal && listener) {\n            fullValues.signal.removeEventListener(\"abort\", listener);\n          }\n        })) as RunOutput;\n      } else {\n        outputValues = await this._call(\n          fullValues as RunInput,\n          runManager,\n          config\n        );\n      }\n    } catch (e) {\n      await runManager?.handleChainError(e);\n      throw e;\n    }\n    if (!(this.memory == null)) {\n      await this.memory.saveContext(\n        this._selectMemoryInputs(input),\n        outputValues\n      );\n    }\n    await runManager?.handleChainEnd(outputValues);\n    // add the runManager's currentRunId to the outputValues\n    Object.defineProperty(outputValues, RUN_KEY, {\n      value: runManager ? { runId: runManager?.runId } : undefined,\n      configurable: true,\n    });\n    return outputValues;\n  }\n\n  private _validateOutputs(outputs: Record<string, unknown>): void {\n    const missingKeys = this.outputKeys.filter((k) => !(k in outputs));\n    if (missingKeys.length) {\n      throw new Error(\n        `Missing output keys: ${missingKeys.join(\n          \", \"\n        )} from chain ${this._chainType()}`\n      );\n    }\n  }\n\n  async prepOutputs(\n    inputs: Record<string, unknown>,\n    outputs: Record<string, unknown>,\n    returnOnlyOutputs = false\n  ) {\n    this._validateOutputs(outputs);\n    if (this.memory) {\n      await this.memory.saveContext(inputs, outputs);\n    }\n    if (returnOnlyOutputs) {\n      return outputs;\n    }\n    return { ...inputs, ...outputs };\n  }\n\n  /**\n   * Run the core logic of this chain and return the output\n   */\n  abstract _call(\n    values: RunInput,\n    runManager?: CallbackManagerForChainRun,\n    config?: RunnableConfig\n  ): Promise<RunOutput>;\n\n  /**\n   * Return the string type key uniquely identifying this class of chain.\n   */\n  abstract _chainType(): string;\n\n  /**\n   * Return a json-like object representing this chain.\n   */\n  serialize(): SerializedBaseChain {\n    throw new Error(\"Method not implemented.\");\n  }\n\n  abstract get inputKeys(): string[];\n\n  abstract get outputKeys(): string[];\n\n  /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */\n  async run(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any,\n    config?: Callbacks | RunnableConfig\n  ): Promise<string> {\n    const inputKeys = this.inputKeys.filter(\n      (k) => !this.memory?.memoryKeys.includes(k)\n    );\n    const isKeylessInput = inputKeys.length <= 1;\n    if (!isKeylessInput) {\n      throw new Error(\n        `Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `\n      );\n    }\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const values = inputKeys.length ? { [inputKeys[0]]: input } : ({} as any);\n    const returnValues = await this.call(values, config);\n    const keys = Object.keys(returnValues);\n\n    if (keys.length === 1) {\n      return returnValues[keys[0]];\n    }\n    throw new Error(\n      \"return values have multiple keys, `run` only supported when one key currently\"\n    );\n  }\n\n  protected async _formatValues(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number }\n  ) {\n    const fullValues = { ...values } as typeof values;\n    if (fullValues.timeout && !fullValues.signal) {\n      fullValues.signal = AbortSignal.timeout(fullValues.timeout);\n      delete fullValues.timeout;\n    }\n    if (!(this.memory == null)) {\n      const newValues = await this.memory.loadMemoryVariables(\n        this._selectMemoryInputs(values)\n      );\n      for (const [key, value] of Object.entries(newValues)) {\n        fullValues[key] = value;\n      }\n    }\n    return fullValues;\n  }\n\n  /**\n   * @deprecated Use .invoke() instead. Will be removed in 0.2.0.\n   *\n   * Run the core logic of this chain and add to output if desired.\n   *\n   * Wraps _call and handles memory.\n   */\n  async call(\n    values: ChainValues & { signal?: AbortSignal; timeout?: number },\n    config?: Callbacks | RunnableConfig,\n    /** @deprecated */\n    tags?: string[]\n  ): Promise<RunOutput> {\n    const parsedConfig = { tags, ...parseCallbackConfigArg(config) };\n    return this.invoke(values as RunInput, parsedConfig);\n  }\n\n  /**\n   * @deprecated Use .batch() instead. Will be removed in 0.2.0.\n   *\n   * Call the chain on all inputs in the list\n   */\n  async apply(\n    inputs: RunInput[],\n    config?: (Callbacks | RunnableConfig)[]\n  ): Promise<RunOutput[]> {\n    return Promise.all(\n      inputs.map(async (i, idx) => this.call(i, config?.[idx]))\n    );\n  }\n\n  /**\n   * Load a chain from a json-like object describing it.\n   */\n  static async deserialize(\n    data: SerializedBaseChain,\n    values: LoadValues = {}\n  ): Promise<BaseChain> {\n    switch (data._type) {\n      case \"llm_chain\": {\n        const { LLMChain } = await import(\"./llm_chain.js\");\n        return LLMChain.deserialize(data);\n      }\n      case \"sequential_chain\": {\n        const { SequentialChain } = await import(\"./sequential_chain.js\");\n        return SequentialChain.deserialize(data);\n      }\n      case \"simple_sequential_chain\": {\n        const { SimpleSequentialChain } = await import(\"./sequential_chain.js\");\n        return SimpleSequentialChain.deserialize(data);\n      }\n      case \"stuff_documents_chain\": {\n        const { StuffDocumentsChain } = await import(\"./combine_docs_chain.js\");\n        return StuffDocumentsChain.deserialize(data);\n      }\n      case \"map_reduce_documents_chain\": {\n        const { MapReduceDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return MapReduceDocumentsChain.deserialize(data);\n      }\n      case \"refine_documents_chain\": {\n        const { RefineDocumentsChain } =\n          await import(\"./combine_docs_chain.js\");\n        return RefineDocumentsChain.deserialize(data);\n      }\n      case \"vector_db_qa\": {\n        const { VectorDBQAChain } = await import(\"./vector_db_qa.js\");\n        return VectorDBQAChain.deserialize(data, values);\n      }\n      case \"api_chain\": {\n        const { APIChain } = await import(\"./api/api_chain.js\");\n        return APIChain.deserialize(data);\n      }\n      default:\n        throw new Error(\n          `Invalid prompt type in config: ${\n            (data as SerializedBaseChain)._type\n          }`\n        );\n    }\n  }\n}\n"],"mappings":";;;;;;;;AA+BA,IAAsB,YAAtB,cAIU,cAEV;CAGE,IAAI,eAAyB;AAC3B,SAAO;GAAC;GAAa;GAAU,KAAK,YAAY;GAAC;;CAGnD,YACE,QAEA,SAEA,WACA;AACA,MACE,UAAU,WAAW,KACrB,OAAO,WAAW,YAClB,EAAE,iBAAiB,SACnB;GAEA,MAAM,EAAE,QAAQ,iBAAiB,GAAG,SAAS;AAC7C,SAAM;IAAE,GAAG;IAAM,WAAW,mBAAmB,KAAK;IAAW,CAAC;AAChE,QAAK,SAAS;SACT;AAEL,SAAM;IAAE;IAAS;IAAW,CAAC;AAC7B,QAAK,SAAS;;;;CAKlB,oBAAoB,QAAkC;EACpD,MAAM,kBAAkB,EAAE,GAAG,QAAQ;AACrC,MAAI,YAAY,gBACd,QAAO,gBAAgB;AAEzB,MAAI,aAAa,gBACf,QAAO,gBAAgB;AAEzB,SAAO;;;;;;;;CAST,MAAM,OAAO,OAAiB,SAA8C;EAC1E,MAAM,SAAS,aAAa,QAAQ;EACpC,MAAM,aAAa,MAAM,KAAK,cAAc,MAAM;EAUlD,MAAM,aAAa,OATM,MAAM,gBAAgB,UAC7C,QAAQ,WACR,KAAK,WACL,QAAQ,MACR,KAAK,MACL,QAAQ,UACR,KAAK,UACL,EAAE,SAAS,KAAK,SAAS,CAC1B,GAC0C,iBACzC,KAAK,QAAQ,EACb,YACA,KAAA,GACA,KAAA,GACA,KAAA,GACA,KAAA,GACA,QAAQ,QACT;EACD,IAAI;AACJ,MAAI;AACF,OAAI,WAAW,QAAQ;IACrB,IAAI;AACJ,mBAAgB,MAAM,QAAQ,KAAK,CACjC,KAAK,MAAM,YAAwB,YAAY,OAAO,EACtD,IAAI,SAAgB,GAAG,WAAW;AAChC,sBAAiB;AACf,6BAAO,IAAI,MAAM,aAAa,CAAC;;AAEjC,gBAAW,QAAQ,iBAAiB,SAAS,UAAU,EACrD,MAAM,MACP,CAAC;MACF,CACH,CAAC,CAAC,cAAc;AACf,SAAI,WAAW,UAAU,SACvB,YAAW,OAAO,oBAAoB,SAAS,SAAS;MAE1D;SAEF,gBAAe,MAAM,KAAK,MACxB,YACA,YACA,OACD;WAEI,GAAG;AACV,SAAM,YAAY,iBAAiB,EAAE;AACrC,SAAM;;AAER,MAAI,EAAE,KAAK,UAAU,MACnB,OAAM,KAAK,OAAO,YAChB,KAAK,oBAAoB,MAAM,EAC/B,aACD;AAEH,QAAM,YAAY,eAAe,aAAa;AAE9C,SAAO,eAAe,cAAc,SAAS;GAC3C,OAAO,aAAa,EAAE,OAAO,YAAY,OAAO,GAAG,KAAA;GACnD,cAAc;GACf,CAAC;AACF,SAAO;;CAGT,iBAAyB,SAAwC;EAC/D,MAAM,cAAc,KAAK,WAAW,QAAQ,MAAM,EAAE,KAAK,SAAS;AAClE,MAAI,YAAY,OACd,OAAM,IAAI,MACR,wBAAwB,YAAY,KAClC,KACD,CAAC,cAAc,KAAK,YAAY,GAClC;;CAIL,MAAM,YACJ,QACA,SACA,oBAAoB,OACpB;AACA,OAAK,iBAAiB,QAAQ;AAC9B,MAAI,KAAK,OACP,OAAM,KAAK,OAAO,YAAY,QAAQ,QAAQ;AAEhD,MAAI,kBACF,QAAO;AAET,SAAO;GAAE,GAAG;GAAQ,GAAG;GAAS;;;;;CAoBlC,YAAiC;AAC/B,QAAM,IAAI,MAAM,0BAA0B;;;CAQ5C,MAAM,IAEJ,OACA,QACiB;EACjB,MAAM,YAAY,KAAK,UAAU,QAC9B,MAAM,CAAC,KAAK,QAAQ,WAAW,SAAS,EAAE,CAC5C;AAED,MAAI,EADmB,UAAU,UAAU,GAEzC,OAAM,IAAI,MACR,SAAS,KAAK,YAAY,CAAC,6CAC5B;EAGH,MAAM,SAAS,UAAU,SAAS,GAAG,UAAU,KAAK,OAAO,GAAI,EAAE;EACjE,MAAM,eAAe,MAAM,KAAK,KAAK,QAAQ,OAAO;EACpD,MAAM,OAAO,OAAO,KAAK,aAAa;AAEtC,MAAI,KAAK,WAAW,EAClB,QAAO,aAAa,KAAK;AAE3B,QAAM,IAAI,MACR,gFACD;;CAGH,MAAgB,cACd,QACA;EACA,MAAM,aAAa,EAAE,GAAG,QAAQ;AAChC,MAAI,WAAW,WAAW,CAAC,WAAW,QAAQ;AAC5C,cAAW,SAAS,YAAY,QAAQ,WAAW,QAAQ;AAC3D,UAAO,WAAW;;AAEpB,MAAI,EAAE,KAAK,UAAU,OAAO;GAC1B,MAAM,YAAY,MAAM,KAAK,OAAO,oBAClC,KAAK,oBAAoB,OAAO,CACjC;AACD,QAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,UAAU,CAClD,YAAW,OAAO;;AAGtB,SAAO;;;;;;;;;CAUT,MAAM,KACJ,QACA,QAEA,MACoB;EACpB,MAAM,eAAe;GAAE;GAAM,GAAG,uBAAuB,OAAO;GAAE;AAChE,SAAO,KAAK,OAAO,QAAoB,aAAa;;;;;;;CAQtD,MAAM,MACJ,QACA,QACsB;AACtB,SAAO,QAAQ,IACb,OAAO,IAAI,OAAO,GAAG,QAAQ,KAAK,KAAK,GAAG,SAAS,KAAK,CAAC,CAC1D;;;;;CAMH,aAAa,YACX,MACA,SAAqB,EAAE,EACH;AACpB,UAAQ,KAAK,OAAb;GACE,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,MAAM,OAAO;AAClC,WAAO,SAAS,YAAY,KAAK;;GAEnC,KAAK,oBAAoB;IACvB,MAAM,EAAE,oBAAoB,MAAM,OAAO;AACzC,WAAO,gBAAgB,YAAY,KAAK;;GAE1C,KAAK,2BAA2B;IAC9B,MAAM,EAAE,0BAA0B,MAAM,OAAO;AAC/C,WAAO,sBAAsB,YAAY,KAAK;;GAEhD,KAAK,yBAAyB;IAC5B,MAAM,EAAE,wBAAwB,MAAM,OAAO;AAC7C,WAAO,oBAAoB,YAAY,KAAK;;GAE9C,KAAK,8BAA8B;IACjC,MAAM,EAAE,4BACN,MAAM,OAAO;AACf,WAAO,wBAAwB,YAAY,KAAK;;GAElD,KAAK,0BAA0B;IAC7B,MAAM,EAAE,yBACN,MAAM,OAAO;AACf,WAAO,qBAAqB,YAAY,KAAK;;GAE/C,KAAK,gBAAgB;IACnB,MAAM,EAAE,oBAAoB,MAAM,OAAO;AACzC,WAAO,gBAAgB,YAAY,MAAM,OAAO;;GAElD,KAAK,aAAa;IAChB,MAAM,EAAE,aAAa,MAAM,OAAO;AAClC,WAAO,SAAS,YAAY,KAAK;;GAEnC,QACE,OAAM,IAAI,MACR,kCACG,KAA6B,QAEjC"}

package/dist/chains/chat_vector_db_chain.cjs

@@ -1,9 +1,8 @@
-const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
-const require_base = require('./base.cjs');
-const require_llm_chain = require('./llm_chain.cjs');
-const require_load = require('./question_answering/load.cjs');
+require("../_virtual/_rolldown/runtime.cjs");
+const require_base = require("./base.cjs");
+const require_llm_chain = require("./llm_chain.cjs");
+const require_load = require("./question_answering/load.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
-
 //#region src/chains/chat_vector_db_chain.ts
 const question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
 
@@ -122,7 +121,7 @@ var ChatVectorDBQAChain = class ChatVectorDBQAChain extends require_base.BaseCha
 		});
 	}
 };
-
 //#endregion
 exports.ChatVectorDBQAChain = ChatVectorDBQAChain;
+
 //# sourceMappingURL=chat_vector_db_chain.cjs.map

package/dist/chains/chat_vector_db_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"chat_vector_db_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate","loadQAStuffChain"],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACHA,uBAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,IAAI,OAAO,KAAK,KAAK;AAC1B,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,aAAa;EAEjE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,WAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,QACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;GAClB;AAEH,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YACX,MACA,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,+DACD;EAEH,MAAM,EAAE,gBAAgB;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAMA,uBAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAMC,2BAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;GACD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;GACT;;;;;;;;;;CAWH,OAAO,QACL,KACA,aACA,UAQI,EAAE,EACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4BC,uCAAe,aAC/C,6BAA6B,4BAC9B;EAGD,MAAM,UAAUC,8BAAiB,KAAK;GAAE,QAFtBD,uCAAe,aAAa,cAAc,YAAY;GAEb;GAAS,CAAC;EACrE,MAAM,yBAAyB,IAAID,2BAAS;GAC1C,QAAQ;GACR;GACA;GACD,CAAC;AAOF,SANiB,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;GACJ,CAAC"}
+{"version":3,"file":"chat_vector_db_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate","loadQAStuffChain"],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACHA,aAAAA,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,IAAI,OAAO,KAAK,KAAK;AAC1B,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,aAAa;EAEjE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,WAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,KAAA,GACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;GAClB;AAEH,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YACX,MACA,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,+DACD;EAEH,MAAM,EAAE,gBAAgB;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAMA,aAAAA,UAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAMC,kBAAAA,SAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;GACD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;GACT;;;;;;;;;;CAWH,OAAO,QACL,KACA,aACA,UAQI,EAAE,EACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4BC,wBAAAA,eAAe,aAC/C,6BAA6B,4BAC9B;EAGD,MAAM,UAAUC,aAAAA,iBAAiB,KAAK;GAAE,QAFtBD,wBAAAA,eAAe,aAAa,cAAc,YAAY;GAEb;GAAS,CAAC;EACrE,MAAM,yBAAyB,IAAID,kBAAAA,SAAS;GAC1C,QAAQ;GACR;GACA;GACD,CAAC;AAOF,SANiB,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;GACJ,CAAC"}

package/dist/chains/chat_vector_db_chain.js

@@ -2,7 +2,6 @@ import { BaseChain } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
 import { loadQAStuffChain } from "./question_answering/load.js";
 import { PromptTemplate } from "@langchain/core/prompts";
-
 //#region src/chains/chat_vector_db_chain.ts
 const question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
 
@@ -121,7 +120,7 @@ var ChatVectorDBQAChain = class ChatVectorDBQAChain extends BaseChain {
 		});
 	}
 };
-
 //#endregion
 export { ChatVectorDBQAChain };
+
 //# sourceMappingURL=chat_vector_db_chain.js.map

package/dist/chains/chat_vector_db_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat_vector_db_chain.js","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,IAAI,OAAO,KAAK,KAAK;AAC1B,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,aAAa;EAEjE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,WAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,QACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;GAClB;AAEH,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YACX,MACA,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,+DACD;EAEH,MAAM,EAAE,gBAAgB;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAM,UAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAM,SAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;GACD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;GACT;;;;;;;;;;CAWH,OAAO,QACL,KACA,aACA,UAQI,EAAE,EACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4B,eAAe,aAC/C,6BAA6B,4BAC9B;EAGD,MAAM,UAAU,iBAAiB,KAAK;GAAE,QAFtB,eAAe,aAAa,cAAc,YAAY;GAEb;GAAS,CAAC;EACrE,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QAAQ;GACR;GACA;GACD,CAAC;AAOF,SANiB,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;GACJ,CAAC"}
+{"version":3,"file":"chat_vector_db_chain.js","names":[],"sources":["../../src/chains/chat_vector_db_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { VectorStoreInterface } from \"@langchain/core/vectorstores\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { loadQAStuffChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\nconst qa_template = `Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.\n\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\n\n/**\n * Interface for the input parameters of the ChatVectorDBQAChain class.\n */\nexport interface ChatVectorDBQAChainInput extends ChainInputs {\n  vectorstore: VectorStoreInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  outputKey?: string;\n  inputKey?: string;\n  k?: number;\n}\n\n/** @deprecated use `ConversationalRetrievalQAChain` instead. */\nexport class ChatVectorDBQAChain\n  extends BaseChain\n  implements ChatVectorDBQAChainInput\n{\n  k = 4;\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  outputKey = \"result\";\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  vectorstore: VectorStoreInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  constructor(fields: ChatVectorDBQAChainInput) {\n    super(fields);\n    this.vectorstore = fields.vectorstore;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.k = fields.k ?? this.k;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`chat history key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string = values[this.chatHistoryKey];\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      console.log(\"_call\", values, keys);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.vectorstore.similaritySearch(\n      newQuestion,\n      this.k,\n      undefined,\n      runManager?.getChild(\"vectorstore\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    const result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      return {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"chat-vector-db\" as const;\n  }\n\n  static async deserialize(\n    data: SerializedChatVectorDBQAChain,\n    values: LoadValues\n  ) {\n    if (!(\"vectorstore\" in values)) {\n      throw new Error(\n        `Need to pass in a vectorstore to deserialize VectorDBQAChain`\n      );\n    }\n    const { vectorstore } = values;\n\n    return new ChatVectorDBQAChain({\n      combineDocumentsChain: await BaseChain.deserialize(\n        data.combine_documents_chain\n      ),\n      questionGeneratorChain: await LLMChain.deserialize(\n        data.question_generator\n      ),\n      k: data.k,\n      vectorstore,\n    });\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    return {\n      _type: this._chainType(),\n      combine_documents_chain: this.combineDocumentsChain.serialize(),\n      question_generator: this.questionGeneratorChain.serialize(),\n      k: this.k,\n    };\n  }\n\n  /**\n   * Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel\n   * and other options.\n   * @param llm Instance of BaseLanguageModel used to generate a new question.\n   * @param vectorstore Instance of VectorStore used for vector operations.\n   * @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.\n   * @returns New instance of ChatVectorDBQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    vectorstore: VectorStoreInterface,\n    options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    } = {}\n  ): ChatVectorDBQAChain {\n    const { questionGeneratorTemplate, qaTemplate, verbose, ...rest } = options;\n    const question_generator_prompt = PromptTemplate.fromTemplate(\n      questionGeneratorTemplate || question_generator_template\n    );\n    const qa_prompt = PromptTemplate.fromTemplate(qaTemplate || qa_template);\n\n    const qaChain = loadQAStuffChain(llm, { prompt: qa_prompt, verbose });\n    const questionGeneratorChain = new LLMChain({\n      prompt: question_generator_prompt,\n      llm,\n      verbose,\n    });\n    const instance = new this({\n      vectorstore,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;AAaA,MAAM,8BAA8B;;;;;;AAOpC,MAAM,cAAc;;;;;;;AAqBpB,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,IAAI,OAAO,KAAK,KAAK;AAC1B,OAAK,wBACH,OAAO,yBAAyB,KAAK;;;CAIzC,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,SAAS,aAAa;EAEjE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,WAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,KAAA,GACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;GAClB;AAEH,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YACX,MACA,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,+DACD;EAEH,MAAM,EAAE,gBAAgB;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAM,UAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAM,SAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;GACD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;GACT;;;;;;;;;;CAWH,OAAO,QACL,KACA,aACA,UAQI,EAAE,EACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,SAAS,GAAG,SAAS;EACpE,MAAM,4BAA4B,eAAe,aAC/C,6BAA6B,4BAC9B;EAGD,MAAM,UAAU,iBAAiB,KAAK;GAAE,QAFtB,eAAe,aAAa,cAAc,YAAY;GAEb;GAAS,CAAC;EACrE,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QAAQ;GACR;GACA;GACD,CAAC;AAOF,SANiB,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;GACJ,CAAC"}

package/dist/chains/combine_docs_chain.cjs

@@ -1,8 +1,7 @@
-const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
-const require_base = require('./base.cjs');
-const require_llm_chain = require('./llm_chain.cjs');
+require("../_virtual/_rolldown/runtime.cjs");
+const require_base = require("./base.cjs");
+const require_llm_chain = require("./llm_chain.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
-
 //#region src/chains/combine_docs_chain.ts
 /**
 * Chain that combines documents by stuffing into context.
@@ -258,9 +257,9 @@ var RefineDocumentsChain = class RefineDocumentsChain extends require_base.BaseC
 		};
 	}
 };
-
 //#endregion
 exports.MapReduceDocumentsChain = MapReduceDocumentsChain;
 exports.RefineDocumentsChain = RefineDocumentsChain;
 exports.StuffDocumentsChain = StuffDocumentsChain;
+
 //# sourceMappingURL=combine_docs_chain.cjs.map

package/dist/chains/combine_docs_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate"],"sources":["../../src/chains/combine_docs_chain.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport type {\n  SerializedStuffDocumentsChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n  /** LLM Wrapper to use after formatting documents */\n  llmChain: LLMChain;\n  inputKey?: string;\n  /** Variable name in the LLM chain to put the documents in */\n  documentVariableName?: string;\n}\n\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class StuffDocumentsChain\n  extends BaseChain\n  implements StuffDocumentsChainInput\n{\n  static lc_name() {\n    return \"StuffDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  get inputKeys() {\n    return [this.inputKey, ...this.llmChain.inputKeys].filter(\n      (key) => key !== this.documentVariableName\n    );\n  }\n\n  get outputKeys() {\n    return this.llmChain.outputKeys;\n  }\n\n  constructor(fields: StuffDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n  }\n\n  /** @ignore */\n  _prepInputs(values: ChainValues): ChainValues {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n    const texts = (docs as Document[]).map(({ pageContent }) => pageContent);\n    const text = texts.join(\"\\n\\n\");\n    return {\n      ...rest,\n      [this.documentVariableName]: text,\n    };\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const result = await this.llmChain.call(\n      this._prepInputs(values),\n      runManager?.getChild(\"combine_documents\")\n    );\n    return result;\n  }\n\n  _chainType() {\n    return \"stuff_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedStuffDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    return new StuffDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n    });\n  }\n\n  serialize(): SerializedStuffDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n  /** The maximum number of tokens before requiring to do the reduction */\n  maxTokens?: number;\n  /** The maximum number of iterations to run through the map */\n  maxIterations?: number;\n  /** Ensures that the map step is taken regardless of max tokens */\n  ensureMapStep?: boolean;\n  /** Chain to use to combine results of applying llm_chain to documents. */\n  combineDocumentChain: StuffDocumentsChain;\n  /** Return the results of the map steps in the output. */\n  returnIntermediateSteps?: boolean;\n}\n\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class MapReduceDocumentsChain\n  extends BaseChain\n  implements MapReduceDocumentsChainInput\n{\n  static lc_name() {\n    return \"MapReduceDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  returnIntermediateSteps = false;\n\n  get inputKeys() {\n    return [this.inputKey, ...this.combineDocumentChain.inputKeys];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentChain.outputKeys;\n  }\n\n  maxTokens = 3000;\n\n  maxIterations = 10;\n\n  ensureMapStep = false;\n\n  combineDocumentChain: StuffDocumentsChain;\n\n  constructor(fields: MapReduceDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.combineDocumentChain = fields.combineDocumentChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.ensureMapStep = fields.ensureMapStep ?? this.ensureMapStep;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.maxTokens = fields.maxTokens ?? this.maxTokens;\n    this.maxIterations = fields.maxIterations ?? this.maxIterations;\n    this.returnIntermediateSteps = fields.returnIntermediateSteps ?? false;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    let currentDocs = docs as Document[];\n    let intermediateSteps: string[] = [];\n\n    // For each iteration, we'll use the `llmChain` to get a new result\n    for (let i = 0; i < this.maxIterations; i += 1) {\n      const inputs = currentDocs.map((d) => ({\n        [this.documentVariableName]: d.pageContent,\n        ...rest,\n      }));\n\n      const canSkipMapStep = i !== 0 || !this.ensureMapStep;\n      if (canSkipMapStep) {\n        // Calculate the total tokens required in the input\n        const formatted =\n          await this.combineDocumentChain.llmChain.prompt.format(\n            this.combineDocumentChain._prepInputs({\n              [this.combineDocumentChain.inputKey]: currentDocs,\n              ...rest,\n            })\n          );\n        const length =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACHA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,UAAU,CAAC,QAChD,QAAQ,QAAQ,KAAK,qBACvB;;CAGH,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;;CAGvB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;;;CAI1C,YAAY,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,YAAY,CACrD,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;GAC9B;;;CAIH,MAAM,MACJ,QACA,YACsB;AAKtB,SAJe,MAAM,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;;CAIH,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAMC,2BAAS,YAAY,KAAK,UAAU,EACrD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACrC;;;;;;;;AA0BL,IAAa,0BAAb,MAAa,gCACHD,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,UAAU;;CAGhE,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;;CAGnC,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;AAChD,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBAAuB,OAAO;AACnC,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,0BAA0B,OAAO,2BAA2B;;;CAInE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,EAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;IACJ,EAAE;AAGH,OADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;KACJ,CAAC,CACH;AAOH,QALE,MAAM,KAAK,qBAAqB,SAAS,cAAc,UAAU,GAEjC,KAAK,UAIrC;;GAIJ,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,QAAQ,GAAG,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,IAAI,CACpC,GACD,OACL;GACD,MAAM,EAAE,cAAc,KAAK;AAG3B,OAAI,KAAK,wBACP,qBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,WAAW,CACjC;AAGH,iBAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,EAAE;IACb,EAAE;;EAKL,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;GACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;GAAmB;AAEzC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM,iCAAiC;AAGnD,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAMC,2BAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;GACF,CAAC;;CAGJ,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;GAC9D;;;;;;;;AAoBL,IAAa,uBAAb,MAAa,6BACHD,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAIE,uCAAe;GACxB,gBAAgB,CAAC,eAAe;GAChC,UAAU;GACX,CAAC;;CAGJ,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;GACxB,CAAC,CACH,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;;CAGH,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAmC;AAC7C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,OAAK,sBACH,OAAO,uBAAuB,KAAK;;;CAIvC,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;AAQF,SADe;IAJZ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC;GAE4B,GAAG;GAAM;;;CAK3C,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;EACF,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC,EACH;AAED,SADe;IAAG,KAAK,sBAAsB;GAAK,GAAG;GAAY;;;CAKnE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,eAAe,EACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,IAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAJI,MAAM,KAAK,uBAC9B,YAAY,IACZ,IACD;IACiC,GAAG;IAAM;AAC3C,SAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,QAAQ,EACb,YAAY,SAAS,SAAS,CAC/B;AACD,eAAY,KAAK,IAAI;;AAGvB,SAAO,GAAG,KAAK,YAAY,KAAK;;CAGlC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM,oBAAoB;EAGtC,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAMD,2BAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAMA,2BAAS,YAAY,8BAA8B;GAC1E,CAAC;;CAGJ,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;GAClD"}
+{"version":3,"file":"combine_docs_chain.cjs","names":["BaseChain","LLMChain","PromptTemplate"],"sources":["../../src/chains/combine_docs_chain.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport type {\n  SerializedStuffDocumentsChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n  /** LLM Wrapper to use after formatting documents */\n  llmChain: LLMChain;\n  inputKey?: string;\n  /** Variable name in the LLM chain to put the documents in */\n  documentVariableName?: string;\n}\n\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class StuffDocumentsChain\n  extends BaseChain\n  implements StuffDocumentsChainInput\n{\n  static lc_name() {\n    return \"StuffDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  get inputKeys() {\n    return [this.inputKey, ...this.llmChain.inputKeys].filter(\n      (key) => key !== this.documentVariableName\n    );\n  }\n\n  get outputKeys() {\n    return this.llmChain.outputKeys;\n  }\n\n  constructor(fields: StuffDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n  }\n\n  /** @ignore */\n  _prepInputs(values: ChainValues): ChainValues {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n    const texts = (docs as Document[]).map(({ pageContent }) => pageContent);\n    const text = texts.join(\"\\n\\n\");\n    return {\n      ...rest,\n      [this.documentVariableName]: text,\n    };\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const result = await this.llmChain.call(\n      this._prepInputs(values),\n      runManager?.getChild(\"combine_documents\")\n    );\n    return result;\n  }\n\n  _chainType() {\n    return \"stuff_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedStuffDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    return new StuffDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n    });\n  }\n\n  serialize(): SerializedStuffDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n  /** The maximum number of tokens before requiring to do the reduction */\n  maxTokens?: number;\n  /** The maximum number of iterations to run through the map */\n  maxIterations?: number;\n  /** Ensures that the map step is taken regardless of max tokens */\n  ensureMapStep?: boolean;\n  /** Chain to use to combine results of applying llm_chain to documents. */\n  combineDocumentChain: StuffDocumentsChain;\n  /** Return the results of the map steps in the output. */\n  returnIntermediateSteps?: boolean;\n}\n\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class MapReduceDocumentsChain\n  extends BaseChain\n  implements MapReduceDocumentsChainInput\n{\n  static lc_name() {\n    return \"MapReduceDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  returnIntermediateSteps = false;\n\n  get inputKeys() {\n    return [this.inputKey, ...this.combineDocumentChain.inputKeys];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentChain.outputKeys;\n  }\n\n  maxTokens = 3000;\n\n  maxIterations = 10;\n\n  ensureMapStep = false;\n\n  combineDocumentChain: StuffDocumentsChain;\n\n  constructor(fields: MapReduceDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.combineDocumentChain = fields.combineDocumentChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.ensureMapStep = fields.ensureMapStep ?? this.ensureMapStep;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.maxTokens = fields.maxTokens ?? this.maxTokens;\n    this.maxIterations = fields.maxIterations ?? this.maxIterations;\n    this.returnIntermediateSteps = fields.returnIntermediateSteps ?? false;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    let currentDocs = docs as Document[];\n    let intermediateSteps: string[] = [];\n\n    // For each iteration, we'll use the `llmChain` to get a new result\n    for (let i = 0; i < this.maxIterations; i += 1) {\n      const inputs = currentDocs.map((d) => ({\n        [this.documentVariableName]: d.pageContent,\n        ...rest,\n      }));\n\n      const canSkipMapStep = i !== 0 || !this.ensureMapStep;\n      if (canSkipMapStep) {\n        // Calculate the total tokens required in the input\n        const formatted =\n          await this.combineDocumentChain.llmChain.prompt.format(\n            this.combineDocumentChain._prepInputs({\n              [this.combineDocumentChain.inputKey]: currentDocs,\n              ...rest,\n            })\n          );\n        const length =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACHA,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,UAAU,CAAC,QAChD,QAAQ,QAAQ,KAAK,qBACvB;;CAGH,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;;CAGvB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;;;CAI1C,YAAY,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,YAAY,CACrD,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;GAC9B;;;CAIH,MAAM,MACJ,QACA,YACsB;AAKtB,SAJe,MAAM,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;;CAIH,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAMC,kBAAAA,SAAS,YAAY,KAAK,UAAU,EACrD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACrC;;;;;;;;AA0BL,IAAa,0BAAb,MAAa,gCACHD,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,UAAU;;CAGhE,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;;CAGnC,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;AAChD,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBAAuB,OAAO;AACnC,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,0BAA0B,OAAO,2BAA2B;;;CAInE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,EAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;IACJ,EAAE;AAGH,OADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;KACJ,CAAC,CACH;AAOH,QALE,MAAM,KAAK,qBAAqB,SAAS,cAAc,UAAU,GAEjC,KAAK,UAIrC;;GAIJ,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,QAAQ,GAAG,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,IAAI,CACpC,GACD,KAAA,EACL;GACD,MAAM,EAAE,cAAc,KAAK;AAG3B,OAAI,KAAK,wBACP,qBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,WAAW,CACjC;AAGH,iBAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,EAAE;IACb,EAAE;;EAKL,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;GACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;GAAmB;AAEzC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM,iCAAiC;AAGnD,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAMC,kBAAAA,SAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;GACF,CAAC;;CAGJ,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;GAC9D;;;;;;;;AAoBL,IAAa,uBAAb,MAAa,6BACHD,aAAAA,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAIE,wBAAAA,eAAe;GACxB,gBAAgB,CAAC,eAAe;GAChC,UAAU;GACX,CAAC;;CAGJ,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;GACxB,CAAC,CACH,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;;CAGH,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAmC;AAC7C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,OAAK,sBACH,OAAO,uBAAuB,KAAK;;;CAIvC,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;AAQF,SADe;IAJZ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC;GAE4B,GAAG;GAAM;;;CAK3C,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;EACF,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC,EACH;AAED,SADe;IAAG,KAAK,sBAAsB;GAAK,GAAG;GAAY;;;CAKnE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,eAAe,EACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,IAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAJI,MAAM,KAAK,uBAC9B,YAAY,IACZ,IACD;IACiC,GAAG;IAAM;AAC3C,SAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,QAAQ,EACb,YAAY,SAAS,SAAS,CAC/B;AACD,eAAY,KAAK,IAAI;;AAGvB,SAAO,GAAG,KAAK,YAAY,KAAK;;CAGlC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM,oBAAoB;EAGtC,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAMD,kBAAAA,SAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAMA,kBAAAA,SAAS,YAAY,8BAA8B;GAC1E,CAAC;;CAGJ,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;GAClD"}

package/dist/chains/combine_docs_chain.js

@@ -1,7 +1,6 @@
 import { BaseChain } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
 import { PromptTemplate } from "@langchain/core/prompts";
-
 //#region src/chains/combine_docs_chain.ts
 /**
 * Chain that combines documents by stuffing into context.
@@ -257,7 +256,7 @@ var RefineDocumentsChain = class RefineDocumentsChain extends BaseChain {
 		};
 	}
 };
-
 //#endregion
 export { MapReduceDocumentsChain, RefineDocumentsChain, StuffDocumentsChain };
+
 //# sourceMappingURL=combine_docs_chain.js.map

package/dist/chains/combine_docs_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"combine_docs_chain.js","names":[],"sources":["../../src/chains/combine_docs_chain.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport type {\n  SerializedStuffDocumentsChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n  /** LLM Wrapper to use after formatting documents */\n  llmChain: LLMChain;\n  inputKey?: string;\n  /** Variable name in the LLM chain to put the documents in */\n  documentVariableName?: string;\n}\n\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class StuffDocumentsChain\n  extends BaseChain\n  implements StuffDocumentsChainInput\n{\n  static lc_name() {\n    return \"StuffDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  get inputKeys() {\n    return [this.inputKey, ...this.llmChain.inputKeys].filter(\n      (key) => key !== this.documentVariableName\n    );\n  }\n\n  get outputKeys() {\n    return this.llmChain.outputKeys;\n  }\n\n  constructor(fields: StuffDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n  }\n\n  /** @ignore */\n  _prepInputs(values: ChainValues): ChainValues {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n    const texts = (docs as Document[]).map(({ pageContent }) => pageContent);\n    const text = texts.join(\"\\n\\n\");\n    return {\n      ...rest,\n      [this.documentVariableName]: text,\n    };\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const result = await this.llmChain.call(\n      this._prepInputs(values),\n      runManager?.getChild(\"combine_documents\")\n    );\n    return result;\n  }\n\n  _chainType() {\n    return \"stuff_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedStuffDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    return new StuffDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n    });\n  }\n\n  serialize(): SerializedStuffDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n  /** The maximum number of tokens before requiring to do the reduction */\n  maxTokens?: number;\n  /** The maximum number of iterations to run through the map */\n  maxIterations?: number;\n  /** Ensures that the map step is taken regardless of max tokens */\n  ensureMapStep?: boolean;\n  /** Chain to use to combine results of applying llm_chain to documents. */\n  combineDocumentChain: StuffDocumentsChain;\n  /** Return the results of the map steps in the output. */\n  returnIntermediateSteps?: boolean;\n}\n\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class MapReduceDocumentsChain\n  extends BaseChain\n  implements MapReduceDocumentsChainInput\n{\n  static lc_name() {\n    return \"MapReduceDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  returnIntermediateSteps = false;\n\n  get inputKeys() {\n    return [this.inputKey, ...this.combineDocumentChain.inputKeys];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentChain.outputKeys;\n  }\n\n  maxTokens = 3000;\n\n  maxIterations = 10;\n\n  ensureMapStep = false;\n\n  combineDocumentChain: StuffDocumentsChain;\n\n  constructor(fields: MapReduceDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.combineDocumentChain = fields.combineDocumentChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.ensureMapStep = fields.ensureMapStep ?? this.ensureMapStep;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.maxTokens = fields.maxTokens ?? this.maxTokens;\n    this.maxIterations = fields.maxIterations ?? this.maxIterations;\n    this.returnIntermediateSteps = fields.returnIntermediateSteps ?? false;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    let currentDocs = docs as Document[];\n    let intermediateSteps: string[] = [];\n\n    // For each iteration, we'll use the `llmChain` to get a new result\n    for (let i = 0; i < this.maxIterations; i += 1) {\n      const inputs = currentDocs.map((d) => ({\n        [this.documentVariableName]: d.pageContent,\n        ...rest,\n      }));\n\n      const canSkipMapStep = i !== 0 || !this.ensureMapStep;\n      if (canSkipMapStep) {\n        // Calculate the total tokens required in the input\n        const formatted =\n          await this.combineDocumentChain.llmChain.prompt.format(\n            this.combineDocumentChain._prepInputs({\n              [this.combineDocumentChain.inputKey]: currentDocs,\n              ...rest,\n            })\n          );\n        const length =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,UAAU,CAAC,QAChD,QAAQ,QAAQ,KAAK,qBACvB;;CAGH,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;;CAGvB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;;;CAI1C,YAAY,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,YAAY,CACrD,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;GAC9B;;;CAIH,MAAM,MACJ,QACA,YACsB;AAKtB,SAJe,MAAM,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;;CAIH,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAM,SAAS,YAAY,KAAK,UAAU,EACrD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACrC;;;;;;;;AA0BL,IAAa,0BAAb,MAAa,gCACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,UAAU;;CAGhE,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;;CAGnC,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;AAChD,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBAAuB,OAAO;AACnC,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,0BAA0B,OAAO,2BAA2B;;;CAInE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,EAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;IACJ,EAAE;AAGH,OADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;KACJ,CAAC,CACH;AAOH,QALE,MAAM,KAAK,qBAAqB,SAAS,cAAc,UAAU,GAEjC,KAAK,UAIrC;;GAIJ,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,QAAQ,GAAG,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,IAAI,CACpC,GACD,OACL;GACD,MAAM,EAAE,cAAc,KAAK;AAG3B,OAAI,KAAK,wBACP,qBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,WAAW,CACjC;AAGH,iBAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,EAAE;IACb,EAAE;;EAKL,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;GACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;GAAmB;AAEzC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM,iCAAiC;AAGnD,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAM,SAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;GACF,CAAC;;CAGJ,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;GAC9D;;;;;;;;AAoBL,IAAa,uBAAb,MAAa,6BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAI,eAAe;GACxB,gBAAgB,CAAC,eAAe;GAChC,UAAU;GACX,CAAC;;CAGJ,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;GACxB,CAAC,CACH,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;;CAGH,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAmC;AAC7C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,OAAK,sBACH,OAAO,uBAAuB,KAAK;;;CAIvC,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;AAQF,SADe;IAJZ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC;GAE4B,GAAG;GAAM;;;CAK3C,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;EACF,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC,EACH;AAED,SADe;IAAG,KAAK,sBAAsB;GAAK,GAAG;GAAY;;;CAKnE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,eAAe,EACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,IAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAJI,MAAM,KAAK,uBAC9B,YAAY,IACZ,IACD;IACiC,GAAG;IAAM;AAC3C,SAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,QAAQ,EACb,YAAY,SAAS,SAAS,CAC/B;AACD,eAAY,KAAK,IAAI;;AAGvB,SAAO,GAAG,KAAK,YAAY,KAAK;;CAGlC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM,oBAAoB;EAGtC,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAM,SAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAM,SAAS,YAAY,8BAA8B;GAC1E,CAAC;;CAGJ,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;GAClD"}
+{"version":3,"file":"combine_docs_chain.js","names":[],"sources":["../../src/chains/combine_docs_chain.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport type {\n  SerializedStuffDocumentsChain,\n  SerializedMapReduceDocumentsChain,\n  SerializedRefineDocumentsChain,\n} from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\n\n/**\n * Interface for the input properties of the StuffDocumentsChain class.\n */\nexport interface StuffDocumentsChainInput extends ChainInputs {\n  /** LLM Wrapper to use after formatting documents */\n  llmChain: LLMChain;\n  inputKey?: string;\n  /** Variable name in the LLM chain to put the documents in */\n  documentVariableName?: string;\n}\n\n/**\n * Chain that combines documents by stuffing into context.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class StuffDocumentsChain\n  extends BaseChain\n  implements StuffDocumentsChainInput\n{\n  static lc_name() {\n    return \"StuffDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  get inputKeys() {\n    return [this.inputKey, ...this.llmChain.inputKeys].filter(\n      (key) => key !== this.documentVariableName\n    );\n  }\n\n  get outputKeys() {\n    return this.llmChain.outputKeys;\n  }\n\n  constructor(fields: StuffDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n  }\n\n  /** @ignore */\n  _prepInputs(values: ChainValues): ChainValues {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n    const texts = (docs as Document[]).map(({ pageContent }) => pageContent);\n    const text = texts.join(\"\\n\\n\");\n    return {\n      ...rest,\n      [this.documentVariableName]: text,\n    };\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const result = await this.llmChain.call(\n      this._prepInputs(values),\n      runManager?.getChild(\"combine_documents\")\n    );\n    return result;\n  }\n\n  _chainType() {\n    return \"stuff_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedStuffDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    return new StuffDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n    });\n  }\n\n  serialize(): SerializedStuffDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the MapReduceDocumentsChain\n * class.\n */\nexport interface MapReduceDocumentsChainInput extends StuffDocumentsChainInput {\n  /** The maximum number of tokens before requiring to do the reduction */\n  maxTokens?: number;\n  /** The maximum number of iterations to run through the map */\n  maxIterations?: number;\n  /** Ensures that the map step is taken regardless of max tokens */\n  ensureMapStep?: boolean;\n  /** Chain to use to combine results of applying llm_chain to documents. */\n  combineDocumentChain: StuffDocumentsChain;\n  /** Return the results of the map steps in the output. */\n  returnIntermediateSteps?: boolean;\n}\n\n/**\n * Combine documents by mapping a chain over them, then combining results.\n * @augments BaseChain\n * @augments StuffDocumentsChainInput\n */\nexport class MapReduceDocumentsChain\n  extends BaseChain\n  implements MapReduceDocumentsChainInput\n{\n  static lc_name() {\n    return \"MapReduceDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  documentVariableName = \"context\";\n\n  returnIntermediateSteps = false;\n\n  get inputKeys() {\n    return [this.inputKey, ...this.combineDocumentChain.inputKeys];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentChain.outputKeys;\n  }\n\n  maxTokens = 3000;\n\n  maxIterations = 10;\n\n  ensureMapStep = false;\n\n  combineDocumentChain: StuffDocumentsChain;\n\n  constructor(fields: MapReduceDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.combineDocumentChain = fields.combineDocumentChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.ensureMapStep = fields.ensureMapStep ?? this.ensureMapStep;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.maxTokens = fields.maxTokens ?? this.maxTokens;\n    this.maxIterations = fields.maxIterations ?? this.maxIterations;\n    this.returnIntermediateSteps = fields.returnIntermediateSteps ?? false;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    let currentDocs = docs as Document[];\n    let intermediateSteps: string[] = [];\n\n    // For each iteration, we'll use the `llmChain` to get a new result\n    for (let i = 0; i < this.maxIterations; i += 1) {\n      const inputs = currentDocs.map((d) => ({\n        [this.documentVariableName]: d.pageContent,\n        ...rest,\n      }));\n\n      const canSkipMapStep = i !== 0 || !this.ensureMapStep;\n      if (canSkipMapStep) {\n        // Calculate the total tokens required in the input\n        const formatted =\n          await this.combineDocumentChain.llmChain.prompt.format(\n            this.combineDocumentChain._prepInputs({\n              [this.combineDocumentChain.inputKey]: currentDocs,\n              ...rest,\n            })\n          );\n        const length =\n          await this.combineDocumentChain.llmChain._getNumTokens(formatted);\n\n        const withinTokenLimit = length < this.maxTokens;\n        // If we can skip the map step, and we're within the token limit, we don't\n        // need to run the map step, so just break out of the loop.\n        if (withinTokenLimit) {\n          break;\n        }\n      }\n\n      const results = await this.llmChain.apply(\n        inputs,\n        // If we have a runManager, then we need to create a child for each input\n        // so that we can track the progress of each input.\n        runManager\n          ? Array.from({ length: inputs.length }, (_, i) =>\n              runManager.getChild(`map_${i + 1}`)\n            )\n          : undefined\n      );\n      const { outputKey } = this.llmChain;\n\n      // If the flag is set, then concat that to the intermediate steps\n      if (this.returnIntermediateSteps) {\n        intermediateSteps = intermediateSteps.concat(\n          results.map((r) => r[outputKey])\n        );\n      }\n\n      currentDocs = results.map((r) => ({\n        pageContent: r[outputKey],\n        metadata: {},\n      }));\n    }\n\n    // Now, with the final result of all the inputs from the `llmChain`, we can\n    // run the `combineDocumentChain` over them.\n    const newInputs = {\n      [this.combineDocumentChain.inputKey]: currentDocs,\n      ...rest,\n    };\n    const result = await this.combineDocumentChain.call(\n      newInputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n\n    // Return the intermediate steps results if the flag is set\n    if (this.returnIntermediateSteps) {\n      return { ...result, intermediateSteps };\n    }\n    return result;\n  }\n\n  _chainType() {\n    return \"map_reduce_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedMapReduceDocumentsChain) {\n    if (!data.llm_chain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    if (!data.combine_document_chain) {\n      throw new Error(\"Missing combine_document_chain\");\n    }\n\n    return new MapReduceDocumentsChain({\n      llmChain: await LLMChain.deserialize(data.llm_chain),\n      combineDocumentChain: await StuffDocumentsChain.deserialize(\n        data.combine_document_chain\n      ),\n    });\n  }\n\n  serialize(): SerializedMapReduceDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      combine_document_chain: this.combineDocumentChain.serialize(),\n    };\n  }\n}\n\n/**\n * Interface for the input properties of the RefineDocumentsChain class.\n */\nexport interface RefineDocumentsChainInput extends StuffDocumentsChainInput {\n  refineLLMChain: LLMChain;\n  documentPrompt?: BasePromptTemplate;\n  initialResponseName?: string;\n  documentVariableName?: string;\n  outputKey?: string;\n}\n\n/**\n * Combine documents by doing a first pass and then refining on more documents.\n * @augments BaseChain\n * @augments RefineDocumentsChainInput\n */\nexport class RefineDocumentsChain\n  extends BaseChain\n  implements RefineDocumentsChainInput\n{\n  static lc_name() {\n    return \"RefineDocumentsChain\";\n  }\n\n  llmChain: LLMChain;\n\n  inputKey = \"input_documents\";\n\n  outputKey = \"output_text\";\n\n  documentVariableName = \"context\";\n\n  initialResponseName = \"existing_answer\";\n\n  refineLLMChain: LLMChain;\n\n  get defaultDocumentPrompt(): BasePromptTemplate {\n    return new PromptTemplate({\n      inputVariables: [\"page_content\"],\n      template: \"{page_content}\",\n    });\n  }\n\n  documentPrompt = this.defaultDocumentPrompt;\n\n  get inputKeys() {\n    return [\n      ...new Set([\n        this.inputKey,\n        ...this.llmChain.inputKeys,\n        ...this.refineLLMChain.inputKeys,\n      ]),\n    ].filter(\n      (key) =>\n        key !== this.documentVariableName && key !== this.initialResponseName\n    );\n  }\n\n  get outputKeys() {\n    return [this.outputKey];\n  }\n\n  constructor(fields: RefineDocumentsChainInput) {\n    super(fields);\n    this.llmChain = fields.llmChain;\n    this.refineLLMChain = fields.refineLLMChain;\n    this.documentVariableName =\n      fields.documentVariableName ?? this.documentVariableName;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.documentPrompt = fields.documentPrompt ?? this.documentPrompt;\n    this.initialResponseName =\n      fields.initialResponseName ?? this.initialResponseName;\n  }\n\n  /** @ignore */\n  async _constructInitialInputs(doc: Document, rest: Record<string, unknown>) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { ...baseInputs, ...rest };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _constructRefineInputs(doc: Document, res: string) {\n    const baseInfo: Record<string, unknown> = {\n      page_content: doc.pageContent,\n      ...doc.metadata,\n    };\n    const documentInfo: Record<string, unknown> = {};\n    this.documentPrompt.inputVariables.forEach((value) => {\n      documentInfo[value] = baseInfo[value];\n    });\n    const baseInputs: Record<string, unknown> = {\n      [this.documentVariableName]: await this.documentPrompt.format({\n        ...documentInfo,\n      }),\n    };\n    const inputs = { [this.initialResponseName]: res, ...baseInputs };\n    return inputs;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Document key ${this.inputKey} not found.`);\n    }\n    const { [this.inputKey]: docs, ...rest } = values;\n\n    const currentDocs = docs as Document[];\n\n    const initialInputs = await this._constructInitialInputs(\n      currentDocs[0],\n      rest\n    );\n    let res = await this.llmChain.predict(\n      { ...initialInputs },\n      runManager?.getChild(\"answer\")\n    );\n\n    const refineSteps = [res];\n\n    for (let i = 1; i < currentDocs.length; i += 1) {\n      const refineInputs = await this._constructRefineInputs(\n        currentDocs[i],\n        res\n      );\n      const inputs = { ...refineInputs, ...rest };\n      res = await this.refineLLMChain.predict(\n        { ...inputs },\n        runManager?.getChild(\"refine\")\n      );\n      refineSteps.push(res);\n    }\n\n    return { [this.outputKey]: res };\n  }\n\n  _chainType() {\n    return \"refine_documents_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedRefineDocumentsChain) {\n    const SerializedLLMChain = data.llm_chain;\n\n    if (!SerializedLLMChain) {\n      throw new Error(\"Missing llm_chain\");\n    }\n\n    const SerializedRefineDocumentChain = data.refine_llm_chain;\n\n    if (!SerializedRefineDocumentChain) {\n      throw new Error(\"Missing refine_llm_chain\");\n    }\n\n    return new RefineDocumentsChain({\n      llmChain: await LLMChain.deserialize(SerializedLLMChain),\n      refineLLMChain: await LLMChain.deserialize(SerializedRefineDocumentChain),\n    });\n  }\n\n  serialize(): SerializedRefineDocumentsChain {\n    return {\n      _type: this._chainType(),\n      llm_chain: this.llmChain.serialize(),\n      refine_llm_chain: this.refineLLMChain.serialize(),\n    };\n  }\n}\n"],"mappings":";;;;;;;;;AA4BA,IAAa,sBAAb,MAAa,4BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,SAAS,UAAU,CAAC,QAChD,QAAQ,QAAQ,KAAK,qBACvB;;CAGH,IAAI,aAAa;AACf,SAAO,KAAK,SAAS;;CAGvB,YAAY,QAAkC;AAC5C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;;;CAI1C,YAAY,QAAkC;AAC5C,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,OADS,KAAoB,KAAK,EAAE,kBAAkB,YAAY,CACrD,KAAK,OAAO;AAC/B,SAAO;GACL,GAAG;IACF,KAAK,uBAAuB;GAC9B;;;CAIH,MAAM,MACJ,QACA,YACsB;AAKtB,SAJe,MAAM,KAAK,SAAS,KACjC,KAAK,YAAY,OAAO,EACxB,YAAY,SAAS,oBAAoB,CAC1C;;CAIH,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAqC;AAC5D,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,SAAO,IAAI,oBAAoB,EAC7B,UAAU,MAAM,SAAS,YAAY,KAAK,UAAU,EACrD,CAAC;;CAGJ,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACrC;;;;;;;;AA0BL,IAAa,0BAAb,MAAa,gCACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,uBAAuB;CAEvB,0BAA0B;CAE1B,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,GAAG,KAAK,qBAAqB,UAAU;;CAGhE,IAAI,aAAa;AACf,SAAO,KAAK,qBAAqB;;CAGnC,YAAY;CAEZ,gBAAgB;CAEhB,gBAAgB;CAEhB;CAEA,YAAY,QAAsC;AAChD,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,uBAAuB,OAAO;AACnC,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,gBAAgB,OAAO,iBAAiB,KAAK;AAClD,OAAK,0BAA0B,OAAO,2BAA2B;;;CAInE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,IAAI,cAAc;EAClB,IAAI,oBAA8B,EAAE;AAGpC,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,eAAe,KAAK,GAAG;GAC9C,MAAM,SAAS,YAAY,KAAK,OAAO;KACpC,KAAK,uBAAuB,EAAE;IAC/B,GAAG;IACJ,EAAE;AAGH,OADuB,MAAM,KAAK,CAAC,KAAK,eACpB;IAElB,MAAM,YACJ,MAAM,KAAK,qBAAqB,SAAS,OAAO,OAC9C,KAAK,qBAAqB,YAAY;MACnC,KAAK,qBAAqB,WAAW;KACtC,GAAG;KACJ,CAAC,CACH;AAOH,QALE,MAAM,KAAK,qBAAqB,SAAS,cAAc,UAAU,GAEjC,KAAK,UAIrC;;GAIJ,MAAM,UAAU,MAAM,KAAK,SAAS,MAClC,QAGA,aACI,MAAM,KAAK,EAAE,QAAQ,OAAO,QAAQ,GAAG,GAAG,MACxC,WAAW,SAAS,OAAO,IAAI,IAAI,CACpC,GACD,KAAA,EACL;GACD,MAAM,EAAE,cAAc,KAAK;AAG3B,OAAI,KAAK,wBACP,qBAAoB,kBAAkB,OACpC,QAAQ,KAAK,MAAM,EAAE,WAAW,CACjC;AAGH,iBAAc,QAAQ,KAAK,OAAO;IAChC,aAAa,EAAE;IACf,UAAU,EAAE;IACb,EAAE;;EAKL,MAAM,YAAY;IACf,KAAK,qBAAqB,WAAW;GACtC,GAAG;GACJ;EACD,MAAM,SAAS,MAAM,KAAK,qBAAqB,KAC7C,WACA,YAAY,SAAS,oBAAoB,CAC1C;AAGD,MAAI,KAAK,wBACP,QAAO;GAAE,GAAG;GAAQ;GAAmB;AAEzC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAyC;AAChE,MAAI,CAAC,KAAK,UACR,OAAM,IAAI,MAAM,oBAAoB;AAGtC,MAAI,CAAC,KAAK,uBACR,OAAM,IAAI,MAAM,iCAAiC;AAGnD,SAAO,IAAI,wBAAwB;GACjC,UAAU,MAAM,SAAS,YAAY,KAAK,UAAU;GACpD,sBAAsB,MAAM,oBAAoB,YAC9C,KAAK,uBACN;GACF,CAAC;;CAGJ,YAA+C;AAC7C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,wBAAwB,KAAK,qBAAqB,WAAW;GAC9D;;;;;;;;AAoBL,IAAa,uBAAb,MAAa,6BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ,uBAAuB;CAEvB,sBAAsB;CAEtB;CAEA,IAAI,wBAA4C;AAC9C,SAAO,IAAI,eAAe;GACxB,gBAAgB,CAAC,eAAe;GAChC,UAAU;GACX,CAAC;;CAGJ,iBAAiB,KAAK;CAEtB,IAAI,YAAY;AACd,SAAO,CACL,GAAG,IAAI,IAAI;GACT,KAAK;GACL,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK,eAAe;GACxB,CAAC,CACH,CAAC,QACC,QACC,QAAQ,KAAK,wBAAwB,QAAQ,KAAK,oBACrD;;CAGH,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAmC;AAC7C,QAAM,OAAO;AACb,OAAK,WAAW,OAAO;AACvB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,uBACH,OAAO,wBAAwB,KAAK;AACtC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,OAAK,sBACH,OAAO,uBAAuB,KAAK;;;CAIvC,MAAM,wBAAwB,KAAe,MAA+B;EAC1E,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;AAQF,SADe;IAJZ,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC;GAE4B,GAAG;GAAM;;;CAK3C,MAAM,uBAAuB,KAAe,KAAa;EACvD,MAAM,WAAoC;GACxC,cAAc,IAAI;GAClB,GAAG,IAAI;GACR;EACD,MAAM,eAAwC,EAAE;AAChD,OAAK,eAAe,eAAe,SAAS,UAAU;AACpD,gBAAa,SAAS,SAAS;IAC/B;EACF,MAAM,aAAsC,GACzC,KAAK,uBAAuB,MAAM,KAAK,eAAe,OAAO,EAC5D,GAAG,cACJ,CAAC,EACH;AAED,SADe;IAAG,KAAK,sBAAsB;GAAK,GAAG;GAAY;;;CAKnE,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAE7D,MAAM,GAAG,KAAK,WAAW,MAAM,GAAG,SAAS;EAE3C,MAAM,cAAc;EAEpB,MAAM,gBAAgB,MAAM,KAAK,wBAC/B,YAAY,IACZ,KACD;EACD,IAAI,MAAM,MAAM,KAAK,SAAS,QAC5B,EAAE,GAAG,eAAe,EACpB,YAAY,SAAS,SAAS,CAC/B;EAED,MAAM,cAAc,CAAC,IAAI;AAEzB,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK,GAAG;GAK9C,MAAM,SAAS;IAAE,GAJI,MAAM,KAAK,uBAC9B,YAAY,IACZ,IACD;IACiC,GAAG;IAAM;AAC3C,SAAM,MAAM,KAAK,eAAe,QAC9B,EAAE,GAAG,QAAQ,EACb,YAAY,SAAS,SAAS,CAC/B;AACD,eAAY,KAAK,IAAI;;AAGvB,SAAO,GAAG,KAAK,YAAY,KAAK;;CAGlC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAsC;EAC7D,MAAM,qBAAqB,KAAK;AAEhC,MAAI,CAAC,mBACH,OAAM,IAAI,MAAM,oBAAoB;EAGtC,MAAM,gCAAgC,KAAK;AAE3C,MAAI,CAAC,8BACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,SAAO,IAAI,qBAAqB;GAC9B,UAAU,MAAM,SAAS,YAAY,mBAAmB;GACxD,gBAAgB,MAAM,SAAS,YAAY,8BAA8B;GAC1E,CAAC;;CAGJ,YAA4C;AAC1C,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,WAAW,KAAK,SAAS,WAAW;GACpC,kBAAkB,KAAK,eAAe,WAAW;GAClD"}

package/dist/chains/combine_documents/base.cjs

@@ -1,8 +1,6 @@
-const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+require("../../_virtual/_rolldown/runtime.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
-
 //#region src/chains/combine_documents/base.ts
-const DEFAULT_DOCUMENT_SEPARATOR = "\n\n";
 const DOCUMENTS_KEY = "context";
 const DEFAULT_DOCUMENT_PROMPT = /* @__PURE__ */ _langchain_core_prompts.PromptTemplate.fromTemplate("{page_content}");
 async function formatDocuments({ documentPrompt, documentSeparator, documents, config }) {
@@ -12,10 +10,9 @@ async function formatDocuments({ documentPrompt, documentSeparator, documents, c
 		page_content: document.pageContent
 	}, config)))).join(documentSeparator);
 }
-
 //#endregion
 exports.DEFAULT_DOCUMENT_PROMPT = DEFAULT_DOCUMENT_PROMPT;
-exports.DEFAULT_DOCUMENT_SEPARATOR = DEFAULT_DOCUMENT_SEPARATOR;
 exports.DOCUMENTS_KEY = DOCUMENTS_KEY;
 exports.formatDocuments = formatDocuments;
+
 //# sourceMappingURL=base.cjs.map

package/dist/chains/combine_documents/base.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"base.cjs","names":[],"sources":["../../../src/chains/combine_documents/base.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport { RunnableConfig } from \"@langchain/core/runnables\";\n\nexport const DEFAULT_DOCUMENT_SEPARATOR = \"\\n\\n\";\n\nexport const DOCUMENTS_KEY = \"context\";\nexport const INTERMEDIATE_STEPS_KEY = \"intermediate_steps\";\n\nexport const DEFAULT_DOCUMENT_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(\"{page_content}\");\n\nexport async function formatDocuments({\n  documentPrompt,\n  documentSeparator,\n  documents,\n  config,\n}: {\n  documentPrompt: BasePromptTemplate;\n  documentSeparator: string;\n  documents: Document[];\n  config?: RunnableConfig;\n}) {\n  if (documents == null || documents.length === 0) {\n    return \"\";\n  }\n  const formattedDocs = await Promise.all(\n    documents.map((document) =>\n      documentPrompt\n        .withConfig({ runName: \"document_formatter\" })\n        .invoke(\n          { ...document.metadata, page_content: document.pageContent },\n          config\n        )\n    )\n  );\n  return formattedDocs.join(documentSeparator);\n}\n"],"mappings":";;;;AAIA,MAAa,6BAA6B;AAE1C,MAAa,gBAAgB;AAG7B,MAAa,0BACK,uDAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;AAYT,SAVsB,MAAM,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,sBAAsB,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;EAAa,EAC5D,OACD,CACJ,CACF,EACoB,KAAK,kBAAkB"}
+{"version":3,"file":"base.cjs","names":[],"sources":["../../../src/chains/combine_documents/base.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport { RunnableConfig } from \"@langchain/core/runnables\";\n\nexport const DEFAULT_DOCUMENT_SEPARATOR = \"\\n\\n\";\n\nexport const DOCUMENTS_KEY = \"context\";\nexport const INTERMEDIATE_STEPS_KEY = \"intermediate_steps\";\n\nexport const DEFAULT_DOCUMENT_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(\"{page_content}\");\n\nexport async function formatDocuments({\n  documentPrompt,\n  documentSeparator,\n  documents,\n  config,\n}: {\n  documentPrompt: BasePromptTemplate;\n  documentSeparator: string;\n  documents: Document[];\n  config?: RunnableConfig;\n}) {\n  if (documents == null || documents.length === 0) {\n    return \"\";\n  }\n  const formattedDocs = await Promise.all(\n    documents.map((document) =>\n      documentPrompt\n        .withConfig({ runName: \"document_formatter\" })\n        .invoke(\n          { ...document.metadata, page_content: document.pageContent },\n          config\n        )\n    )\n  );\n  return formattedDocs.join(documentSeparator);\n}\n"],"mappings":";;;AAMA,MAAa,gBAAgB;AAG7B,MAAa,0BACK,wCAAA,eAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;AAYT,SAVsB,MAAM,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,sBAAsB,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;EAAa,EAC5D,OACD,CACJ,CACF,EACoB,KAAK,kBAAkB"}

package/dist/chains/combine_documents/base.js

@@ -1,7 +1,5 @@
 import { PromptTemplate } from "@langchain/core/prompts";
-
 //#region src/chains/combine_documents/base.ts
-const DEFAULT_DOCUMENT_SEPARATOR = "\n\n";
 const DOCUMENTS_KEY = "context";
 const DEFAULT_DOCUMENT_PROMPT = /* @__PURE__ */ PromptTemplate.fromTemplate("{page_content}");
 async function formatDocuments({ documentPrompt, documentSeparator, documents, config }) {
@@ -11,7 +9,7 @@ async function formatDocuments({ documentPrompt, documentSeparator, documents, c
 		page_content: document.pageContent
 	}, config)))).join(documentSeparator);
 }
-
 //#endregion
-export { DEFAULT_DOCUMENT_PROMPT, DEFAULT_DOCUMENT_SEPARATOR, DOCUMENTS_KEY, formatDocuments };
+export { DEFAULT_DOCUMENT_PROMPT, DOCUMENTS_KEY, formatDocuments };
+
 //# sourceMappingURL=base.js.map

package/dist/chains/combine_documents/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","names":[],"sources":["../../../src/chains/combine_documents/base.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport { RunnableConfig } from \"@langchain/core/runnables\";\n\nexport const DEFAULT_DOCUMENT_SEPARATOR = \"\\n\\n\";\n\nexport const DOCUMENTS_KEY = \"context\";\nexport const INTERMEDIATE_STEPS_KEY = \"intermediate_steps\";\n\nexport const DEFAULT_DOCUMENT_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(\"{page_content}\");\n\nexport async function formatDocuments({\n  documentPrompt,\n  documentSeparator,\n  documents,\n  config,\n}: {\n  documentPrompt: BasePromptTemplate;\n  documentSeparator: string;\n  documents: Document[];\n  config?: RunnableConfig;\n}) {\n  if (documents == null || documents.length === 0) {\n    return \"\";\n  }\n  const formattedDocs = await Promise.all(\n    documents.map((document) =>\n      documentPrompt\n        .withConfig({ runName: \"document_formatter\" })\n        .invoke(\n          { ...document.metadata, page_content: document.pageContent },\n          config\n        )\n    )\n  );\n  return formattedDocs.join(documentSeparator);\n}\n"],"mappings":";;;AAIA,MAAa,6BAA6B;AAE1C,MAAa,gBAAgB;AAG7B,MAAa,0BACK,+BAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;AAYT,SAVsB,MAAM,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,sBAAsB,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;EAAa,EAC5D,OACD,CACJ,CACF,EACoB,KAAK,kBAAkB"}
+{"version":3,"file":"base.js","names":[],"sources":["../../../src/chains/combine_documents/base.ts"],"sourcesContent":["import { Document } from \"@langchain/core/documents\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport { RunnableConfig } from \"@langchain/core/runnables\";\n\nexport const DEFAULT_DOCUMENT_SEPARATOR = \"\\n\\n\";\n\nexport const DOCUMENTS_KEY = \"context\";\nexport const INTERMEDIATE_STEPS_KEY = \"intermediate_steps\";\n\nexport const DEFAULT_DOCUMENT_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(\"{page_content}\");\n\nexport async function formatDocuments({\n  documentPrompt,\n  documentSeparator,\n  documents,\n  config,\n}: {\n  documentPrompt: BasePromptTemplate;\n  documentSeparator: string;\n  documents: Document[];\n  config?: RunnableConfig;\n}) {\n  if (documents == null || documents.length === 0) {\n    return \"\";\n  }\n  const formattedDocs = await Promise.all(\n    documents.map((document) =>\n      documentPrompt\n        .withConfig({ runName: \"document_formatter\" })\n        .invoke(\n          { ...document.metadata, page_content: document.pageContent },\n          config\n        )\n    )\n  );\n  return formattedDocs.join(documentSeparator);\n}\n"],"mappings":";;AAMA,MAAa,gBAAgB;AAG7B,MAAa,0BACK,+BAAe,aAAa,iBAAiB;AAE/D,eAAsB,gBAAgB,EACpC,gBACA,mBACA,WACA,UAMC;AACD,KAAI,aAAa,QAAQ,UAAU,WAAW,EAC5C,QAAO;AAYT,SAVsB,MAAM,QAAQ,IAClC,UAAU,KAAK,aACb,eACG,WAAW,EAAE,SAAS,sBAAsB,CAAC,CAC7C,OACC;EAAE,GAAG,SAAS;EAAU,cAAc,SAAS;EAAa,EAC5D,OACD,CACJ,CACF,EACoB,KAAK,kBAAkB"}