@langchain/classic 1.0.17 → 1.0.18

package/dist/chains/router/multi_route.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"multi_route.d.ts","names":["ChainValues","CallbackManagerForChainRun","Callbacks","BaseChain","ChainInputs","Inputs","Route","MultiRouteChainInput","RouterChain","Promise","MultiRouteChain"],"sources":["../../../src/chains/router/multi_route.d.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\n/**\n * A type that represents the inputs for the MultiRouteChain. It is a\n * recursive type that can contain nested objects, arrays, strings, and\n * numbers.\n */\ntype Inputs = {\n    [key: string]: Inputs | Inputs[] | string | string[] | number | number[];\n};\n/**\n * An interface that represents the route returned by the RouterChain. It\n * includes optional fields for the destination and nextInputs.\n */\nexport interface Route {\n    destination?: string;\n    nextInputs: {\n        [key: string]: Inputs;\n    };\n}\n/**\n * An interface that extends the ChainInputs interface and adds additional\n * properties for the routerChain, destinationChains, defaultChain, and\n * silentErrors. It represents the input expected by the MultiRouteChain\n * class.\n */\nexport interface MultiRouteChainInput extends ChainInputs {\n    routerChain: RouterChain;\n    destinationChains: {\n        [name: string]: BaseChain;\n    };\n    defaultChain: BaseChain;\n    silentErrors?: boolean;\n}\n/**\n * A class that represents a router chain. It\n * extends the BaseChain class and provides functionality for routing\n * inputs to different chains.\n */\nexport declare abstract class RouterChain extends BaseChain {\n    get outputKeys(): string[];\n    route(inputs: ChainValues, callbacks?: Callbacks): Promise<Route>;\n}\n/**\n * A class that represents a multi-route chain.\n * It extends the BaseChain class and provides functionality for routing\n * inputs to different chains based on a router chain.\n */\nexport declare class MultiRouteChain extends BaseChain {\n    static lc_name(): string;\n    routerChain: RouterChain;\n    destinationChains: {\n        [name: string]: BaseChain;\n    };\n    defaultChain: BaseChain;\n    silentErrors: boolean;\n    constructor(fields: MultiRouteChainInput);\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): string;\n}\nexport {};\n//# sourceMappingURL=multi_route.d.ts.map"],"mappings":";;;;;;;;AAE8D;AAa9D;AAYA;KAnBKK,MAAAA,GAoBYG;EAEOL,CAAAA,GAAAA,EAAAA,MAAAA,CAAAA,EArBLE,MAqBKF,GArBIE,MAqBJF,EAAAA,GAAAA,MAAAA,GAAAA,MAAAA,EAAAA,GAAAA,MAAAA,GAAAA,MAAAA,EAAAA;CAENA;;AALuC;AAazD;;AAE2CD,UA3B1BI,KAAAA,CA2B0BJ;EAAoBI,WAAAA,CAAAA,EAAAA,MAAAA;EAARG,UAAAA,EAAAA;IAFLN,CAAAA,GAAAA,EAAAA,MAAAA,CAAAA,EAtB3BE,MAsB2BF;EAAS,CAAA;AAS3D;;;;;;;AAWiFH,UAjChEO,oBAAAA,SAA6BH,WAiCmCJ,CAAAA;EAARS,WAAAA,EAhCxDD,WAgCwDC;EAX5BN,iBAAAA,EAAAA;IAAS,CAAA,IAAA,EAAA,MAAA,CAAA,EAnB9BA,SAmB8B;;gBAjBpCA;;;;;;;;uBAQYK,WAAAA,SAAoBL,SAAAA;;gBAEhCH,yBAAyBE,YAAYO,QAAQH;;;;;;;cAO1CI,eAAAA,SAAwBP,SAAAA;;eAE5BK;;oBAEOL;;gBAENA;;sBAEMI;;;gBAGNP,0BAA0BC,6BAA6BQ,QAAQT"}
+{"version":3,"file":"multi_route.d.ts","names":[],"sources":["../../../src/chains/router/multi_route.ts"],"mappings":";;;;;;;AAK8D;;;KAOzD,MAAA;EAAA,CACF,GAAA,WAAc,MAAA,GAAS,MAAA;AAAA;;;;AAO1B;UAAiB,KAAA;EACf,WAAA;EACA,UAAA;IAAA,CAAe,GAAA,WAAc,MAAA;EAAA;AAAA;;;;AAS/B;;;UAAiB,oBAAA,SAA6B,WAAA;EAC5C,WAAA,EAAa,WAAA;EACb,iBAAA;IAAA,CAAsB,IAAA,WAAe,SAAA;EAAA;EACrC,YAAA,EAAc,SAAA;EACd,YAAA;AAAA;;;;;;uBAQoB,WAAA,SAAoB,SAAA;EAAA,IACpC,UAAA,CAAA;EAIE,KAAA,CAAM,MAAA,EAAQ,WAAA,EAAa,SAAA,GAAY,SAAA,GAAY,OAAA,CAAQ,KAAA;AAAA;AALnE;;;;;AAAA,cAmBa,eAAA,SAAwB,SAAA;EAAA,OAC5B,OAAA,CAAA;EAIP,WAAA,EAAa,WAAA;EAEb,iBAAA;IAAA,CAAsB,IAAA,WAAe,SAAA;EAAA;EAErC,YAAA,EAAc,SAAA;EAEd,YAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,oBAAA;EAAA,IAQhB,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIE,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAoCX,UAAA,CAAA;AAAA"}

package/dist/chains/router/multi_route.js.map

@@ -1 +1 @@
-{"version":3,"file":"multi_route.js","names":["inputs: ChainValues","callbacks?: Callbacks","fields: MultiRouteChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun"],"sources":["../../../src/chains/router/multi_route.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport {\n  CallbackManagerForChainRun,\n  Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\n\n/**\n * A type that represents the inputs for the MultiRouteChain. It is a\n * recursive type that can contain nested objects, arrays, strings, and\n * numbers.\n */\ntype Inputs = {\n  [key: string]: Inputs | Inputs[] | string | string[] | number | number[];\n};\n\n/**\n * An interface that represents the route returned by the RouterChain. It\n * includes optional fields for the destination and nextInputs.\n */\nexport interface Route {\n  destination?: string;\n  nextInputs: { [key: string]: Inputs };\n}\n\n/**\n * An interface that extends the ChainInputs interface and adds additional\n * properties for the routerChain, destinationChains, defaultChain, and\n * silentErrors. It represents the input expected by the MultiRouteChain\n * class.\n */\nexport interface MultiRouteChainInput extends ChainInputs {\n  routerChain: RouterChain;\n  destinationChains: { [name: string]: BaseChain };\n  defaultChain: BaseChain;\n  silentErrors?: boolean;\n}\n\n/**\n * A class that represents a router chain. It\n * extends the BaseChain class and provides functionality for routing\n * inputs to different chains.\n */\nexport abstract class RouterChain extends BaseChain {\n  get outputKeys(): string[] {\n    return [\"destination\", \"next_inputs\"];\n  }\n\n  async route(inputs: ChainValues, callbacks?: Callbacks): Promise<Route> {\n    const result = await this.call(inputs, callbacks);\n    return {\n      destination: result.destination,\n      nextInputs: result.next_inputs,\n    };\n  }\n}\n\n/**\n * A class that represents a multi-route chain.\n * It extends the BaseChain class and provides functionality for routing\n * inputs to different chains based on a router chain.\n */\nexport class MultiRouteChain extends BaseChain {\n  static lc_name() {\n    return \"MultiRouteChain\";\n  }\n\n  routerChain: RouterChain;\n\n  destinationChains: { [name: string]: BaseChain };\n\n  defaultChain: BaseChain;\n\n  silentErrors = false;\n\n  constructor(fields: MultiRouteChainInput) {\n    super(fields);\n    this.routerChain = fields.routerChain;\n    this.destinationChains = fields.destinationChains;\n    this.defaultChain = fields.defaultChain;\n    this.silentErrors = fields.silentErrors ?? this.silentErrors;\n  }\n\n  get inputKeys(): string[] {\n    return this.routerChain.inputKeys;\n  }\n\n  get outputKeys(): string[] {\n    return [];\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const { destination, nextInputs } = await this.routerChain.route(\n      values,\n      runManager?.getChild()\n    );\n    await runManager?.handleText(\n      `${destination}: ${JSON.stringify(nextInputs)}`\n    );\n    if (!destination) {\n      return this.defaultChain\n        .call(nextInputs, runManager?.getChild())\n        .catch((err) => {\n          throw new Error(`Error in default chain: ${err}`);\n        });\n    }\n    if (destination in this.destinationChains) {\n      return this.destinationChains[destination]\n        .call(nextInputs, runManager?.getChild())\n        .catch((err) => {\n          throw new Error(`Error in ${destination} chain: ${err}`);\n        });\n    }\n    if (this.silentErrors) {\n      return this.defaultChain\n        .call(nextInputs, runManager?.getChild())\n        .catch((err) => {\n          throw new Error(`Error in default chain: ${err}`);\n        });\n    }\n    throw new Error(\n      `Destination ${destination} not found in destination chains with keys ${Object.keys(\n        this.destinationChains\n      )}`\n    );\n  }\n\n  _chainType(): string {\n    return \"multi_route_chain\";\n  }\n}\n"],"mappings":";;;;;;;;AA2CA,IAAsB,cAAtB,cAA0C,UAAU;CAClD,IAAI,aAAuB;AACzB,SAAO,CAAC,eAAe,aAAc;CACtC;CAED,MAAM,MAAMA,QAAqBC,WAAuC;EACtE,MAAM,SAAS,MAAM,KAAK,KAAK,QAAQ,UAAU;AACjD,SAAO;GACL,aAAa,OAAO;GACpB,YAAY,OAAO;EACpB;CACF;AACF;;;;;;AAOD,IAAa,kBAAb,cAAqC,UAAU;CAC7C,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA,eAAe;CAEf,YAAYC,QAA8B;EACxC,MAAM,OAAO;EACb,KAAK,cAAc,OAAO;EAC1B,KAAK,oBAAoB,OAAO;EAChC,KAAK,eAAe,OAAO;EAC3B,KAAK,eAAe,OAAO,gBAAgB,KAAK;CACjD;CAED,IAAI,YAAsB;AACxB,SAAO,KAAK,YAAY;CACzB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAE;CACV;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,MAAM,EAAE,aAAa,YAAY,GAAG,MAAM,KAAK,YAAY,MACzD,QACA,YAAY,UAAU,CACvB;EACD,MAAM,YAAY,WAChB,GAAG,YAAY,EAAE,EAAE,KAAK,UAAU,WAAW,EAAE,CAChD;AACD,MAAI,CAAC,YACH,QAAO,KAAK,aACT,KAAK,YAAY,YAAY,UAAU,CAAC,CACxC,MAAM,CAAC,QAAQ;AACd,SAAM,IAAI,MAAM,CAAC,wBAAwB,EAAE,KAAK;EACjD,EAAC;AAEN,MAAI,eAAe,KAAK,kBACtB,QAAO,KAAK,kBAAkB,aAC3B,KAAK,YAAY,YAAY,UAAU,CAAC,CACxC,MAAM,CAAC,QAAQ;AACd,SAAM,IAAI,MAAM,CAAC,SAAS,EAAE,YAAY,QAAQ,EAAE,KAAK;EACxD,EAAC;AAEN,MAAI,KAAK,aACP,QAAO,KAAK,aACT,KAAK,YAAY,YAAY,UAAU,CAAC,CACxC,MAAM,CAAC,QAAQ;AACd,SAAM,IAAI,MAAM,CAAC,wBAAwB,EAAE,KAAK;EACjD,EAAC;AAEN,QAAM,IAAI,MACR,CAAC,YAAY,EAAE,YAAY,2CAA2C,EAAE,OAAO,KAC7E,KAAK,kBACN,EAAE;CAEN;CAED,aAAqB;AACnB,SAAO;CACR;AACF"}
+{"version":3,"file":"multi_route.js","names":[],"sources":["../../../src/chains/router/multi_route.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport {\n  CallbackManagerForChainRun,\n  Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"../../chains/base.js\";\n\n/**\n * A type that represents the inputs for the MultiRouteChain. It is a\n * recursive type that can contain nested objects, arrays, strings, and\n * numbers.\n */\ntype Inputs = {\n  [key: string]: Inputs | Inputs[] | string | string[] | number | number[];\n};\n\n/**\n * An interface that represents the route returned by the RouterChain. It\n * includes optional fields for the destination and nextInputs.\n */\nexport interface Route {\n  destination?: string;\n  nextInputs: { [key: string]: Inputs };\n}\n\n/**\n * An interface that extends the ChainInputs interface and adds additional\n * properties for the routerChain, destinationChains, defaultChain, and\n * silentErrors. It represents the input expected by the MultiRouteChain\n * class.\n */\nexport interface MultiRouteChainInput extends ChainInputs {\n  routerChain: RouterChain;\n  destinationChains: { [name: string]: BaseChain };\n  defaultChain: BaseChain;\n  silentErrors?: boolean;\n}\n\n/**\n * A class that represents a router chain. It\n * extends the BaseChain class and provides functionality for routing\n * inputs to different chains.\n */\nexport abstract class RouterChain extends BaseChain {\n  get outputKeys(): string[] {\n    return [\"destination\", \"next_inputs\"];\n  }\n\n  async route(inputs: ChainValues, callbacks?: Callbacks): Promise<Route> {\n    const result = await this.call(inputs, callbacks);\n    return {\n      destination: result.destination,\n      nextInputs: result.next_inputs,\n    };\n  }\n}\n\n/**\n * A class that represents a multi-route chain.\n * It extends the BaseChain class and provides functionality for routing\n * inputs to different chains based on a router chain.\n */\nexport class MultiRouteChain extends BaseChain {\n  static lc_name() {\n    return \"MultiRouteChain\";\n  }\n\n  routerChain: RouterChain;\n\n  destinationChains: { [name: string]: BaseChain };\n\n  defaultChain: BaseChain;\n\n  silentErrors = false;\n\n  constructor(fields: MultiRouteChainInput) {\n    super(fields);\n    this.routerChain = fields.routerChain;\n    this.destinationChains = fields.destinationChains;\n    this.defaultChain = fields.defaultChain;\n    this.silentErrors = fields.silentErrors ?? this.silentErrors;\n  }\n\n  get inputKeys(): string[] {\n    return this.routerChain.inputKeys;\n  }\n\n  get outputKeys(): string[] {\n    return [];\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const { destination, nextInputs } = await this.routerChain.route(\n      values,\n      runManager?.getChild()\n    );\n    await runManager?.handleText(\n      `${destination}: ${JSON.stringify(nextInputs)}`\n    );\n    if (!destination) {\n      return this.defaultChain\n        .call(nextInputs, runManager?.getChild())\n        .catch((err) => {\n          throw new Error(`Error in default chain: ${err}`);\n        });\n    }\n    if (destination in this.destinationChains) {\n      return this.destinationChains[destination]\n        .call(nextInputs, runManager?.getChild())\n        .catch((err) => {\n          throw new Error(`Error in ${destination} chain: ${err}`);\n        });\n    }\n    if (this.silentErrors) {\n      return this.defaultChain\n        .call(nextInputs, runManager?.getChild())\n        .catch((err) => {\n          throw new Error(`Error in default chain: ${err}`);\n        });\n    }\n    throw new Error(\n      `Destination ${destination} not found in destination chains with keys ${Object.keys(\n        this.destinationChains\n      )}`\n    );\n  }\n\n  _chainType(): string {\n    return \"multi_route_chain\";\n  }\n}\n"],"mappings":";;;;;;;;AA2CA,IAAsB,cAAtB,cAA0C,UAAU;CAClD,IAAI,aAAuB;AACzB,SAAO,CAAC,eAAe,cAAc;;CAGvC,MAAM,MAAM,QAAqB,WAAuC;EACtE,MAAM,SAAS,MAAM,KAAK,KAAK,QAAQ,UAAU;AACjD,SAAO;GACL,aAAa,OAAO;GACpB,YAAY,OAAO;GACpB;;;;;;;;AASL,IAAa,kBAAb,cAAqC,UAAU;CAC7C,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA;CAEA;CAEA,eAAe;CAEf,YAAY,QAA8B;AACxC,QAAM,OAAO;AACb,OAAK,cAAc,OAAO;AAC1B,OAAK,oBAAoB,OAAO;AAChC,OAAK,eAAe,OAAO;AAC3B,OAAK,eAAe,OAAO,gBAAgB,KAAK;;CAGlD,IAAI,YAAsB;AACxB,SAAO,KAAK,YAAY;;CAG1B,IAAI,aAAuB;AACzB,SAAO,EAAE;;CAGX,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,EAAE,aAAa,eAAe,MAAM,KAAK,YAAY,MACzD,QACA,YAAY,UAAU,CACvB;AACD,QAAM,YAAY,WAChB,GAAG,YAAY,IAAI,KAAK,UAAU,WAAW,GAC9C;AACD,MAAI,CAAC,YACH,QAAO,KAAK,aACT,KAAK,YAAY,YAAY,UAAU,CAAC,CACxC,OAAO,QAAQ;AACd,SAAM,IAAI,MAAM,2BAA2B,MAAM;IACjD;AAEN,MAAI,eAAe,KAAK,kBACtB,QAAO,KAAK,kBAAkB,aAC3B,KAAK,YAAY,YAAY,UAAU,CAAC,CACxC,OAAO,QAAQ;AACd,SAAM,IAAI,MAAM,YAAY,YAAY,UAAU,MAAM;IACxD;AAEN,MAAI,KAAK,aACP,QAAO,KAAK,aACT,KAAK,YAAY,YAAY,UAAU,CAAC,CACxC,OAAO,QAAQ;AACd,SAAM,IAAI,MAAM,2BAA2B,MAAM;IACjD;AAEN,QAAM,IAAI,MACR,eAAe,YAAY,6CAA6C,OAAO,KAC7E,KAAK,kBACN,GACF;;CAGH,aAAqB;AACnB,SAAO"}

package/dist/chains/router/utils.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"utils.cjs","names":["zipped: T[]","zippedElement: T[keyof T][]"],"sources":["../../../src/chains/router/utils.ts"],"sourcesContent":["// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function zipEntries<T extends any[]>(\n  ...arrays: {\n    [P in keyof T]: T[P][];\n  }\n): T[] {\n  // Check for empty input\n  if (arrays.length === 0) {\n    return [];\n  }\n\n  // Find the length of the first input array\n  const firstArrayLength = arrays[0].length;\n\n  // Ensure all input arrays have the same length\n  for (const array of arrays) {\n    if (array.length !== firstArrayLength) {\n      throw new Error(\"All input arrays must have the same length.\");\n    }\n  }\n\n  // Create an empty array to store the zipped arrays\n  const zipped: T[] = [];\n\n  // Iterate through each element of the first input array\n  for (let i = 0; i < firstArrayLength; i += 1) {\n    // Create an array to store the zipped elements at the current index\n    const zippedElement: T[keyof T][] = [];\n\n    // Iterate through each input array\n    for (const array of arrays) {\n      // Add the element at the current index to the zipped element array\n      zippedElement.push(array[i]);\n    }\n\n    // Add the zipped element array to the zipped array\n    zipped.push(zippedElement as T);\n  }\n\n  return zipped;\n}\n"],"mappings":";;AACA,SAAgB,WACd,GAAG,QAGE;AAEL,KAAI,OAAO,WAAW,EACpB,QAAO,CAAE;CAIX,MAAM,mBAAmB,OAAO,GAAG;AAGnC,MAAK,MAAM,SAAS,OAClB,KAAI,MAAM,WAAW,iBACnB,OAAM,IAAI,MAAM;CAKpB,MAAMA,SAAc,CAAE;AAGtB,MAAK,IAAI,IAAI,GAAG,IAAI,kBAAkB,KAAK,GAAG;EAE5C,MAAMC,gBAA8B,CAAE;AAGtC,OAAK,MAAM,SAAS,QAElB,cAAc,KAAK,MAAM,GAAG;EAI9B,OAAO,KAAK,cAAmB;CAChC;AAED,QAAO;AACR"}
+{"version":3,"file":"utils.cjs","names":[],"sources":["../../../src/chains/router/utils.ts"],"sourcesContent":["// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function zipEntries<T extends any[]>(\n  ...arrays: {\n    [P in keyof T]: T[P][];\n  }\n): T[] {\n  // Check for empty input\n  if (arrays.length === 0) {\n    return [];\n  }\n\n  // Find the length of the first input array\n  const firstArrayLength = arrays[0].length;\n\n  // Ensure all input arrays have the same length\n  for (const array of arrays) {\n    if (array.length !== firstArrayLength) {\n      throw new Error(\"All input arrays must have the same length.\");\n    }\n  }\n\n  // Create an empty array to store the zipped arrays\n  const zipped: T[] = [];\n\n  // Iterate through each element of the first input array\n  for (let i = 0; i < firstArrayLength; i += 1) {\n    // Create an array to store the zipped elements at the current index\n    const zippedElement: T[keyof T][] = [];\n\n    // Iterate through each input array\n    for (const array of arrays) {\n      // Add the element at the current index to the zipped element array\n      zippedElement.push(array[i]);\n    }\n\n    // Add the zipped element array to the zipped array\n    zipped.push(zippedElement as T);\n  }\n\n  return zipped;\n}\n"],"mappings":";;AACA,SAAgB,WACd,GAAG,QAGE;AAEL,KAAI,OAAO,WAAW,EACpB,QAAO,EAAE;CAIX,MAAM,mBAAmB,OAAO,GAAG;AAGnC,MAAK,MAAM,SAAS,OAClB,KAAI,MAAM,WAAW,iBACnB,OAAM,IAAI,MAAM,8CAA8C;CAKlE,MAAM,SAAc,EAAE;AAGtB,MAAK,IAAI,IAAI,GAAG,IAAI,kBAAkB,KAAK,GAAG;EAE5C,MAAM,gBAA8B,EAAE;AAGtC,OAAK,MAAM,SAAS,OAElB,eAAc,KAAK,MAAM,GAAG;AAI9B,SAAO,KAAK,cAAmB;;AAGjC,QAAO"}

package/dist/chains/router/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","names":["zipped: T[]","zippedElement: T[keyof T][]"],"sources":["../../../src/chains/router/utils.ts"],"sourcesContent":["// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function zipEntries<T extends any[]>(\n  ...arrays: {\n    [P in keyof T]: T[P][];\n  }\n): T[] {\n  // Check for empty input\n  if (arrays.length === 0) {\n    return [];\n  }\n\n  // Find the length of the first input array\n  const firstArrayLength = arrays[0].length;\n\n  // Ensure all input arrays have the same length\n  for (const array of arrays) {\n    if (array.length !== firstArrayLength) {\n      throw new Error(\"All input arrays must have the same length.\");\n    }\n  }\n\n  // Create an empty array to store the zipped arrays\n  const zipped: T[] = [];\n\n  // Iterate through each element of the first input array\n  for (let i = 0; i < firstArrayLength; i += 1) {\n    // Create an array to store the zipped elements at the current index\n    const zippedElement: T[keyof T][] = [];\n\n    // Iterate through each input array\n    for (const array of arrays) {\n      // Add the element at the current index to the zipped element array\n      zippedElement.push(array[i]);\n    }\n\n    // Add the zipped element array to the zipped array\n    zipped.push(zippedElement as T);\n  }\n\n  return zipped;\n}\n"],"mappings":";AACA,SAAgB,WACd,GAAG,QAGE;AAEL,KAAI,OAAO,WAAW,EACpB,QAAO,CAAE;CAIX,MAAM,mBAAmB,OAAO,GAAG;AAGnC,MAAK,MAAM,SAAS,OAClB,KAAI,MAAM,WAAW,iBACnB,OAAM,IAAI,MAAM;CAKpB,MAAMA,SAAc,CAAE;AAGtB,MAAK,IAAI,IAAI,GAAG,IAAI,kBAAkB,KAAK,GAAG;EAE5C,MAAMC,gBAA8B,CAAE;AAGtC,OAAK,MAAM,SAAS,QAElB,cAAc,KAAK,MAAM,GAAG;EAI9B,OAAO,KAAK,cAAmB;CAChC;AAED,QAAO;AACR"}
+{"version":3,"file":"utils.js","names":[],"sources":["../../../src/chains/router/utils.ts"],"sourcesContent":["// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function zipEntries<T extends any[]>(\n  ...arrays: {\n    [P in keyof T]: T[P][];\n  }\n): T[] {\n  // Check for empty input\n  if (arrays.length === 0) {\n    return [];\n  }\n\n  // Find the length of the first input array\n  const firstArrayLength = arrays[0].length;\n\n  // Ensure all input arrays have the same length\n  for (const array of arrays) {\n    if (array.length !== firstArrayLength) {\n      throw new Error(\"All input arrays must have the same length.\");\n    }\n  }\n\n  // Create an empty array to store the zipped arrays\n  const zipped: T[] = [];\n\n  // Iterate through each element of the first input array\n  for (let i = 0; i < firstArrayLength; i += 1) {\n    // Create an array to store the zipped elements at the current index\n    const zippedElement: T[keyof T][] = [];\n\n    // Iterate through each input array\n    for (const array of arrays) {\n      // Add the element at the current index to the zipped element array\n      zippedElement.push(array[i]);\n    }\n\n    // Add the zipped element array to the zipped array\n    zipped.push(zippedElement as T);\n  }\n\n  return zipped;\n}\n"],"mappings":";AACA,SAAgB,WACd,GAAG,QAGE;AAEL,KAAI,OAAO,WAAW,EACpB,QAAO,EAAE;CAIX,MAAM,mBAAmB,OAAO,GAAG;AAGnC,MAAK,MAAM,SAAS,OAClB,KAAI,MAAM,WAAW,iBACnB,OAAM,IAAI,MAAM,8CAA8C;CAKlE,MAAM,SAAc,EAAE;AAGtB,MAAK,IAAI,IAAI,GAAG,IAAI,kBAAkB,KAAK,GAAG;EAE5C,MAAM,gBAA8B,EAAE;AAGtC,OAAK,MAAM,SAAS,OAElB,eAAc,KAAK,MAAM,GAAG;AAI9B,SAAO,KAAK,cAAmB;;AAGjC,QAAO"}

package/dist/chains/sequential_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"sequential_chain.cjs","names":["input: Set<string>","BaseChain","fields: SequentialChainInput","intersection","union","difference","values: ChainValues","runManager?: CallbackManagerForChainRun","input: ChainValues","allChainValues: ChainValues","output: ChainValues","data: SerializedSequentialChain","chains: BaseChain[]","inputVariables: string[]","outputVariables: string[]","chains: SerializedBaseChain[]","fields: SimpleSequentialChainInput","input: string","data: SerializedSimpleSequentialChain","chains: Array<BaseChain>","chains: Array<SerializedBaseChain>"],"sources":["../../src/chains/sequential_chain.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport {\n  SerializedBaseChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n} from \"./serde.js\";\nimport { intersection, union, difference } from \"../util/set.js\";\n\nfunction formatSet(input: Set<string>) {\n  return Array.from(input)\n    .map((i) => `\"${i}\"`)\n    .join(\", \");\n}\n\n/**\n * Interface for the input parameters of the SequentialChain class.\n */\nexport interface SequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: BaseChain[];\n  /** Defines which variables should be passed as initial input to the first chain. */\n  inputVariables: string[];\n  /** Which variables should be returned as a result of executing the chain. If not specified, output of the last of the chains is used. */\n  outputVariables?: string[];\n  /** Whether or not to return all intermediate outputs and variables (excluding initial input variables). */\n  returnAll?: boolean;\n}\n\n/**\n * Chain where the outputs of one chain feed directly into next.\n * @example\n * ```typescript\n * const promptTemplate = new PromptTemplate({\n *   template: `You are a playwright. Given the title of play and the era it is set in, it is your job to write a synopsis for that title.\n * Title: {title}\n * Era: {era}\n * Playwright: This is a synopsis for the above play:`,\n *   inputVariables: [\"title\", \"era\"],\n * });\n\n * const reviewPromptTemplate = new PromptTemplate({\n *   template: `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *   \n *     Play Synopsis:\n *     {synopsis}\n *     Review from a New York Times play critic of the above play:`,\n *   inputVariables: [\"synopsis\"],\n * });\n\n * const overallChain = new SequentialChain({\n *   chains: [\n *     new LLMChain({\n *       llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: promptTemplate,\n *       outputKey: \"synopsis\",\n *     }),\n *     new LLMChain({\n *       llm: new OpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: reviewPromptTemplate,\n *       outputKey: \"review\",\n *     }),\n *   ],\n *   inputVariables: [\"era\", \"title\"],\n *   outputVariables: [\"synopsis\", \"review\"],\n *   verbose: true,\n * });\n\n * const chainExecutionResult = await overallChain.call({\n *   title: \"Tragedy at sunset on the beach\",\n *   era: \"Victorian England\",\n * });\n * console.log(chainExecutionResult);\n * ```\n */\nexport class SequentialChain extends BaseChain implements SequentialChainInput {\n  static lc_name() {\n    return \"SequentialChain\";\n  }\n\n  chains: BaseChain[];\n\n  inputVariables: string[];\n\n  outputVariables: string[];\n\n  returnAll?: boolean | undefined;\n\n  get inputKeys() {\n    return this.inputVariables;\n  }\n\n  get outputKeys(): string[] {\n    return this.outputVariables;\n  }\n\n  constructor(fields: SequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.inputVariables = fields.inputVariables;\n    this.outputVariables = fields.outputVariables ?? [];\n    if (this.outputVariables.length > 0 && fields.returnAll) {\n      throw new Error(\n        \"Either specify variables to return using `outputVariables` or use `returnAll` param. Cannot apply both conditions at the same time.\"\n      );\n    }\n    this.returnAll = fields.returnAll ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    if (this.chains.length === 0) {\n      throw new Error(\"Sequential chain must have at least one chain.\");\n    }\n\n    const memoryKeys = this.memory?.memoryKeys ?? [];\n    const inputKeysSet = new Set(this.inputKeys);\n    const memoryKeysSet = new Set(memoryKeys);\n    const keysIntersection = intersection(inputKeysSet, memoryKeysSet);\n    if (keysIntersection.size > 0) {\n      throw new Error(\n        `The following keys: ${formatSet(\n          keysIntersection\n        )} are overlapping between memory and input keys of the chain variables. This can lead to unexpected behaviour. Please use input and memory keys that don't overlap.`\n      );\n    }\n\n    const availableKeys = union(inputKeysSet, memoryKeysSet);\n    for (const chain of this.chains) {\n      let missingKeys = difference(new Set(chain.inputKeys), availableKeys);\n\n      if (chain.memory) {\n        missingKeys = difference(missingKeys, new Set(chain.memory.memoryKeys));\n      }\n\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `Missing variables for chain \"${chain._chainType()}\": ${formatSet(\n            missingKeys\n          )}. Only got the following variables: ${formatSet(availableKeys)}.`\n        );\n      }\n      const outputKeysSet = new Set(chain.outputKeys);\n      const overlappingOutputKeys = intersection(availableKeys, outputKeysSet);\n      if (overlappingOutputKeys.size > 0) {\n        throw new Error(\n          `The following output variables for chain \"${chain._chainType()}\" are overlapping: ${formatSet(\n            overlappingOutputKeys\n          )}. This can lead to unexpected behaviour.`\n        );\n      }\n\n      for (const outputKey of outputKeysSet) {\n        availableKeys.add(outputKey);\n      }\n    }\n\n    if (this.outputVariables.length === 0) {\n      if (this.returnAll) {\n        const outputKeys = difference(availableKeys, inputKeysSet);\n        this.outputVariables = Array.from(outputKeys);\n      } else {\n        this.outputVariables = this.chains[this.chains.length - 1].outputKeys;\n      }\n    } else {\n      const missingKeys = difference(\n        new Set(this.outputVariables),\n        new Set(availableKeys)\n      );\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `The following output variables were expected to be in the final chain output but were not found: ${formatSet(\n            missingKeys\n          )}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: ChainValues = {};\n    const allChainValues: ChainValues = values;\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = await chain.call(\n        allChainValues,\n        runManager?.getChild(`step_${i}`)\n      );\n      for (const key of Object.keys(input)) {\n        allChainValues[key] = input[key];\n      }\n    }\n    const output: ChainValues = {};\n    for (const key of this.outputVariables) {\n      output[key] = allChainValues[key];\n    }\n\n    return output;\n  }\n\n  _chainType() {\n    return \"sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSequentialChain) {\n    const chains: BaseChain[] = [];\n    const inputVariables: string[] = data.input_variables;\n    const outputVariables: string[] = data.output_variables;\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SequentialChain({ chains, inputVariables, outputVariables });\n  }\n\n  serialize(): SerializedSequentialChain {\n    const chains: SerializedBaseChain[] = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      input_variables: this.inputVariables,\n      output_variables: this.outputVariables,\n      chains,\n    };\n  }\n}\n\n/**\n * Interface for the input parameters of the SimpleSequentialChain class.\n */\nexport interface SimpleSequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: Array<BaseChain>;\n  /** Whether or not to trim the intermediate outputs. */\n  trimOutputs?: boolean;\n}\n\n/**\n * Simple chain where a single string output of one chain is fed directly into the next.\n * @augments BaseChain\n * @augments SimpleSequentialChainInput\n *\n * @example\n * ```ts\n * import { SimpleSequentialChain, LLMChain } from \"@langchain/classic/chains\";\n * import { OpenAI } from \"langchain/llms/openai\";\n * import { PromptTemplate } from \"langchain/prompts\";\n *\n * // This is an LLMChain to write a synopsis given a title of a play.\n * const llm = new OpenAI({ temperature: 0 });\n * const template = `You are a playwright. Given the title of play, it is your job to write a synopsis for that title.\n *\n * Title: {title}\n * Playwright: This is a synopsis for the above play:`\n * const promptTemplate = new PromptTemplate({ template, inputVariables: [\"title\"] });\n * const synopsisChain = new LLMChain({ llm, prompt: promptTemplate });\n *\n *\n * // This is an LLMChain to write a review of a play given a synopsis.\n * const reviewLLM = new OpenAI({ temperature: 0 })\n * const reviewTemplate = `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n * Play Synopsis:\n * {synopsis}\n * Review from a New York Times play critic of the above play:`\n * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: [\"synopsis\"] });\n * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });\n *\n * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})\n * const review = await overallChain.run(\"Tragedy at sunset on the beach\")\n * // the variable review contains resulting play review.\n * ```\n */\nexport class SimpleSequentialChain\n  extends BaseChain\n  implements SimpleSequentialChainInput\n{\n  static lc_name() {\n    return \"SimpleSequentialChain\";\n  }\n\n  chains: Array<BaseChain>;\n\n  inputKey = \"input\";\n\n  outputKey = \"output\";\n\n  trimOutputs: boolean;\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  constructor(fields: SimpleSequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.trimOutputs = fields.trimOutputs ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    for (const chain of this.chains) {\n      if (\n        chain.inputKeys.filter((k) => !chain.memory?.memoryKeys.includes(k))\n          .length !== 1\n      ) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one input, got ${\n            chain.inputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n      if (chain.outputKeys.length !== 1) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one output, got ${\n            chain.outputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: string = values[this.inputKey];\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = (\n        await chain.call(\n          { [chain.inputKeys[0]]: input, signal: values.signal },\n          runManager?.getChild(`step_${i}`)\n        )\n      )[chain.outputKeys[0]];\n      if (this.trimOutputs) {\n        input = input.trim();\n      }\n      await runManager?.handleText(input);\n    }\n    return { [this.outputKey]: input };\n  }\n\n  _chainType() {\n    return \"simple_sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSimpleSequentialChain) {\n    const chains: Array<BaseChain> = [];\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SimpleSequentialChain({ chains });\n  }\n\n  serialize(): SerializedSimpleSequentialChain {\n    const chains: Array<SerializedBaseChain> = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      chains,\n    };\n  }\n}\n"],"mappings":";;;;AAUA,SAAS,UAAUA,OAAoB;AACrC,QAAO,MAAM,KAAK,MAAM,CACrB,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CACpB,KAAK,KAAK;AACd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8DD,IAAa,kBAAb,MAAa,wBAAwBC,uBAA0C;CAC7E,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA;CAEA,IAAI,YAAY;AACd,SAAO,KAAK;CACb;CAED,IAAI,aAAuB;AACzB,SAAO,KAAK;CACb;CAED,YAAYC,QAA8B;EACxC,MAAM,OAAO;EACb,KAAK,SAAS,OAAO;EACrB,KAAK,iBAAiB,OAAO;EAC7B,KAAK,kBAAkB,OAAO,mBAAmB,CAAE;AACnD,MAAI,KAAK,gBAAgB,SAAS,KAAK,OAAO,UAC5C,OAAM,IAAI,MACR;EAGJ,KAAK,YAAY,OAAO,aAAa;EACrC,KAAK,iBAAiB;CACvB;;CAGD,kBAAkB;AAChB,MAAI,KAAK,OAAO,WAAW,EACzB,OAAM,IAAI,MAAM;EAGlB,MAAM,aAAa,KAAK,QAAQ,cAAc,CAAE;EAChD,MAAM,eAAe,IAAI,IAAI,KAAK;EAClC,MAAM,gBAAgB,IAAI,IAAI;EAC9B,MAAM,mBAAmBC,yBAAa,cAAc,cAAc;AAClE,MAAI,iBAAiB,OAAO,EAC1B,OAAM,IAAI,MACR,CAAC,oBAAoB,EAAE,UACrB,iBACD,CAAC,kKAAkK,CAAC;EAIzK,MAAM,gBAAgBC,kBAAM,cAAc,cAAc;AACxD,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,IAAI,cAAcC,uBAAW,IAAI,IAAI,MAAM,YAAY,cAAc;AAErE,OAAI,MAAM,QACR,cAAcA,uBAAW,aAAa,IAAI,IAAI,MAAM,OAAO,YAAY;AAGzE,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,CAAC,6BAA6B,EAAE,MAAM,YAAY,CAAC,GAAG,EAAE,UACtD,YACD,CAAC,oCAAoC,EAAE,UAAU,cAAc,CAAC,CAAC,CAAC;GAGvE,MAAM,gBAAgB,IAAI,IAAI,MAAM;GACpC,MAAM,wBAAwBF,yBAAa,eAAe,cAAc;AACxE,OAAI,sBAAsB,OAAO,EAC/B,OAAM,IAAI,MACR,CAAC,0CAA0C,EAAE,MAAM,YAAY,CAAC,mBAAmB,EAAE,UACnF,sBACD,CAAC,wCAAwC,CAAC;AAI/C,QAAK,MAAM,aAAa,eACtB,cAAc,IAAI,UAAU;EAE/B;AAED,MAAI,KAAK,gBAAgB,WAAW,EAClC,KAAI,KAAK,WAAW;GAClB,MAAM,aAAaE,uBAAW,eAAe,aAAa;GAC1D,KAAK,kBAAkB,MAAM,KAAK,WAAW;EAC9C,OACC,KAAK,kBAAkB,KAAK,OAAO,KAAK,OAAO,SAAS,GAAG;OAExD;GACL,MAAM,cAAcA,uBAClB,IAAI,IAAI,KAAK,kBACb,IAAI,IAAI,eACT;AACD,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,CAAC,iGAAiG,EAAE,UAClG,YACD,CAAC,CAAC,CAAC;EAGT;CACF;;CAGD,MAAM,MACJC,QACAC,YACsB;EACtB,IAAIC,QAAqB,CAAE;EAC3B,MAAMC,iBAA8B;EACpC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,KAAK;GACL,QAAQ,MAAM,MAAM,KAClB,gBACA,YAAY,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAClC;AACD,QAAK,MAAM,OAAO,OAAO,KAAK,MAAM,EAClC,eAAe,OAAO,MAAM;EAE/B;EACD,MAAMC,SAAsB,CAAE;AAC9B,OAAK,MAAM,OAAO,KAAK,iBACrB,OAAO,OAAO,eAAe;AAG/B,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYC,MAAiC;EACxD,MAAMC,SAAsB,CAAE;EAC9B,MAAMC,iBAA2B,KAAK;EACtC,MAAMC,kBAA4B,KAAK;EACvC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAMb,uBAAU,YAAY,gBAAgB;GACtE,OAAO,KAAK,kBAAkB;EAC/B;AACD,SAAO,IAAI,gBAAgB;GAAE;GAAQ;GAAgB;EAAiB;CACvE;CAED,YAAuC;EACrC,MAAMc,SAAgC,CAAE;AACxC,OAAK,MAAM,SAAS,KAAK,QACvB,OAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,iBAAiB,KAAK;GACtB,kBAAkB,KAAK;GACvB;EACD;CACF;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDD,IAAa,wBAAb,MAAa,8BACHd,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA,WAAW;CAEX,YAAY;CAEZ;CAEA,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,YAAYe,QAAoC;EAC9C,MAAM,OAAO;EACb,KAAK,SAAS,OAAO;EACrB,KAAK,cAAc,OAAO,eAAe;EACzC,KAAK,iBAAiB;CACvB;;CAGD,kBAAkB;AAChB,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,OACE,MAAM,UAAU,OAAO,CAAC,MAAM,CAAC,MAAM,QAAQ,WAAW,SAAS,EAAE,CAAC,CACjE,WAAW,EAEd,OAAM,IAAI,MACR,CAAC,oEAAoE,EACnE,MAAM,UAAU,OACjB,KAAK,EAAE,MAAM,YAAY,CAAC,CAAC,CAAC;AAGjC,OAAI,MAAM,WAAW,WAAW,EAC9B,OAAM,IAAI,MACR,CAAC,qEAAqE,EACpE,MAAM,WAAW,OAClB,KAAK,EAAE,MAAM,YAAY,CAAC,CAAC,CAAC;EAGlC;CACF;;CAGD,MAAM,MACJV,QACAC,YACsB;EACtB,IAAIU,QAAgB,OAAO,KAAK;EAChC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,KAAK;GACL,SACE,MAAM,MAAM,KACV;KAAG,MAAM,UAAU,KAAK;IAAO,QAAQ,OAAO;GAAQ,GACtD,YAAY,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAClC,EACD,MAAM,WAAW;AACnB,OAAI,KAAK,aACP,QAAQ,MAAM,MAAM;GAEtB,MAAM,YAAY,WAAW,MAAM;EACpC;AACD,SAAO,GAAG,KAAK,YAAY,MAAO;CACnC;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYC,MAAuC;EAC9D,MAAMC,SAA2B,CAAE;EACnC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAMlB,uBAAU,YAAY,gBAAgB;GACtE,OAAO,KAAK,kBAAkB;EAC/B;AACD,SAAO,IAAI,sBAAsB,EAAE,OAAQ;CAC5C;CAED,YAA6C;EAC3C,MAAMmB,SAAqC,CAAE;AAC7C,OAAK,MAAM,SAAS,KAAK,QACvB,OAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB;EACD;CACF;AACF"}
+{"version":3,"file":"sequential_chain.cjs","names":["BaseChain","intersection","union","difference"],"sources":["../../src/chains/sequential_chain.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport {\n  SerializedBaseChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n} from \"./serde.js\";\nimport { intersection, union, difference } from \"../util/set.js\";\n\nfunction formatSet(input: Set<string>) {\n  return Array.from(input)\n    .map((i) => `\"${i}\"`)\n    .join(\", \");\n}\n\n/**\n * Interface for the input parameters of the SequentialChain class.\n */\nexport interface SequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: BaseChain[];\n  /** Defines which variables should be passed as initial input to the first chain. */\n  inputVariables: string[];\n  /** Which variables should be returned as a result of executing the chain. If not specified, output of the last of the chains is used. */\n  outputVariables?: string[];\n  /** Whether or not to return all intermediate outputs and variables (excluding initial input variables). */\n  returnAll?: boolean;\n}\n\n/**\n * Chain where the outputs of one chain feed directly into next.\n * @example\n * ```typescript\n * const promptTemplate = new PromptTemplate({\n *   template: `You are a playwright. Given the title of play and the era it is set in, it is your job to write a synopsis for that title.\n * Title: {title}\n * Era: {era}\n * Playwright: This is a synopsis for the above play:`,\n *   inputVariables: [\"title\", \"era\"],\n * });\n\n * const reviewPromptTemplate = new PromptTemplate({\n *   template: `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *   \n *     Play Synopsis:\n *     {synopsis}\n *     Review from a New York Times play critic of the above play:`,\n *   inputVariables: [\"synopsis\"],\n * });\n\n * const overallChain = new SequentialChain({\n *   chains: [\n *     new LLMChain({\n *       llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: promptTemplate,\n *       outputKey: \"synopsis\",\n *     }),\n *     new LLMChain({\n *       llm: new OpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: reviewPromptTemplate,\n *       outputKey: \"review\",\n *     }),\n *   ],\n *   inputVariables: [\"era\", \"title\"],\n *   outputVariables: [\"synopsis\", \"review\"],\n *   verbose: true,\n * });\n\n * const chainExecutionResult = await overallChain.call({\n *   title: \"Tragedy at sunset on the beach\",\n *   era: \"Victorian England\",\n * });\n * console.log(chainExecutionResult);\n * ```\n */\nexport class SequentialChain extends BaseChain implements SequentialChainInput {\n  static lc_name() {\n    return \"SequentialChain\";\n  }\n\n  chains: BaseChain[];\n\n  inputVariables: string[];\n\n  outputVariables: string[];\n\n  returnAll?: boolean | undefined;\n\n  get inputKeys() {\n    return this.inputVariables;\n  }\n\n  get outputKeys(): string[] {\n    return this.outputVariables;\n  }\n\n  constructor(fields: SequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.inputVariables = fields.inputVariables;\n    this.outputVariables = fields.outputVariables ?? [];\n    if (this.outputVariables.length > 0 && fields.returnAll) {\n      throw new Error(\n        \"Either specify variables to return using `outputVariables` or use `returnAll` param. Cannot apply both conditions at the same time.\"\n      );\n    }\n    this.returnAll = fields.returnAll ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    if (this.chains.length === 0) {\n      throw new Error(\"Sequential chain must have at least one chain.\");\n    }\n\n    const memoryKeys = this.memory?.memoryKeys ?? [];\n    const inputKeysSet = new Set(this.inputKeys);\n    const memoryKeysSet = new Set(memoryKeys);\n    const keysIntersection = intersection(inputKeysSet, memoryKeysSet);\n    if (keysIntersection.size > 0) {\n      throw new Error(\n        `The following keys: ${formatSet(\n          keysIntersection\n        )} are overlapping between memory and input keys of the chain variables. This can lead to unexpected behaviour. Please use input and memory keys that don't overlap.`\n      );\n    }\n\n    const availableKeys = union(inputKeysSet, memoryKeysSet);\n    for (const chain of this.chains) {\n      let missingKeys = difference(new Set(chain.inputKeys), availableKeys);\n\n      if (chain.memory) {\n        missingKeys = difference(missingKeys, new Set(chain.memory.memoryKeys));\n      }\n\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `Missing variables for chain \"${chain._chainType()}\": ${formatSet(\n            missingKeys\n          )}. Only got the following variables: ${formatSet(availableKeys)}.`\n        );\n      }\n      const outputKeysSet = new Set(chain.outputKeys);\n      const overlappingOutputKeys = intersection(availableKeys, outputKeysSet);\n      if (overlappingOutputKeys.size > 0) {\n        throw new Error(\n          `The following output variables for chain \"${chain._chainType()}\" are overlapping: ${formatSet(\n            overlappingOutputKeys\n          )}. This can lead to unexpected behaviour.`\n        );\n      }\n\n      for (const outputKey of outputKeysSet) {\n        availableKeys.add(outputKey);\n      }\n    }\n\n    if (this.outputVariables.length === 0) {\n      if (this.returnAll) {\n        const outputKeys = difference(availableKeys, inputKeysSet);\n        this.outputVariables = Array.from(outputKeys);\n      } else {\n        this.outputVariables = this.chains[this.chains.length - 1].outputKeys;\n      }\n    } else {\n      const missingKeys = difference(\n        new Set(this.outputVariables),\n        new Set(availableKeys)\n      );\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `The following output variables were expected to be in the final chain output but were not found: ${formatSet(\n            missingKeys\n          )}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: ChainValues = {};\n    const allChainValues: ChainValues = values;\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = await chain.call(\n        allChainValues,\n        runManager?.getChild(`step_${i}`)\n      );\n      for (const key of Object.keys(input)) {\n        allChainValues[key] = input[key];\n      }\n    }\n    const output: ChainValues = {};\n    for (const key of this.outputVariables) {\n      output[key] = allChainValues[key];\n    }\n\n    return output;\n  }\n\n  _chainType() {\n    return \"sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSequentialChain) {\n    const chains: BaseChain[] = [];\n    const inputVariables: string[] = data.input_variables;\n    const outputVariables: string[] = data.output_variables;\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SequentialChain({ chains, inputVariables, outputVariables });\n  }\n\n  serialize(): SerializedSequentialChain {\n    const chains: SerializedBaseChain[] = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      input_variables: this.inputVariables,\n      output_variables: this.outputVariables,\n      chains,\n    };\n  }\n}\n\n/**\n * Interface for the input parameters of the SimpleSequentialChain class.\n */\nexport interface SimpleSequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: Array<BaseChain>;\n  /** Whether or not to trim the intermediate outputs. */\n  trimOutputs?: boolean;\n}\n\n/**\n * Simple chain where a single string output of one chain is fed directly into the next.\n * @augments BaseChain\n * @augments SimpleSequentialChainInput\n *\n * @example\n * ```ts\n * import { SimpleSequentialChain, LLMChain } from \"@langchain/classic/chains\";\n * import { OpenAI } from \"langchain/llms/openai\";\n * import { PromptTemplate } from \"langchain/prompts\";\n *\n * // This is an LLMChain to write a synopsis given a title of a play.\n * const llm = new OpenAI({ temperature: 0 });\n * const template = `You are a playwright. Given the title of play, it is your job to write a synopsis for that title.\n *\n * Title: {title}\n * Playwright: This is a synopsis for the above play:`\n * const promptTemplate = new PromptTemplate({ template, inputVariables: [\"title\"] });\n * const synopsisChain = new LLMChain({ llm, prompt: promptTemplate });\n *\n *\n * // This is an LLMChain to write a review of a play given a synopsis.\n * const reviewLLM = new OpenAI({ temperature: 0 })\n * const reviewTemplate = `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n * Play Synopsis:\n * {synopsis}\n * Review from a New York Times play critic of the above play:`\n * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: [\"synopsis\"] });\n * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });\n *\n * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})\n * const review = await overallChain.run(\"Tragedy at sunset on the beach\")\n * // the variable review contains resulting play review.\n * ```\n */\nexport class SimpleSequentialChain\n  extends BaseChain\n  implements SimpleSequentialChainInput\n{\n  static lc_name() {\n    return \"SimpleSequentialChain\";\n  }\n\n  chains: Array<BaseChain>;\n\n  inputKey = \"input\";\n\n  outputKey = \"output\";\n\n  trimOutputs: boolean;\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  constructor(fields: SimpleSequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.trimOutputs = fields.trimOutputs ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    for (const chain of this.chains) {\n      if (\n        chain.inputKeys.filter((k) => !chain.memory?.memoryKeys.includes(k))\n          .length !== 1\n      ) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one input, got ${\n            chain.inputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n      if (chain.outputKeys.length !== 1) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one output, got ${\n            chain.outputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: string = values[this.inputKey];\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = (\n        await chain.call(\n          { [chain.inputKeys[0]]: input, signal: values.signal },\n          runManager?.getChild(`step_${i}`)\n        )\n      )[chain.outputKeys[0]];\n      if (this.trimOutputs) {\n        input = input.trim();\n      }\n      await runManager?.handleText(input);\n    }\n    return { [this.outputKey]: input };\n  }\n\n  _chainType() {\n    return \"simple_sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSimpleSequentialChain) {\n    const chains: Array<BaseChain> = [];\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SimpleSequentialChain({ chains });\n  }\n\n  serialize(): SerializedSimpleSequentialChain {\n    const chains: Array<SerializedBaseChain> = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      chains,\n    };\n  }\n}\n"],"mappings":";;;;AAUA,SAAS,UAAU,OAAoB;AACrC,QAAO,MAAM,KAAK,MAAM,CACrB,KAAK,MAAM,IAAI,EAAE,GAAG,CACpB,KAAK,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+Df,IAAa,kBAAb,MAAa,wBAAwBA,uBAA0C;CAC7E,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA;CAEA;CAEA;CAEA,IAAI,YAAY;AACd,SAAO,KAAK;;CAGd,IAAI,aAAuB;AACzB,SAAO,KAAK;;CAGd,YAAY,QAA8B;AACxC,QAAM,OAAO;AACb,OAAK,SAAS,OAAO;AACrB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,kBAAkB,OAAO,mBAAmB,EAAE;AACnD,MAAI,KAAK,gBAAgB,SAAS,KAAK,OAAO,UAC5C,OAAM,IAAI,MACR,sIACD;AAEH,OAAK,YAAY,OAAO,aAAa;AACrC,OAAK,iBAAiB;;;CAIxB,kBAAkB;AAChB,MAAI,KAAK,OAAO,WAAW,EACzB,OAAM,IAAI,MAAM,iDAAiD;EAGnE,MAAM,aAAa,KAAK,QAAQ,cAAc,EAAE;EAChD,MAAM,eAAe,IAAI,IAAI,KAAK,UAAU;EAC5C,MAAM,gBAAgB,IAAI,IAAI,WAAW;EACzC,MAAM,mBAAmBC,yBAAa,cAAc,cAAc;AAClE,MAAI,iBAAiB,OAAO,EAC1B,OAAM,IAAI,MACR,uBAAuB,UACrB,iBACD,CAAC,oKACH;EAGH,MAAM,gBAAgBC,kBAAM,cAAc,cAAc;AACxD,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,IAAI,cAAcC,uBAAW,IAAI,IAAI,MAAM,UAAU,EAAE,cAAc;AAErE,OAAI,MAAM,OACR,eAAcA,uBAAW,aAAa,IAAI,IAAI,MAAM,OAAO,WAAW,CAAC;AAGzE,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,gCAAgC,MAAM,YAAY,CAAC,KAAK,UACtD,YACD,CAAC,sCAAsC,UAAU,cAAc,CAAC,GAClE;GAEH,MAAM,gBAAgB,IAAI,IAAI,MAAM,WAAW;GAC/C,MAAM,wBAAwBF,yBAAa,eAAe,cAAc;AACxE,OAAI,sBAAsB,OAAO,EAC/B,OAAM,IAAI,MACR,6CAA6C,MAAM,YAAY,CAAC,qBAAqB,UACnF,sBACD,CAAC,0CACH;AAGH,QAAK,MAAM,aAAa,cACtB,eAAc,IAAI,UAAU;;AAIhC,MAAI,KAAK,gBAAgB,WAAW,EAClC,KAAI,KAAK,WAAW;GAClB,MAAM,aAAaE,uBAAW,eAAe,aAAa;AAC1D,QAAK,kBAAkB,MAAM,KAAK,WAAW;QAE7C,MAAK,kBAAkB,KAAK,OAAO,KAAK,OAAO,SAAS,GAAG;OAExD;GACL,MAAM,cAAcA,uBAClB,IAAI,IAAI,KAAK,gBAAgB,EAC7B,IAAI,IAAI,cAAc,CACvB;AACD,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,oGAAoG,UAClG,YACD,CAAC,GACH;;;;CAMP,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,QAAqB,EAAE;EAC3B,MAAM,iBAA8B;EACpC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,QAAK;AACL,WAAQ,MAAM,MAAM,KAClB,gBACA,YAAY,SAAS,QAAQ,IAAI,CAClC;AACD,QAAK,MAAM,OAAO,OAAO,KAAK,MAAM,CAClC,gBAAe,OAAO,MAAM;;EAGhC,MAAM,SAAsB,EAAE;AAC9B,OAAK,MAAM,OAAO,KAAK,gBACrB,QAAO,OAAO,eAAe;AAG/B,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAiC;EACxD,MAAM,SAAsB,EAAE;EAC9B,MAAM,iBAA2B,KAAK;EACtC,MAAM,kBAA4B,KAAK;EACvC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAMH,uBAAU,YAAY,gBAAgB;AACtE,UAAO,KAAK,kBAAkB;;AAEhC,SAAO,IAAI,gBAAgB;GAAE;GAAQ;GAAgB;GAAiB,CAAC;;CAGzE,YAAuC;EACrC,MAAM,SAAgC,EAAE;AACxC,OAAK,MAAM,SAAS,KAAK,OACvB,QAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,iBAAiB,KAAK;GACtB,kBAAkB,KAAK;GACvB;GACD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDL,IAAa,wBAAb,MAAa,8BACHA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ;CAEA,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAoC;AAC9C,QAAM,OAAO;AACb,OAAK,SAAS,OAAO;AACrB,OAAK,cAAc,OAAO,eAAe;AACzC,OAAK,iBAAiB;;;CAIxB,kBAAkB;AAChB,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,OACE,MAAM,UAAU,QAAQ,MAAM,CAAC,MAAM,QAAQ,WAAW,SAAS,EAAE,CAAC,CACjE,WAAW,EAEd,OAAM,IAAI,MACR,uEACE,MAAM,UAAU,OACjB,OAAO,MAAM,YAAY,CAAC,GAC5B;AAEH,OAAI,MAAM,WAAW,WAAW,EAC9B,OAAM,IAAI,MACR,wEACE,MAAM,WAAW,OAClB,OAAO,MAAM,YAAY,CAAC,GAC5B;;;;CAMP,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,QAAgB,OAAO,KAAK;EAChC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,QAAK;AACL,YACE,MAAM,MAAM,KACV;KAAG,MAAM,UAAU,KAAK;IAAO,QAAQ,OAAO;IAAQ,EACtD,YAAY,SAAS,QAAQ,IAAI,CAClC,EACD,MAAM,WAAW;AACnB,OAAI,KAAK,YACP,SAAQ,MAAM,MAAM;AAEtB,SAAM,YAAY,WAAW,MAAM;;AAErC,SAAO,GAAG,KAAK,YAAY,OAAO;;CAGpC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAuC;EAC9D,MAAM,SAA2B,EAAE;EACnC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAMA,uBAAU,YAAY,gBAAgB;AACtE,UAAO,KAAK,kBAAkB;;AAEhC,SAAO,IAAI,sBAAsB,EAAE,QAAQ,CAAC;;CAG9C,YAA6C;EAC3C,MAAM,SAAqC,EAAE;AAC7C,OAAK,MAAM,SAAS,KAAK,OACvB,QAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB;GACD"}

package/dist/chains/sequential_chain.d.cts

@@ -4,7 +4,6 @@ import { ChainValues } from "@langchain/core/utils/types";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 
 //#region src/chains/sequential_chain.d.ts
-
 /**
  * Interface for the input parameters of the SequentialChain class.
  */

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

@@ -1 +1 @@
-{"version":3,"file":"sequential_chain.d.cts","names":["ChainValues","CallbackManagerForChainRun","BaseChain","ChainInputs","SerializedSequentialChain","SerializedSimpleSequentialChain","SequentialChainInput","SequentialChain","Promise","SimpleSequentialChainInput","Array","SimpleSequentialChain"],"sources":["../../src/chains/sequential_chain.d.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedSequentialChain, SerializedSimpleSequentialChain } from \"./serde.js\";\n/**\n * Interface for the input parameters of the SequentialChain class.\n */\nexport interface SequentialChainInput extends ChainInputs {\n    /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n    chains: BaseChain[];\n    /** Defines which variables should be passed as initial input to the first chain. */\n    inputVariables: string[];\n    /** Which variables should be returned as a result of executing the chain. If not specified, output of the last of the chains is used. */\n    outputVariables?: string[];\n    /** Whether or not to return all intermediate outputs and variables (excluding initial input variables). */\n    returnAll?: boolean;\n}\n/**\n * Chain where the outputs of one chain feed directly into next.\n * @example\n * ```typescript\n * const promptTemplate = new PromptTemplate({\n *   template: `You are a playwright. Given the title of play and the era it is set in, it is your job to write a synopsis for that title.\n * Title: {title}\n * Era: {era}\n * Playwright: This is a synopsis for the above play:`,\n *   inputVariables: [\"title\", \"era\"],\n * });\n\n * const reviewPromptTemplate = new PromptTemplate({\n *   template: `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n *     Play Synopsis:\n *     {synopsis}\n *     Review from a New York Times play critic of the above play:`,\n *   inputVariables: [\"synopsis\"],\n * });\n\n * const overallChain = new SequentialChain({\n *   chains: [\n *     new LLMChain({\n *       llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: promptTemplate,\n *       outputKey: \"synopsis\",\n *     }),\n *     new LLMChain({\n *       llm: new OpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: reviewPromptTemplate,\n *       outputKey: \"review\",\n *     }),\n *   ],\n *   inputVariables: [\"era\", \"title\"],\n *   outputVariables: [\"synopsis\", \"review\"],\n *   verbose: true,\n * });\n\n * const chainExecutionResult = await overallChain.call({\n *   title: \"Tragedy at sunset on the beach\",\n *   era: \"Victorian England\",\n * });\n * console.log(chainExecutionResult);\n * ```\n */\nexport declare class SequentialChain extends BaseChain implements SequentialChainInput {\n    static lc_name(): string;\n    chains: BaseChain[];\n    inputVariables: string[];\n    outputVariables: string[];\n    returnAll?: boolean | undefined;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: SequentialChainInput);\n    /** @ignore */\n    _validateChains(): void;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"sequential_chain\";\n    static deserialize(data: SerializedSequentialChain): Promise<SequentialChain>;\n    serialize(): SerializedSequentialChain;\n}\n/**\n * Interface for the input parameters of the SimpleSequentialChain class.\n */\nexport interface SimpleSequentialChainInput extends ChainInputs {\n    /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n    chains: Array<BaseChain>;\n    /** Whether or not to trim the intermediate outputs. */\n    trimOutputs?: boolean;\n}\n/**\n * Simple chain where a single string output of one chain is fed directly into the next.\n * @augments BaseChain\n * @augments SimpleSequentialChainInput\n *\n * @example\n * ```ts\n * import { SimpleSequentialChain, LLMChain } from \"@langchain/classic/chains\";\n * import { OpenAI } from \"langchain/llms/openai\";\n * import { PromptTemplate } from \"langchain/prompts\";\n *\n * // This is an LLMChain to write a synopsis given a title of a play.\n * const llm = new OpenAI({ temperature: 0 });\n * const template = `You are a playwright. Given the title of play, it is your job to write a synopsis for that title.\n *\n * Title: {title}\n * Playwright: This is a synopsis for the above play:`\n * const promptTemplate = new PromptTemplate({ template, inputVariables: [\"title\"] });\n * const synopsisChain = new LLMChain({ llm, prompt: promptTemplate });\n *\n *\n * // This is an LLMChain to write a review of a play given a synopsis.\n * const reviewLLM = new OpenAI({ temperature: 0 })\n * const reviewTemplate = `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n * Play Synopsis:\n * {synopsis}\n * Review from a New York Times play critic of the above play:`\n * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: [\"synopsis\"] });\n * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });\n *\n * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})\n * const review = await overallChain.run(\"Tragedy at sunset on the beach\")\n * // the variable review contains resulting play review.\n * ```\n */\nexport declare class SimpleSequentialChain extends BaseChain implements SimpleSequentialChainInput {\n    static lc_name(): string;\n    chains: Array<BaseChain>;\n    inputKey: string;\n    outputKey: string;\n    trimOutputs: boolean;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: SimpleSequentialChainInput);\n    /** @ignore */\n    _validateChains(): void;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"simple_sequential_chain\";\n    static deserialize(data: SerializedSimpleSequentialChain): Promise<SimpleSequentialChain>;\n    serialize(): SerializedSimpleSequentialChain;\n}\n//# sourceMappingURL=sequential_chain.d.ts.map"],"mappings":";;;;;;;;;AAOA;AAwDqBO,UAxDJD,oBAAAA,SAA6BH,WAwDV,CAAA;EAExBD;EAMYI,MAAAA,EA9DZJ,SA8DYI,EAAAA;EAINN;EAA0BC,cAAAA,EAAAA,MAAAA,EAAAA;EAAqCD;EAARQ,eAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAE5CJ;EAAoCG,SAAAA,CAAAA,EAAAA,OAAAA;;;;;AAdqB;AAoBtF;;;;AAA+D;AA0C/D;;;;;;;;;;;;;;AAAkG;;;;;;;;;;;;;;;;;;;;;;;cA9D7EA,eAAAA,SAAwBL,SAAAA,YAAqBI;;UAEtDJ;;;;;;sBAMYI;;;;gBAINN,0BAA0BC,6BAA6BO,QAAQR;;2BAEpDI,4BAA4BI,QAAQD;eAChDH;;;;;UAKAK,0BAAAA,SAAmCN;;UAExCO,MAAMR;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAwCGS,qBAAAA,SAA8BT,SAAAA,YAAqBO;;UAE5DC,MAAMR;;;;;;sBAMMO;;;;gBAINT,0BAA0BC,6BAA6BO,QAAQR;;2BAEpDK,kCAAkCG,QAAQG;eACtDN"}
+{"version":3,"file":"sequential_chain.d.cts","names":[],"sources":["../../src/chains/sequential_chain.ts"],"mappings":";;;;;;;;AAmBA;UAAiB,oBAAA,SAA6B,WAAA;;EAE5C,MAAA,EAAQ,SAAA;EAFoC;EAI5C,cAAA;EAFQ;EAIR,eAAA;EAAA;EAEA,SAAA;AAAA;;AAiDF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAAa,eAAA,SAAwB,SAAA,YAAqB,oBAAA;EAAA,OACjD,OAAA,CAAA;EAIP,MAAA,EAAQ,SAAA;EAER,cAAA;EAEA,eAAA;EAEA,SAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,oBAAA;EA+IyC;EAhI7D,eAAA,CAAA;EAgIkD;EA1D5C,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAsBX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,yBAAA,GAAyB,OAAA,CAAA,eAAA;EAYxD,SAAA,CAAA,GAAa,yBAAA;AAAA;;AA4Df;;UA3CiB,0BAAA,SAAmC,WAAA;EAmDpC;EAjDd,MAAA,EAAQ,KAAA,CAAM,SAAA;EAiEM;EA/DpB,WAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCW,qBAAA,SACH,SAAA,YACG,0BAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,MAAA,EAAQ,KAAA,CAAM,SAAA;EAEd,QAAA;EAEA,SAAA;EAEA,WAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,0BAAA;;EAQpB,eAAA,CAAA;;EAuBM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAmBX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,+BAAA,GAA+B,OAAA,CAAA,qBAAA;EAU9D,SAAA,CAAA,GAAa,+BAAA;AAAA"}

package/dist/chains/sequential_chain.d.ts

@@ -4,7 +4,6 @@ import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { ChainValues } from "@langchain/core/utils/types";
 
 //#region src/chains/sequential_chain.d.ts
-
 /**
  * Interface for the input parameters of the SequentialChain class.
  */

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

@@ -1 +1 @@
-{"version":3,"file":"sequential_chain.d.ts","names":["ChainValues","CallbackManagerForChainRun","BaseChain","ChainInputs","SerializedSequentialChain","SerializedSimpleSequentialChain","SequentialChainInput","SequentialChain","Promise","SimpleSequentialChainInput","Array","SimpleSequentialChain"],"sources":["../../src/chains/sequential_chain.d.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { SerializedSequentialChain, SerializedSimpleSequentialChain } from \"./serde.js\";\n/**\n * Interface for the input parameters of the SequentialChain class.\n */\nexport interface SequentialChainInput extends ChainInputs {\n    /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n    chains: BaseChain[];\n    /** Defines which variables should be passed as initial input to the first chain. */\n    inputVariables: string[];\n    /** Which variables should be returned as a result of executing the chain. If not specified, output of the last of the chains is used. */\n    outputVariables?: string[];\n    /** Whether or not to return all intermediate outputs and variables (excluding initial input variables). */\n    returnAll?: boolean;\n}\n/**\n * Chain where the outputs of one chain feed directly into next.\n * @example\n * ```typescript\n * const promptTemplate = new PromptTemplate({\n *   template: `You are a playwright. Given the title of play and the era it is set in, it is your job to write a synopsis for that title.\n * Title: {title}\n * Era: {era}\n * Playwright: This is a synopsis for the above play:`,\n *   inputVariables: [\"title\", \"era\"],\n * });\n\n * const reviewPromptTemplate = new PromptTemplate({\n *   template: `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n *     Play Synopsis:\n *     {synopsis}\n *     Review from a New York Times play critic of the above play:`,\n *   inputVariables: [\"synopsis\"],\n * });\n\n * const overallChain = new SequentialChain({\n *   chains: [\n *     new LLMChain({\n *       llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: promptTemplate,\n *       outputKey: \"synopsis\",\n *     }),\n *     new LLMChain({\n *       llm: new OpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: reviewPromptTemplate,\n *       outputKey: \"review\",\n *     }),\n *   ],\n *   inputVariables: [\"era\", \"title\"],\n *   outputVariables: [\"synopsis\", \"review\"],\n *   verbose: true,\n * });\n\n * const chainExecutionResult = await overallChain.call({\n *   title: \"Tragedy at sunset on the beach\",\n *   era: \"Victorian England\",\n * });\n * console.log(chainExecutionResult);\n * ```\n */\nexport declare class SequentialChain extends BaseChain implements SequentialChainInput {\n    static lc_name(): string;\n    chains: BaseChain[];\n    inputVariables: string[];\n    outputVariables: string[];\n    returnAll?: boolean | undefined;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: SequentialChainInput);\n    /** @ignore */\n    _validateChains(): void;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"sequential_chain\";\n    static deserialize(data: SerializedSequentialChain): Promise<SequentialChain>;\n    serialize(): SerializedSequentialChain;\n}\n/**\n * Interface for the input parameters of the SimpleSequentialChain class.\n */\nexport interface SimpleSequentialChainInput extends ChainInputs {\n    /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n    chains: Array<BaseChain>;\n    /** Whether or not to trim the intermediate outputs. */\n    trimOutputs?: boolean;\n}\n/**\n * Simple chain where a single string output of one chain is fed directly into the next.\n * @augments BaseChain\n * @augments SimpleSequentialChainInput\n *\n * @example\n * ```ts\n * import { SimpleSequentialChain, LLMChain } from \"@langchain/classic/chains\";\n * import { OpenAI } from \"langchain/llms/openai\";\n * import { PromptTemplate } from \"langchain/prompts\";\n *\n * // This is an LLMChain to write a synopsis given a title of a play.\n * const llm = new OpenAI({ temperature: 0 });\n * const template = `You are a playwright. Given the title of play, it is your job to write a synopsis for that title.\n *\n * Title: {title}\n * Playwright: This is a synopsis for the above play:`\n * const promptTemplate = new PromptTemplate({ template, inputVariables: [\"title\"] });\n * const synopsisChain = new LLMChain({ llm, prompt: promptTemplate });\n *\n *\n * // This is an LLMChain to write a review of a play given a synopsis.\n * const reviewLLM = new OpenAI({ temperature: 0 })\n * const reviewTemplate = `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n * Play Synopsis:\n * {synopsis}\n * Review from a New York Times play critic of the above play:`\n * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: [\"synopsis\"] });\n * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });\n *\n * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})\n * const review = await overallChain.run(\"Tragedy at sunset on the beach\")\n * // the variable review contains resulting play review.\n * ```\n */\nexport declare class SimpleSequentialChain extends BaseChain implements SimpleSequentialChainInput {\n    static lc_name(): string;\n    chains: Array<BaseChain>;\n    inputKey: string;\n    outputKey: string;\n    trimOutputs: boolean;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    constructor(fields: SimpleSequentialChainInput);\n    /** @ignore */\n    _validateChains(): void;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): \"simple_sequential_chain\";\n    static deserialize(data: SerializedSimpleSequentialChain): Promise<SimpleSequentialChain>;\n    serialize(): SerializedSimpleSequentialChain;\n}\n//# sourceMappingURL=sequential_chain.d.ts.map"],"mappings":";;;;;;;;;AAOA;AAwDqBO,UAxDJD,oBAAAA,SAA6BH,WAwDV,CAAA;EAExBD;EAMYI,MAAAA,EA9DZJ,SA8DYI,EAAAA;EAINN;EAA0BC,cAAAA,EAAAA,MAAAA,EAAAA;EAAqCD;EAARQ,eAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAE5CJ;EAAoCG,SAAAA,CAAAA,EAAAA,OAAAA;;;;;AAdqB;AAoBtF;;;;AAA+D;AA0C/D;;;;;;;;;;;;;;AAAkG;;;;;;;;;;;;;;;;;;;;;;;cA9D7EA,eAAAA,SAAwBL,SAAAA,YAAqBI;;UAEtDJ;;;;;;sBAMYI;;;;gBAINN,0BAA0BC,6BAA6BO,QAAQR;;2BAEpDI,4BAA4BI,QAAQD;eAChDH;;;;;UAKAK,0BAAAA,SAAmCN;;UAExCO,MAAMR;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAwCGS,qBAAAA,SAA8BT,SAAAA,YAAqBO;;UAE5DC,MAAMR;;;;;;sBAMMO;;;;gBAINT,0BAA0BC,6BAA6BO,QAAQR;;2BAEpDK,kCAAkCG,QAAQG;eACtDN"}
+{"version":3,"file":"sequential_chain.d.ts","names":[],"sources":["../../src/chains/sequential_chain.ts"],"mappings":";;;;;;;;AAmBA;UAAiB,oBAAA,SAA6B,WAAA;;EAE5C,MAAA,EAAQ,SAAA;EAFoC;EAI5C,cAAA;EAFQ;EAIR,eAAA;EAAA;EAEA,SAAA;AAAA;;AAiDF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAAa,eAAA,SAAwB,SAAA,YAAqB,oBAAA;EAAA,OACjD,OAAA,CAAA;EAIP,MAAA,EAAQ,SAAA;EAER,cAAA;EAEA,eAAA;EAEA,SAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,oBAAA;EA+IyC;EAhI7D,eAAA,CAAA;EAgIkD;EA1D5C,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAsBX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,yBAAA,GAAyB,OAAA,CAAA,eAAA;EAYxD,SAAA,CAAA,GAAa,yBAAA;AAAA;;AA4Df;;UA3CiB,0BAAA,SAAmC,WAAA;EAmDpC;EAjDd,MAAA,EAAQ,KAAA,CAAM,SAAA;EAiEM;EA/DpB,WAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCW,qBAAA,SACH,SAAA,YACG,0BAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,MAAA,EAAQ,KAAA,CAAM,SAAA;EAEd,QAAA;EAEA,SAAA;EAEA,WAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,WAAA,CAAY,MAAA,EAAQ,0BAAA;;EAQpB,eAAA,CAAA;;EAuBM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAmBX,UAAA,CAAA;EAAA,OAIa,WAAA,CAAY,IAAA,EAAM,+BAAA,GAA+B,OAAA,CAAA,qBAAA;EAU9D,SAAA,CAAA,GAAa,+BAAA;AAAA"}

package/dist/chains/sequential_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"sequential_chain.js","names":["input: Set<string>","fields: SequentialChainInput","values: ChainValues","runManager?: CallbackManagerForChainRun","input: ChainValues","allChainValues: ChainValues","output: ChainValues","data: SerializedSequentialChain","chains: BaseChain[]","inputVariables: string[]","outputVariables: string[]","chains: SerializedBaseChain[]","fields: SimpleSequentialChainInput","input: string","data: SerializedSimpleSequentialChain","chains: Array<BaseChain>","chains: Array<SerializedBaseChain>"],"sources":["../../src/chains/sequential_chain.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport {\n  SerializedBaseChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n} from \"./serde.js\";\nimport { intersection, union, difference } from \"../util/set.js\";\n\nfunction formatSet(input: Set<string>) {\n  return Array.from(input)\n    .map((i) => `\"${i}\"`)\n    .join(\", \");\n}\n\n/**\n * Interface for the input parameters of the SequentialChain class.\n */\nexport interface SequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: BaseChain[];\n  /** Defines which variables should be passed as initial input to the first chain. */\n  inputVariables: string[];\n  /** Which variables should be returned as a result of executing the chain. If not specified, output of the last of the chains is used. */\n  outputVariables?: string[];\n  /** Whether or not to return all intermediate outputs and variables (excluding initial input variables). */\n  returnAll?: boolean;\n}\n\n/**\n * Chain where the outputs of one chain feed directly into next.\n * @example\n * ```typescript\n * const promptTemplate = new PromptTemplate({\n *   template: `You are a playwright. Given the title of play and the era it is set in, it is your job to write a synopsis for that title.\n * Title: {title}\n * Era: {era}\n * Playwright: This is a synopsis for the above play:`,\n *   inputVariables: [\"title\", \"era\"],\n * });\n\n * const reviewPromptTemplate = new PromptTemplate({\n *   template: `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *   \n *     Play Synopsis:\n *     {synopsis}\n *     Review from a New York Times play critic of the above play:`,\n *   inputVariables: [\"synopsis\"],\n * });\n\n * const overallChain = new SequentialChain({\n *   chains: [\n *     new LLMChain({\n *       llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: promptTemplate,\n *       outputKey: \"synopsis\",\n *     }),\n *     new LLMChain({\n *       llm: new OpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: reviewPromptTemplate,\n *       outputKey: \"review\",\n *     }),\n *   ],\n *   inputVariables: [\"era\", \"title\"],\n *   outputVariables: [\"synopsis\", \"review\"],\n *   verbose: true,\n * });\n\n * const chainExecutionResult = await overallChain.call({\n *   title: \"Tragedy at sunset on the beach\",\n *   era: \"Victorian England\",\n * });\n * console.log(chainExecutionResult);\n * ```\n */\nexport class SequentialChain extends BaseChain implements SequentialChainInput {\n  static lc_name() {\n    return \"SequentialChain\";\n  }\n\n  chains: BaseChain[];\n\n  inputVariables: string[];\n\n  outputVariables: string[];\n\n  returnAll?: boolean | undefined;\n\n  get inputKeys() {\n    return this.inputVariables;\n  }\n\n  get outputKeys(): string[] {\n    return this.outputVariables;\n  }\n\n  constructor(fields: SequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.inputVariables = fields.inputVariables;\n    this.outputVariables = fields.outputVariables ?? [];\n    if (this.outputVariables.length > 0 && fields.returnAll) {\n      throw new Error(\n        \"Either specify variables to return using `outputVariables` or use `returnAll` param. Cannot apply both conditions at the same time.\"\n      );\n    }\n    this.returnAll = fields.returnAll ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    if (this.chains.length === 0) {\n      throw new Error(\"Sequential chain must have at least one chain.\");\n    }\n\n    const memoryKeys = this.memory?.memoryKeys ?? [];\n    const inputKeysSet = new Set(this.inputKeys);\n    const memoryKeysSet = new Set(memoryKeys);\n    const keysIntersection = intersection(inputKeysSet, memoryKeysSet);\n    if (keysIntersection.size > 0) {\n      throw new Error(\n        `The following keys: ${formatSet(\n          keysIntersection\n        )} are overlapping between memory and input keys of the chain variables. This can lead to unexpected behaviour. Please use input and memory keys that don't overlap.`\n      );\n    }\n\n    const availableKeys = union(inputKeysSet, memoryKeysSet);\n    for (const chain of this.chains) {\n      let missingKeys = difference(new Set(chain.inputKeys), availableKeys);\n\n      if (chain.memory) {\n        missingKeys = difference(missingKeys, new Set(chain.memory.memoryKeys));\n      }\n\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `Missing variables for chain \"${chain._chainType()}\": ${formatSet(\n            missingKeys\n          )}. Only got the following variables: ${formatSet(availableKeys)}.`\n        );\n      }\n      const outputKeysSet = new Set(chain.outputKeys);\n      const overlappingOutputKeys = intersection(availableKeys, outputKeysSet);\n      if (overlappingOutputKeys.size > 0) {\n        throw new Error(\n          `The following output variables for chain \"${chain._chainType()}\" are overlapping: ${formatSet(\n            overlappingOutputKeys\n          )}. This can lead to unexpected behaviour.`\n        );\n      }\n\n      for (const outputKey of outputKeysSet) {\n        availableKeys.add(outputKey);\n      }\n    }\n\n    if (this.outputVariables.length === 0) {\n      if (this.returnAll) {\n        const outputKeys = difference(availableKeys, inputKeysSet);\n        this.outputVariables = Array.from(outputKeys);\n      } else {\n        this.outputVariables = this.chains[this.chains.length - 1].outputKeys;\n      }\n    } else {\n      const missingKeys = difference(\n        new Set(this.outputVariables),\n        new Set(availableKeys)\n      );\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `The following output variables were expected to be in the final chain output but were not found: ${formatSet(\n            missingKeys\n          )}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: ChainValues = {};\n    const allChainValues: ChainValues = values;\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = await chain.call(\n        allChainValues,\n        runManager?.getChild(`step_${i}`)\n      );\n      for (const key of Object.keys(input)) {\n        allChainValues[key] = input[key];\n      }\n    }\n    const output: ChainValues = {};\n    for (const key of this.outputVariables) {\n      output[key] = allChainValues[key];\n    }\n\n    return output;\n  }\n\n  _chainType() {\n    return \"sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSequentialChain) {\n    const chains: BaseChain[] = [];\n    const inputVariables: string[] = data.input_variables;\n    const outputVariables: string[] = data.output_variables;\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SequentialChain({ chains, inputVariables, outputVariables });\n  }\n\n  serialize(): SerializedSequentialChain {\n    const chains: SerializedBaseChain[] = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      input_variables: this.inputVariables,\n      output_variables: this.outputVariables,\n      chains,\n    };\n  }\n}\n\n/**\n * Interface for the input parameters of the SimpleSequentialChain class.\n */\nexport interface SimpleSequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: Array<BaseChain>;\n  /** Whether or not to trim the intermediate outputs. */\n  trimOutputs?: boolean;\n}\n\n/**\n * Simple chain where a single string output of one chain is fed directly into the next.\n * @augments BaseChain\n * @augments SimpleSequentialChainInput\n *\n * @example\n * ```ts\n * import { SimpleSequentialChain, LLMChain } from \"@langchain/classic/chains\";\n * import { OpenAI } from \"langchain/llms/openai\";\n * import { PromptTemplate } from \"langchain/prompts\";\n *\n * // This is an LLMChain to write a synopsis given a title of a play.\n * const llm = new OpenAI({ temperature: 0 });\n * const template = `You are a playwright. Given the title of play, it is your job to write a synopsis for that title.\n *\n * Title: {title}\n * Playwright: This is a synopsis for the above play:`\n * const promptTemplate = new PromptTemplate({ template, inputVariables: [\"title\"] });\n * const synopsisChain = new LLMChain({ llm, prompt: promptTemplate });\n *\n *\n * // This is an LLMChain to write a review of a play given a synopsis.\n * const reviewLLM = new OpenAI({ temperature: 0 })\n * const reviewTemplate = `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n * Play Synopsis:\n * {synopsis}\n * Review from a New York Times play critic of the above play:`\n * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: [\"synopsis\"] });\n * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });\n *\n * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})\n * const review = await overallChain.run(\"Tragedy at sunset on the beach\")\n * // the variable review contains resulting play review.\n * ```\n */\nexport class SimpleSequentialChain\n  extends BaseChain\n  implements SimpleSequentialChainInput\n{\n  static lc_name() {\n    return \"SimpleSequentialChain\";\n  }\n\n  chains: Array<BaseChain>;\n\n  inputKey = \"input\";\n\n  outputKey = \"output\";\n\n  trimOutputs: boolean;\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  constructor(fields: SimpleSequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.trimOutputs = fields.trimOutputs ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    for (const chain of this.chains) {\n      if (\n        chain.inputKeys.filter((k) => !chain.memory?.memoryKeys.includes(k))\n          .length !== 1\n      ) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one input, got ${\n            chain.inputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n      if (chain.outputKeys.length !== 1) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one output, got ${\n            chain.outputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: string = values[this.inputKey];\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = (\n        await chain.call(\n          { [chain.inputKeys[0]]: input, signal: values.signal },\n          runManager?.getChild(`step_${i}`)\n        )\n      )[chain.outputKeys[0]];\n      if (this.trimOutputs) {\n        input = input.trim();\n      }\n      await runManager?.handleText(input);\n    }\n    return { [this.outputKey]: input };\n  }\n\n  _chainType() {\n    return \"simple_sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSimpleSequentialChain) {\n    const chains: Array<BaseChain> = [];\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SimpleSequentialChain({ chains });\n  }\n\n  serialize(): SerializedSimpleSequentialChain {\n    const chains: Array<SerializedBaseChain> = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      chains,\n    };\n  }\n}\n"],"mappings":";;;;AAUA,SAAS,UAAUA,OAAoB;AACrC,QAAO,MAAM,KAAK,MAAM,CACrB,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CACpB,KAAK,KAAK;AACd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8DD,IAAa,kBAAb,MAAa,wBAAwB,UAA0C;CAC7E,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA;CAEA;CAEA;CAEA,IAAI,YAAY;AACd,SAAO,KAAK;CACb;CAED,IAAI,aAAuB;AACzB,SAAO,KAAK;CACb;CAED,YAAYC,QAA8B;EACxC,MAAM,OAAO;EACb,KAAK,SAAS,OAAO;EACrB,KAAK,iBAAiB,OAAO;EAC7B,KAAK,kBAAkB,OAAO,mBAAmB,CAAE;AACnD,MAAI,KAAK,gBAAgB,SAAS,KAAK,OAAO,UAC5C,OAAM,IAAI,MACR;EAGJ,KAAK,YAAY,OAAO,aAAa;EACrC,KAAK,iBAAiB;CACvB;;CAGD,kBAAkB;AAChB,MAAI,KAAK,OAAO,WAAW,EACzB,OAAM,IAAI,MAAM;EAGlB,MAAM,aAAa,KAAK,QAAQ,cAAc,CAAE;EAChD,MAAM,eAAe,IAAI,IAAI,KAAK;EAClC,MAAM,gBAAgB,IAAI,IAAI;EAC9B,MAAM,mBAAmB,aAAa,cAAc,cAAc;AAClE,MAAI,iBAAiB,OAAO,EAC1B,OAAM,IAAI,MACR,CAAC,oBAAoB,EAAE,UACrB,iBACD,CAAC,kKAAkK,CAAC;EAIzK,MAAM,gBAAgB,MAAM,cAAc,cAAc;AACxD,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,IAAI,cAAc,WAAW,IAAI,IAAI,MAAM,YAAY,cAAc;AAErE,OAAI,MAAM,QACR,cAAc,WAAW,aAAa,IAAI,IAAI,MAAM,OAAO,YAAY;AAGzE,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,CAAC,6BAA6B,EAAE,MAAM,YAAY,CAAC,GAAG,EAAE,UACtD,YACD,CAAC,oCAAoC,EAAE,UAAU,cAAc,CAAC,CAAC,CAAC;GAGvE,MAAM,gBAAgB,IAAI,IAAI,MAAM;GACpC,MAAM,wBAAwB,aAAa,eAAe,cAAc;AACxE,OAAI,sBAAsB,OAAO,EAC/B,OAAM,IAAI,MACR,CAAC,0CAA0C,EAAE,MAAM,YAAY,CAAC,mBAAmB,EAAE,UACnF,sBACD,CAAC,wCAAwC,CAAC;AAI/C,QAAK,MAAM,aAAa,eACtB,cAAc,IAAI,UAAU;EAE/B;AAED,MAAI,KAAK,gBAAgB,WAAW,EAClC,KAAI,KAAK,WAAW;GAClB,MAAM,aAAa,WAAW,eAAe,aAAa;GAC1D,KAAK,kBAAkB,MAAM,KAAK,WAAW;EAC9C,OACC,KAAK,kBAAkB,KAAK,OAAO,KAAK,OAAO,SAAS,GAAG;OAExD;GACL,MAAM,cAAc,WAClB,IAAI,IAAI,KAAK,kBACb,IAAI,IAAI,eACT;AACD,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,CAAC,iGAAiG,EAAE,UAClG,YACD,CAAC,CAAC,CAAC;EAGT;CACF;;CAGD,MAAM,MACJC,QACAC,YACsB;EACtB,IAAIC,QAAqB,CAAE;EAC3B,MAAMC,iBAA8B;EACpC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,KAAK;GACL,QAAQ,MAAM,MAAM,KAClB,gBACA,YAAY,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAClC;AACD,QAAK,MAAM,OAAO,OAAO,KAAK,MAAM,EAClC,eAAe,OAAO,MAAM;EAE/B;EACD,MAAMC,SAAsB,CAAE;AAC9B,OAAK,MAAM,OAAO,KAAK,iBACrB,OAAO,OAAO,eAAe;AAG/B,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYC,MAAiC;EACxD,MAAMC,SAAsB,CAAE;EAC9B,MAAMC,iBAA2B,KAAK;EACtC,MAAMC,kBAA4B,KAAK;EACvC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAM,UAAU,YAAY,gBAAgB;GACtE,OAAO,KAAK,kBAAkB;EAC/B;AACD,SAAO,IAAI,gBAAgB;GAAE;GAAQ;GAAgB;EAAiB;CACvE;CAED,YAAuC;EACrC,MAAMC,SAAgC,CAAE;AACxC,OAAK,MAAM,SAAS,KAAK,QACvB,OAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,iBAAiB,KAAK;GACtB,kBAAkB,KAAK;GACvB;EACD;CACF;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDD,IAAa,wBAAb,MAAa,8BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED;CAEA,WAAW;CAEX,YAAY;CAEZ;CAEA,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,YAAYC,QAAoC;EAC9C,MAAM,OAAO;EACb,KAAK,SAAS,OAAO;EACrB,KAAK,cAAc,OAAO,eAAe;EACzC,KAAK,iBAAiB;CACvB;;CAGD,kBAAkB;AAChB,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,OACE,MAAM,UAAU,OAAO,CAAC,MAAM,CAAC,MAAM,QAAQ,WAAW,SAAS,EAAE,CAAC,CACjE,WAAW,EAEd,OAAM,IAAI,MACR,CAAC,oEAAoE,EACnE,MAAM,UAAU,OACjB,KAAK,EAAE,MAAM,YAAY,CAAC,CAAC,CAAC;AAGjC,OAAI,MAAM,WAAW,WAAW,EAC9B,OAAM,IAAI,MACR,CAAC,qEAAqE,EACpE,MAAM,WAAW,OAClB,KAAK,EAAE,MAAM,YAAY,CAAC,CAAC,CAAC;EAGlC;CACF;;CAGD,MAAM,MACJV,QACAC,YACsB;EACtB,IAAIU,QAAgB,OAAO,KAAK;EAChC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,KAAK;GACL,SACE,MAAM,MAAM,KACV;KAAG,MAAM,UAAU,KAAK;IAAO,QAAQ,OAAO;GAAQ,GACtD,YAAY,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAClC,EACD,MAAM,WAAW;AACnB,OAAI,KAAK,aACP,QAAQ,MAAM,MAAM;GAEtB,MAAM,YAAY,WAAW,MAAM;EACpC;AACD,SAAO,GAAG,KAAK,YAAY,MAAO;CACnC;CAED,aAAa;AACX,SAAO;CACR;CAED,aAAa,YAAYC,MAAuC;EAC9D,MAAMC,SAA2B,CAAE;EACnC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAM,UAAU,YAAY,gBAAgB;GACtE,OAAO,KAAK,kBAAkB;EAC/B;AACD,SAAO,IAAI,sBAAsB,EAAE,OAAQ;CAC5C;CAED,YAA6C;EAC3C,MAAMC,SAAqC,CAAE;AAC7C,OAAK,MAAM,SAAS,KAAK,QACvB,OAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB;EACD;CACF;AACF"}
+{"version":3,"file":"sequential_chain.js","names":[],"sources":["../../src/chains/sequential_chain.ts"],"sourcesContent":["import { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport {\n  SerializedBaseChain,\n  SerializedSequentialChain,\n  SerializedSimpleSequentialChain,\n} from \"./serde.js\";\nimport { intersection, union, difference } from \"../util/set.js\";\n\nfunction formatSet(input: Set<string>) {\n  return Array.from(input)\n    .map((i) => `\"${i}\"`)\n    .join(\", \");\n}\n\n/**\n * Interface for the input parameters of the SequentialChain class.\n */\nexport interface SequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: BaseChain[];\n  /** Defines which variables should be passed as initial input to the first chain. */\n  inputVariables: string[];\n  /** Which variables should be returned as a result of executing the chain. If not specified, output of the last of the chains is used. */\n  outputVariables?: string[];\n  /** Whether or not to return all intermediate outputs and variables (excluding initial input variables). */\n  returnAll?: boolean;\n}\n\n/**\n * Chain where the outputs of one chain feed directly into next.\n * @example\n * ```typescript\n * const promptTemplate = new PromptTemplate({\n *   template: `You are a playwright. Given the title of play and the era it is set in, it is your job to write a synopsis for that title.\n * Title: {title}\n * Era: {era}\n * Playwright: This is a synopsis for the above play:`,\n *   inputVariables: [\"title\", \"era\"],\n * });\n\n * const reviewPromptTemplate = new PromptTemplate({\n *   template: `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *   \n *     Play Synopsis:\n *     {synopsis}\n *     Review from a New York Times play critic of the above play:`,\n *   inputVariables: [\"synopsis\"],\n * });\n\n * const overallChain = new SequentialChain({\n *   chains: [\n *     new LLMChain({\n *       llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: promptTemplate,\n *       outputKey: \"synopsis\",\n *     }),\n *     new LLMChain({\n *       llm: new OpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *       prompt: reviewPromptTemplate,\n *       outputKey: \"review\",\n *     }),\n *   ],\n *   inputVariables: [\"era\", \"title\"],\n *   outputVariables: [\"synopsis\", \"review\"],\n *   verbose: true,\n * });\n\n * const chainExecutionResult = await overallChain.call({\n *   title: \"Tragedy at sunset on the beach\",\n *   era: \"Victorian England\",\n * });\n * console.log(chainExecutionResult);\n * ```\n */\nexport class SequentialChain extends BaseChain implements SequentialChainInput {\n  static lc_name() {\n    return \"SequentialChain\";\n  }\n\n  chains: BaseChain[];\n\n  inputVariables: string[];\n\n  outputVariables: string[];\n\n  returnAll?: boolean | undefined;\n\n  get inputKeys() {\n    return this.inputVariables;\n  }\n\n  get outputKeys(): string[] {\n    return this.outputVariables;\n  }\n\n  constructor(fields: SequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.inputVariables = fields.inputVariables;\n    this.outputVariables = fields.outputVariables ?? [];\n    if (this.outputVariables.length > 0 && fields.returnAll) {\n      throw new Error(\n        \"Either specify variables to return using `outputVariables` or use `returnAll` param. Cannot apply both conditions at the same time.\"\n      );\n    }\n    this.returnAll = fields.returnAll ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    if (this.chains.length === 0) {\n      throw new Error(\"Sequential chain must have at least one chain.\");\n    }\n\n    const memoryKeys = this.memory?.memoryKeys ?? [];\n    const inputKeysSet = new Set(this.inputKeys);\n    const memoryKeysSet = new Set(memoryKeys);\n    const keysIntersection = intersection(inputKeysSet, memoryKeysSet);\n    if (keysIntersection.size > 0) {\n      throw new Error(\n        `The following keys: ${formatSet(\n          keysIntersection\n        )} are overlapping between memory and input keys of the chain variables. This can lead to unexpected behaviour. Please use input and memory keys that don't overlap.`\n      );\n    }\n\n    const availableKeys = union(inputKeysSet, memoryKeysSet);\n    for (const chain of this.chains) {\n      let missingKeys = difference(new Set(chain.inputKeys), availableKeys);\n\n      if (chain.memory) {\n        missingKeys = difference(missingKeys, new Set(chain.memory.memoryKeys));\n      }\n\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `Missing variables for chain \"${chain._chainType()}\": ${formatSet(\n            missingKeys\n          )}. Only got the following variables: ${formatSet(availableKeys)}.`\n        );\n      }\n      const outputKeysSet = new Set(chain.outputKeys);\n      const overlappingOutputKeys = intersection(availableKeys, outputKeysSet);\n      if (overlappingOutputKeys.size > 0) {\n        throw new Error(\n          `The following output variables for chain \"${chain._chainType()}\" are overlapping: ${formatSet(\n            overlappingOutputKeys\n          )}. This can lead to unexpected behaviour.`\n        );\n      }\n\n      for (const outputKey of outputKeysSet) {\n        availableKeys.add(outputKey);\n      }\n    }\n\n    if (this.outputVariables.length === 0) {\n      if (this.returnAll) {\n        const outputKeys = difference(availableKeys, inputKeysSet);\n        this.outputVariables = Array.from(outputKeys);\n      } else {\n        this.outputVariables = this.chains[this.chains.length - 1].outputKeys;\n      }\n    } else {\n      const missingKeys = difference(\n        new Set(this.outputVariables),\n        new Set(availableKeys)\n      );\n      if (missingKeys.size > 0) {\n        throw new Error(\n          `The following output variables were expected to be in the final chain output but were not found: ${formatSet(\n            missingKeys\n          )}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: ChainValues = {};\n    const allChainValues: ChainValues = values;\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = await chain.call(\n        allChainValues,\n        runManager?.getChild(`step_${i}`)\n      );\n      for (const key of Object.keys(input)) {\n        allChainValues[key] = input[key];\n      }\n    }\n    const output: ChainValues = {};\n    for (const key of this.outputVariables) {\n      output[key] = allChainValues[key];\n    }\n\n    return output;\n  }\n\n  _chainType() {\n    return \"sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSequentialChain) {\n    const chains: BaseChain[] = [];\n    const inputVariables: string[] = data.input_variables;\n    const outputVariables: string[] = data.output_variables;\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SequentialChain({ chains, inputVariables, outputVariables });\n  }\n\n  serialize(): SerializedSequentialChain {\n    const chains: SerializedBaseChain[] = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      input_variables: this.inputVariables,\n      output_variables: this.outputVariables,\n      chains,\n    };\n  }\n}\n\n/**\n * Interface for the input parameters of the SimpleSequentialChain class.\n */\nexport interface SimpleSequentialChainInput extends ChainInputs {\n  /** Array of chains to run as a sequence. The chains are run in order they appear in the array. */\n  chains: Array<BaseChain>;\n  /** Whether or not to trim the intermediate outputs. */\n  trimOutputs?: boolean;\n}\n\n/**\n * Simple chain where a single string output of one chain is fed directly into the next.\n * @augments BaseChain\n * @augments SimpleSequentialChainInput\n *\n * @example\n * ```ts\n * import { SimpleSequentialChain, LLMChain } from \"@langchain/classic/chains\";\n * import { OpenAI } from \"langchain/llms/openai\";\n * import { PromptTemplate } from \"langchain/prompts\";\n *\n * // This is an LLMChain to write a synopsis given a title of a play.\n * const llm = new OpenAI({ temperature: 0 });\n * const template = `You are a playwright. Given the title of play, it is your job to write a synopsis for that title.\n *\n * Title: {title}\n * Playwright: This is a synopsis for the above play:`\n * const promptTemplate = new PromptTemplate({ template, inputVariables: [\"title\"] });\n * const synopsisChain = new LLMChain({ llm, prompt: promptTemplate });\n *\n *\n * // This is an LLMChain to write a review of a play given a synopsis.\n * const reviewLLM = new OpenAI({ temperature: 0 })\n * const reviewTemplate = `You are a play critic from the New York Times. Given the synopsis of play, it is your job to write a review for that play.\n *\n * Play Synopsis:\n * {synopsis}\n * Review from a New York Times play critic of the above play:`\n * const reviewPromptTemplate = new PromptTemplate({ template: reviewTemplate, inputVariables: [\"synopsis\"] });\n * const reviewChain = new LLMChain({ llm: reviewLLM, prompt: reviewPromptTemplate });\n *\n * const overallChain = new SimpleSequentialChain({chains: [synopsisChain, reviewChain], verbose:true})\n * const review = await overallChain.run(\"Tragedy at sunset on the beach\")\n * // the variable review contains resulting play review.\n * ```\n */\nexport class SimpleSequentialChain\n  extends BaseChain\n  implements SimpleSequentialChainInput\n{\n  static lc_name() {\n    return \"SimpleSequentialChain\";\n  }\n\n  chains: Array<BaseChain>;\n\n  inputKey = \"input\";\n\n  outputKey = \"output\";\n\n  trimOutputs: boolean;\n\n  get inputKeys() {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  constructor(fields: SimpleSequentialChainInput) {\n    super(fields);\n    this.chains = fields.chains;\n    this.trimOutputs = fields.trimOutputs ?? false;\n    this._validateChains();\n  }\n\n  /** @ignore */\n  _validateChains() {\n    for (const chain of this.chains) {\n      if (\n        chain.inputKeys.filter((k) => !chain.memory?.memoryKeys.includes(k))\n          .length !== 1\n      ) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one input, got ${\n            chain.inputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n      if (chain.outputKeys.length !== 1) {\n        throw new Error(\n          `Chains used in SimpleSequentialChain should all have one output, got ${\n            chain.outputKeys.length\n          } for ${chain._chainType()}.`\n        );\n      }\n    }\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    let input: string = values[this.inputKey];\n    let i = 0;\n    for (const chain of this.chains) {\n      i += 1;\n      input = (\n        await chain.call(\n          { [chain.inputKeys[0]]: input, signal: values.signal },\n          runManager?.getChild(`step_${i}`)\n        )\n      )[chain.outputKeys[0]];\n      if (this.trimOutputs) {\n        input = input.trim();\n      }\n      await runManager?.handleText(input);\n    }\n    return { [this.outputKey]: input };\n  }\n\n  _chainType() {\n    return \"simple_sequential_chain\" as const;\n  }\n\n  static async deserialize(data: SerializedSimpleSequentialChain) {\n    const chains: Array<BaseChain> = [];\n    const serializedChains = data.chains;\n    for (const serializedChain of serializedChains) {\n      const deserializedChain = await BaseChain.deserialize(serializedChain);\n      chains.push(deserializedChain);\n    }\n    return new SimpleSequentialChain({ chains });\n  }\n\n  serialize(): SerializedSimpleSequentialChain {\n    const chains: Array<SerializedBaseChain> = [];\n    for (const chain of this.chains) {\n      chains.push(chain.serialize());\n    }\n    return {\n      _type: this._chainType(),\n      chains,\n    };\n  }\n}\n"],"mappings":";;;;AAUA,SAAS,UAAU,OAAoB;AACrC,QAAO,MAAM,KAAK,MAAM,CACrB,KAAK,MAAM,IAAI,EAAE,GAAG,CACpB,KAAK,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+Df,IAAa,kBAAb,MAAa,wBAAwB,UAA0C;CAC7E,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA;CAEA;CAEA;CAEA,IAAI,YAAY;AACd,SAAO,KAAK;;CAGd,IAAI,aAAuB;AACzB,SAAO,KAAK;;CAGd,YAAY,QAA8B;AACxC,QAAM,OAAO;AACb,OAAK,SAAS,OAAO;AACrB,OAAK,iBAAiB,OAAO;AAC7B,OAAK,kBAAkB,OAAO,mBAAmB,EAAE;AACnD,MAAI,KAAK,gBAAgB,SAAS,KAAK,OAAO,UAC5C,OAAM,IAAI,MACR,sIACD;AAEH,OAAK,YAAY,OAAO,aAAa;AACrC,OAAK,iBAAiB;;;CAIxB,kBAAkB;AAChB,MAAI,KAAK,OAAO,WAAW,EACzB,OAAM,IAAI,MAAM,iDAAiD;EAGnE,MAAM,aAAa,KAAK,QAAQ,cAAc,EAAE;EAChD,MAAM,eAAe,IAAI,IAAI,KAAK,UAAU;EAC5C,MAAM,gBAAgB,IAAI,IAAI,WAAW;EACzC,MAAM,mBAAmB,aAAa,cAAc,cAAc;AAClE,MAAI,iBAAiB,OAAO,EAC1B,OAAM,IAAI,MACR,uBAAuB,UACrB,iBACD,CAAC,oKACH;EAGH,MAAM,gBAAgB,MAAM,cAAc,cAAc;AACxD,OAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,IAAI,cAAc,WAAW,IAAI,IAAI,MAAM,UAAU,EAAE,cAAc;AAErE,OAAI,MAAM,OACR,eAAc,WAAW,aAAa,IAAI,IAAI,MAAM,OAAO,WAAW,CAAC;AAGzE,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,gCAAgC,MAAM,YAAY,CAAC,KAAK,UACtD,YACD,CAAC,sCAAsC,UAAU,cAAc,CAAC,GAClE;GAEH,MAAM,gBAAgB,IAAI,IAAI,MAAM,WAAW;GAC/C,MAAM,wBAAwB,aAAa,eAAe,cAAc;AACxE,OAAI,sBAAsB,OAAO,EAC/B,OAAM,IAAI,MACR,6CAA6C,MAAM,YAAY,CAAC,qBAAqB,UACnF,sBACD,CAAC,0CACH;AAGH,QAAK,MAAM,aAAa,cACtB,eAAc,IAAI,UAAU;;AAIhC,MAAI,KAAK,gBAAgB,WAAW,EAClC,KAAI,KAAK,WAAW;GAClB,MAAM,aAAa,WAAW,eAAe,aAAa;AAC1D,QAAK,kBAAkB,MAAM,KAAK,WAAW;QAE7C,MAAK,kBAAkB,KAAK,OAAO,KAAK,OAAO,SAAS,GAAG;OAExD;GACL,MAAM,cAAc,WAClB,IAAI,IAAI,KAAK,gBAAgB,EAC7B,IAAI,IAAI,cAAc,CACvB;AACD,OAAI,YAAY,OAAO,EACrB,OAAM,IAAI,MACR,oGAAoG,UAClG,YACD,CAAC,GACH;;;;CAMP,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,QAAqB,EAAE;EAC3B,MAAM,iBAA8B;EACpC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,QAAK;AACL,WAAQ,MAAM,MAAM,KAClB,gBACA,YAAY,SAAS,QAAQ,IAAI,CAClC;AACD,QAAK,MAAM,OAAO,OAAO,KAAK,MAAM,CAClC,gBAAe,OAAO,MAAM;;EAGhC,MAAM,SAAsB,EAAE;AAC9B,OAAK,MAAM,OAAO,KAAK,gBACrB,QAAO,OAAO,eAAe;AAG/B,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAiC;EACxD,MAAM,SAAsB,EAAE;EAC9B,MAAM,iBAA2B,KAAK;EACtC,MAAM,kBAA4B,KAAK;EACvC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAM,UAAU,YAAY,gBAAgB;AACtE,UAAO,KAAK,kBAAkB;;AAEhC,SAAO,IAAI,gBAAgB;GAAE;GAAQ;GAAgB;GAAiB,CAAC;;CAGzE,YAAuC;EACrC,MAAM,SAAgC,EAAE;AACxC,OAAK,MAAM,SAAS,KAAK,OACvB,QAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB,iBAAiB,KAAK;GACtB,kBAAkB,KAAK;GACvB;GACD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDL,IAAa,wBAAb,MAAa,8BACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT;CAEA,WAAW;CAEX,YAAY;CAEZ;CAEA,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,UAAU;;CAGzB,YAAY,QAAoC;AAC9C,QAAM,OAAO;AACb,OAAK,SAAS,OAAO;AACrB,OAAK,cAAc,OAAO,eAAe;AACzC,OAAK,iBAAiB;;;CAIxB,kBAAkB;AAChB,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,OACE,MAAM,UAAU,QAAQ,MAAM,CAAC,MAAM,QAAQ,WAAW,SAAS,EAAE,CAAC,CACjE,WAAW,EAEd,OAAM,IAAI,MACR,uEACE,MAAM,UAAU,OACjB,OAAO,MAAM,YAAY,CAAC,GAC5B;AAEH,OAAI,MAAM,WAAW,WAAW,EAC9B,OAAM,IAAI,MACR,wEACE,MAAM,WAAW,OAClB,OAAO,MAAM,YAAY,CAAC,GAC5B;;;;CAMP,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,QAAgB,OAAO,KAAK;EAChC,IAAI,IAAI;AACR,OAAK,MAAM,SAAS,KAAK,QAAQ;AAC/B,QAAK;AACL,YACE,MAAM,MAAM,KACV;KAAG,MAAM,UAAU,KAAK;IAAO,QAAQ,OAAO;IAAQ,EACtD,YAAY,SAAS,QAAQ,IAAI,CAClC,EACD,MAAM,WAAW;AACnB,OAAI,KAAK,YACP,SAAQ,MAAM,MAAM;AAEtB,SAAM,YAAY,WAAW,MAAM;;AAErC,SAAO,GAAG,KAAK,YAAY,OAAO;;CAGpC,aAAa;AACX,SAAO;;CAGT,aAAa,YAAY,MAAuC;EAC9D,MAAM,SAA2B,EAAE;EACnC,MAAM,mBAAmB,KAAK;AAC9B,OAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAM,UAAU,YAAY,gBAAgB;AACtE,UAAO,KAAK,kBAAkB;;AAEhC,SAAO,IAAI,sBAAsB,EAAE,QAAQ,CAAC;;CAG9C,YAA6C;EAC3C,MAAM,SAAqC,EAAE;AAC7C,OAAK,MAAM,SAAS,KAAK,OACvB,QAAO,KAAK,MAAM,WAAW,CAAC;AAEhC,SAAO;GACL,OAAO,KAAK,YAAY;GACxB;GACD"}

package/dist/chains/serde.d.cts

@@ -2,7 +2,6 @@ import { SerializedLLM } from "@langchain/core/language_models/llms";
 import { SerializedBasePromptTemplate } from "@langchain/core/prompts";
 
 //#region src/chains/serde.d.ts
-
 /**
  * Represents the serialized form of an LLMChain. It includes properties
  * such as `_type`, `llm`, and `prompt`.

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

@@ -1 +1 @@
-{"version":3,"file":"serde.d.cts","names":["SerializedLLM","SerializedBasePromptTemplate","SerializedLLMChain","SerializedSequentialChain","SerializedBaseChain","SerializedSimpleSequentialChain","Array","SerializedVectorDBQAChain","SerializedAPIChain","SerializedStuffDocumentsChain","SerializedChatVectorDBQAChain","SerializedMapReduceDocumentsChain","SerializedRefineDocumentsChain","SerializedAnalyzeDocumentChain","SerializedConstitutionalPrinciple","SerializedConstitutionalChain"],"sources":["../../src/chains/serde.d.ts"],"sourcesContent":["import type { SerializedLLM } from \"@langchain/core/language_models/llms\";\nimport { SerializedBasePromptTemplate } from \"@langchain/core/prompts\";\n/**\n * Represents the serialized form of an LLMChain. It includes properties\n * such as `_type`, `llm`, and `prompt`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedLLMChain = {\n    _type: \"llm_chain\";\n    llm?: SerializedLLM;\n    prompt?: SerializedBasePromptTemplate;\n};\n/**\n * Represents the serialized form of a SequentialChain. It includes\n * properties such as `_type`, `input_variables`, `output_variables`, and\n * `chains`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedSequentialChain = {\n    _type: \"sequential_chain\";\n    input_variables: string[];\n    output_variables: string[];\n    chains: SerializedBaseChain[];\n};\n/**\n * Represents the serialized form of a SimpleSequentialChain. It includes\n * properties such as `_type` and `chains`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedSimpleSequentialChain = {\n    _type: \"simple_sequential_chain\";\n    chains: Array<SerializedBaseChain>;\n};\n/**\n * Represents the serialized form of a VectorDBQAChain. It includes\n * properties such as `_type`, `k`, and `combine_documents_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedVectorDBQAChain = {\n    _type: \"vector_db_qa\";\n    k: number;\n    combine_documents_chain: SerializedBaseChain;\n};\n/**\n * Represents the serialized form of an APIChain. It includes properties\n * such as `_type`, `api_request_chain`, `api_answer_chain`, and\n * `api_docs`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedAPIChain = {\n    _type: \"api_chain\";\n    api_request_chain: SerializedLLMChain;\n    api_answer_chain: SerializedLLMChain;\n    api_docs: string;\n};\n/**\n * Represents the serialized form of a StuffDocumentsChain. It includes\n * properties such as `_type` and `llm_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedStuffDocumentsChain = {\n    _type: \"stuff_documents_chain\";\n    llm_chain?: SerializedLLMChain;\n};\n/**\n * Represents the serialized form of a ChatVectorDBQAChain. It includes\n * properties such as `_type`, `k`, `combine_documents_chain`, and\n * `question_generator`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedChatVectorDBQAChain = {\n    _type: \"chat-vector-db\";\n    k: number;\n    combine_documents_chain: SerializedBaseChain;\n    question_generator: SerializedLLMChain;\n};\n/**\n * Represents the serialized form of a MapReduceDocumentsChain. It\n * includes properties such as `_type`, `llm_chain`, and\n * `combine_document_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedMapReduceDocumentsChain = {\n    _type: \"map_reduce_documents_chain\";\n    llm_chain?: SerializedLLMChain;\n    combine_document_chain?: SerializedStuffDocumentsChain;\n};\n/**\n * Represents the serialized form of a RefineDocumentsChain. It includes\n * properties such as `_type`, `llm_chain`, and `refine_llm_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedRefineDocumentsChain = {\n    _type: \"refine_documents_chain\";\n    llm_chain?: SerializedLLMChain;\n    refine_llm_chain?: SerializedLLMChain;\n};\n/**\n * Represents the serialized form of an AnalyzeDocumentChain. It includes\n * properties such as `_type` and `combine_document_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedAnalyzeDocumentChain = {\n    _type: \"analyze_document_chain\";\n    combine_document_chain?: SerializedBaseChain;\n};\n/**\n * Represents the serialized form of a ConstitutionalPrinciple. It\n * includes properties such as `_type`, `critiqueRequest`,\n * `revisionRequest`, and `name`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedConstitutionalPrinciple = {\n    _type: \"constitutional_principle\";\n    critiqueRequest: string;\n    revisionRequest: string;\n    name: string;\n};\n/**\n * Represents the serialized form of a ConstitutionalChain. It includes\n * properties such as `_type`, `chain`, `critiqueChain`, `revisionChain`,\n * and `ConstitutionalPrinciple`.\n *\n * @deprecated\n */\nexport type SerializedConstitutionalChain = {\n    _type: \"constitutional_chain\";\n    chain?: SerializedLLMChain;\n    critiqueChain?: SerializedBaseChain;\n    revisionChain?: SerializedBaseChain;\n    ConstitutionalPrinciple?: SerializedConstitutionalPrinciple[];\n};\n/**\n * Represents the serialized form of a BaseChain. It can be one of the\n * above serialized chain types.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedBaseChain = SerializedLLMChain | SerializedSequentialChain | SerializedSimpleSequentialChain | SerializedVectorDBQAChain | SerializedAPIChain | SerializedStuffDocumentsChain | SerializedChatVectorDBQAChain | SerializedMapReduceDocumentsChain | SerializedAnalyzeDocumentChain | SerializedRefineDocumentsChain | SerializedConstitutionalChain;\n//# sourceMappingURL=serde.d.ts.map"],"mappings":";;;;;;;AAQA;AAYA;AAYA;AAUA;AAYYQ,KA9CAN,kBAAAA,GA8CkB;EAYlBO,KAAAA,EAAAA,WAAAA;EAWAC,GAAAA,CAAAA,EAnEFV,aAmEEU;EAaAC,MAAAA,CAAAA,EA/ECV,4BA+EgC;AAW7C,CAAA;AAWA;AAWA;AAaA;;;;;AAK+D,KAzHnDE,yBAAAA,GAyHmD;EAQnDC,KAAAA,EAAAA,kBAAmB;EAAGF,eAAAA,EAAAA,MAAAA,EAAAA;EAAqBC,gBAAAA,EAAAA,MAAAA,EAAAA;EAA4BE,MAAAA,EA7HvED,mBA6HuEC,EAAAA;CAAkCE;;;;;;;AAAuNQ,KArHhUV,+BAAAA,GAqHgUU;EAA6B,KAAA,EAAA,yBAAA;UAnH7VT,MAAMF;;;;;;;;KAQNG,yBAAAA;;;2BAGiBH;;;;;;;;;KASjBI,kBAAAA;;qBAEWN;oBACDA;;;;;;;;;KASVO,6BAAAA;;cAEIP;;;;;;;;;KASJQ,6BAAAA;;;2BAGiBN;sBACLF;;;;;;;;;KASZS,iCAAAA;;cAEIT;2BACaO;;;;;;;;KAQjBG,8BAAAA;;cAEIV;qBACOA;;;;;;;;KAQXW,8BAAAA;;2BAEiBT;;;;;;;;;KASjBU,iCAAAA;;;;;;;;;;;;;KAaAC,6BAAAA;;UAEAb;kBACQE;kBACAA;4BACUU;;;;;;;;KAQlBV,mBAAAA,GAAsBF,qBAAqBC,4BAA4BE,kCAAkCE,4BAA4BC,qBAAqBC,gCAAgCC,gCAAgCC,oCAAoCE,iCAAiCD,iCAAiCG"}
+{"version":3,"file":"serde.d.cts","names":[],"sources":["../../src/chains/serde.ts"],"mappings":";;;;;;AASA;;;;KAAY,kBAAA;EACV,KAAA;EACA,GAAA,GAAM,aAAA;EACN,MAAA,GAAS,4BAAA;AAAA;;;AAUX;;;;;KAAY,yBAAA;EACV,KAAA;EACA,eAAA;EACA,gBAAA;EACA,MAAA,EAAQ,mBAAA;AAAA;AASV;;;;;;AAAA,KAAY,+BAAA;EACV,KAAA;EACA,MAAA,EAAQ,KAAA,CAAM,mBAAA;AAAA;AAShB;;;;;;AAAA,KAAY,yBAAA;EACV,KAAA;EACA,CAAA;EACA,uBAAA,EAAyB,mBAAA;AAAA;;;;;;;;KAUf,kBAAA;EACV,KAAA;EACA,iBAAA,EAAmB,kBAAA;EACnB,gBAAA,EAAkB,kBAAA;EAClB,QAAA;AAAA;;;;;;;KASU,6BAAA;EACV,KAAA;EACA,SAAA,GAAY,kBAAA;AAAA;;;;;;;;KAUF,6BAAA;EACV,KAAA;EACA,CAAA;EACA,uBAAA,EAAyB,mBAAA;EACzB,kBAAA,EAAoB,kBAAA;AAAA;;;;;;;;KAUV,iCAAA;EACV,KAAA;EACA,SAAA,GAAY,kBAAA;EACZ,sBAAA,GAAyB,6BAAA;AAAA;;;;;;;KASf,8BAAA;EACV,KAAA;EACA,SAAA,GAAY,kBAAA;EACZ,gBAAA,GAAmB,kBAAA;AAAA;;;;;AAqBrB;;KAZY,8BAAA;EACV,KAAA;EACA,sBAAA,GAAyB,mBAAA;AAAA;;;;;AAwB3B;;;KAdY,iCAAA;EACV,KAAA;EACA,eAAA;EACA,eAAA;EACA,IAAA;AAAA;;;;;;;;KAUU,6BAAA;EACV,KAAA;EACA,KAAA,GAAQ,kBAAA;EACR,aAAA,GAAgB,mBAAA;EAChB,aAAA,GAAgB,mBAAA;EAChB,uBAAA,GAA0B,iCAAA;AAAA;;;;;;;KAShB,mBAAA,GACR,kBAAA,GACA,yBAAA,GACA,+BAAA,GACA,yBAAA,GACA,kBAAA,GACA,6BAAA,GACA,6BAAA,GACA,iCAAA,GACA,8BAAA,GACA,8BAAA,GACA,6BAAA"}

package/dist/chains/serde.d.ts

@@ -2,7 +2,6 @@ import { SerializedBasePromptTemplate } from "@langchain/core/prompts";
 import { SerializedLLM } from "@langchain/core/language_models/llms";
 
 //#region src/chains/serde.d.ts
-
 /**
  * Represents the serialized form of an LLMChain. It includes properties
  * such as `_type`, `llm`, and `prompt`.

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

@@ -1 +1 @@
-{"version":3,"file":"serde.d.ts","names":["SerializedLLM","SerializedBasePromptTemplate","SerializedLLMChain","SerializedSequentialChain","SerializedBaseChain","SerializedSimpleSequentialChain","Array","SerializedVectorDBQAChain","SerializedAPIChain","SerializedStuffDocumentsChain","SerializedChatVectorDBQAChain","SerializedMapReduceDocumentsChain","SerializedRefineDocumentsChain","SerializedAnalyzeDocumentChain","SerializedConstitutionalPrinciple","SerializedConstitutionalChain"],"sources":["../../src/chains/serde.d.ts"],"sourcesContent":["import type { SerializedLLM } from \"@langchain/core/language_models/llms\";\nimport { SerializedBasePromptTemplate } from \"@langchain/core/prompts\";\n/**\n * Represents the serialized form of an LLMChain. It includes properties\n * such as `_type`, `llm`, and `prompt`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedLLMChain = {\n    _type: \"llm_chain\";\n    llm?: SerializedLLM;\n    prompt?: SerializedBasePromptTemplate;\n};\n/**\n * Represents the serialized form of a SequentialChain. It includes\n * properties such as `_type`, `input_variables`, `output_variables`, and\n * `chains`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedSequentialChain = {\n    _type: \"sequential_chain\";\n    input_variables: string[];\n    output_variables: string[];\n    chains: SerializedBaseChain[];\n};\n/**\n * Represents the serialized form of a SimpleSequentialChain. It includes\n * properties such as `_type` and `chains`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedSimpleSequentialChain = {\n    _type: \"simple_sequential_chain\";\n    chains: Array<SerializedBaseChain>;\n};\n/**\n * Represents the serialized form of a VectorDBQAChain. It includes\n * properties such as `_type`, `k`, and `combine_documents_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedVectorDBQAChain = {\n    _type: \"vector_db_qa\";\n    k: number;\n    combine_documents_chain: SerializedBaseChain;\n};\n/**\n * Represents the serialized form of an APIChain. It includes properties\n * such as `_type`, `api_request_chain`, `api_answer_chain`, and\n * `api_docs`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedAPIChain = {\n    _type: \"api_chain\";\n    api_request_chain: SerializedLLMChain;\n    api_answer_chain: SerializedLLMChain;\n    api_docs: string;\n};\n/**\n * Represents the serialized form of a StuffDocumentsChain. It includes\n * properties such as `_type` and `llm_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedStuffDocumentsChain = {\n    _type: \"stuff_documents_chain\";\n    llm_chain?: SerializedLLMChain;\n};\n/**\n * Represents the serialized form of a ChatVectorDBQAChain. It includes\n * properties such as `_type`, `k`, `combine_documents_chain`, and\n * `question_generator`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedChatVectorDBQAChain = {\n    _type: \"chat-vector-db\";\n    k: number;\n    combine_documents_chain: SerializedBaseChain;\n    question_generator: SerializedLLMChain;\n};\n/**\n * Represents the serialized form of a MapReduceDocumentsChain. It\n * includes properties such as `_type`, `llm_chain`, and\n * `combine_document_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedMapReduceDocumentsChain = {\n    _type: \"map_reduce_documents_chain\";\n    llm_chain?: SerializedLLMChain;\n    combine_document_chain?: SerializedStuffDocumentsChain;\n};\n/**\n * Represents the serialized form of a RefineDocumentsChain. It includes\n * properties such as `_type`, `llm_chain`, and `refine_llm_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedRefineDocumentsChain = {\n    _type: \"refine_documents_chain\";\n    llm_chain?: SerializedLLMChain;\n    refine_llm_chain?: SerializedLLMChain;\n};\n/**\n * Represents the serialized form of an AnalyzeDocumentChain. It includes\n * properties such as `_type` and `combine_document_chain`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedAnalyzeDocumentChain = {\n    _type: \"analyze_document_chain\";\n    combine_document_chain?: SerializedBaseChain;\n};\n/**\n * Represents the serialized form of a ConstitutionalPrinciple. It\n * includes properties such as `_type`, `critiqueRequest`,\n * `revisionRequest`, and `name`.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedConstitutionalPrinciple = {\n    _type: \"constitutional_principle\";\n    critiqueRequest: string;\n    revisionRequest: string;\n    name: string;\n};\n/**\n * Represents the serialized form of a ConstitutionalChain. It includes\n * properties such as `_type`, `chain`, `critiqueChain`, `revisionChain`,\n * and `ConstitutionalPrinciple`.\n *\n * @deprecated\n */\nexport type SerializedConstitutionalChain = {\n    _type: \"constitutional_chain\";\n    chain?: SerializedLLMChain;\n    critiqueChain?: SerializedBaseChain;\n    revisionChain?: SerializedBaseChain;\n    ConstitutionalPrinciple?: SerializedConstitutionalPrinciple[];\n};\n/**\n * Represents the serialized form of a BaseChain. It can be one of the\n * above serialized chain types.\n *\n * @deprecated Use newer {@link https://api.js.langchain.com/functions/langchain.load.load.html | serialization methods}.\n */\nexport type SerializedBaseChain = SerializedLLMChain | SerializedSequentialChain | SerializedSimpleSequentialChain | SerializedVectorDBQAChain | SerializedAPIChain | SerializedStuffDocumentsChain | SerializedChatVectorDBQAChain | SerializedMapReduceDocumentsChain | SerializedAnalyzeDocumentChain | SerializedRefineDocumentsChain | SerializedConstitutionalChain;\n//# sourceMappingURL=serde.d.ts.map"],"mappings":";;;;;;;AAQA;AAYA;AAYA;AAUA;AAYYQ,KA9CAN,kBAAAA,GA8CkB;EAYlBO,KAAAA,EAAAA,WAAAA;EAWAC,GAAAA,CAAAA,EAnEFV,aAmEEU;EAaAC,MAAAA,CAAAA,EA/ECV,4BA+EgC;AAW7C,CAAA;AAWA;AAWA;AAaA;;;;;AAK+D,KAzHnDE,yBAAAA,GAyHmD;EAQnDC,KAAAA,EAAAA,kBAAmB;EAAGF,eAAAA,EAAAA,MAAAA,EAAAA;EAAqBC,gBAAAA,EAAAA,MAAAA,EAAAA;EAA4BE,MAAAA,EA7HvED,mBA6HuEC,EAAAA;CAAkCE;;;;;;;AAAuNQ,KArHhUV,+BAAAA,GAqHgUU;EAA6B,KAAA,EAAA,yBAAA;UAnH7VT,MAAMF;;;;;;;;KAQNG,yBAAAA;;;2BAGiBH;;;;;;;;;KASjBI,kBAAAA;;qBAEWN;oBACDA;;;;;;;;;KASVO,6BAAAA;;cAEIP;;;;;;;;;KASJQ,6BAAAA;;;2BAGiBN;sBACLF;;;;;;;;;KASZS,iCAAAA;;cAEIT;2BACaO;;;;;;;;KAQjBG,8BAAAA;;cAEIV;qBACOA;;;;;;;;KAQXW,8BAAAA;;2BAEiBT;;;;;;;;;KASjBU,iCAAAA;;;;;;;;;;;;;KAaAC,6BAAAA;;UAEAb;kBACQE;kBACAA;4BACUU;;;;;;;;KAQlBV,mBAAAA,GAAsBF,qBAAqBC,4BAA4BE,kCAAkCE,4BAA4BC,qBAAqBC,gCAAgCC,gCAAgCC,oCAAoCE,iCAAiCD,iCAAiCG"}
+{"version":3,"file":"serde.d.ts","names":[],"sources":["../../src/chains/serde.ts"],"mappings":";;;;;;AASA;;;;KAAY,kBAAA;EACV,KAAA;EACA,GAAA,GAAM,aAAA;EACN,MAAA,GAAS,4BAAA;AAAA;;;AAUX;;;;;KAAY,yBAAA;EACV,KAAA;EACA,eAAA;EACA,gBAAA;EACA,MAAA,EAAQ,mBAAA;AAAA;AASV;;;;;;AAAA,KAAY,+BAAA;EACV,KAAA;EACA,MAAA,EAAQ,KAAA,CAAM,mBAAA;AAAA;AAShB;;;;;;AAAA,KAAY,yBAAA;EACV,KAAA;EACA,CAAA;EACA,uBAAA,EAAyB,mBAAA;AAAA;;;;;;;;KAUf,kBAAA;EACV,KAAA;EACA,iBAAA,EAAmB,kBAAA;EACnB,gBAAA,EAAkB,kBAAA;EAClB,QAAA;AAAA;;;;;;;KASU,6BAAA;EACV,KAAA;EACA,SAAA,GAAY,kBAAA;AAAA;;;;;;;;KAUF,6BAAA;EACV,KAAA;EACA,CAAA;EACA,uBAAA,EAAyB,mBAAA;EACzB,kBAAA,EAAoB,kBAAA;AAAA;;;;;;;;KAUV,iCAAA;EACV,KAAA;EACA,SAAA,GAAY,kBAAA;EACZ,sBAAA,GAAyB,6BAAA;AAAA;;;;;;;KASf,8BAAA;EACV,KAAA;EACA,SAAA,GAAY,kBAAA;EACZ,gBAAA,GAAmB,kBAAA;AAAA;;;;;AAqBrB;;KAZY,8BAAA;EACV,KAAA;EACA,sBAAA,GAAyB,mBAAA;AAAA;;;;;AAwB3B;;;KAdY,iCAAA;EACV,KAAA;EACA,eAAA;EACA,eAAA;EACA,IAAA;AAAA;;;;;;;;KAUU,6BAAA;EACV,KAAA;EACA,KAAA,GAAQ,kBAAA;EACR,aAAA,GAAgB,mBAAA;EAChB,aAAA,GAAgB,mBAAA;EAChB,uBAAA,GAA0B,iCAAA;AAAA;;;;;;;KAShB,mBAAA,GACR,kBAAA,GACA,yBAAA,GACA,+BAAA,GACA,yBAAA,GACA,kBAAA,GACA,6BAAA,GACA,6BAAA,GACA,iCAAA,GACA,8BAAA,GACA,8BAAA,GACA,6BAAA"}

package/dist/chains/sql_db/index.cjs

@@ -1,10 +1,10 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
 const require_sql_db_prompt = require('./sql_db_prompt.cjs');
 const require_sql_db_chain = require('./sql_db_chain.cjs');
 
 //#region src/chains/sql_db/index.ts
-var sql_db_exports = {};
-require_rolldown_runtime.__export(sql_db_exports, {
+var sql_db_exports = /* @__PURE__ */ require_runtime.__exportAll({
 	DEFAULT_SQL_DATABASE_PROMPT: () => require_sql_db_prompt.DEFAULT_SQL_DATABASE_PROMPT,
 	SQL_MSSQL_PROMPT: () => require_sql_db_prompt.SQL_MSSQL_PROMPT,
 	SQL_MYSQL_PROMPT: () => require_sql_db_prompt.SQL_MYSQL_PROMPT,

package/dist/chains/sql_db/index.js

@@ -1,10 +1,9 @@
-import { __export } from "../../_virtual/rolldown_runtime.js";
+import { __exportAll } from "../../_virtual/_rolldown/runtime.js";
 import { DEFAULT_SQL_DATABASE_PROMPT, SQL_MSSQL_PROMPT, SQL_MYSQL_PROMPT, SQL_POSTGRES_PROMPT, SQL_PROMPTS_MAP, SQL_SAP_HANA_PROMPT, SQL_SQLITE_PROMPT } from "./sql_db_prompt.js";
 import { SqlDatabaseChain, createSqlQueryChain } from "./sql_db_chain.js";
 
 //#region src/chains/sql_db/index.ts
-var sql_db_exports = {};
-__export(sql_db_exports, {
+var sql_db_exports = /* @__PURE__ */ __exportAll({
 	DEFAULT_SQL_DATABASE_PROMPT: () => DEFAULT_SQL_DATABASE_PROMPT,
 	SQL_MSSQL_PROMPT: () => SQL_MSSQL_PROMPT,
 	SQL_MYSQL_PROMPT: () => SQL_MYSQL_PROMPT,

package/dist/chains/sql_db/sql_db_chain.cjs

@@ -1,11 +1,11 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const require_sql_db_prompt = require('./sql_db_prompt.cjs');
-const require_sql_utils = require('../../util/sql_utils.cjs');
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
 const require_base = require('../base.cjs');
 const require_llm_chain = require('../llm_chain.cjs');
-const __langchain_core_runnables = require_rolldown_runtime.__toESM(require("@langchain/core/runnables"));
-const __langchain_core_language_models_base = require_rolldown_runtime.__toESM(require("@langchain/core/language_models/base"));
-const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
+const require_sql_db_prompt = require('./sql_db_prompt.cjs');
+const require_sql_utils = require('../../util/sql_utils.cjs');
+let _langchain_core_runnables = require("@langchain/core/runnables");
+let _langchain_core_language_models_base = require("@langchain/core/language_models/base");
+let _langchain_core_output_parsers = require("@langchain/core/output_parsers");
 
 //#region src/chains/sql_db/sql_db_chain.ts
 /**
@@ -64,8 +64,7 @@ var SqlDatabaseChain = class extends require_base.BaseChain {
 			memory: this.memory
 		});
 		if (!(this.inputKey in values)) throw new Error(`Question key ${this.inputKey} not found.`);
-		const question = values[this.inputKey];
-		let inputText = `${question}\nSQLQuery:`;
+		let inputText = `${values[this.inputKey]}\nSQLQuery:`;
 		const tablesToUse = values.table_names_to_use;
 		const tableInfo = await this.database.getTableInfo(tablesToUse);
 		const llmInputs = {
@@ -114,13 +113,10 @@ var SqlDatabaseChain = class extends require_base.BaseChain {
 	async verifyNumberOfTokens(inputText, tableinfo) {
 		if (this.llm._llmType() !== "openai") return;
 		const llm = this.llm;
-		const promptTemplate = this.prompt.template;
-		const stringWeSend = `${inputText}${promptTemplate}${tableinfo}`;
-		const maxToken = await (0, __langchain_core_language_models_base.calculateMaxTokens)({
-			prompt: stringWeSend,
+		if (await (0, _langchain_core_language_models_base.calculateMaxTokens)({
+			prompt: `${inputText}${this.prompt.template}${tableinfo}`,
 			modelName: llm.model
-		});
-		if (maxToken < (llm.maxTokens ?? -1)) throw new Error(`The combination of the database structure and your question is too big for the model ${llm.model} which can compute only a max tokens of ${(0, __langchain_core_language_models_base.getModelContextSize)(llm.model)}.
+		}) < (llm.maxTokens ?? -1)) throw new Error(`The combination of the database structure and your question is too big for the model ${llm.model} which can compute only a max tokens of ${(0, _langchain_core_language_models_base.getModelContextSize)(llm.model)}.
       We suggest you to use the includeTables parameters when creating the SqlDatabase object to select only a subset of the tables. You can also use a model which can handle more tokens.`);
 	}
 };
@@ -168,15 +164,14 @@ async function createSqlQueryChain({ llm, db, prompt, k = 5, dialect }) {
 	]), new Set(promptToUse.inputVariables)).size > 0) throw new Error(`Prompt must have input variables: 'input', 'top_k', 'table_info'. Received prompt with input variables: ${promptToUse.inputVariables}. Full prompt:\n\n${promptToUse}`);
 	if (promptToUse.inputVariables.includes("dialect")) promptToUse = await promptToUse.partial({ dialect });
 	promptToUse = await promptToUse.partial({ top_k: k.toString() });
-	const inputs = {
-		input: (x) => {
-			if ("question" in x) return `${x.question}\nSQLQuery: `;
-			throw new Error("Input must include a question property.");
-		},
-		table_info: async (x) => db.getTableInfo(x.tableNamesToUse)
-	};
-	return __langchain_core_runnables.RunnableSequence.from([
-		__langchain_core_runnables.RunnablePassthrough.assign(inputs),
+	return _langchain_core_runnables.RunnableSequence.from([
+		_langchain_core_runnables.RunnablePassthrough.assign({
+			input: (x) => {
+				if ("question" in x) return `${x.question}\nSQLQuery: `;
+				throw new Error("Input must include a question property.");
+			},
+			table_info: async (x) => db.getTableInfo(x.tableNamesToUse)
+		}),
 		(x) => {
 			const newInputs = { ...x };
 			delete newInputs.question;
@@ -185,7 +180,7 @@ async function createSqlQueryChain({ llm, db, prompt, k = 5, dialect }) {
 		},
 		promptToUse,
 		llm.withConfig({ stop: ["\nSQLResult:"] }),
-		new __langchain_core_output_parsers.StringOutputParser(),
+		new _langchain_core_output_parsers.StringOutputParser(),
 		strip
 	]);
 }

package/dist/chains/sql_db/sql_db_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"sql_db_chain.cjs","names":["BaseChain","DEFAULT_SQL_DATABASE_PROMPT","fields: SqlDatabaseChainInput","getPromptTemplateFromDataSource","values: ChainValues","runManager?: CallbackManagerForChainRun","LLMChain","question: string","inputText: string","tableinfo: string","text: string","setA: Set<string>","setB: Set<string>","promptToUse: BasePromptTemplate","SQL_PROMPTS_MAP","x: Record<string, unknown>","RunnableSequence","RunnablePassthrough","StringOutputParser"],"sources":["../../../src/chains/sql_db/sql_db_chain.ts"],"sourcesContent":["import type {\n  BaseLanguageModel,\n  BaseLanguageModelInterface,\n} from \"@langchain/core/language_models/base\";\nimport type { OpenAI, TiktokenModel } from \"@langchain/openai\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport {\n  calculateMaxTokens,\n  getModelContextSize,\n} from \"@langchain/core/language_models/base\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport {\n  RunnablePassthrough,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport {\n  DEFAULT_SQL_DATABASE_PROMPT,\n  SQL_PROMPTS_MAP,\n  SqlDialect,\n} from \"./sql_db_prompt.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport type { SqlDatabase } from \"../../sql_db.js\";\nimport { getPromptTemplateFromDataSource } from \"../../util/sql_utils.js\";\n\n/**\n * Interface that extends the ChainInputs interface and defines additional\n * fields specific to a SQL database chain. It represents the input fields\n * for a SQL database chain.\n */\nexport interface SqlDatabaseChainInput extends ChainInputs {\n  llm: BaseLanguageModelInterface;\n  database: SqlDatabase;\n  topK?: number;\n  inputKey?: string;\n  outputKey?: string;\n  sqlOutputKey?: string;\n  prompt?: PromptTemplate;\n}\n\n/**\n * Class that represents a SQL database chain in the LangChain framework.\n * It extends the BaseChain class and implements the functionality\n * specific to a SQL database chain.\n *\n * @security **Security Notice**\n * This chain generates SQL queries for the given database.\n * The SQLDatabase class provides a getTableInfo method that can be used\n * to get column information as well as sample data from the table.\n * To mitigate risk of leaking sensitive data, limit permissions\n * to read and scope to the tables that are needed.\n * Optionally, use the includesTables or ignoreTables class parameters\n * to limit which tables can/cannot be accessed.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n * @example\n * ```typescript\n * const chain = new SqlDatabaseChain({\n *   llm: new OpenAI({ temperature: 0 }),\n *   database: new SqlDatabase({ ...config }),\n * });\n *\n * const result = await chain.run(\"How many tracks are there?\");\n * ```\n */\nexport class SqlDatabaseChain extends BaseChain {\n  static lc_name() {\n    return \"SqlDatabaseChain\";\n  }\n\n  // LLM wrapper to use\n  llm: BaseLanguageModelInterface;\n\n  // SQL Database to connect to.\n  database: SqlDatabase;\n\n  // Prompt to use to translate natural language to SQL.\n  prompt = DEFAULT_SQL_DATABASE_PROMPT;\n\n  // Number of results to return from the query\n  topK = 5;\n\n  inputKey = \"query\";\n\n  outputKey = \"result\";\n\n  sqlOutputKey: string | undefined = undefined;\n\n  // Whether to return the result of querying the SQL table directly.\n  returnDirect = false;\n\n  constructor(fields: SqlDatabaseChainInput) {\n    super(fields);\n    this.llm = fields.llm;\n    this.database = fields.database;\n    this.topK = fields.topK ?? this.topK;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.sqlOutputKey = fields.sqlOutputKey ?? this.sqlOutputKey;\n    this.prompt =\n      fields.prompt ??\n      getPromptTemplateFromDataSource(this.database.appDataSource);\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const llmChain = new LLMChain({\n      prompt: this.prompt,\n      llm: this.llm,\n      outputKey: this.outputKey,\n      memory: this.memory,\n    });\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    let inputText = `${question}\\nSQLQuery:`;\n    const tablesToUse = values.table_names_to_use;\n    const tableInfo = await this.database.getTableInfo(tablesToUse);\n\n    const llmInputs = {\n      input: inputText,\n      top_k: this.topK,\n      dialect: this.database.appDataSourceOptions.type,\n      table_info: tableInfo,\n      stop: [\"\\nSQLResult:\"],\n    };\n    await this.verifyNumberOfTokens(inputText, tableInfo);\n\n    const sqlCommand = await llmChain.predict(\n      llmInputs,\n      runManager?.getChild(\"sql_generation\")\n    );\n    let queryResult = \"\";\n    try {\n      queryResult = await this.database.appDataSource.query(sqlCommand);\n    } catch (error) {\n      console.error(error);\n    }\n\n    let finalResult;\n    if (this.returnDirect) {\n      finalResult = { [this.outputKey]: queryResult };\n    } else {\n      inputText += `${sqlCommand}\\nSQLResult: ${JSON.stringify(\n        queryResult\n      )}\\nAnswer:`;\n      llmInputs.input = inputText;\n      finalResult = {\n        [this.outputKey]: await llmChain.predict(\n          llmInputs,\n          runManager?.getChild(\"result_generation\")\n        ),\n      };\n    }\n\n    if (this.sqlOutputKey != null) {\n      finalResult[this.sqlOutputKey] = sqlCommand;\n    }\n\n    return finalResult;\n  }\n\n  _chainType() {\n    return \"sql_database_chain\" as const;\n  }\n\n  get inputKeys(): string[] {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    if (this.sqlOutputKey != null) {\n      return [this.outputKey, this.sqlOutputKey];\n    }\n    return [this.outputKey];\n  }\n\n  /**\n   * Private method that verifies the number of tokens in the input text and\n   * table information. It throws an error if the number of tokens exceeds\n   * the maximum allowed by the language model.\n   * @param inputText The input text.\n   * @param tableinfo The table information.\n   * @returns A promise that resolves when the verification is complete.\n   */\n  private async verifyNumberOfTokens(\n    inputText: string,\n    tableinfo: string\n  ): Promise<void> {\n    // We verify it only for OpenAI for the moment\n    if (this.llm._llmType() !== \"openai\") {\n      return;\n    }\n    const llm = this.llm as OpenAI;\n    const promptTemplate = this.prompt.template;\n    const stringWeSend = `${inputText}${promptTemplate}${tableinfo}`;\n\n    const maxToken = await calculateMaxTokens({\n      prompt: stringWeSend,\n      // Cast here to allow for other models that may not fit the union\n      modelName: llm.model as TiktokenModel,\n    });\n\n    if (maxToken < (llm.maxTokens ?? -1)) {\n      throw new Error(`The combination of the database structure and your question is too big for the model ${\n        llm.model\n      } which can compute only a max tokens of ${getModelContextSize(\n        llm.model\n      )}.\n      We suggest you to use the includeTables parameters when creating the SqlDatabase object to select only a subset of the tables. You can also use a model which can handle more tokens.`);\n    }\n  }\n}\n\nexport interface CreateSqlQueryChainFields {\n  llm: BaseLanguageModel;\n  db: SqlDatabase;\n  prompt?: BasePromptTemplate;\n  /**\n   * @default 5\n   */\n  k?: number;\n  dialect: SqlDialect;\n}\n\ntype SqlInput = {\n  question: string;\n};\n\ntype SqlInoutWithTables = SqlInput & {\n  tableNamesToUse: string[];\n};\n\nconst strip = (text: string) => {\n  // Replace escaped quotes with actual quotes\n  let newText = text.replace(/\\\\\"/g, '\"').trim();\n  // Remove wrapping quotes if the entire string is wrapped in quotes\n  if (newText.startsWith('\"') && newText.endsWith('\"')) {\n    newText = newText.substring(1, newText.length - 1);\n  }\n  return newText;\n};\n\nconst difference = (setA: Set<string>, setB: Set<string>) =>\n  new Set([...setA].filter((x) => !setB.has(x)));\n\n/**\n * Create a SQL query chain that can create SQL queries for the given database.\n * Returns a Runnable.\n *\n * @param {BaseLanguageModel} llm The language model to use in the chain.\n * @param {SqlDatabase} db The database to use in the chain.\n * @param {BasePromptTemplate | undefined} prompt The prompt to use in the chain.\n * @param {BaseLanguageModel | undefined} k The amount of docs/results to return. Passed through the prompt input value `top_k`.\n * @param {SqlDialect} dialect The SQL dialect to use in the chain.\n * @returns {Promise<RunnableSequence<Record<string, unknown>, string>>} A runnable sequence representing the chain.\n * @example ```typescript\n * const datasource = new DataSource({\n *   type: \"sqlite\",\n *   database: \"../../../../Chinook.db\",\n * });\n * const db = await SqlDatabase.fromDataSourceParams({\n *   appDataSource: datasource,\n * });\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 });\n * const chain = await createSqlQueryChain({\n *   llm,\n *   db,\n *   dialect: \"sqlite\",\n * });\n * ```\n */\nexport async function createSqlQueryChain({\n  llm,\n  db,\n  prompt,\n  k = 5,\n  dialect,\n}: CreateSqlQueryChainFields) {\n  let promptToUse: BasePromptTemplate;\n  if (prompt) {\n    promptToUse = prompt;\n  } else if (SQL_PROMPTS_MAP[dialect]) {\n    promptToUse = SQL_PROMPTS_MAP[dialect];\n  } else {\n    promptToUse = DEFAULT_SQL_DATABASE_PROMPT;\n  }\n\n  if (\n    difference(\n      new Set([\"input\", \"top_k\", \"table_info\"]),\n      new Set(promptToUse.inputVariables)\n    ).size > 0\n  ) {\n    throw new Error(\n      `Prompt must have input variables: 'input', 'top_k', 'table_info'. Received prompt with input variables: ` +\n        `${promptToUse.inputVariables}. Full prompt:\\n\\n${promptToUse}`\n    );\n  }\n  if (promptToUse.inputVariables.includes(\"dialect\")) {\n    promptToUse = await promptToUse.partial({ dialect });\n  }\n\n  promptToUse = await promptToUse.partial({ top_k: k.toString() });\n\n  const inputs = {\n    input: (x: Record<string, unknown>) => {\n      if (\"question\" in x) {\n        return `${(x as SqlInput).question}\\nSQLQuery: `;\n      }\n      throw new Error(\"Input must include a question property.\");\n    },\n    table_info: async (x: Record<string, unknown>) =>\n      db.getTableInfo((x as SqlInoutWithTables).tableNamesToUse),\n  };\n\n  return RunnableSequence.from([\n    RunnablePassthrough.assign(inputs),\n    (x) => {\n      const newInputs = { ...x };\n      delete newInputs.question;\n      delete newInputs.tableNamesToUse;\n      return newInputs;\n    },\n    promptToUse,\n    llm.withConfig({ stop: [\"\\nSQLResult:\"] }),\n    new StringOutputParser(),\n    strip,\n  ]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmEA,IAAa,mBAAb,cAAsCA,uBAAU;CAC9C,OAAO,UAAU;AACf,SAAO;CACR;CAGD;CAGA;CAGA,SAASC;CAGT,OAAO;CAEP,WAAW;CAEX,YAAY;CAEZ,eAAmC;CAGnC,eAAe;CAEf,YAAYC,QAA+B;EACzC,MAAM,OAAO;EACb,KAAK,MAAM,OAAO;EAClB,KAAK,WAAW,OAAO;EACvB,KAAK,OAAO,OAAO,QAAQ,KAAK;EAChC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,eAAe,OAAO,gBAAgB,KAAK;EAChD,KAAK,SACH,OAAO,UACPC,kDAAgC,KAAK,SAAS,cAAc;CAC/D;;CAGD,MAAM,MACJC,QACAC,YACsB;EACtB,MAAM,WAAW,IAAIC,2BAAS;GAC5B,QAAQ,KAAK;GACb,KAAK,KAAK;GACV,WAAW,KAAK;GAChB,QAAQ,KAAK;EACd;AACD,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;EAE5D,MAAMC,WAAmB,OAAO,KAAK;EACrC,IAAI,YAAY,GAAG,SAAS,WAAW,CAAC;EACxC,MAAM,cAAc,OAAO;EAC3B,MAAM,YAAY,MAAM,KAAK,SAAS,aAAa,YAAY;EAE/D,MAAM,YAAY;GAChB,OAAO;GACP,OAAO,KAAK;GACZ,SAAS,KAAK,SAAS,qBAAqB;GAC5C,YAAY;GACZ,MAAM,CAAC,cAAe;EACvB;EACD,MAAM,KAAK,qBAAqB,WAAW,UAAU;EAErD,MAAM,aAAa,MAAM,SAAS,QAChC,WACA,YAAY,SAAS,iBAAiB,CACvC;EACD,IAAI,cAAc;AAClB,MAAI;GACF,cAAc,MAAM,KAAK,SAAS,cAAc,MAAM,WAAW;EAClE,SAAQ,OAAO;GACd,QAAQ,MAAM,MAAM;EACrB;EAED,IAAI;AACJ,MAAI,KAAK,cACP,cAAc,GAAG,KAAK,YAAY,YAAa;OAC1C;GACL,aAAa,GAAG,WAAW,aAAa,EAAE,KAAK,UAC7C,YACD,CAAC,SAAS,CAAC;GACZ,UAAU,QAAQ;GAClB,cAAc,GACX,KAAK,YAAY,MAAM,SAAS,QAC/B,WACA,YAAY,SAAS,oBAAoB,CAC1C,CACF;EACF;AAED,MAAI,KAAK,gBAAgB,MACvB,YAAY,KAAK,gBAAgB;AAGnC,SAAO;CACR;CAED,aAAa;AACX,SAAO;CACR;CAED,IAAI,YAAsB;AACxB,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAuB;AACzB,MAAI,KAAK,gBAAgB,KACvB,QAAO,CAAC,KAAK,WAAW,KAAK,YAAa;AAE5C,SAAO,CAAC,KAAK,SAAU;CACxB;;;;;;;;;CAUD,MAAc,qBACZC,WACAC,WACe;AAEf,MAAI,KAAK,IAAI,UAAU,KAAK,SAC1B;EAEF,MAAM,MAAM,KAAK;EACjB,MAAM,iBAAiB,KAAK,OAAO;EACnC,MAAM,eAAe,GAAG,YAAY,iBAAiB,WAAW;EAEhE,MAAM,WAAW,oEAAyB;GACxC,QAAQ;GAER,WAAW,IAAI;EAChB,EAAC;AAEF,MAAI,YAAY,IAAI,aAAa,IAC/B,OAAM,IAAI,MAAM,CAAC,qFAAqF,EACpG,IAAI,MACL,wCAAwC,iEACvC,IAAI,MACL,CAAC;2LACmL,CAAC;CAEzL;AACF;AAqBD,MAAM,QAAQ,CAACC,SAAiB;CAE9B,IAAI,UAAU,KAAK,QAAQ,QAAQ,KAAI,CAAC,MAAM;AAE9C,KAAI,QAAQ,WAAW,KAAI,IAAI,QAAQ,SAAS,KAAI,EAClD,UAAU,QAAQ,UAAU,GAAG,QAAQ,SAAS,EAAE;AAEpD,QAAO;AACR;AAED,MAAM,aAAa,CAACC,MAAmBC,SACrC,IAAI,IAAI,CAAC,GAAG,IAAK,EAAC,OAAO,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B/C,eAAsB,oBAAoB,EACxC,KACA,IACA,QACA,IAAI,GACJ,SAC0B,EAAE;CAC5B,IAAIC;AACJ,KAAI,QACF,cAAc;UACLC,sCAAgB,UACzB,cAAcA,sCAAgB;MAE9B,cAAcb;AAGhB,KACE,WACE,IAAI,IAAI;EAAC;EAAS;EAAS;CAAa,IACxC,IAAI,IAAI,YAAY,gBACrB,CAAC,OAAO,EAET,OAAM,IAAI,MACR,CAAC,wGAAwG,EACpG,YAAY,eAAe,kBAAkB,EAAE,aAAa;AAGrE,KAAI,YAAY,eAAe,SAAS,UAAU,EAChD,cAAc,MAAM,YAAY,QAAQ,EAAE,QAAS,EAAC;CAGtD,cAAc,MAAM,YAAY,QAAQ,EAAE,OAAO,EAAE,UAAU,CAAE,EAAC;CAEhE,MAAM,SAAS;EACb,OAAO,CAACc,MAA+B;AACrC,OAAI,cAAc,EAChB,QAAO,GAAI,EAAe,SAAS,YAAY,CAAC;AAElD,SAAM,IAAI,MAAM;EACjB;EACD,YAAY,OAAOA,MACjB,GAAG,aAAc,EAAyB,gBAAgB;CAC7D;AAED,QAAOC,4CAAiB,KAAK;EAC3BC,+CAAoB,OAAO,OAAO;EAClC,CAAC,MAAM;GACL,MAAM,YAAY,EAAE,GAAG,EAAG;GAC1B,OAAO,UAAU;GACjB,OAAO,UAAU;AACjB,UAAO;EACR;EACD;EACA,IAAI,WAAW,EAAE,MAAM,CAAC,cAAe,EAAE,EAAC;EAC1C,IAAIC;EACJ;CACD,EAAC;AACH"}
+{"version":3,"file":"sql_db_chain.cjs","names":["BaseChain","DEFAULT_SQL_DATABASE_PROMPT","getPromptTemplateFromDataSource","LLMChain","SQL_PROMPTS_MAP","RunnableSequence","RunnablePassthrough","StringOutputParser"],"sources":["../../../src/chains/sql_db/sql_db_chain.ts"],"sourcesContent":["import type {\n  BaseLanguageModel,\n  BaseLanguageModelInterface,\n} from \"@langchain/core/language_models/base\";\nimport type { OpenAI, TiktokenModel } from \"@langchain/openai\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\nimport {\n  calculateMaxTokens,\n  getModelContextSize,\n} from \"@langchain/core/language_models/base\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport {\n  RunnablePassthrough,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport {\n  DEFAULT_SQL_DATABASE_PROMPT,\n  SQL_PROMPTS_MAP,\n  SqlDialect,\n} from \"./sql_db_prompt.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport type { SqlDatabase } from \"../../sql_db.js\";\nimport { getPromptTemplateFromDataSource } from \"../../util/sql_utils.js\";\n\n/**\n * Interface that extends the ChainInputs interface and defines additional\n * fields specific to a SQL database chain. It represents the input fields\n * for a SQL database chain.\n */\nexport interface SqlDatabaseChainInput extends ChainInputs {\n  llm: BaseLanguageModelInterface;\n  database: SqlDatabase;\n  topK?: number;\n  inputKey?: string;\n  outputKey?: string;\n  sqlOutputKey?: string;\n  prompt?: PromptTemplate;\n}\n\n/**\n * Class that represents a SQL database chain in the LangChain framework.\n * It extends the BaseChain class and implements the functionality\n * specific to a SQL database chain.\n *\n * @security **Security Notice**\n * This chain generates SQL queries for the given database.\n * The SQLDatabase class provides a getTableInfo method that can be used\n * to get column information as well as sample data from the table.\n * To mitigate risk of leaking sensitive data, limit permissions\n * to read and scope to the tables that are needed.\n * Optionally, use the includesTables or ignoreTables class parameters\n * to limit which tables can/cannot be accessed.\n *\n * @link See https://js.langchain.com/docs/security for more information.\n * @example\n * ```typescript\n * const chain = new SqlDatabaseChain({\n *   llm: new OpenAI({ temperature: 0 }),\n *   database: new SqlDatabase({ ...config }),\n * });\n *\n * const result = await chain.run(\"How many tracks are there?\");\n * ```\n */\nexport class SqlDatabaseChain extends BaseChain {\n  static lc_name() {\n    return \"SqlDatabaseChain\";\n  }\n\n  // LLM wrapper to use\n  llm: BaseLanguageModelInterface;\n\n  // SQL Database to connect to.\n  database: SqlDatabase;\n\n  // Prompt to use to translate natural language to SQL.\n  prompt = DEFAULT_SQL_DATABASE_PROMPT;\n\n  // Number of results to return from the query\n  topK = 5;\n\n  inputKey = \"query\";\n\n  outputKey = \"result\";\n\n  sqlOutputKey: string | undefined = undefined;\n\n  // Whether to return the result of querying the SQL table directly.\n  returnDirect = false;\n\n  constructor(fields: SqlDatabaseChainInput) {\n    super(fields);\n    this.llm = fields.llm;\n    this.database = fields.database;\n    this.topK = fields.topK ?? this.topK;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.sqlOutputKey = fields.sqlOutputKey ?? this.sqlOutputKey;\n    this.prompt =\n      fields.prompt ??\n      getPromptTemplateFromDataSource(this.database.appDataSource);\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const llmChain = new LLMChain({\n      prompt: this.prompt,\n      llm: this.llm,\n      outputKey: this.outputKey,\n      memory: this.memory,\n    });\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    let inputText = `${question}\\nSQLQuery:`;\n    const tablesToUse = values.table_names_to_use;\n    const tableInfo = await this.database.getTableInfo(tablesToUse);\n\n    const llmInputs = {\n      input: inputText,\n      top_k: this.topK,\n      dialect: this.database.appDataSourceOptions.type,\n      table_info: tableInfo,\n      stop: [\"\\nSQLResult:\"],\n    };\n    await this.verifyNumberOfTokens(inputText, tableInfo);\n\n    const sqlCommand = await llmChain.predict(\n      llmInputs,\n      runManager?.getChild(\"sql_generation\")\n    );\n    let queryResult = \"\";\n    try {\n      queryResult = await this.database.appDataSource.query(sqlCommand);\n    } catch (error) {\n      console.error(error);\n    }\n\n    let finalResult;\n    if (this.returnDirect) {\n      finalResult = { [this.outputKey]: queryResult };\n    } else {\n      inputText += `${sqlCommand}\\nSQLResult: ${JSON.stringify(\n        queryResult\n      )}\\nAnswer:`;\n      llmInputs.input = inputText;\n      finalResult = {\n        [this.outputKey]: await llmChain.predict(\n          llmInputs,\n          runManager?.getChild(\"result_generation\")\n        ),\n      };\n    }\n\n    if (this.sqlOutputKey != null) {\n      finalResult[this.sqlOutputKey] = sqlCommand;\n    }\n\n    return finalResult;\n  }\n\n  _chainType() {\n    return \"sql_database_chain\" as const;\n  }\n\n  get inputKeys(): string[] {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    if (this.sqlOutputKey != null) {\n      return [this.outputKey, this.sqlOutputKey];\n    }\n    return [this.outputKey];\n  }\n\n  /**\n   * Private method that verifies the number of tokens in the input text and\n   * table information. It throws an error if the number of tokens exceeds\n   * the maximum allowed by the language model.\n   * @param inputText The input text.\n   * @param tableinfo The table information.\n   * @returns A promise that resolves when the verification is complete.\n   */\n  private async verifyNumberOfTokens(\n    inputText: string,\n    tableinfo: string\n  ): Promise<void> {\n    // We verify it only for OpenAI for the moment\n    if (this.llm._llmType() !== \"openai\") {\n      return;\n    }\n    const llm = this.llm as OpenAI;\n    const promptTemplate = this.prompt.template;\n    const stringWeSend = `${inputText}${promptTemplate}${tableinfo}`;\n\n    const maxToken = await calculateMaxTokens({\n      prompt: stringWeSend,\n      // Cast here to allow for other models that may not fit the union\n      modelName: llm.model as TiktokenModel,\n    });\n\n    if (maxToken < (llm.maxTokens ?? -1)) {\n      throw new Error(`The combination of the database structure and your question is too big for the model ${\n        llm.model\n      } which can compute only a max tokens of ${getModelContextSize(\n        llm.model\n      )}.\n      We suggest you to use the includeTables parameters when creating the SqlDatabase object to select only a subset of the tables. You can also use a model which can handle more tokens.`);\n    }\n  }\n}\n\nexport interface CreateSqlQueryChainFields {\n  llm: BaseLanguageModel;\n  db: SqlDatabase;\n  prompt?: BasePromptTemplate;\n  /**\n   * @default 5\n   */\n  k?: number;\n  dialect: SqlDialect;\n}\n\ntype SqlInput = {\n  question: string;\n};\n\ntype SqlInoutWithTables = SqlInput & {\n  tableNamesToUse: string[];\n};\n\nconst strip = (text: string) => {\n  // Replace escaped quotes with actual quotes\n  let newText = text.replace(/\\\\\"/g, '\"').trim();\n  // Remove wrapping quotes if the entire string is wrapped in quotes\n  if (newText.startsWith('\"') && newText.endsWith('\"')) {\n    newText = newText.substring(1, newText.length - 1);\n  }\n  return newText;\n};\n\nconst difference = (setA: Set<string>, setB: Set<string>) =>\n  new Set([...setA].filter((x) => !setB.has(x)));\n\n/**\n * Create a SQL query chain that can create SQL queries for the given database.\n * Returns a Runnable.\n *\n * @param {BaseLanguageModel} llm The language model to use in the chain.\n * @param {SqlDatabase} db The database to use in the chain.\n * @param {BasePromptTemplate | undefined} prompt The prompt to use in the chain.\n * @param {BaseLanguageModel | undefined} k The amount of docs/results to return. Passed through the prompt input value `top_k`.\n * @param {SqlDialect} dialect The SQL dialect to use in the chain.\n * @returns {Promise<RunnableSequence<Record<string, unknown>, string>>} A runnable sequence representing the chain.\n * @example ```typescript\n * const datasource = new DataSource({\n *   type: \"sqlite\",\n *   database: \"../../../../Chinook.db\",\n * });\n * const db = await SqlDatabase.fromDataSourceParams({\n *   appDataSource: datasource,\n * });\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 });\n * const chain = await createSqlQueryChain({\n *   llm,\n *   db,\n *   dialect: \"sqlite\",\n * });\n * ```\n */\nexport async function createSqlQueryChain({\n  llm,\n  db,\n  prompt,\n  k = 5,\n  dialect,\n}: CreateSqlQueryChainFields) {\n  let promptToUse: BasePromptTemplate;\n  if (prompt) {\n    promptToUse = prompt;\n  } else if (SQL_PROMPTS_MAP[dialect]) {\n    promptToUse = SQL_PROMPTS_MAP[dialect];\n  } else {\n    promptToUse = DEFAULT_SQL_DATABASE_PROMPT;\n  }\n\n  if (\n    difference(\n      new Set([\"input\", \"top_k\", \"table_info\"]),\n      new Set(promptToUse.inputVariables)\n    ).size > 0\n  ) {\n    throw new Error(\n      `Prompt must have input variables: 'input', 'top_k', 'table_info'. Received prompt with input variables: ` +\n        `${promptToUse.inputVariables}. Full prompt:\\n\\n${promptToUse}`\n    );\n  }\n  if (promptToUse.inputVariables.includes(\"dialect\")) {\n    promptToUse = await promptToUse.partial({ dialect });\n  }\n\n  promptToUse = await promptToUse.partial({ top_k: k.toString() });\n\n  const inputs = {\n    input: (x: Record<string, unknown>) => {\n      if (\"question\" in x) {\n        return `${(x as SqlInput).question}\\nSQLQuery: `;\n      }\n      throw new Error(\"Input must include a question property.\");\n    },\n    table_info: async (x: Record<string, unknown>) =>\n      db.getTableInfo((x as SqlInoutWithTables).tableNamesToUse),\n  };\n\n  return RunnableSequence.from([\n    RunnablePassthrough.assign(inputs),\n    (x) => {\n      const newInputs = { ...x };\n      delete newInputs.question;\n      delete newInputs.tableNamesToUse;\n      return newInputs;\n    },\n    promptToUse,\n    llm.withConfig({ stop: [\"\\nSQLResult:\"] }),\n    new StringOutputParser(),\n    strip,\n  ]);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmEA,IAAa,mBAAb,cAAsCA,uBAAU;CAC9C,OAAO,UAAU;AACf,SAAO;;CAIT;CAGA;CAGA,SAASC;CAGT,OAAO;CAEP,WAAW;CAEX,YAAY;CAEZ,eAAmC;CAGnC,eAAe;CAEf,YAAY,QAA+B;AACzC,QAAM,OAAO;AACb,OAAK,MAAM,OAAO;AAClB,OAAK,WAAW,OAAO;AACvB,OAAK,OAAO,OAAO,QAAQ,KAAK;AAChC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,YAAY,OAAO,aAAa,KAAK;AAC1C,OAAK,eAAe,OAAO,gBAAgB,KAAK;AAChD,OAAK,SACH,OAAO,UACPC,kDAAgC,KAAK,SAAS,cAAc;;;CAIhE,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,WAAW,IAAIC,2BAAS;GAC5B,QAAQ,KAAK;GACb,KAAK,KAAK;GACV,WAAW,KAAK;GAChB,QAAQ,KAAK;GACd,CAAC;AACF,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;EAG7D,IAAI,YAAY,GADS,OAAO,KAAK,UACT;EAC5B,MAAM,cAAc,OAAO;EAC3B,MAAM,YAAY,MAAM,KAAK,SAAS,aAAa,YAAY;EAE/D,MAAM,YAAY;GAChB,OAAO;GACP,OAAO,KAAK;GACZ,SAAS,KAAK,SAAS,qBAAqB;GAC5C,YAAY;GACZ,MAAM,CAAC,eAAe;GACvB;AACD,QAAM,KAAK,qBAAqB,WAAW,UAAU;EAErD,MAAM,aAAa,MAAM,SAAS,QAChC,WACA,YAAY,SAAS,iBAAiB,CACvC;EACD,IAAI,cAAc;AAClB,MAAI;AACF,iBAAc,MAAM,KAAK,SAAS,cAAc,MAAM,WAAW;WAC1D,OAAO;AACd,WAAQ,MAAM,MAAM;;EAGtB,IAAI;AACJ,MAAI,KAAK,aACP,eAAc,GAAG,KAAK,YAAY,aAAa;OAC1C;AACL,gBAAa,GAAG,WAAW,eAAe,KAAK,UAC7C,YACD,CAAC;AACF,aAAU,QAAQ;AAClB,iBAAc,GACX,KAAK,YAAY,MAAM,SAAS,QAC/B,WACA,YAAY,SAAS,oBAAoB,CAC1C,EACF;;AAGH,MAAI,KAAK,gBAAgB,KACvB,aAAY,KAAK,gBAAgB;AAGnC,SAAO;;CAGT,aAAa;AACX,SAAO;;CAGT,IAAI,YAAsB;AACxB,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAuB;AACzB,MAAI,KAAK,gBAAgB,KACvB,QAAO,CAAC,KAAK,WAAW,KAAK,aAAa;AAE5C,SAAO,CAAC,KAAK,UAAU;;;;;;;;;;CAWzB,MAAc,qBACZ,WACA,WACe;AAEf,MAAI,KAAK,IAAI,UAAU,KAAK,SAC1B;EAEF,MAAM,MAAM,KAAK;AAUjB,MANiB,mEAAyB;GACxC,QAHmB,GAAG,YADD,KAAK,OAAO,WACkB;GAKnD,WAAW,IAAI;GAChB,CAAC,IAEc,IAAI,aAAa,IAC/B,OAAM,IAAI,MAAM,wFACd,IAAI,MACL,wGACC,IAAI,MACL,CAAC;6LACqL;;;AAwB7L,MAAM,SAAS,SAAiB;CAE9B,IAAI,UAAU,KAAK,QAAQ,QAAQ,KAAI,CAAC,MAAM;AAE9C,KAAI,QAAQ,WAAW,KAAI,IAAI,QAAQ,SAAS,KAAI,CAClD,WAAU,QAAQ,UAAU,GAAG,QAAQ,SAAS,EAAE;AAEpD,QAAO;;AAGT,MAAM,cAAc,MAAmB,SACrC,IAAI,IAAI,CAAC,GAAG,KAAK,CAAC,QAAQ,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4BhD,eAAsB,oBAAoB,EACxC,KACA,IACA,QACA,IAAI,GACJ,WAC4B;CAC5B,IAAI;AACJ,KAAI,OACF,eAAc;UACLC,sCAAgB,SACzB,eAAcA,sCAAgB;KAE9B,eAAcH;AAGhB,KACE,WACE,IAAI,IAAI;EAAC;EAAS;EAAS;EAAa,CAAC,EACzC,IAAI,IAAI,YAAY,eAAe,CACpC,CAAC,OAAO,EAET,OAAM,IAAI,MACR,2GACK,YAAY,eAAe,oBAAoB,cACrD;AAEH,KAAI,YAAY,eAAe,SAAS,UAAU,CAChD,eAAc,MAAM,YAAY,QAAQ,EAAE,SAAS,CAAC;AAGtD,eAAc,MAAM,YAAY,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;AAahE,QAAOI,2CAAiB,KAAK;EAC3BC,8CAAoB,OAZP;GACb,QAAQ,MAA+B;AACrC,QAAI,cAAc,EAChB,QAAO,GAAI,EAAe,SAAS;AAErC,UAAM,IAAI,MAAM,0CAA0C;;GAE5D,YAAY,OAAO,MACjB,GAAG,aAAc,EAAyB,gBAAgB;GAC7D,CAGmC;GACjC,MAAM;GACL,MAAM,YAAY,EAAE,GAAG,GAAG;AAC1B,UAAO,UAAU;AACjB,UAAO,UAAU;AACjB,UAAO;;EAET;EACA,IAAI,WAAW,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;EAC1C,IAAIC,mDAAoB;EACxB;EACD,CAAC"}

package/dist/chains/sql_db/sql_db_chain.d.cts

@@ -8,7 +8,6 @@ import { BaseLanguageModel, BaseLanguageModelInterface } from "@langchain/core/l
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 
 //#region src/chains/sql_db/sql_db_chain.d.ts
-
 /**
  * Interface that extends the ChainInputs interface and defines additional
  * fields specific to a SQL database chain. It represents the input fields