@langchain/classic 1.0.39 → 1.0.41

package/dist/chains/sequential_chain.d.cts

@@ -2,7 +2,6 @@ import { SerializedSequentialChain, SerializedSimpleSequentialChain } from "./se
 import { BaseChain, ChainInputs } from "./base.cjs";
 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.
@@ -28,7 +27,6 @@ interface SequentialChainInput extends ChainInputs {
  * Playwright: This is a synopsis for the above play:`,
  *   inputVariables: ["title", "era"],
  * });
-
  * const reviewPromptTemplate = new PromptTemplate({
  *   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.
  *
@@ -37,7 +35,6 @@ interface SequentialChainInput extends ChainInputs {
  *     Review from a New York Times play critic of the above play:`,
  *   inputVariables: ["synopsis"],
  * });
-
  * const overallChain = new SequentialChain({
  *   chains: [
  *     new LLMChain({
@@ -55,7 +52,6 @@ interface SequentialChainInput extends ChainInputs {
  *   outputVariables: ["synopsis", "review"],
  *   verbose: true,
  * });
-
  * const chainExecutionResult = await overallChain.call({
  *   title: "Tragedy at sunset on the beach",
  *   era: "Victorian England",

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

@@ -1 +1 @@
-{"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"}
+{"version":3,"file":"sequential_chain.d.cts","names":[],"sources":["../../src/chains/sequential_chain.ts"],"mappings":";;;;;;;;UAmBiB,6BAA6B;;EAE5C,QAAQ;;EAER;;EAEA;;EAEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAiDW,wBAAwB,qBAAqB;SACjD;EAIP,QAAQ;EAER;EAEA;EAEA;MAEI;MAIA;EAIJ,YAAY,QAAQ;;EAepB;;EAsEM,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAsBX;SAIa,YAAY,MAAM,4BAAyB,QAAA;EAYxD,aAAa;;;;;UAiBE,mCAAmC;;EAElD,QAAQ,MAAM;;EAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCW,8BACH,qBACG;SAEJ;EAIP,QAAQ,MAAM;EAEd;EAEA;EAEA;MAEI;MAIA;EAIJ,YAAY,QAAQ;;EAQpB;;EAuBM,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAmBX;SAIa,YAAY,MAAM,kCAA+B,QAAA;EAU9D,aAAa"}

package/dist/chains/sequential_chain.d.ts

@@ -2,7 +2,6 @@ import { SerializedSequentialChain, SerializedSimpleSequentialChain } from "./se
 import { BaseChain, ChainInputs } from "./base.js";
 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.
@@ -28,7 +27,6 @@ interface SequentialChainInput extends ChainInputs {
  * Playwright: This is a synopsis for the above play:`,
  *   inputVariables: ["title", "era"],
  * });
-
  * const reviewPromptTemplate = new PromptTemplate({
  *   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.
  *
@@ -37,7 +35,6 @@ interface SequentialChainInput extends ChainInputs {
  *     Review from a New York Times play critic of the above play:`,
  *   inputVariables: ["synopsis"],
  * });
-
  * const overallChain = new SequentialChain({
  *   chains: [
  *     new LLMChain({
@@ -55,7 +52,6 @@ interface SequentialChainInput extends ChainInputs {
  *   outputVariables: ["synopsis", "review"],
  *   verbose: true,
  * });
-
  * const chainExecutionResult = await overallChain.call({
  *   title: "Tragedy at sunset on the beach",
  *   era: "Victorian England",

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

@@ -1 +1 @@
-{"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"}
+{"version":3,"file":"sequential_chain.d.ts","names":[],"sources":["../../src/chains/sequential_chain.ts"],"mappings":";;;;;;;;UAmBiB,6BAA6B;;EAE5C,QAAQ;;EAER;;EAEA;;EAEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAiDW,wBAAwB,qBAAqB;SACjD;EAIP,QAAQ;EAER;EAEA;EAEA;MAEI;MAIA;EAIJ,YAAY,QAAQ;;EAepB;;EAsEM,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAsBX;SAIa,YAAY,MAAM,4BAAyB,QAAA;EAYxD,aAAa;;;;;UAiBE,mCAAmC;;EAElD,QAAQ,MAAM;;EAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuCW,8BACH,qBACG;SAEJ;EAIP,QAAQ,MAAM;EAEd;EAEA;EAEA;MAEI;MAIA;EAIJ,YAAY,QAAQ;;EAQpB;;EAuBM,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EAmBX;SAIa,YAAY,MAAM,kCAA+B,QAAA;EAU9D,aAAa"}

package/dist/chains/sequential_chain.js.map

@@ -1 +1 @@
-{"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"}
+{"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;CACrC,OAAO,MAAM,KAAK,KAAK,CAAC,CACrB,KAAK,MAAM,IAAI,EAAE,EAAE,CAAC,CACpB,KAAK,IAAI;AACd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8DA,IAAa,kBAAb,MAAa,wBAAwB,UAA0C;CAC7E,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA;CAEA;CAEA;CAEA,IAAI,YAAY;EACd,OAAO,KAAK;CACd;CAEA,IAAI,aAAuB;EACzB,OAAO,KAAK;CACd;CAEA,YAAY,QAA8B;EACxC,MAAM,MAAM;EACZ,KAAK,SAAS,OAAO;EACrB,KAAK,iBAAiB,OAAO;EAC7B,KAAK,kBAAkB,OAAO,mBAAmB,CAAC;EAClD,IAAI,KAAK,gBAAgB,SAAS,KAAK,OAAO,WAC5C,MAAM,IAAI,MACR,qIACF;EAEF,KAAK,YAAY,OAAO,aAAa;EACrC,KAAK,gBAAgB;CACvB;;CAGA,kBAAkB;EAChB,IAAI,KAAK,OAAO,WAAW,GACzB,MAAM,IAAI,MAAM,gDAAgD;EAGlE,MAAM,aAAa,KAAK,QAAQ,cAAc,CAAC;EAC/C,MAAM,eAAe,IAAI,IAAI,KAAK,SAAS;EAC3C,MAAM,gBAAgB,IAAI,IAAI,UAAU;EACxC,MAAM,mBAAmB,aAAa,cAAc,aAAa;EACjE,IAAI,iBAAiB,OAAO,GAC1B,MAAM,IAAI,MACR,uBAAuB,UACrB,gBACF,EAAE,mKACJ;EAGF,MAAM,gBAAgB,MAAM,cAAc,aAAa;EACvD,KAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,IAAI,cAAc,WAAW,IAAI,IAAI,MAAM,SAAS,GAAG,aAAa;GAEpE,IAAI,MAAM,QACR,cAAc,WAAW,aAAa,IAAI,IAAI,MAAM,OAAO,UAAU,CAAC;GAGxE,IAAI,YAAY,OAAO,GACrB,MAAM,IAAI,MACR,gCAAgC,MAAM,WAAW,EAAE,KAAK,UACtD,WACF,EAAE,sCAAsC,UAAU,aAAa,EAAE,EACnE;GAEF,MAAM,gBAAgB,IAAI,IAAI,MAAM,UAAU;GAC9C,MAAM,wBAAwB,aAAa,eAAe,aAAa;GACvE,IAAI,sBAAsB,OAAO,GAC/B,MAAM,IAAI,MACR,6CAA6C,MAAM,WAAW,EAAE,qBAAqB,UACnF,qBACF,EAAE,yCACJ;GAGF,KAAK,MAAM,aAAa,eACtB,cAAc,IAAI,SAAS;EAE/B;EAEA,IAAI,KAAK,gBAAgB,WAAW,GAClC,IAAI,KAAK,WAAW;GAClB,MAAM,aAAa,WAAW,eAAe,YAAY;GACzD,KAAK,kBAAkB,MAAM,KAAK,UAAU;EAC9C,OACE,KAAK,kBAAkB,KAAK,OAAO,KAAK,OAAO,SAAS,EAAE,CAAC;OAExD;GACL,MAAM,cAAc,WAClB,IAAI,IAAI,KAAK,eAAe,GAC5B,IAAI,IAAI,aAAa,CACvB;GACA,IAAI,YAAY,OAAO,GACrB,MAAM,IAAI,MACR,oGAAoG,UAClG,WACF,EAAE,EACJ;EAEJ;CACF;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,QAAqB,CAAC;EAC1B,MAAM,iBAA8B;EACpC,IAAI,IAAI;EACR,KAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,KAAK;GACL,QAAQ,MAAM,MAAM,KAClB,gBACA,YAAY,SAAS,QAAQ,GAAG,CAClC;GACA,KAAK,MAAM,OAAO,OAAO,KAAK,KAAK,GACjC,eAAe,OAAO,MAAM;EAEhC;EACA,MAAM,SAAsB,CAAC;EAC7B,KAAK,MAAM,OAAO,KAAK,iBACrB,OAAO,OAAO,eAAe;EAG/B,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAiC;EACxD,MAAM,SAAsB,CAAC;EAC7B,MAAM,iBAA2B,KAAK;EACtC,MAAM,kBAA4B,KAAK;EACvC,MAAM,mBAAmB,KAAK;EAC9B,KAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAM,UAAU,YAAY,eAAe;GACrE,OAAO,KAAK,iBAAiB;EAC/B;EACA,OAAO,IAAI,gBAAgB;GAAE;GAAQ;GAAgB;EAAgB,CAAC;CACxE;CAEA,YAAuC;EACrC,MAAM,SAAgC,CAAC;EACvC,KAAK,MAAM,SAAS,KAAK,QACvB,OAAO,KAAK,MAAM,UAAU,CAAC;EAE/B,OAAO;GACL,OAAO,KAAK,WAAW;GACvB,iBAAiB,KAAK;GACtB,kBAAkB,KAAK;GACvB;EACF;CACF;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgDA,IAAa,wBAAb,MAAa,8BACH,UAEV;CACE,OAAO,UAAU;EACf,OAAO;CACT;CAEA;CAEA,WAAW;CAEX,YAAY;CAEZ;CAEA,IAAI,YAAY;EACd,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAuB;EACzB,OAAO,CAAC,KAAK,SAAS;CACxB;CAEA,YAAY,QAAoC;EAC9C,MAAM,MAAM;EACZ,KAAK,SAAS,OAAO;EACrB,KAAK,cAAc,OAAO,eAAe;EACzC,KAAK,gBAAgB;CACvB;;CAGA,kBAAkB;EAChB,KAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,IACE,MAAM,UAAU,QAAQ,MAAM,CAAC,MAAM,QAAQ,WAAW,SAAS,CAAC,CAAC,CAAC,CACjE,WAAW,GAEd,MAAM,IAAI,MACR,uEACE,MAAM,UAAU,OACjB,OAAO,MAAM,WAAW,EAAE,EAC7B;GAEF,IAAI,MAAM,WAAW,WAAW,GAC9B,MAAM,IAAI,MACR,wEACE,MAAM,WAAW,OAClB,OAAO,MAAM,WAAW,EAAE,EAC7B;EAEJ;CACF;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,IAAI,QAAgB,OAAO,KAAK;EAChC,IAAI,IAAI;EACR,KAAK,MAAM,SAAS,KAAK,QAAQ;GAC/B,KAAK;GACL,SACE,MAAM,MAAM,KACV;KAAG,MAAM,UAAU,KAAK;IAAO,QAAQ,OAAO;GAAO,GACrD,YAAY,SAAS,QAAQ,GAAG,CAClC,EAAA,CACA,MAAM,WAAW;GACnB,IAAI,KAAK,aACP,QAAQ,MAAM,KAAK;GAErB,MAAM,YAAY,WAAW,KAAK;EACpC;EACA,OAAO,GAAG,KAAK,YAAY,MAAM;CACnC;CAEA,aAAa;EACX,OAAO;CACT;CAEA,aAAa,YAAY,MAAuC;EAC9D,MAAM,SAA2B,CAAC;EAClC,MAAM,mBAAmB,KAAK;EAC9B,KAAK,MAAM,mBAAmB,kBAAkB;GAC9C,MAAM,oBAAoB,MAAM,UAAU,YAAY,eAAe;GACrE,OAAO,KAAK,iBAAiB;EAC/B;EACA,OAAO,IAAI,sBAAsB,EAAE,OAAO,CAAC;CAC7C;CAEA,YAA6C;EAC3C,MAAM,SAAqC,CAAC;EAC5C,KAAK,MAAM,SAAS,KAAK,QACvB,OAAO,KAAK,MAAM,UAAU,CAAC;EAE/B,OAAO;GACL,OAAO,KAAK,WAAW;GACvB;EACF;CACF;AACF"}

package/dist/chains/serde.d.cts

@@ -1,6 +1,5 @@
 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
@@ -151,5 +150,5 @@ type SerializedConstitutionalChain = {
  */
 type SerializedBaseChain = SerializedLLMChain | SerializedSequentialChain | SerializedSimpleSequentialChain | SerializedVectorDBQAChain | SerializedAPIChain | SerializedStuffDocumentsChain | SerializedChatVectorDBQAChain | SerializedMapReduceDocumentsChain | SerializedAnalyzeDocumentChain | SerializedRefineDocumentsChain | SerializedConstitutionalChain;
 //#endregion
-export { SerializedAPIChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedConstitutionalPrinciple, SerializedLLMChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, SerializedSequentialChain, SerializedSimpleSequentialChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain };
+export { SerializedAPIChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedConstitutionalChain, SerializedConstitutionalPrinciple, SerializedLLMChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, SerializedSequentialChain, SerializedSimpleSequentialChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain };
 //# sourceMappingURL=serde.d.cts.map

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

@@ -1 +1 @@
-{"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"}
+{"version":3,"file":"serde.d.cts","names":[],"sources":["../../src/chains/serde.ts"],"mappings":";;;;;;;;;KASY;EACV;EACA,MAAM;EACN,SAAS;;;;;;;;;KAUC;EACV;EACA;EACA;EACA,QAAQ;;;;;;;;KASE;EACV;EACA,QAAQ,MAAM;;;;;;;;KASJ;EACV;EACA;EACA,yBAAyB;;;;;;;;;KAUf;EACV;EACA,mBAAmB;EACnB,kBAAkB;EAClB;;;;;;;;KASU;EACV;EACA,YAAY;;;;;;;;;KAUF;EACV;EACA;EACA,yBAAyB;EACzB,oBAAoB;;;;;;;;;KAUV;EACV;EACA,YAAY;EACZ,yBAAyB;;;;;;;;KASf;EACV;EACA,YAAY;EACZ,mBAAmB;;;;;;;;KAST;EACV;EACA,yBAAyB;;;;;;;;;KAUf;EACV;EACA;EACA;EACA;;;;;;;;;KAUU;EACV;EACA,QAAQ;EACR,gBAAgB;EAChB,gBAAgB;EAChB,0BAA0B;;;;;;;;KAShB,sBACR,qBACA,4BACA,kCACA,4BACA,qBACA,gCACA,gCACA,oCACA,iCACA,iCACA"}

package/dist/chains/serde.d.ts

@@ -1,6 +1,5 @@
 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
@@ -151,5 +150,5 @@ type SerializedConstitutionalChain = {
  */
 type SerializedBaseChain = SerializedLLMChain | SerializedSequentialChain | SerializedSimpleSequentialChain | SerializedVectorDBQAChain | SerializedAPIChain | SerializedStuffDocumentsChain | SerializedChatVectorDBQAChain | SerializedMapReduceDocumentsChain | SerializedAnalyzeDocumentChain | SerializedRefineDocumentsChain | SerializedConstitutionalChain;
 //#endregion
-export { SerializedAPIChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedConstitutionalPrinciple, SerializedLLMChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, SerializedSequentialChain, SerializedSimpleSequentialChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain };
+export { SerializedAPIChain, SerializedAnalyzeDocumentChain, SerializedBaseChain, SerializedChatVectorDBQAChain, SerializedConstitutionalChain, SerializedConstitutionalPrinciple, SerializedLLMChain, SerializedMapReduceDocumentsChain, SerializedRefineDocumentsChain, SerializedSequentialChain, SerializedSimpleSequentialChain, SerializedStuffDocumentsChain, SerializedVectorDBQAChain };
 //# sourceMappingURL=serde.d.ts.map

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

@@ -1 +1 @@
-{"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"}
+{"version":3,"file":"serde.d.ts","names":[],"sources":["../../src/chains/serde.ts"],"mappings":";;;;;;;;;KASY;EACV;EACA,MAAM;EACN,SAAS;;;;;;;;;KAUC;EACV;EACA;EACA;EACA,QAAQ;;;;;;;;KASE;EACV;EACA,QAAQ,MAAM;;;;;;;;KASJ;EACV;EACA;EACA,yBAAyB;;;;;;;;;KAUf;EACV;EACA,mBAAmB;EACnB,kBAAkB;EAClB;;;;;;;;KASU;EACV;EACA,YAAY;;;;;;;;;KAUF;EACV;EACA;EACA,yBAAyB;EACzB,oBAAoB;;;;;;;;;KAUV;EACV;EACA,YAAY;EACZ,yBAAyB;;;;;;;;KASf;EACV;EACA,YAAY;EACZ,mBAAmB;;;;;;;;KAST;EACV;EACA,yBAAyB;;;;;;;;;KAUf;EACV;EACA;EACA;EACA;;;;;;;;;KAUU;EACV;EACA,QAAQ;EACR,gBAAgB;EAChB,gBAAgB;EAChB,0BAA0B;;;;;;;;KAShB,sBACR,qBACA,4BACA,kCACA,4BACA,qBACA,gCACA,gCACA,oCACA,iCACA,iCACA"}

package/dist/chains/sql_db/sql_db_chain.cjs

@@ -1,4 +1,3 @@
-require("../../_virtual/_rolldown/runtime.cjs");
 const require_base = require("../base.cjs");
 const require_llm_chain = require("../llm_chain.cjs");
 const require_sql_db_prompt = require("./sql_db_prompt.cjs");
@@ -156,7 +155,7 @@ async function createSqlQueryChain({ llm, db, prompt, k = 5, dialect }) {
 	if (prompt) promptToUse = prompt;
 	else if (require_sql_db_prompt.SQL_PROMPTS_MAP[dialect]) promptToUse = require_sql_db_prompt.SQL_PROMPTS_MAP[dialect];
 	else promptToUse = require_sql_db_prompt.DEFAULT_SQL_DATABASE_PROMPT;
-	if (difference(new Set([
+	if (difference(/* @__PURE__ */ new Set([
 		"input",
 		"top_k",
 		"table_info"

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","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,aAAAA,UAAU;CAC9C,OAAO,UAAU;AACf,SAAO;;CAIT;CAGA;CAGA,SAASC,sBAAAA;CAGT,OAAO;CAEP,WAAW;CAEX,YAAY;CAEZ,eAAmC,KAAA;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,kBAAAA,gCAAgC,KAAK,SAAS,cAAc;;;CAIhE,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,WAAW,IAAIC,kBAAAA,SAAS;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,MAAI,OAAA,GAAA,qCAAA,oBANsC;GACxC,QAAQ,GAHc,YADD,KAAK,OAAO,WACkB;GAKnD,WAAW,IAAI;GAChB,CAAC,IAEc,IAAI,aAAa,IAC/B,OAAM,IAAI,MAAM,wFACd,IAAI,MACL,2CAAA,GAAA,qCAAA,qBACC,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,sBAAAA,gBAAgB,SACzB,eAAcA,sBAAAA,gBAAgB;KAE9B,eAAcH,sBAAAA;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,0BAAAA,iBAAiB,KAAK;EAC3BC,0BAAAA,oBAAoB,OAAO;GAX3B,QAAQ,MAA+B;AACrC,QAAI,cAAc,EAChB,QAAO,GAAI,EAAe,SAAS;AAErC,UAAM,IAAI,MAAM,0CAA0C;;GAE5D,YAAY,OAAO,MACjB,GAAG,aAAc,EAAyB,gBAAgB;GAI3B,CAAC;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,+BAAAA,oBAAoB;EACxB;EACD,CAAC"}
+{"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,aAAAA,UAAU;CAC9C,OAAO,UAAU;EACf,OAAO;CACT;CAGA;CAGA;CAGA,SAASC,sBAAAA;CAGT,OAAO;CAEP,WAAW;CAEX,YAAY;CAEZ,eAAmC,KAAA;CAGnC,eAAe;CAEf,YAAY,QAA+B;EACzC,MAAM,MAAM;EACZ,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,kBAAAA,gCAAgC,KAAK,SAAS,aAAa;CAC/D;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,WAAW,IAAIC,kBAAAA,SAAS;GAC5B,QAAQ,KAAK;GACb,KAAK,KAAK;GACV,WAAW,KAAK;GAChB,QAAQ,KAAK;EACf,CAAC;EACD,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAG5D,IAAI,YAAY,GADS,OAAO,KAAK,UACT;EAC5B,MAAM,cAAc,OAAO;EAC3B,MAAM,YAAY,MAAM,KAAK,SAAS,aAAa,WAAW;EAE9D,MAAM,YAAY;GAChB,OAAO;GACP,OAAO,KAAK;GACZ,SAAS,KAAK,SAAS,qBAAqB;GAC5C,YAAY;GACZ,MAAM,CAAC,cAAc;EACvB;EACA,MAAM,KAAK,qBAAqB,WAAW,SAAS;EAEpD,MAAM,aAAa,MAAM,SAAS,QAChC,WACA,YAAY,SAAS,gBAAgB,CACvC;EACA,IAAI,cAAc;EAClB,IAAI;GACF,cAAc,MAAM,KAAK,SAAS,cAAc,MAAM,UAAU;EAClE,SAAS,OAAO;GACd,QAAQ,MAAM,KAAK;EACrB;EAEA,IAAI;EACJ,IAAI,KAAK,cACP,cAAc,GAAG,KAAK,YAAY,YAAY;OACzC;GACL,aAAa,GAAG,WAAW,eAAe,KAAK,UAC7C,WACF,EAAE;GACF,UAAU,QAAQ;GAClB,cAAc,GACX,KAAK,YAAY,MAAM,SAAS,QAC/B,WACA,YAAY,SAAS,mBAAmB,CAC1C,EACF;EACF;EAEA,IAAI,KAAK,gBAAgB,MACvB,YAAY,KAAK,gBAAgB;EAGnC,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,IAAI,YAAsB;EACxB,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAuB;EACzB,IAAI,KAAK,gBAAgB,MACvB,OAAO,CAAC,KAAK,WAAW,KAAK,YAAY;EAE3C,OAAO,CAAC,KAAK,SAAS;CACxB;;;;;;;;;CAUA,MAAc,qBACZ,WACA,WACe;EAEf,IAAI,KAAK,IAAI,SAAS,MAAM,UAC1B;EAEF,MAAM,MAAM,KAAK;EAUjB,IAAI,OAAA,GAAA,qCAAA,mBAAA,CANsC;GACxC,QAAQ,GAHc,YADD,KAAK,OAAO,WACkB;GAKnD,WAAW,IAAI;EACjB,CAAC,KAEe,IAAI,aAAa,KAC/B,MAAM,IAAI,MAAM,wFACd,IAAI,MACL,2CAAA,GAAA,qCAAA,oBAAA,CACC,IAAI,KACN,EAAE;4LACoL;CAE1L;AACF;AAqBA,MAAM,SAAS,SAAiB;CAE9B,IAAI,UAAU,KAAK,QAAQ,QAAQ,IAAG,CAAC,CAAC,KAAK;CAE7C,IAAI,QAAQ,WAAW,IAAG,KAAK,QAAQ,SAAS,IAAG,GACjD,UAAU,QAAQ,UAAU,GAAG,QAAQ,SAAS,CAAC;CAEnD,OAAO;AACT;AAEA,MAAM,cAAc,MAAmB,SACrC,IAAI,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,QAAQ,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B/C,eAAsB,oBAAoB,EACxC,KACA,IACA,QACA,IAAI,GACJ,WAC4B;CAC5B,IAAI;CACJ,IAAI,QACF,cAAc;MACT,IAAIC,sBAAAA,gBAAgB,UACzB,cAAcA,sBAAAA,gBAAgB;MAE9B,cAAcH,sBAAAA;CAGhB,IACE,2BACE,IAAI,IAAI;EAAC;EAAS;EAAS;CAAY,CAAC,GACxC,IAAI,IAAI,YAAY,cAAc,CACpC,CAAC,CAAC,OAAO,GAET,MAAM,IAAI,MACR,2GACK,YAAY,eAAe,oBAAoB,aACtD;CAEF,IAAI,YAAY,eAAe,SAAS,SAAS,GAC/C,cAAc,MAAM,YAAY,QAAQ,EAAE,QAAQ,CAAC;CAGrD,cAAc,MAAM,YAAY,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;CAa/D,OAAOI,0BAAAA,iBAAiB,KAAK;EAC3BC,0BAAAA,oBAAoB,OAAO;GAX3B,QAAQ,MAA+B;IACrC,IAAI,cAAc,GAChB,OAAO,GAAI,EAAe,SAAS;IAErC,MAAM,IAAI,MAAM,yCAAyC;GAC3D;GACA,YAAY,OAAO,MACjB,GAAG,aAAc,EAAyB,eAAe;EAI3B,CAAC;GAChC,MAAM;GACL,MAAM,YAAY,EAAE,GAAG,EAAE;GACzB,OAAO,UAAU;GACjB,OAAO,UAAU;GACjB,OAAO;EACT;EACA;EACA,IAAI,WAAW,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC;EACzC,IAAIC,+BAAAA,mBAAmB;EACvB;CACF,CAAC;AACH"}

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

@@ -6,7 +6,6 @@ import { ChainValues } from "@langchain/core/utils/types";
 import { BasePromptTemplate, PromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModel, BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 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
@@ -68,6 +67,14 @@ declare class SqlDatabaseChain extends BaseChain {
   _chainType(): "sql_database_chain";
   get inputKeys(): string[];
   get outputKeys(): string[];
+  /**
+   * Private method that verifies the number of tokens in the input text and
+   * table information. It throws an error if the number of tokens exceeds
+   * the maximum allowed by the language model.
+   * @param inputText The input text.
+   * @param tableinfo The table information.
+   * @returns A promise that resolves when the verification is complete.
+   */
   private verifyNumberOfTokens;
 }
 interface CreateSqlQueryChainFields {
@@ -106,13 +113,7 @@ interface CreateSqlQueryChainFields {
  * });
  * ```
  */
-declare function createSqlQueryChain({
-  llm,
-  db,
-  prompt,
-  k,
-  dialect
-}: CreateSqlQueryChainFields): Promise<RunnableSequence<Record<string, unknown>, string>>;
+declare function createSqlQueryChain({ llm, db, prompt, k, dialect }: CreateSqlQueryChainFields): Promise<RunnableSequence<Record<string, unknown>, string>>;
 //#endregion
 export { CreateSqlQueryChainFields, SqlDatabaseChain, SqlDatabaseChainInput, createSqlQueryChain };
 //# sourceMappingURL=sql_db_chain.d.cts.map

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

@@ -1 +1 @@
-{"version":3,"file":"sql_db_chain.d.cts","names":[],"sources":["../../../src/chains/sql_db/sql_db_chain.ts"],"mappings":";;;;;;;;;;;;AAgCA;;;UAAiB,qBAAA,SAA8B,WAAA;EAC7C,GAAA,EAAK,0BAAA;EACL,QAAA,EAAU,WAAA;EACV,IAAA;EACA,QAAA;EACA,SAAA;EACA,YAAA;EACA,MAAA,GAAS,cAAA;AAAA;;;;;;;;;;;AA4BX;;;;;;;;;;;;;;;cAAa,gBAAA,SAAyB,SAAA;EAAA,OAC7B,OAAA,CAAA;EAKP,GAAA,EAAK,0BAAA;EAGL,QAAA,EAAU,WAAA;EAGV,MAAA,EAAM,cAAA;;;;;;EAGN,IAAA;EAEA,QAAA;EAEA,SAAA;EAEA,YAAA;EAGA,YAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,qBAAA;EAApB;EAcM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EA0DX,UAAA,CAAA;EAAA,IAII,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAAA,QAeU,oBAAA;AAAA;AAAA,UA6BC,yBAAA;EACf,GAAA,EAAK,iBAAA;EACL,EAAA,EAAI,WAAA;EACJ,MAAA,GAAS,kBAAA;EAnDL;;;EAuDJ,CAAA;EACA,OAAA,EAAS,UAAA;AAAA;;;;;;;;;;;;;;;;;;;;AAkDX;;;;;;;iBAAsB,mBAAA,CAAA;EACpB,GAAA;EACA,EAAA;EACA,MAAA;EACA,CAAA;EACA;AAAA,GACC,yBAAA,GAAyB,OAAA,CAAA,gBAAA,CAAA,MAAA"}
+{"version":3,"file":"sql_db_chain.d.cts","names":[],"sources":["../../../src/chains/sql_db/sql_db_chain.ts"],"mappings":";;;;;;;;;;;;;;UAgCiB,8BAA8B;EAC7C,KAAK;EACL,UAAU;EACV;EACA;EACA;EACA;EACA,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;cA4BE,yBAAyB;SAC7B;EAKP,KAAK;EAGL,UAAU;EAGV,QAAM;;;;;;EAGN;EAEA;EAEA;EAEA;EAGA;EAEA,YAAY,QAAQ;;EAcd,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EA0DX;MAII;MAIA;;;;;;;;;UAeU;;UA6BC;EACf,KAAK;EACL,IAAI;EACJ,SAAS;;;;EAIT;EACA,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAkDW,sBACpB,KACA,IACA,QACA,GACA,WACC,4BAAyB,QAAA,iBAAA"}

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

@@ -6,7 +6,6 @@ import { RunnableSequence } from "@langchain/core/runnables";
 import { BaseLanguageModel, BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { ChainValues } from "@langchain/core/utils/types";
-
 //#region src/chains/sql_db/sql_db_chain.d.ts
 /**
  * Interface that extends the ChainInputs interface and defines additional
@@ -68,6 +67,14 @@ declare class SqlDatabaseChain extends BaseChain {
   _chainType(): "sql_database_chain";
   get inputKeys(): string[];
   get outputKeys(): string[];
+  /**
+   * Private method that verifies the number of tokens in the input text and
+   * table information. It throws an error if the number of tokens exceeds
+   * the maximum allowed by the language model.
+   * @param inputText The input text.
+   * @param tableinfo The table information.
+   * @returns A promise that resolves when the verification is complete.
+   */
   private verifyNumberOfTokens;
 }
 interface CreateSqlQueryChainFields {
@@ -106,13 +113,7 @@ interface CreateSqlQueryChainFields {
  * });
  * ```
  */
-declare function createSqlQueryChain({
-  llm,
-  db,
-  prompt,
-  k,
-  dialect
-}: CreateSqlQueryChainFields): Promise<RunnableSequence<Record<string, unknown>, string>>;
+declare function createSqlQueryChain({ llm, db, prompt, k, dialect }: CreateSqlQueryChainFields): Promise<RunnableSequence<Record<string, unknown>, string>>;
 //#endregion
 export { CreateSqlQueryChainFields, SqlDatabaseChain, SqlDatabaseChainInput, createSqlQueryChain };
 //# sourceMappingURL=sql_db_chain.d.ts.map

package/dist/chains/sql_db/sql_db_chain.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sql_db_chain.d.ts","names":[],"sources":["../../../src/chains/sql_db/sql_db_chain.ts"],"mappings":";;;;;;;;;;;;AAgCA;;;UAAiB,qBAAA,SAA8B,WAAA;EAC7C,GAAA,EAAK,0BAAA;EACL,QAAA,EAAU,WAAA;EACV,IAAA;EACA,QAAA;EACA,SAAA;EACA,YAAA;EACA,MAAA,GAAS,cAAA;AAAA;;;;;;;;;;;AA4BX;;;;;;;;;;;;;;;cAAa,gBAAA,SAAyB,SAAA;EAAA,OAC7B,OAAA,CAAA;EAKP,GAAA,EAAK,0BAAA;EAGL,QAAA,EAAU,WAAA;EAGV,MAAA,EAAM,cAAA;;;;;;EAGN,IAAA;EAEA,QAAA;EAEA,SAAA;EAEA,YAAA;EAGA,YAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,qBAAA;EAApB;EAcM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EA0DX,UAAA,CAAA;EAAA,IAII,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAAA,QAeU,oBAAA;AAAA;AAAA,UA6BC,yBAAA;EACf,GAAA,EAAK,iBAAA;EACL,EAAA,EAAI,WAAA;EACJ,MAAA,GAAS,kBAAA;EAnDL;;;EAuDJ,CAAA;EACA,OAAA,EAAS,UAAA;AAAA;;;;;;;;;;;;;;;;;;;;AAkDX;;;;;;;iBAAsB,mBAAA,CAAA;EACpB,GAAA;EACA,EAAA;EACA,MAAA;EACA,CAAA;EACA;AAAA,GACC,yBAAA,GAAyB,OAAA,CAAA,gBAAA,CAAA,MAAA"}
+{"version":3,"file":"sql_db_chain.d.ts","names":[],"sources":["../../../src/chains/sql_db/sql_db_chain.ts"],"mappings":";;;;;;;;;;;;;;UAgCiB,8BAA8B;EAC7C,KAAK;EACL,UAAU;EACV;EACA;EACA;EACA;EACA,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;cA4BE,yBAAyB;SAC7B;EAKP,KAAK;EAGL,UAAU;EAGV,QAAM;;;;;;EAGN;EAEA;EAEA;EAEA;EAGA;EAEA,YAAY,QAAQ;;EAcd,MACJ,QAAQ,aACR,aAAa,6BACZ,QAAQ;EA0DX;MAII;MAIA;;;;;;;;;UAeU;;UA6BC;EACf,KAAK;EACL,IAAI;EACJ,SAAS;;;;EAIT;EACA,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAkDW,sBACpB,KACA,IACA,QACA,GACA,WACC,4BAAyB,QAAA,iBAAA"}

package/dist/chains/sql_db/sql_db_chain.js

@@ -155,7 +155,7 @@ async function createSqlQueryChain({ llm, db, prompt, k = 5, dialect }) {
 	if (prompt) promptToUse = prompt;
 	else if (SQL_PROMPTS_MAP[dialect]) promptToUse = SQL_PROMPTS_MAP[dialect];
 	else promptToUse = DEFAULT_SQL_DATABASE_PROMPT;
-	if (difference(new Set([
+	if (difference(/* @__PURE__ */ new Set([
 		"input",
 		"top_k",
 		"table_info"

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

@@ -1 +1 @@
-{"version":3,"file":"sql_db_chain.js","names":[],"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,cAAsC,UAAU;CAC9C,OAAO,UAAU;AACf,SAAO;;CAIT;CAGA;CAGA,SAAS;CAGT,OAAO;CAEP,WAAW;CAEX,YAAY;CAEZ,eAAmC,KAAA;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,UACP,gCAAgC,KAAK,SAAS,cAAc;;;CAIhE,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,WAAW,IAAI,SAAS;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,MAAI,MANmB,mBAAmB;GACxC,QAAQ,GAHc,YADD,KAAK,OAAO,WACkB;GAKnD,WAAW,IAAI;GAChB,CAAC,IAEc,IAAI,aAAa,IAC/B,OAAM,IAAI,MAAM,wFACd,IAAI,MACL,0CAA0C,oBACzC,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;UACL,gBAAgB,SACzB,eAAc,gBAAgB;KAE9B,eAAc;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,QAAO,iBAAiB,KAAK;EAC3B,oBAAoB,OAAO;GAX3B,QAAQ,MAA+B;AACrC,QAAI,cAAc,EAChB,QAAO,GAAI,EAAe,SAAS;AAErC,UAAM,IAAI,MAAM,0CAA0C;;GAE5D,YAAY,OAAO,MACjB,GAAG,aAAc,EAAyB,gBAAgB;GAI3B,CAAC;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,IAAI,oBAAoB;EACxB;EACD,CAAC"}
+{"version":3,"file":"sql_db_chain.js","names":[],"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,cAAsC,UAAU;CAC9C,OAAO,UAAU;EACf,OAAO;CACT;CAGA;CAGA;CAGA,SAAS;CAGT,OAAO;CAEP,WAAW;CAEX,YAAY;CAEZ,eAAmC,KAAA;CAGnC,eAAe;CAEf,YAAY,QAA+B;EACzC,MAAM,MAAM;EACZ,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,UACP,gCAAgC,KAAK,SAAS,aAAa;CAC/D;;CAGA,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,WAAW,IAAI,SAAS;GAC5B,QAAQ,KAAK;GACb,KAAK,KAAK;GACV,WAAW,KAAK;GAChB,QAAQ,KAAK;EACf,CAAC;EACD,IAAI,EAAE,KAAK,YAAY,SACrB,MAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,YAAY;EAG5D,IAAI,YAAY,GADS,OAAO,KAAK,UACT;EAC5B,MAAM,cAAc,OAAO;EAC3B,MAAM,YAAY,MAAM,KAAK,SAAS,aAAa,WAAW;EAE9D,MAAM,YAAY;GAChB,OAAO;GACP,OAAO,KAAK;GACZ,SAAS,KAAK,SAAS,qBAAqB;GAC5C,YAAY;GACZ,MAAM,CAAC,cAAc;EACvB;EACA,MAAM,KAAK,qBAAqB,WAAW,SAAS;EAEpD,MAAM,aAAa,MAAM,SAAS,QAChC,WACA,YAAY,SAAS,gBAAgB,CACvC;EACA,IAAI,cAAc;EAClB,IAAI;GACF,cAAc,MAAM,KAAK,SAAS,cAAc,MAAM,UAAU;EAClE,SAAS,OAAO;GACd,QAAQ,MAAM,KAAK;EACrB;EAEA,IAAI;EACJ,IAAI,KAAK,cACP,cAAc,GAAG,KAAK,YAAY,YAAY;OACzC;GACL,aAAa,GAAG,WAAW,eAAe,KAAK,UAC7C,WACF,EAAE;GACF,UAAU,QAAQ;GAClB,cAAc,GACX,KAAK,YAAY,MAAM,SAAS,QAC/B,WACA,YAAY,SAAS,mBAAmB,CAC1C,EACF;EACF;EAEA,IAAI,KAAK,gBAAgB,MACvB,YAAY,KAAK,gBAAgB;EAGnC,OAAO;CACT;CAEA,aAAa;EACX,OAAO;CACT;CAEA,IAAI,YAAsB;EACxB,OAAO,CAAC,KAAK,QAAQ;CACvB;CAEA,IAAI,aAAuB;EACzB,IAAI,KAAK,gBAAgB,MACvB,OAAO,CAAC,KAAK,WAAW,KAAK,YAAY;EAE3C,OAAO,CAAC,KAAK,SAAS;CACxB;;;;;;;;;CAUA,MAAc,qBACZ,WACA,WACe;EAEf,IAAI,KAAK,IAAI,SAAS,MAAM,UAC1B;EAEF,MAAM,MAAM,KAAK;EAUjB,IAAI,MANmB,mBAAmB;GACxC,QAAQ,GAHc,YADD,KAAK,OAAO,WACkB;GAKnD,WAAW,IAAI;EACjB,CAAC,KAEe,IAAI,aAAa,KAC/B,MAAM,IAAI,MAAM,wFACd,IAAI,MACL,0CAA0C,oBACzC,IAAI,KACN,EAAE;4LACoL;CAE1L;AACF;AAqBA,MAAM,SAAS,SAAiB;CAE9B,IAAI,UAAU,KAAK,QAAQ,QAAQ,IAAG,CAAC,CAAC,KAAK;CAE7C,IAAI,QAAQ,WAAW,IAAG,KAAK,QAAQ,SAAS,IAAG,GACjD,UAAU,QAAQ,UAAU,GAAG,QAAQ,SAAS,CAAC;CAEnD,OAAO;AACT;AAEA,MAAM,cAAc,MAAmB,SACrC,IAAI,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,QAAQ,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4B/C,eAAsB,oBAAoB,EACxC,KACA,IACA,QACA,IAAI,GACJ,WAC4B;CAC5B,IAAI;CACJ,IAAI,QACF,cAAc;MACT,IAAI,gBAAgB,UACzB,cAAc,gBAAgB;MAE9B,cAAc;CAGhB,IACE,2BACE,IAAI,IAAI;EAAC;EAAS;EAAS;CAAY,CAAC,GACxC,IAAI,IAAI,YAAY,cAAc,CACpC,CAAC,CAAC,OAAO,GAET,MAAM,IAAI,MACR,2GACK,YAAY,eAAe,oBAAoB,aACtD;CAEF,IAAI,YAAY,eAAe,SAAS,SAAS,GAC/C,cAAc,MAAM,YAAY,QAAQ,EAAE,QAAQ,CAAC;CAGrD,cAAc,MAAM,YAAY,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;CAa/D,OAAO,iBAAiB,KAAK;EAC3B,oBAAoB,OAAO;GAX3B,QAAQ,MAA+B;IACrC,IAAI,cAAc,GAChB,OAAO,GAAI,EAAe,SAAS;IAErC,MAAM,IAAI,MAAM,yCAAyC;GAC3D;GACA,YAAY,OAAO,MACjB,GAAG,aAAc,EAAyB,eAAe;EAI3B,CAAC;GAChC,MAAM;GACL,MAAM,YAAY,EAAE,GAAG,EAAE;GACzB,OAAO,UAAU;GACjB,OAAO,UAAU;GACjB,OAAO;EACT;EACA;EACA,IAAI,WAAW,EAAE,MAAM,CAAC,cAAc,EAAE,CAAC;EACzC,IAAI,mBAAmB;EACvB;CACF,CAAC;AACH"}

package/dist/chains/sql_db/sql_db_prompt.cjs

@@ -1,7 +1,6 @@
-require("../../_virtual/_rolldown/runtime.cjs");
 let _langchain_core_prompts = require("@langchain/core/prompts");
 //#region src/chains/sql_db/sql_db_prompt.ts
-const DEFAULT_SQL_DATABASE_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const DEFAULT_SQL_DATABASE_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `Given an input question, first create a syntactically correct {dialect} query to run, then look at the results of the query and return the answer. Unless the user specifies in his question a specific number of examples he wishes to obtain, always limit your query to at most {top_k} results. You can order the results by a relevant column to return the most interesting examples in the database.
 
 Never query for all the columns from a specific table, only ask for a the few relevant columns given the question.
@@ -27,7 +26,7 @@ Question: {input}`,
 		"top_k"
 	]
 });
-const SQL_POSTGRES_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const SQL_POSTGRES_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `You are a PostgreSQL expert. Given an input question, first create a syntactically correct PostgreSQL query to run, then look at the results of the query and return the answer to the input question.
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per PostgreSQL. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (") to denote them as delimited identifiers.
@@ -51,7 +50,7 @@ Question: {input}`,
 		"top_k"
 	]
 });
-const SQL_SQLITE_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const SQL_SQLITE_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `You are a SQLite expert. Given an input question, first create a syntactically correct SQLite query to run, then look at the results of the query and return the answer to the input question.
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per SQLite. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (") to denote them as delimited identifiers.
@@ -75,7 +74,7 @@ Question: {input}`,
 		"top_k"
 	]
 });
-const SQL_MYSQL_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const SQL_MYSQL_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `You are a MySQL expert. Given an input question, first create a syntactically correct MySQL query to run, then look at the results of the query and return the answer to the input question.
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per MySQL. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in backticks (\`) to denote them as delimited identifiers.
@@ -99,7 +98,7 @@ Question: {input}`,
 		"top_k"
 	]
 });
-const SQL_MSSQL_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const SQL_MSSQL_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `You are an MS SQL expert. Given an input question, first create a syntactically correct MS SQL query to run, then look at the results of the query and return the answer to the input question.
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the TOP clause as per MS SQL. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in square brackets ([]) to denote them as delimited identifiers.
@@ -123,7 +122,7 @@ Question: {input}`,
 		"top_k"
 	]
 });
-const SQL_SAP_HANA_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const SQL_SAP_HANA_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `You are a SAP HANA expert. Given an input question, first create a syntactically correct SAP HANA query to run, then look at the results of the query and return the answer to the input question.
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per SAP HANA. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (") to denote them as delimited identifiers.
@@ -148,7 +147,7 @@ Question: {input}`,
 		"top_k"
 	]
 });
-const SQL_ORACLE_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
+const SQL_ORACLE_PROMPT = /*#__PURE__*/ new _langchain_core_prompts.PromptTemplate({
 	template: `You are a ORACLE expert. Given an input question, first create a syntactically correct ORACLE query to run, then look at the results of the query and return the answer to the input question.
 Unless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the ROWNUM clause as per ORACLE. You can order the results to return the most informative data in the database.
 Never query for all columns from a table. You must query only the columns that are needed to answer the question.

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

@@ -1 +1 @@
-{"version":3,"file":"sql_db_prompt.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/sql_db/sql_db_prompt.ts"],"sourcesContent":["import { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const DEFAULT_SQL_DATABASE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `Given an input question, first create a syntactically correct {dialect} query to run, then look at the results of the query and return the answer. Unless the user specifies in his question a specific number of examples he wishes to obtain, always limit your query to at most {top_k} results. You can order the results by a relevant column to return the most interesting examples in the database.\n\nNever query for all the columns from a specific table, only ask for a the few relevant columns given the question.\n\nPay attention to use only the column names that you can see in the schema description. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the tables listed below.\n\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_POSTGRES_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a PostgreSQL expert. Given an input question, first create a syntactically correct PostgreSQL query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per PostgreSQL. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (\") to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_SQLITE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a SQLite expert. Given an input question, first create a syntactically correct SQLite query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per SQLite. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (\") to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_MYSQL_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a MySQL expert. Given an input question, first create a syntactically correct MySQL query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per MySQL. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in backticks (\\`) to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_MSSQL_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are an MS SQL expert. Given an input question, first create a syntactically correct MS SQL query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the TOP clause as per MS SQL. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in square brackets ([]) to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_SAP_HANA_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a SAP HANA expert. Given an input question, first create a syntactically correct SAP HANA query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per SAP HANA. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (\") to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\nAlways use a schema name when executing a query.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_ORACLE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a ORACLE expert. Given an input question, first create a syntactically correct ORACLE query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the ROWNUM clause as per ORACLE. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport type SqlDialect =\n  | \"oracle\"\n  | \"postgres\"\n  | \"sqlite\"\n  | \"mysql\"\n  | \"mssql\"\n  | \"sap hana\";\n\nexport const SQL_PROMPTS_MAP: Record<SqlDialect, BasePromptTemplate> = {\n  oracle: SQL_ORACLE_PROMPT,\n  postgres: SQL_POSTGRES_PROMPT,\n  sqlite: SQL_SQLITE_PROMPT,\n  mysql: SQL_MYSQL_PROMPT,\n  mssql: SQL_MSSQL_PROMPT,\n  \"sap hana\": SQL_SAP_HANA_PROMPT,\n};\n"],"mappings":";;;AAEA,MAAa,8CAA4C,IAAIA,wBAAAA,eAAe;CAC1E,UAAU;;;;;;;;;;;;;;;;;;CAkBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAEF,MAAa,sCAAoC,IAAIA,wBAAAA,eAAe;CAClE,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAEF,MAAa,oCAAkC,IAAIA,wBAAAA,eAAe;CAChE,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAEF,MAAa,mCAAiC,IAAIA,wBAAAA,eAAe;CAC/D,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAEF,MAAa,mCAAiC,IAAIA,wBAAAA,eAAe;CAC/D,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAEF,MAAa,sCAAoC,IAAIA,wBAAAA,eAAe;CAClE,UAAU;;;;;;;;;;;;;;;;;CAiBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAEF,MAAa,oCAAkC,IAAIA,wBAAAA,eAAe;CAChE,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;EAAQ;CAC5D,CAAC;AAUF,MAAa,kBAA0D;CACrE,QAAQ;CACR,UAAU;CACV,QAAQ;CACR,OAAO;CACP,OAAO;CACP,YAAY;CACb"}
+{"version":3,"file":"sql_db_prompt.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/sql_db/sql_db_prompt.ts"],"sourcesContent":["import { BasePromptTemplate, PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const DEFAULT_SQL_DATABASE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `Given an input question, first create a syntactically correct {dialect} query to run, then look at the results of the query and return the answer. Unless the user specifies in his question a specific number of examples he wishes to obtain, always limit your query to at most {top_k} results. You can order the results by a relevant column to return the most interesting examples in the database.\n\nNever query for all the columns from a specific table, only ask for a the few relevant columns given the question.\n\nPay attention to use only the column names that you can see in the schema description. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the tables listed below.\n\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_POSTGRES_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a PostgreSQL expert. Given an input question, first create a syntactically correct PostgreSQL query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per PostgreSQL. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (\") to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_SQLITE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a SQLite expert. Given an input question, first create a syntactically correct SQLite query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per SQLite. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (\") to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_MYSQL_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a MySQL expert. Given an input question, first create a syntactically correct MySQL query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per MySQL. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in backticks (\\`) to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_MSSQL_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are an MS SQL expert. Given an input question, first create a syntactically correct MS SQL query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the TOP clause as per MS SQL. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in square brackets ([]) to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_SAP_HANA_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a SAP HANA expert. Given an input question, first create a syntactically correct SAP HANA query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the LIMIT clause as per SAP HANA. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question. Wrap each column name in double quotes (\") to denote them as delimited identifiers.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\nAlways use a schema name when executing a query.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport const SQL_ORACLE_PROMPT = /*#__PURE__*/ new PromptTemplate({\n  template: `You are a ORACLE expert. Given an input question, first create a syntactically correct ORACLE query to run, then look at the results of the query and return the answer to the input question.\nUnless the user specifies in the question a specific number of examples to obtain, query for at most {top_k} results using the ROWNUM clause as per ORACLE. You can order the results to return the most informative data in the database.\nNever query for all columns from a table. You must query only the columns that are needed to answer the question.\nPay attention to use only the column names you can see in the tables below. Be careful to not query for columns that do not exist. Also, pay attention to which column is in which table.\n\nUse the following format:\n\nQuestion: \"Question here\"\nSQLQuery: \"SQL Query to run\"\nSQLResult: \"Result of the SQLQuery\"\nAnswer: \"Final answer here\"\n\nOnly use the following tables:\n{table_info}\n\nQuestion: {input}`,\n  inputVariables: [\"dialect\", \"table_info\", \"input\", \"top_k\"],\n});\n\nexport type SqlDialect =\n  | \"oracle\"\n  | \"postgres\"\n  | \"sqlite\"\n  | \"mysql\"\n  | \"mssql\"\n  | \"sap hana\";\n\nexport const SQL_PROMPTS_MAP: Record<SqlDialect, BasePromptTemplate> = {\n  oracle: SQL_ORACLE_PROMPT,\n  postgres: SQL_POSTGRES_PROMPT,\n  sqlite: SQL_SQLITE_PROMPT,\n  mysql: SQL_MYSQL_PROMPT,\n  mssql: SQL_MSSQL_PROMPT,\n  \"sap hana\": SQL_SAP_HANA_PROMPT,\n};\n"],"mappings":";;AAEA,MAAa,4CAA4C,IAAIA,wBAAAA,eAAe;CAC1E,UAAU;;;;;;;;;;;;;;;;;;CAkBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAED,MAAa,oCAAoC,IAAIA,wBAAAA,eAAe;CAClE,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAED,MAAa,kCAAkC,IAAIA,wBAAAA,eAAe;CAChE,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAED,MAAa,iCAAiC,IAAIA,wBAAAA,eAAe;CAC/D,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAED,MAAa,iCAAiC,IAAIA,wBAAAA,eAAe;CAC/D,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAED,MAAa,oCAAoC,IAAIA,wBAAAA,eAAe;CAClE,UAAU;;;;;;;;;;;;;;;;;CAiBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAED,MAAa,kCAAkC,IAAIA,wBAAAA,eAAe;CAChE,UAAU;;;;;;;;;;;;;;;;CAgBV,gBAAgB;EAAC;EAAW;EAAc;EAAS;CAAO;AAC5D,CAAC;AAUD,MAAa,kBAA0D;CACrE,QAAQ;CACR,UAAU;CACV,QAAQ;CACR,OAAO;CACP,OAAO;CACP,YAAY;AACd"}

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

@@ -1,5 +1,4 @@
 import { BasePromptTemplate, PromptTemplate } from "@langchain/core/prompts";
-
 //#region src/chains/sql_db/sql_db_prompt.d.ts
 declare const DEFAULT_SQL_DATABASE_PROMPT: PromptTemplate<{
   dialect: any;