@langchain/classic 1.0.0-alpha.1

package/dist/chains/base.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.cjs","names":["BaseLangChain","fields?: BaseMemory | ChainInputs","verbose?: boolean","callbacks?: Callbacks","values: ChainValues","input: RunInput","options?: RunnableConfig","CallbackManager","outputValues: RunOutput","RUN_KEY","outputs: Record<string, unknown>","inputs: Record<string, unknown>","input: any","config?: Callbacks | RunnableConfig","values: ChainValues & { signal?: AbortSignal; timeout?: number }","tags?: string[]","inputs: RunInput[]","config?: (Callbacks | RunnableConfig)[]","data: SerializedBaseChain","values: LoadValues"],"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      outputValues = await (fullValues.signal\n        ? (Promise.race([\n            this._call(fullValues as RunInput, runManager, config),\n            new Promise((_, reject) => {\n              fullValues.signal?.addEventListener(\"abort\", () => {\n                reject(new Error(\"AbortError\"));\n              });\n            }),\n          ]) as Promise<RunOutput>)\n        : this._call(fullValues as RunInput, runManager, config));\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 } = await import(\n          \"./combine_docs_chain.js\"\n        );\n        return MapReduceDocumentsChain.deserialize(data);\n      }\n      case \"refine_documents_chain\": {\n        const { RefineDocumentsChain } = await import(\n          \"./combine_docs_chain.js\"\n        );\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,oDAEV;CAGE,IAAI,eAAyB;AAC3B,SAAO;GAAC;GAAa;GAAU,KAAK,YAAY;EAAC;CAClD;CAED,YACEC,QAEAC,SAEAC,WACA;AACA,MACE,UAAU,WAAW,KACrB,OAAO,WAAW,YAClB,EAAE,iBAAiB,SACnB;GAEA,MAAM,EAAE,QAAQ,gBAAiB,GAAG,MAAM,GAAG;GAC7C,MAAM;IAAE,GAAG;IAAM,WAAW,mBAAmB,KAAK;GAAW,EAAC;GAChE,KAAK,SAAS;EACf,OAAM;GAEL,MAAM;IAAE;IAAS;GAAW,EAAC;GAC7B,KAAK,SAAS;EACf;CACF;;CAGD,oBAAoBC,QAAkC;EACpD,MAAM,kBAAkB,EAAE,GAAG,OAAQ;AACrC,MAAI,YAAY,iBACd,OAAO,gBAAgB;AAEzB,MAAI,aAAa,iBACf,OAAO,gBAAgB;AAEzB,SAAO;CACR;;;;;;;CAQD,MAAM,OAAOC,OAAiBC,SAA8C;EAC1E,MAAM,sDAAsB,QAAQ;EACpC,MAAM,aAAa,MAAM,KAAK,cAAc,MAAM;EAClD,MAAM,mBAAmB,MAAMC,mDAAgB,UAC7C,QAAQ,WACR,KAAK,WACL,QAAQ,MACR,KAAK,MACL,QAAQ,UACR,KAAK,UACL,EAAE,SAAS,KAAK,QAAS,EAC1B;EACD,MAAM,aAAa,MAAM,kBAAkB,iBACzC,KAAK,QAAQ,EACb,YACA,QACA,QACA,QACA,QACA,QAAQ,QACT;EACD,IAAIC;AACJ,MAAI;GACF,eAAe,OAAO,WAAW,SAC5B,QAAQ,KAAK,CACZ,KAAK,MAAM,YAAwB,YAAY,OAAO,EACtD,IAAI,QAAQ,CAAC,GAAG,WAAW;IACzB,WAAW,QAAQ,iBAAiB,SAAS,MAAM;KACjD,uBAAO,IAAI,MAAM,cAAc;IAChC,EAAC;GACH,EACF,EAAC,GACF,KAAK,MAAM,YAAwB,YAAY,OAAO;EAC3D,SAAQ,GAAG;GACV,MAAM,YAAY,iBAAiB,EAAE;AACrC,SAAM;EACP;AACD,MAAI,EAAE,KAAK,UAAU,OACnB,MAAM,KAAK,OAAO,YAChB,KAAK,oBAAoB,MAAM,EAC/B,aACD;EAEH,MAAM,YAAY,eAAe,aAAa;EAE9C,OAAO,eAAe,cAAcC,kCAAS;GAC3C,OAAO,aAAa,EAAE,OAAO,YAAY,MAAO,IAAG;GACnD,cAAc;EACf,EAAC;AACF,SAAO;CACR;CAED,AAAQ,iBAAiBC,SAAwC;EAC/D,MAAM,cAAc,KAAK,WAAW,OAAO,CAAC,MAAM,EAAE,KAAK,SAAS;AAClE,MAAI,YAAY,OACd,OAAM,IAAI,MACR,CAAC,qBAAqB,EAAE,YAAY,KAClC,KACD,CAAC,YAAY,EAAE,KAAK,YAAY,EAAE;CAGxC;CAED,MAAM,YACJC,QACAD,SACA,oBAAoB,OACpB;EACA,KAAK,iBAAiB,QAAQ;AAC9B,MAAI,KAAK,QACP,MAAM,KAAK,OAAO,YAAY,QAAQ,QAAQ;AAEhD,MAAI,kBACF,QAAO;AAET,SAAO;GAAE,GAAG;GAAQ,GAAG;EAAS;CACjC;;;;CAmBD,YAAiC;AAC/B,QAAM,IAAI,MAAM;CACjB;;CAOD,MAAM,IAEJE,OACAC,QACiB;EACjB,MAAM,YAAY,KAAK,UAAU,OAC/B,CAAC,MAAM,CAAC,KAAK,QAAQ,WAAW,SAAS,EAAE,CAC5C;EACD,MAAM,iBAAiB,UAAU,UAAU;AAC3C,MAAI,CAAC,eACH,OAAM,IAAI,MACR,CAAC,MAAM,EAAE,KAAK,YAAY,CAAC,2CAA2C,CAAC;EAI3E,MAAM,SAAS,UAAU,SAAS,GAAG,UAAU,KAAK,MAAO,IAAI,CAAE;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;CAEH;CAED,MAAgB,cACdC,QACA;EACA,MAAM,aAAa,EAAE,GAAG,OAAQ;AAChC,MAAI,WAAW,WAAW,CAAC,WAAW,QAAQ;GAC5C,WAAW,SAAS,YAAY,QAAQ,WAAW,QAAQ;GAC3D,OAAO,WAAW;EACnB;AACD,MAAI,EAAE,KAAK,UAAU,OAAO;GAC1B,MAAM,YAAY,MAAM,KAAK,OAAO,oBAClC,KAAK,oBAAoB,OAAO,CACjC;AACD,QAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,UAAU,EAClD,WAAW,OAAO;EAErB;AACD,SAAO;CACR;;;;;;;;CASD,MAAM,KACJA,QACAD,QAEAE,MACoB;EACpB,MAAM,eAAe;GAAE;GAAM,kEAA0B,OAAO;EAAE;AAChE,SAAO,KAAK,OAAO,QAAoB,aAAa;CACrD;;;;;;CAOD,MAAM,MACJC,QACAC,QACsB;AACtB,SAAO,QAAQ,IACb,OAAO,IAAI,OAAO,GAAG,QAAQ,KAAK,KAAK,GAAG,SAAS,KAAK,CAAC,CAC1D;CACF;;;;CAKD,aAAa,YACXC,MACAC,SAAqB,CAAE,GACH;AACpB,UAAQ,KAAK,OAAb;GACE,KAAK,aAAa;IAChB,MAAM,EAAE,UAAU,GAAG,2CAAM;AAC3B,WAAO,SAAS,YAAY,KAAK;GAClC;GACD,KAAK,oBAAoB;IACvB,MAAM,EAAE,iBAAiB,GAAG,2CAAM;AAClC,WAAO,gBAAgB,YAAY,KAAK;GACzC;GACD,KAAK,2BAA2B;IAC9B,MAAM,EAAE,uBAAuB,GAAG,2CAAM;AACxC,WAAO,sBAAsB,YAAY,KAAK;GAC/C;GACD,KAAK,yBAAyB;IAC5B,MAAM,EAAE,qBAAqB,GAAG,2CAAM;AACtC,WAAO,oBAAoB,YAAY,KAAK;GAC7C;GACD,KAAK,8BAA8B;IACjC,MAAM,EAAE,yBAAyB,GAAG,2CAAM;AAG1C,WAAO,wBAAwB,YAAY,KAAK;GACjD;GACD,KAAK,0BAA0B;IAC7B,MAAM,EAAE,sBAAsB,GAAG,2CAAM;AAGvC,WAAO,qBAAqB,YAAY,KAAK;GAC9C;GACD,KAAK,gBAAgB;IACnB,MAAM,EAAE,iBAAiB,GAAG,2CAAM;AAClC,WAAO,gBAAgB,YAAY,MAAM,OAAO;GACjD;GACD,KAAK,aAAa;IAChB,MAAM,EAAE,UAAU,GAAG,2CAAM;AAC3B,WAAO,SAAS,YAAY,KAAK;GAClC;GACD,QACE,OAAM,IAAI,MACR,CAAC,+BAA+B,EAC7B,KAA6B,OAC9B;EAEP;CACF;AACF"}

package/dist/chains/base.d.cts

@@ -0,0 +1,88 @@
+import { SerializedBaseChain } from "./serde.cjs";
+import { RunnableConfig } from "@langchain/core/runnables";
+import { ChainValues } from "@langchain/core/utils/types";
+import { BaseLangChain, BaseLangChainParams } from "@langchain/core/language_models/base";
+import { CallbackManager, CallbackManagerForChainRun, Callbacks } from "@langchain/core/callbacks/manager";
+import { BaseMemory } from "@langchain/core/memory";
+
+//#region src/chains/base.d.ts
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+type LoadValues = Record<string, any>;
+interface ChainInputs extends BaseLangChainParams {
+  memory?: BaseMemory;
+  /**
+   * @deprecated Use `callbacks` instead
+   */
+  callbackManager?: CallbackManager;
+}
+/**
+ * Base interface that all chains must implement.
+ */
+declare abstract class BaseChain<RunInput extends ChainValues = ChainValues, RunOutput extends ChainValues = ChainValues> extends BaseLangChain<RunInput, RunOutput> implements ChainInputs {
+  memory?: BaseMemory;
+  get lc_namespace(): string[];
+  constructor(fields?: BaseMemory | ChainInputs, /** @deprecated */
+  verbose?: boolean, /** @deprecated */
+  callbacks?: Callbacks);
+  /** @ignore */
+  _selectMemoryInputs(values: ChainValues): ChainValues;
+  /**
+   * Invoke the chain with the provided input and returns the output.
+   * @param input Input values for the chain run.
+   * @param config Optional configuration for the Runnable.
+   * @returns Promise that resolves with the output of the chain run.
+   */
+  invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput>;
+  private _validateOutputs;
+  prepOutputs(inputs: Record<string, unknown>, outputs: Record<string, unknown>, returnOnlyOutputs?: boolean): Promise<Record<string, unknown>>;
+  /**
+   * Run the core logic of this chain and return the output
+   */
+  abstract _call(values: RunInput, runManager?: CallbackManagerForChainRun, config?: RunnableConfig): Promise<RunOutput>;
+  /**
+   * Return the string type key uniquely identifying this class of chain.
+   */
+  abstract _chainType(): string;
+  /**
+   * Return a json-like object representing this chain.
+   */
+  serialize(): SerializedBaseChain;
+  abstract get inputKeys(): string[];
+  abstract get outputKeys(): string[];
+  /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */
+  run(
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  input: any, config?: Callbacks | RunnableConfig): Promise<string>;
+  protected _formatValues(values: ChainValues & {
+    signal?: AbortSignal;
+    timeout?: number;
+  }): Promise<ChainValues & {
+    signal?: AbortSignal | undefined;
+    timeout?: number | undefined;
+  }>;
+  /**
+   * @deprecated Use .invoke() instead. Will be removed in 0.2.0.
+   *
+   * Run the core logic of this chain and add to output if desired.
+   *
+   * Wraps _call and handles memory.
+   */
+  call(values: ChainValues & {
+    signal?: AbortSignal;
+    timeout?: number;
+  }, config?: Callbacks | RunnableConfig, /** @deprecated */
+  tags?: string[]): Promise<RunOutput>;
+  /**
+   * @deprecated Use .batch() instead. Will be removed in 0.2.0.
+   *
+   * Call the chain on all inputs in the list
+   */
+  apply(inputs: RunInput[], config?: (Callbacks | RunnableConfig)[]): Promise<RunOutput[]>;
+  /**
+   * Load a chain from a json-like object describing it.
+   */
+  static deserialize(data: SerializedBaseChain, values?: LoadValues): Promise<BaseChain>;
+}
+//#endregion
+export { BaseChain, ChainInputs };
+//# sourceMappingURL=base.d.cts.map

package/dist/chains/base.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.cts","names":["BaseMemory","ChainValues","CallbackManagerForChainRun","CallbackManager","Callbacks","RunnableConfig","BaseLangChain","BaseLangChainParams","SerializedBaseChain","LoadValues","Record","ChainInputs","BaseChain","RunInput","RunOutput","Promise","AbortSignal"],"sources":["../../src/chains/base.d.ts"],"sourcesContent":["import { BaseMemory } from \"@langchain/core/memory\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun, CallbackManager, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { type RunnableConfig } from \"@langchain/core/runnables\";\nimport { BaseLangChain, BaseLangChainParams } from \"@langchain/core/language_models/base\";\nimport { SerializedBaseChain } from \"./serde.js\";\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\nexport interface ChainInputs extends BaseLangChainParams {\n    memory?: BaseMemory;\n    /**\n     * @deprecated Use `callbacks` instead\n     */\n    callbackManager?: CallbackManager;\n}\n/**\n * Base interface that all chains must implement.\n */\nexport declare abstract class BaseChain<RunInput extends ChainValues = ChainValues, RunOutput extends ChainValues = ChainValues> extends BaseLangChain<RunInput, RunOutput> implements ChainInputs {\n    memory?: BaseMemory;\n    get lc_namespace(): string[];\n    constructor(fields?: BaseMemory | ChainInputs, \n    /** @deprecated */\n    verbose?: boolean, \n    /** @deprecated */\n    callbacks?: Callbacks);\n    /** @ignore */\n    _selectMemoryInputs(values: ChainValues): ChainValues;\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    invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput>;\n    private _validateOutputs;\n    prepOutputs(inputs: Record<string, unknown>, outputs: Record<string, unknown>, returnOnlyOutputs?: boolean): Promise<Record<string, unknown>>;\n    /**\n     * Run the core logic of this chain and return the output\n     */\n    abstract _call(values: RunInput, runManager?: CallbackManagerForChainRun, config?: RunnableConfig): Promise<RunOutput>;\n    /**\n     * Return the string type key uniquely identifying this class of chain.\n     */\n    abstract _chainType(): string;\n    /**\n     * Return a json-like object representing this chain.\n     */\n    serialize(): SerializedBaseChain;\n    abstract get inputKeys(): string[];\n    abstract get outputKeys(): string[];\n    /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */\n    run(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any, config?: Callbacks | RunnableConfig): Promise<string>;\n    protected _formatValues(values: ChainValues & {\n        signal?: AbortSignal;\n        timeout?: number;\n    }): Promise<ChainValues & {\n        signal?: AbortSignal | undefined;\n        timeout?: number | undefined;\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    call(values: ChainValues & {\n        signal?: AbortSignal;\n        timeout?: number;\n    }, config?: Callbacks | RunnableConfig, \n    /** @deprecated */\n    tags?: string[]): Promise<RunOutput>;\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    apply(inputs: RunInput[], config?: (Callbacks | RunnableConfig)[]): Promise<RunOutput[]>;\n    /**\n     * Load a chain from a json-like object describing it.\n     */\n    static deserialize(data: SerializedBaseChain, values?: LoadValues): Promise<BaseChain>;\n}\n"],"mappings":";;;;;;;;;KAOYS,UAAAA,GAAaC;AAAbD,UACKE,WAAAA,SAAoBJ,mBADN,CAAA;EACdI,MAAAA,CAAAA,EACJX,UADe;EAAA;;;EAKS,eALAO,CAAAA,EAKfJ,eALeI;AAAmB;AAUxD;;;AAAuEN,uBAAzCW,SAAyCX,CAAAA,iBAAdA,WAAcA,GAAAA,WAAAA,EAAAA,kBAA+BA,WAA/BA,GAA6CA,WAA7CA,CAAAA,SAAkEK,aAAlEL,CAAgFY,QAAhFZ,EAA0Fa,SAA1Fb,CAAAA,YAAgHU,WAAhHV,CAAAA;EAAW,MAAoBA,CAAAA,EACzFD,UADyFC;EAAW,IAAGA,YAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAW,WAAwBY,CAAAA,MAAAA,CAAAA,EAG9Hb,UAH8Ha,GAGjHF,WAHiHE,EAAQ;EAAW,OAC7Jb,CAAAA,EAAAA,OAAAA,EAAU;EAEY,SAAGW,CAAAA,EAItBP,SAJsBO;EAAW;EAIxB,mBAEOV,CAAAA,MAAAA,EAAAA,WAAAA,CAAAA,EAAcA,WAAdA;EAAW;;;;;;EASb,MAA4BS,CAAAA,KAAAA,EAFxCG,QAEwCH,EAAAA,OAAAA,CAAAA,EAFpBL,cAEoBK,CAAAA,EAFHK,OAEGL,CAFKI,SAELJ,CAAAA;EAAM,QAAyDA,gBAAAA;EAAM,WAAdK,CAAAA,MAAAA,EAAzFL,MAAyFK,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,OAAAA,EAAvDL,MAAuDK,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,iBAAAA,CAAAA,EAAAA,OAAAA,CAAAA,EAAAA,OAAAA,CAAQL,MAARK,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,CAAAA;EAAO;;;EAInB,SAAWD,KAAAA,CAAAA,MAAAA,EAArFD,QAAqFC,EAAAA,UAAAA,CAAAA,EAA9DZ,0BAA8DY,EAAAA,MAAAA,CAAAA,EAAzBT,cAAyBS,CAAAA,EAARC,OAAQD,CAAAA,SAAAA,CAAAA;EAAS;;;EAcvF,SAAGT,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAc;;;EAEvB,SAEZJ,CAAAA,CAAAA,EAVCO,mBAUDP;EAAW,aACVe,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAW,aADpBD,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAO;EAWa,GACXC;EAAW;EAEH,KAAGX,EAAAA,GAAAA,EAAAA,MAAAA,CAAAA,EAlBHD,SAkBGC,GAlBSA,cAkBTA,CAAAA,EAlB0BU,OAkB1BV,CAAAA,MAAAA,CAAAA;EAAc,UAEZS,aAAAA,CAAAA,MAAAA,EAnBMb,WAmBNa,GAAAA;IAARC,MAAAA,CAAAA,EAlBLC,WAkBKD;IAMJF,OAAAA,CAAAA,EAAAA,MAAAA;EAAQ,CAAA,CAAA,EAtBlBE,OAsBgCX,CAtBxBH,WAsBwBG,GAAAA;IAAYC,MAAAA,CAAAA,EArBnCW,WAqBmCX,GAAAA,SAAAA;IAA4BS,OAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAS,CAAA,CAAA;EAAV;;;;;;AA9DmH;eAmDjLb;aACAe;;cAEDZ,YAAYC;oBAENU,QAAQD;;;;;;gBAMZD,sBAAsBT,YAAYC,oBAAoBU,QAAQD;;;;2BAInDN,8BAA8BC,aAAaM,QAAQH"}

package/dist/chains/base.d.ts

@@ -0,0 +1,88 @@
+import { SerializedBaseChain } from "./serde.js";
+import { RunnableConfig } from "@langchain/core/runnables";
+import { BaseLangChain, BaseLangChainParams } from "@langchain/core/language_models/base";
+import { CallbackManager, CallbackManagerForChainRun, Callbacks } from "@langchain/core/callbacks/manager";
+import { BaseMemory } from "@langchain/core/memory";
+import { ChainValues } from "@langchain/core/utils/types";
+
+//#region src/chains/base.d.ts
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+type LoadValues = Record<string, any>;
+interface ChainInputs extends BaseLangChainParams {
+  memory?: BaseMemory;
+  /**
+   * @deprecated Use `callbacks` instead
+   */
+  callbackManager?: CallbackManager;
+}
+/**
+ * Base interface that all chains must implement.
+ */
+declare abstract class BaseChain<RunInput extends ChainValues = ChainValues, RunOutput extends ChainValues = ChainValues> extends BaseLangChain<RunInput, RunOutput> implements ChainInputs {
+  memory?: BaseMemory;
+  get lc_namespace(): string[];
+  constructor(fields?: BaseMemory | ChainInputs, /** @deprecated */
+  verbose?: boolean, /** @deprecated */
+  callbacks?: Callbacks);
+  /** @ignore */
+  _selectMemoryInputs(values: ChainValues): ChainValues;
+  /**
+   * Invoke the chain with the provided input and returns the output.
+   * @param input Input values for the chain run.
+   * @param config Optional configuration for the Runnable.
+   * @returns Promise that resolves with the output of the chain run.
+   */
+  invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput>;
+  private _validateOutputs;
+  prepOutputs(inputs: Record<string, unknown>, outputs: Record<string, unknown>, returnOnlyOutputs?: boolean): Promise<Record<string, unknown>>;
+  /**
+   * Run the core logic of this chain and return the output
+   */
+  abstract _call(values: RunInput, runManager?: CallbackManagerForChainRun, config?: RunnableConfig): Promise<RunOutput>;
+  /**
+   * Return the string type key uniquely identifying this class of chain.
+   */
+  abstract _chainType(): string;
+  /**
+   * Return a json-like object representing this chain.
+   */
+  serialize(): SerializedBaseChain;
+  abstract get inputKeys(): string[];
+  abstract get outputKeys(): string[];
+  /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */
+  run(
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  input: any, config?: Callbacks | RunnableConfig): Promise<string>;
+  protected _formatValues(values: ChainValues & {
+    signal?: AbortSignal;
+    timeout?: number;
+  }): Promise<ChainValues & {
+    signal?: AbortSignal | undefined;
+    timeout?: number | undefined;
+  }>;
+  /**
+   * @deprecated Use .invoke() instead. Will be removed in 0.2.0.
+   *
+   * Run the core logic of this chain and add to output if desired.
+   *
+   * Wraps _call and handles memory.
+   */
+  call(values: ChainValues & {
+    signal?: AbortSignal;
+    timeout?: number;
+  }, config?: Callbacks | RunnableConfig, /** @deprecated */
+  tags?: string[]): Promise<RunOutput>;
+  /**
+   * @deprecated Use .batch() instead. Will be removed in 0.2.0.
+   *
+   * Call the chain on all inputs in the list
+   */
+  apply(inputs: RunInput[], config?: (Callbacks | RunnableConfig)[]): Promise<RunOutput[]>;
+  /**
+   * Load a chain from a json-like object describing it.
+   */
+  static deserialize(data: SerializedBaseChain, values?: LoadValues): Promise<BaseChain>;
+}
+//#endregion
+export { BaseChain, ChainInputs };
+//# sourceMappingURL=base.d.ts.map

package/dist/chains/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","names":["BaseMemory","ChainValues","CallbackManagerForChainRun","CallbackManager","Callbacks","RunnableConfig","BaseLangChain","BaseLangChainParams","SerializedBaseChain","LoadValues","Record","ChainInputs","BaseChain","RunInput","RunOutput","Promise","AbortSignal"],"sources":["../../src/chains/base.d.ts"],"sourcesContent":["import { BaseMemory } from \"@langchain/core/memory\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun, CallbackManager, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { type RunnableConfig } from \"@langchain/core/runnables\";\nimport { BaseLangChain, BaseLangChainParams } from \"@langchain/core/language_models/base\";\nimport { SerializedBaseChain } from \"./serde.js\";\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\nexport interface ChainInputs extends BaseLangChainParams {\n    memory?: BaseMemory;\n    /**\n     * @deprecated Use `callbacks` instead\n     */\n    callbackManager?: CallbackManager;\n}\n/**\n * Base interface that all chains must implement.\n */\nexport declare abstract class BaseChain<RunInput extends ChainValues = ChainValues, RunOutput extends ChainValues = ChainValues> extends BaseLangChain<RunInput, RunOutput> implements ChainInputs {\n    memory?: BaseMemory;\n    get lc_namespace(): string[];\n    constructor(fields?: BaseMemory | ChainInputs, \n    /** @deprecated */\n    verbose?: boolean, \n    /** @deprecated */\n    callbacks?: Callbacks);\n    /** @ignore */\n    _selectMemoryInputs(values: ChainValues): ChainValues;\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    invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput>;\n    private _validateOutputs;\n    prepOutputs(inputs: Record<string, unknown>, outputs: Record<string, unknown>, returnOnlyOutputs?: boolean): Promise<Record<string, unknown>>;\n    /**\n     * Run the core logic of this chain and return the output\n     */\n    abstract _call(values: RunInput, runManager?: CallbackManagerForChainRun, config?: RunnableConfig): Promise<RunOutput>;\n    /**\n     * Return the string type key uniquely identifying this class of chain.\n     */\n    abstract _chainType(): string;\n    /**\n     * Return a json-like object representing this chain.\n     */\n    serialize(): SerializedBaseChain;\n    abstract get inputKeys(): string[];\n    abstract get outputKeys(): string[];\n    /** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */\n    run(\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    input: any, config?: Callbacks | RunnableConfig): Promise<string>;\n    protected _formatValues(values: ChainValues & {\n        signal?: AbortSignal;\n        timeout?: number;\n    }): Promise<ChainValues & {\n        signal?: AbortSignal | undefined;\n        timeout?: number | undefined;\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    call(values: ChainValues & {\n        signal?: AbortSignal;\n        timeout?: number;\n    }, config?: Callbacks | RunnableConfig, \n    /** @deprecated */\n    tags?: string[]): Promise<RunOutput>;\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    apply(inputs: RunInput[], config?: (Callbacks | RunnableConfig)[]): Promise<RunOutput[]>;\n    /**\n     * Load a chain from a json-like object describing it.\n     */\n    static deserialize(data: SerializedBaseChain, values?: LoadValues): Promise<BaseChain>;\n}\n"],"mappings":";;;;;;;;;KAOYS,UAAAA,GAAaC;AAAbD,UACKE,WAAAA,SAAoBJ,mBADN,CAAA;EACdI,MAAAA,CAAAA,EACJX,UADe;EAAA;;;EAKS,eALAO,CAAAA,EAKfJ,eALeI;AAAmB;AAUxD;;;AAAuEN,uBAAzCW,SAAyCX,CAAAA,iBAAdA,WAAcA,GAAAA,WAAAA,EAAAA,kBAA+BA,WAA/BA,GAA6CA,WAA7CA,CAAAA,SAAkEK,aAAlEL,CAAgFY,QAAhFZ,EAA0Fa,SAA1Fb,CAAAA,YAAgHU,WAAhHV,CAAAA;EAAW,MAAoBA,CAAAA,EACzFD,UADyFC;EAAW,IAAGA,YAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAW,WAAwBY,CAAAA,MAAAA,CAAAA,EAG9Hb,UAH8Ha,GAGjHF,WAHiHE,EAAQ;EAAW,OAC7Jb,CAAAA,EAAAA,OAAAA,EAAU;EAEY,SAAGW,CAAAA,EAItBP,SAJsBO;EAAW;EAIxB,mBAEOV,CAAAA,MAAAA,EAAAA,WAAAA,CAAAA,EAAcA,WAAdA;EAAW;;;;;;EASb,MAA4BS,CAAAA,KAAAA,EAFxCG,QAEwCH,EAAAA,OAAAA,CAAAA,EAFpBL,cAEoBK,CAAAA,EAFHK,OAEGL,CAFKI,SAELJ,CAAAA;EAAM,QAAyDA,gBAAAA;EAAM,WAAdK,CAAAA,MAAAA,EAAzFL,MAAyFK,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,OAAAA,EAAvDL,MAAuDK,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,EAAAA,iBAAAA,CAAAA,EAAAA,OAAAA,CAAAA,EAAAA,OAAAA,CAAQL,MAARK,CAAAA,MAAAA,EAAAA,OAAAA,CAAAA,CAAAA;EAAO;;;EAInB,SAAWD,KAAAA,CAAAA,MAAAA,EAArFD,QAAqFC,EAAAA,UAAAA,CAAAA,EAA9DZ,0BAA8DY,EAAAA,MAAAA,CAAAA,EAAzBT,cAAyBS,CAAAA,EAARC,OAAQD,CAAAA,SAAAA,CAAAA;EAAS;;;EAcvF,SAAGT,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAc;;;EAEvB,SAEZJ,CAAAA,CAAAA,EAVCO,mBAUDP;EAAW,aACVe,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAW,aADpBD,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAO;EAWa,GACXC;EAAW;EAEH,KAAGX,EAAAA,GAAAA,EAAAA,MAAAA,CAAAA,EAlBHD,SAkBGC,GAlBSA,cAkBTA,CAAAA,EAlB0BU,OAkB1BV,CAAAA,MAAAA,CAAAA;EAAc,UAEZS,aAAAA,CAAAA,MAAAA,EAnBMb,WAmBNa,GAAAA;IAARC,MAAAA,CAAAA,EAlBLC,WAkBKD;IAMJF,OAAAA,CAAAA,EAAAA,MAAAA;EAAQ,CAAA,CAAA,EAtBlBE,OAsBgCX,CAtBxBH,WAsBwBG,GAAAA;IAAYC,MAAAA,CAAAA,EArBnCW,WAqBmCX,GAAAA,SAAAA;IAA4BS,OAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAS,CAAA,CAAA;EAAV;;;;;;AA9DmH;eAmDjLb;aACAe;;cAEDZ,YAAYC;oBAENU,QAAQD;;;;;;gBAMZD,sBAAsBT,YAAYC,oBAAoBU,QAAQD;;;;2BAInDN,8BAA8BC,aAAaM,QAAQH"}

package/dist/chains/base.js

@@ -0,0 +1,179 @@
+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.
+*/
+var BaseChain = class extends BaseLangChain {
+	get lc_namespace() {
+		return [
+			"langchain",
+			"chains",
+			this._chainType()
+		];
+	}
+	constructor(fields, verbose, callbacks) {
+		if (arguments.length === 1 && typeof fields === "object" && !("saveContext" in fields)) {
+			const { memory, callbackManager,...rest } = fields;
+			super({
+				...rest,
+				callbacks: callbackManager ?? rest.callbacks
+			});
+			this.memory = memory;
+		} else {
+			super({
+				verbose,
+				callbacks
+			});
+			this.memory = fields;
+		}
+	}
+	/** @ignore */
+	_selectMemoryInputs(values) {
+		const valuesForMemory = { ...values };
+		if ("signal" in valuesForMemory) delete valuesForMemory.signal;
+		if ("timeout" in valuesForMemory) delete valuesForMemory.timeout;
+		return valuesForMemory;
+	}
+	/**
+	* Invoke the chain with the provided input and returns the output.
+	* @param input Input values for the chain run.
+	* @param config Optional configuration for the Runnable.
+	* @returns Promise that resolves with the output of the chain run.
+	*/
+	async invoke(input, options) {
+		const config = ensureConfig(options);
+		const fullValues = await this._formatValues(input);
+		const callbackManager_ = await CallbackManager.configure(config?.callbacks, this.callbacks, config?.tags, this.tags, config?.metadata, this.metadata, { verbose: this.verbose });
+		const runManager = await callbackManager_?.handleChainStart(this.toJSON(), fullValues, void 0, void 0, void 0, void 0, config?.runName);
+		let outputValues;
+		try {
+			outputValues = await (fullValues.signal ? Promise.race([this._call(fullValues, runManager, config), new Promise((_, reject) => {
+				fullValues.signal?.addEventListener("abort", () => {
+					reject(/* @__PURE__ */ new Error("AbortError"));
+				});
+			})]) : this._call(fullValues, runManager, config));
+		} catch (e) {
+			await runManager?.handleChainError(e);
+			throw e;
+		}
+		if (!(this.memory == null)) await this.memory.saveContext(this._selectMemoryInputs(input), outputValues);
+		await runManager?.handleChainEnd(outputValues);
+		Object.defineProperty(outputValues, RUN_KEY, {
+			value: runManager ? { runId: runManager?.runId } : void 0,
+			configurable: true
+		});
+		return outputValues;
+	}
+	_validateOutputs(outputs) {
+		const missingKeys = this.outputKeys.filter((k) => !(k in outputs));
+		if (missingKeys.length) throw new Error(`Missing output keys: ${missingKeys.join(", ")} from chain ${this._chainType()}`);
+	}
+	async prepOutputs(inputs, outputs, returnOnlyOutputs = false) {
+		this._validateOutputs(outputs);
+		if (this.memory) await this.memory.saveContext(inputs, outputs);
+		if (returnOnlyOutputs) return outputs;
+		return {
+			...inputs,
+			...outputs
+		};
+	}
+	/**
+	* Return a json-like object representing this chain.
+	*/
+	serialize() {
+		throw new Error("Method not implemented.");
+	}
+	/** @deprecated Use .invoke() instead. Will be removed in 0.2.0. */
+	async run(input, config) {
+		const inputKeys = this.inputKeys.filter((k) => !this.memory?.memoryKeys.includes(k));
+		const isKeylessInput = inputKeys.length <= 1;
+		if (!isKeylessInput) throw new Error(`Chain ${this._chainType()} expects multiple inputs, cannot use 'run' `);
+		const values = inputKeys.length ? { [inputKeys[0]]: input } : {};
+		const returnValues = await this.call(values, config);
+		const keys = Object.keys(returnValues);
+		if (keys.length === 1) return returnValues[keys[0]];
+		throw new Error("return values have multiple keys, `run` only supported when one key currently");
+	}
+	async _formatValues(values) {
+		const fullValues = { ...values };
+		if (fullValues.timeout && !fullValues.signal) {
+			fullValues.signal = AbortSignal.timeout(fullValues.timeout);
+			delete fullValues.timeout;
+		}
+		if (!(this.memory == null)) {
+			const newValues = await this.memory.loadMemoryVariables(this._selectMemoryInputs(values));
+			for (const [key, value] of Object.entries(newValues)) fullValues[key] = value;
+		}
+		return fullValues;
+	}
+	/**
+	* @deprecated Use .invoke() instead. Will be removed in 0.2.0.
+	*
+	* Run the core logic of this chain and add to output if desired.
+	*
+	* Wraps _call and handles memory.
+	*/
+	async call(values, config, tags) {
+		const parsedConfig = {
+			tags,
+			...parseCallbackConfigArg(config)
+		};
+		return this.invoke(values, parsedConfig);
+	}
+	/**
+	* @deprecated Use .batch() instead. Will be removed in 0.2.0.
+	*
+	* Call the chain on all inputs in the list
+	*/
+	async apply(inputs, config) {
+		return Promise.all(inputs.map(async (i, idx) => this.call(i, config?.[idx])));
+	}
+	/**
+	* Load a chain from a json-like object describing it.
+	*/
+	static async deserialize(data, values = {}) {
+		switch (data._type) {
+			case "llm_chain": {
+				const { LLMChain } = await import("./llm_chain.js");
+				return LLMChain.deserialize(data);
+			}
+			case "sequential_chain": {
+				const { SequentialChain } = await import("./sequential_chain.js");
+				return SequentialChain.deserialize(data);
+			}
+			case "simple_sequential_chain": {
+				const { SimpleSequentialChain } = await import("./sequential_chain.js");
+				return SimpleSequentialChain.deserialize(data);
+			}
+			case "stuff_documents_chain": {
+				const { StuffDocumentsChain } = await import("./combine_docs_chain.js");
+				return StuffDocumentsChain.deserialize(data);
+			}
+			case "map_reduce_documents_chain": {
+				const { MapReduceDocumentsChain } = await import("./combine_docs_chain.js");
+				return MapReduceDocumentsChain.deserialize(data);
+			}
+			case "refine_documents_chain": {
+				const { RefineDocumentsChain } = await import("./combine_docs_chain.js");
+				return RefineDocumentsChain.deserialize(data);
+			}
+			case "vector_db_qa": {
+				const { VectorDBQAChain } = await import("./vector_db_qa.js");
+				return VectorDBQAChain.deserialize(data, values);
+			}
+			case "api_chain": {
+				const { APIChain } = await import("./api/api_chain.js");
+				return APIChain.deserialize(data);
+			}
+			default: throw new Error(`Invalid prompt type in config: ${data._type}`);
+		}
+	}
+};
+
+//#endregion
+export { BaseChain };
+//# sourceMappingURL=base.js.map

package/dist/chains/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","names":["fields?: BaseMemory | ChainInputs","verbose?: boolean","callbacks?: Callbacks","values: ChainValues","input: RunInput","options?: RunnableConfig","outputValues: RunOutput","outputs: Record<string, unknown>","inputs: Record<string, unknown>","input: any","config?: Callbacks | RunnableConfig","values: ChainValues & { signal?: AbortSignal; timeout?: number }","tags?: string[]","inputs: RunInput[]","config?: (Callbacks | RunnableConfig)[]","data: SerializedBaseChain","values: LoadValues"],"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      outputValues = await (fullValues.signal\n        ? (Promise.race([\n            this._call(fullValues as RunInput, runManager, config),\n            new Promise((_, reject) => {\n              fullValues.signal?.addEventListener(\"abort\", () => {\n                reject(new Error(\"AbortError\"));\n              });\n            }),\n          ]) as Promise<RunOutput>)\n        : this._call(fullValues as RunInput, runManager, config));\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 } = await import(\n          \"./combine_docs_chain.js\"\n        );\n        return MapReduceDocumentsChain.deserialize(data);\n      }\n      case \"refine_documents_chain\": {\n        const { RefineDocumentsChain } = await import(\n          \"./combine_docs_chain.js\"\n        );\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;EAAC;CAClD;CAED,YACEA,QAEAC,SAEAC,WACA;AACA,MACE,UAAU,WAAW,KACrB,OAAO,WAAW,YAClB,EAAE,iBAAiB,SACnB;GAEA,MAAM,EAAE,QAAQ,gBAAiB,GAAG,MAAM,GAAG;GAC7C,MAAM;IAAE,GAAG;IAAM,WAAW,mBAAmB,KAAK;GAAW,EAAC;GAChE,KAAK,SAAS;EACf,OAAM;GAEL,MAAM;IAAE;IAAS;GAAW,EAAC;GAC7B,KAAK,SAAS;EACf;CACF;;CAGD,oBAAoBC,QAAkC;EACpD,MAAM,kBAAkB,EAAE,GAAG,OAAQ;AACrC,MAAI,YAAY,iBACd,OAAO,gBAAgB;AAEzB,MAAI,aAAa,iBACf,OAAO,gBAAgB;AAEzB,SAAO;CACR;;;;;;;CAQD,MAAM,OAAOC,OAAiBC,SAA8C;EAC1E,MAAM,SAAS,aAAa,QAAQ;EACpC,MAAM,aAAa,MAAM,KAAK,cAAc,MAAM;EAClD,MAAM,mBAAmB,MAAM,gBAAgB,UAC7C,QAAQ,WACR,KAAK,WACL,QAAQ,MACR,KAAK,MACL,QAAQ,UACR,KAAK,UACL,EAAE,SAAS,KAAK,QAAS,EAC1B;EACD,MAAM,aAAa,MAAM,kBAAkB,iBACzC,KAAK,QAAQ,EACb,YACA,QACA,QACA,QACA,QACA,QAAQ,QACT;EACD,IAAIC;AACJ,MAAI;GACF,eAAe,OAAO,WAAW,SAC5B,QAAQ,KAAK,CACZ,KAAK,MAAM,YAAwB,YAAY,OAAO,EACtD,IAAI,QAAQ,CAAC,GAAG,WAAW;IACzB,WAAW,QAAQ,iBAAiB,SAAS,MAAM;KACjD,uBAAO,IAAI,MAAM,cAAc;IAChC,EAAC;GACH,EACF,EAAC,GACF,KAAK,MAAM,YAAwB,YAAY,OAAO;EAC3D,SAAQ,GAAG;GACV,MAAM,YAAY,iBAAiB,EAAE;AACrC,SAAM;EACP;AACD,MAAI,EAAE,KAAK,UAAU,OACnB,MAAM,KAAK,OAAO,YAChB,KAAK,oBAAoB,MAAM,EAC/B,aACD;EAEH,MAAM,YAAY,eAAe,aAAa;EAE9C,OAAO,eAAe,cAAc,SAAS;GAC3C,OAAO,aAAa,EAAE,OAAO,YAAY,MAAO,IAAG;GACnD,cAAc;EACf,EAAC;AACF,SAAO;CACR;CAED,AAAQ,iBAAiBC,SAAwC;EAC/D,MAAM,cAAc,KAAK,WAAW,OAAO,CAAC,MAAM,EAAE,KAAK,SAAS;AAClE,MAAI,YAAY,OACd,OAAM,IAAI,MACR,CAAC,qBAAqB,EAAE,YAAY,KAClC,KACD,CAAC,YAAY,EAAE,KAAK,YAAY,EAAE;CAGxC;CAED,MAAM,YACJC,QACAD,SACA,oBAAoB,OACpB;EACA,KAAK,iBAAiB,QAAQ;AAC9B,MAAI,KAAK,QACP,MAAM,KAAK,OAAO,YAAY,QAAQ,QAAQ;AAEhD,MAAI,kBACF,QAAO;AAET,SAAO;GAAE,GAAG;GAAQ,GAAG;EAAS;CACjC;;;;CAmBD,YAAiC;AAC/B,QAAM,IAAI,MAAM;CACjB;;CAOD,MAAM,IAEJE,OACAC,QACiB;EACjB,MAAM,YAAY,KAAK,UAAU,OAC/B,CAAC,MAAM,CAAC,KAAK,QAAQ,WAAW,SAAS,EAAE,CAC5C;EACD,MAAM,iBAAiB,UAAU,UAAU;AAC3C,MAAI,CAAC,eACH,OAAM,IAAI,MACR,CAAC,MAAM,EAAE,KAAK,YAAY,CAAC,2CAA2C,CAAC;EAI3E,MAAM,SAAS,UAAU,SAAS,GAAG,UAAU,KAAK,MAAO,IAAI,CAAE;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;CAEH;CAED,MAAgB,cACdC,QACA;EACA,MAAM,aAAa,EAAE,GAAG,OAAQ;AAChC,MAAI,WAAW,WAAW,CAAC,WAAW,QAAQ;GAC5C,WAAW,SAAS,YAAY,QAAQ,WAAW,QAAQ;GAC3D,OAAO,WAAW;EACnB;AACD,MAAI,EAAE,KAAK,UAAU,OAAO;GAC1B,MAAM,YAAY,MAAM,KAAK,OAAO,oBAClC,KAAK,oBAAoB,OAAO,CACjC;AACD,QAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,UAAU,EAClD,WAAW,OAAO;EAErB;AACD,SAAO;CACR;;;;;;;;CASD,MAAM,KACJA,QACAD,QAEAE,MACoB;EACpB,MAAM,eAAe;GAAE;GAAM,GAAG,uBAAuB,OAAO;EAAE;AAChE,SAAO,KAAK,OAAO,QAAoB,aAAa;CACrD;;;;;;CAOD,MAAM,MACJC,QACAC,QACsB;AACtB,SAAO,QAAQ,IACb,OAAO,IAAI,OAAO,GAAG,QAAQ,KAAK,KAAK,GAAG,SAAS,KAAK,CAAC,CAC1D;CACF;;;;CAKD,aAAa,YACXC,MACAC,SAAqB,CAAE,GACH;AACpB,UAAQ,KAAK,OAAb;GACE,KAAK,aAAa;IAChB,MAAM,EAAE,UAAU,GAAG,MAAM,OAAO;AAClC,WAAO,SAAS,YAAY,KAAK;GAClC;GACD,KAAK,oBAAoB;IACvB,MAAM,EAAE,iBAAiB,GAAG,MAAM,OAAO;AACzC,WAAO,gBAAgB,YAAY,KAAK;GACzC;GACD,KAAK,2BAA2B;IAC9B,MAAM,EAAE,uBAAuB,GAAG,MAAM,OAAO;AAC/C,WAAO,sBAAsB,YAAY,KAAK;GAC/C;GACD,KAAK,yBAAyB;IAC5B,MAAM,EAAE,qBAAqB,GAAG,MAAM,OAAO;AAC7C,WAAO,oBAAoB,YAAY,KAAK;GAC7C;GACD,KAAK,8BAA8B;IACjC,MAAM,EAAE,yBAAyB,GAAG,MAAM,OACxC;AAEF,WAAO,wBAAwB,YAAY,KAAK;GACjD;GACD,KAAK,0BAA0B;IAC7B,MAAM,EAAE,sBAAsB,GAAG,MAAM,OACrC;AAEF,WAAO,qBAAqB,YAAY,KAAK;GAC9C;GACD,KAAK,gBAAgB;IACnB,MAAM,EAAE,iBAAiB,GAAG,MAAM,OAAO;AACzC,WAAO,gBAAgB,YAAY,MAAM,OAAO;GACjD;GACD,KAAK,aAAa;IAChB,MAAM,EAAE,UAAU,GAAG,MAAM,OAAO;AAClC,WAAO,SAAS,YAAY,KAAK;GAClC;GACD,QACE,OAAM,IAAI,MACR,CAAC,+BAA+B,EAC7B,KAA6B,OAC9B;EAEP;CACF;AACF"}

package/dist/chains/chat_vector_db_chain.cjs

@@ -0,0 +1,130 @@
+const require_rolldown_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');
+const __langchain_core_prompts = require_rolldown_runtime.__toESM(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.
+
+Chat History:
+{chat_history}
+Follow Up Input: {question}
+Standalone question:`;
+const 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.
+
+{context}
+
+Question: {question}
+Helpful Answer:`;
+/** @deprecated use `ConversationalRetrievalQAChain` instead. */
+var ChatVectorDBQAChain = class ChatVectorDBQAChain extends require_base.BaseChain {
+	k = 4;
+	inputKey = "question";
+	chatHistoryKey = "chat_history";
+	get inputKeys() {
+		return [this.inputKey, this.chatHistoryKey];
+	}
+	outputKey = "result";
+	get outputKeys() {
+		return [this.outputKey];
+	}
+	vectorstore;
+	combineDocumentsChain;
+	questionGeneratorChain;
+	returnSourceDocuments = false;
+	constructor(fields) {
+		super(fields);
+		this.vectorstore = fields.vectorstore;
+		this.combineDocumentsChain = fields.combineDocumentsChain;
+		this.questionGeneratorChain = fields.questionGeneratorChain;
+		this.inputKey = fields.inputKey ?? this.inputKey;
+		this.outputKey = fields.outputKey ?? this.outputKey;
+		this.k = fields.k ?? this.k;
+		this.returnSourceDocuments = fields.returnSourceDocuments ?? this.returnSourceDocuments;
+	}
+	/** @ignore */
+	async _call(values, runManager) {
+		if (!(this.inputKey in values)) throw new Error(`Question key ${this.inputKey} not found.`);
+		if (!(this.chatHistoryKey in values)) throw new Error(`chat history key ${this.inputKey} not found.`);
+		const question = values[this.inputKey];
+		const chatHistory = values[this.chatHistoryKey];
+		let newQuestion = question;
+		if (chatHistory.length > 0) {
+			const result$1 = await this.questionGeneratorChain.call({
+				question,
+				chat_history: chatHistory
+			}, runManager?.getChild("question_generator"));
+			const keys = Object.keys(result$1);
+			console.log("_call", values, keys);
+			if (keys.length === 1) newQuestion = result$1[keys[0]];
+			else throw new Error("Return from llm chain has multiple values, only single values supported.");
+		}
+		const docs = await this.vectorstore.similaritySearch(newQuestion, this.k, void 0, runManager?.getChild("vectorstore"));
+		const inputs = {
+			question: newQuestion,
+			input_documents: docs,
+			chat_history: chatHistory
+		};
+		const result = await this.combineDocumentsChain.call(inputs, runManager?.getChild("combine_documents"));
+		if (this.returnSourceDocuments) return {
+			...result,
+			sourceDocuments: docs
+		};
+		return result;
+	}
+	_chainType() {
+		return "chat-vector-db";
+	}
+	static async deserialize(data, values) {
+		if (!("vectorstore" in values)) throw new Error(`Need to pass in a vectorstore to deserialize VectorDBQAChain`);
+		const { vectorstore } = values;
+		return new ChatVectorDBQAChain({
+			combineDocumentsChain: await require_base.BaseChain.deserialize(data.combine_documents_chain),
+			questionGeneratorChain: await require_llm_chain.LLMChain.deserialize(data.question_generator),
+			k: data.k,
+			vectorstore
+		});
+	}
+	serialize() {
+		return {
+			_type: this._chainType(),
+			combine_documents_chain: this.combineDocumentsChain.serialize(),
+			question_generator: this.questionGeneratorChain.serialize(),
+			k: this.k
+		};
+	}
+	/**
+	* Creates an instance of ChatVectorDBQAChain using a BaseLanguageModel
+	* and other options.
+	* @param llm Instance of BaseLanguageModel used to generate a new question.
+	* @param vectorstore Instance of VectorStore used for vector operations.
+	* @param options (Optional) Additional options for creating the ChatVectorDBQAChain instance.
+	* @returns New instance of ChatVectorDBQAChain.
+	*/
+	static fromLLM(llm, vectorstore, options = {}) {
+		const { questionGeneratorTemplate, qaTemplate, verbose,...rest } = options;
+		const question_generator_prompt = __langchain_core_prompts.PromptTemplate.fromTemplate(questionGeneratorTemplate || question_generator_template);
+		const qa_prompt = __langchain_core_prompts.PromptTemplate.fromTemplate(qaTemplate || qa_template);
+		const qaChain = require_load.loadQAStuffChain(llm, {
+			prompt: qa_prompt,
+			verbose
+		});
+		const questionGeneratorChain = new require_llm_chain.LLMChain({
+			prompt: question_generator_prompt,
+			llm,
+			verbose
+		});
+		const instance = new this({
+			vectorstore,
+			combineDocumentsChain: qaChain,
+			questionGeneratorChain,
+			...rest
+		});
+		return instance;
+	}
+};
+
+//#endregion
+exports.ChatVectorDBQAChain = ChatVectorDBQAChain;
+//# sourceMappingURL=chat_vector_db_chain.cjs.map

package/dist/chains/chat_vector_db_chain.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_vector_db_chain.cjs","names":["BaseChain","fields: ChatVectorDBQAChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","question: string","chatHistory: string","result","data: SerializedChatVectorDBQAChain","values: LoadValues","LLMChain","llm: BaseLanguageModelInterface","vectorstore: VectorStoreInterface","options: {\n      inputKey?: string;\n      outputKey?: string;\n      k?: number;\n      returnSourceDocuments?: boolean;\n      questionGeneratorTemplate?: string;\n      qaTemplate?: string;\n      verbose?: boolean;\n    }","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,CAAC;;;;;oBAKjB,CAAC;AAErB,MAAM,cAAc,CAAC;;;;;eAKN,CAAC;;AAgBhB,IAAa,sBAAb,MAAa,4BACHA,uBAEV;CACE,IAAI;CAEJ,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,cAAe;CAC5C;CAED,YAAY;CAEZ,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,SAAU;CACxB;CAED;CAEA;CAEA;CAEA,wBAAwB;CAExB,YAAYC,QAAkC;EAC5C,MAAM,OAAO;EACb,KAAK,cAAc,OAAO;EAC1B,KAAK,wBAAwB,OAAO;EACpC,KAAK,yBAAyB,OAAO;EACrC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,IAAI,OAAO,KAAK,KAAK;EAC1B,KAAK,wBACH,OAAO,yBAAyB,KAAK;CACxC;;CAGD,MAAM,MACJC,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;AAE5D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,CAAC,iBAAiB,EAAE,KAAK,SAAS,WAAW,CAAC;EAEhE,MAAMC,WAAmB,OAAO,KAAK;EACrC,MAAMC,cAAsB,OAAO,KAAK;EACxC,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAMC,WAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;GACf,GACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAKA,SAAO;GAChC,QAAQ,IAAI,SAAS,QAAQ,KAAK;AAClC,OAAI,KAAK,WAAW,GAClB,cAAcA,SAAO,KAAK;OAE1B,OAAM,IAAI,MACR;EAGL;EACD,MAAM,OAAO,MAAM,KAAK,YAAY,iBAClC,aACA,KAAK,GACL,QACA,YAAY,SAAS,cAAc,CACpC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;EACf;EACD,MAAM,SAAS,MAAM,KAAK,sBAAsB,KAC9C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,QAAO;GACL,GAAG;GACH,iBAAiB;EAClB;AAEH,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YACXC,MACAC,QACA;AACA,MAAI,EAAE,iBAAiB,QACrB,OAAM,IAAI,MACR,CAAC,4DAA4D,CAAC;EAGlE,MAAM,EAAE,aAAa,GAAG;AAExB,SAAO,IAAI,oBAAoB;GAC7B,uBAAuB,MAAMR,uBAAU,YACrC,KAAK,wBACN;GACD,wBAAwB,MAAMS,2BAAS,YACrC,KAAK,mBACN;GACD,GAAG,KAAK;GACR;EACD;CACF;CAED,YAA2C;AACzC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,yBAAyB,KAAK,sBAAsB,WAAW;GAC/D,oBAAoB,KAAK,uBAAuB,WAAW;GAC3D,GAAG,KAAK;EACT;CACF;;;;;;;;;CAUD,OAAO,QACLC,KACAC,aACAC,UAQI,CAAE,GACe;EACrB,MAAM,EAAE,2BAA2B,YAAY,QAAS,GAAG,MAAM,GAAG;EACpE,MAAM,4BAA4BC,wCAAe,aAC/C,6BAA6B,4BAC9B;EACD,MAAM,YAAYA,wCAAe,aAAa,cAAc,YAAY;EAExE,MAAM,UAAUC,8BAAiB,KAAK;GAAE,QAAQ;GAAW;EAAS,EAAC;EACrE,MAAM,yBAAyB,IAAIL,2BAAS;GAC1C,QAAQ;GACR;GACA;EACD;EACD,MAAM,WAAW,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA,GAAG;EACJ;AACD,SAAO;CACR;AACF"}