@langchain/classic 1.0.17 → 1.0.18

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

@@ -1 +1 @@
-{"version":3,"file":"conversational_retrieval_chain.d.ts","names":["BaseLanguageModelInterface","BaseRetrieverInterface","BaseMessage","ChainValues","CallbackManagerForChainRun","SerializedChatVectorDBQAChain","BaseChain","ChainInputs","LLMChain","QAChainParams","LoadValues","Record","ConversationalRetrievalQAChainInput","ConversationalRetrievalQAChain","Promise","Omit"],"sources":["../../src/chains/conversational_retrieval_chain.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { QAChainParams } from \"./question_answering/load.js\";\nexport type LoadValues = Record<string, any>;\n/**\n * Interface for the input parameters of the\n * ConversationalRetrievalQAChain class.\n */\nexport interface ConversationalRetrievalQAChainInput extends ChainInputs {\n    retriever: BaseRetrieverInterface;\n    combineDocumentsChain: BaseChain;\n    questionGeneratorChain: LLMChain;\n    returnSourceDocuments?: boolean;\n    returnGeneratedQuestion?: boolean;\n    inputKey?: string;\n}\n/**\n * Class for conducting conversational question-answering tasks with a\n * retrieval component. Extends the BaseChain class and implements the\n * ConversationalRetrievalQAChainInput interface.\n * @example\n * ```typescript\n * import { ChatAnthropic } from \"@langchain/anthropic\";\n * import {\n *   ChatPromptTemplate,\n *   MessagesPlaceholder,\n * } from \"@langchain/core/prompts\";\n * import { BaseMessage } from \"@langchain/core/messages\";\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n *\n * const retriever = ...your retriever;\n * const llm = new ChatAnthropic();\n *\n * // Contextualize question\n * const contextualizeQSystemPrompt = `\n * Given a chat history and the latest user question\n * which might reference context in the chat history,\n * formulate a standalone question which can be understood\n * without the chat history. Do NOT answer the question, just\n * reformulate it if needed and otherwise return it as is.`;\n * const contextualizeQPrompt = ChatPromptTemplate.fromMessages([\n *   [\"system\", contextualizeQSystemPrompt],\n *   new MessagesPlaceholder(\"chat_history\"),\n *   [\"human\", \"{input}\"],\n * ]);\n * const historyAwareRetriever = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt: contextualizeQPrompt,\n * });\n *\n * // Answer question\n * const qaSystemPrompt = `\n * You are an assistant for question-answering tasks. Use\n * the following pieces of retrieved context to answer the\n * question. If you don't know the answer, just say that you\n * don't know. Use three sentences maximum and keep the answer\n * concise.\n * \\n\\n\n * {context}`;\n * const qaPrompt = ChatPromptTemplate.fromMessages([\n *   [\"system\", qaSystemPrompt],\n *   new MessagesPlaceholder(\"chat_history\"),\n *   [\"human\", \"{input}\"],\n * ]);\n *\n * // Below we use createStuffDocuments_chain to feed all retrieved context\n * // into the LLM. Note that we can also use StuffDocumentsChain and other\n * // instances of BaseCombineDocumentsChain.\n * const questionAnswerChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt: qaPrompt,\n * });\n *\n * const ragChain = await createRetrievalChain({\n *   retriever: historyAwareRetriever,\n *   combineDocsChain: questionAnswerChain,\n * });\n *\n * // Usage:\n * const chat_history: BaseMessage[] = [];\n * const response = await ragChain.invoke({\n *   chat_history,\n *   input: \"...\",\n * });\n * ```\n */\nexport declare class ConversationalRetrievalQAChain extends BaseChain implements ConversationalRetrievalQAChainInput {\n    static lc_name(): string;\n    inputKey: string;\n    chatHistoryKey: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    retriever: BaseRetrieverInterface;\n    combineDocumentsChain: BaseChain;\n    questionGeneratorChain: LLMChain;\n    returnSourceDocuments: boolean;\n    returnGeneratedQuestion: boolean;\n    constructor(fields: ConversationalRetrievalQAChainInput);\n    /**\n     * Static method to convert the chat history input into a formatted\n     * string.\n     * @param chatHistory Chat history input which can be a string, an array of BaseMessage instances, or an array of string arrays.\n     * @returns A formatted string representing the chat history.\n     */\n    static getChatHistoryString(chatHistory: string | BaseMessage[] | string[][]): string;\n    /** @ignore */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    _chainType(): string;\n    static deserialize(_data: SerializedChatVectorDBQAChain, _values: LoadValues): Promise<ConversationalRetrievalQAChain>;\n    serialize(): SerializedChatVectorDBQAChain;\n    /**\n     * Static method to create a new ConversationalRetrievalQAChain from a\n     * BaseLanguageModel and a BaseRetriever.\n     * @param llm {@link BaseLanguageModelInterface} instance used to generate a new question.\n     * @param retriever {@link BaseRetrieverInterface} instance used to retrieve relevant documents.\n     * @param options.returnSourceDocuments Whether to return source documents in the final output\n     * @param options.questionGeneratorChainOptions Options to initialize the standalone question generation chain used as the first internal step\n     * @param options.qaChainOptions {@link QAChainParams} used to initialize the QA chain used as the second internal step\n     * @returns A new instance of ConversationalRetrievalQAChain.\n     */\n    static fromLLM(llm: BaseLanguageModelInterface, retriever: BaseRetrieverInterface, options?: {\n        outputKey?: string;\n        returnSourceDocuments?: boolean;\n        /** @deprecated Pass in questionGeneratorChainOptions.template instead */\n        questionGeneratorTemplate?: string;\n        /** @deprecated Pass in qaChainOptions.prompt instead */\n        qaTemplate?: string;\n        questionGeneratorChainOptions?: {\n            llm?: BaseLanguageModelInterface;\n            template?: string;\n        };\n        qaChainOptions?: QAChainParams;\n    } & Omit<ConversationalRetrievalQAChainInput, \"retriever\" | \"combineDocumentsChain\" | \"questionGeneratorChain\">): ConversationalRetrievalQAChain;\n}\n//# sourceMappingURL=conversational_retrieval_chain.d.ts.map"],"mappings":";;;;;;;;;;;KASYU,UAAAA,GAAaC;;AAAzB;AAKA;;AAE2BL,UAFVM,mCAAAA,SAA4CL,WAElCD,CAAAA;EACCE,SAAAA,EAFbP,sBAEaO;EAHiCD,qBAAAA,EAElCD,SAFkCC;EAAW,sBAAA,EAG5CC,QAH4C;EAiFnDK,qBAAAA,CAAAA,EAAAA,OAAAA;EAMNZ,uBAAAA,CAAAA,EAAAA,OAAAA;EACYK,QAAAA,CAAAA,EAAAA,MAAAA;;;;;;;;;;;;;;;;;;;;;;AAPyF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAA/FO,8BAAAA,SAAuCP,SAAAA,YAAqBM;;;;;;aAMlEX;yBACYK;0BACCE;;;sBAGJI;;;;;;;oDAO8BV;;gBAEpCC,0BAA0BC,6BAA6BU,QAAQX;;4BAEnDE,wCAAwCK,aAAaI,QAAQD;eAC1ER;;;;;;;;;;;sBAWOL,uCAAuCC;;;;;;;;YAQ7CD;;;qBAGOS;MACjBM,KAAKH,yGAAyGC"}
+{"version":3,"file":"conversational_retrieval_chain.d.ts","names":[],"sources":["../../src/chains/conversational_retrieval_chain.ts"],"mappings":";;;;;;;;;;;KAYY,UAAA,GAAa,MAAA;;AAAzB;;;UAaiB,mCAAA,SAA4C,WAAA;EAC3D,SAAA,EAAW,sBAAA;EACX,qBAAA,EAAuB,SAAA;EACvB,sBAAA,EAAwB,QAAA;EACxB,qBAAA;EACA,uBAAA;EACA,QAAA;AAAA;;;;;;;;;;;;;;;;AA4EF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAAa,8BAAA,SACH,SAAA,YACG,mCAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,QAAA;EAEA,cAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAMJ,SAAA,EAAW,sBAAA;EAEX,qBAAA,EAAuB,SAAA;EAEvB,sBAAA,EAAwB,QAAA;EAExB,qBAAA;EAEA,uBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,mCAAA;EAiJlB;;;;;;EAAA,OA/HK,oBAAA,CACL,WAAA,WAAsB,WAAA;EAuIZ;EAjGN,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EA0DX,UAAA,CAAA;EAAA,OAIa,WAAA,CACX,KAAA,EAAO,6BAAA,EACP,OAAA,EAAS,UAAA,GACR,OAAA,CAAQ,8BAAA;EAIX,SAAA,CAAA,GAAa,6BAAA;EA8BT;;;;;;;;;;EAAA,OAhBG,OAAA,CACL,GAAA,EAAK,0BAAA,EACL,SAAA,EAAW,sBAAA,EACX,OAAA;IACE,SAAA;IACA,qBAAA;IAEA,yBAAA;IAEA,UAAA;IACA,6BAAA;MACE,GAAA,GAAM,0BAAA;MACN,QAAA;IAAA;IAEF,cAAA,GAAiB,aAAA;EAAA,IACf,IAAA,CACF,mCAAA,sEAGD,8BAAA;AAAA"}

package/dist/chains/conversational_retrieval_chain.js

@@ -1,8 +1,8 @@
 import { BaseChain } from "./base.js";
 import { LLMChain } from "./llm_chain.js";
 import { loadQAChain } from "./question_answering/load.js";
-import { PromptTemplate } from "@langchain/core/prompts";
 import { AIMessage, HumanMessage } from "@langchain/core/messages";
+import { PromptTemplate } from "@langchain/core/prompts";
 
 //#region src/chains/conversational_retrieval_chain.ts
 const question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
@@ -142,12 +142,12 @@ var ConversationalRetrievalQAChain = class ConversationalRetrievalQAChain extend
 		const chatHistory = ConversationalRetrievalQAChain.getChatHistoryString(values[this.chatHistoryKey]);
 		let newQuestion = question;
 		if (chatHistory.length > 0) {
-			const result$1 = await this.questionGeneratorChain.call({
+			const result = await this.questionGeneratorChain.call({
 				question,
 				chat_history: chatHistory
 			}, runManager?.getChild("question_generator"));
-			const keys = Object.keys(result$1);
-			if (keys.length === 1) newQuestion = result$1[keys[0]];
+			const keys = Object.keys(result);
+			if (keys.length === 1) newQuestion = result[keys[0]];
 			else throw new Error("Return from llm chain has multiple values, only single values supported.");
 		}
 		const docs = await this.retriever.invoke(newQuestion, runManager?.getChild("retriever"));
@@ -190,22 +190,20 @@ var ConversationalRetrievalQAChain = class ConversationalRetrievalQAChain extend
 		const { questionGeneratorTemplate, qaTemplate, qaChainOptions = {
 			type: "stuff",
 			prompt: qaTemplate ? PromptTemplate.fromTemplate(qaTemplate) : void 0
-		}, questionGeneratorChainOptions, verbose,...rest } = options;
+		}, questionGeneratorChainOptions, verbose, ...rest } = options;
 		const qaChain = loadQAChain(llm, qaChainOptions);
-		const questionGeneratorChainPrompt = PromptTemplate.fromTemplate(questionGeneratorChainOptions?.template ?? questionGeneratorTemplate ?? question_generator_template);
 		const questionGeneratorChain = new LLMChain({
-			prompt: questionGeneratorChainPrompt,
+			prompt: PromptTemplate.fromTemplate(questionGeneratorChainOptions?.template ?? questionGeneratorTemplate ?? question_generator_template),
 			llm: questionGeneratorChainOptions?.llm ?? llm,
 			verbose
 		});
-		const instance = new this({
+		return new this({
 			retriever,
 			combineDocumentsChain: qaChain,
 			questionGeneratorChain,
 			verbose,
 			...rest
 		});
-		return instance;
 	}
 };
 

package/dist/chains/conversational_retrieval_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"conversational_retrieval_chain.js","names":["fields: ConversationalRetrievalQAChainInput","chatHistory: string | BaseMessage[] | string[][]","historyMessages: BaseMessage[]","values: ChainValues","runManager?: CallbackManagerForChainRun","question: string","chatHistory: string","result","_data: SerializedChatVectorDBQAChain","_values: LoadValues","llm: BaseLanguageModelInterface","retriever: BaseRetrieverInterface","options: {\n      outputKey?: string; // not used\n      returnSourceDocuments?: boolean;\n      /** @deprecated Pass in questionGeneratorChainOptions.template instead */\n      questionGeneratorTemplate?: string;\n      /** @deprecated Pass in qaChainOptions.prompt instead */\n      qaTemplate?: string;\n      questionGeneratorChainOptions?: {\n        llm?: BaseLanguageModelInterface;\n        template?: string;\n      };\n      qaChainOptions?: QAChainParams;\n    } & Omit<\n      ConversationalRetrievalQAChainInput,\n      \"retriever\" | \"combineDocumentsChain\" | \"questionGeneratorChain\"\n    >"],"sources":["../../src/chains/conversational_retrieval_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { BaseMessage, HumanMessage, AIMessage } from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { QAChainParams, loadQAChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\n/**\n * Interface for the input parameters of the\n * ConversationalRetrievalQAChain class.\n */\nexport interface ConversationalRetrievalQAChainInput extends ChainInputs {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  returnGeneratedQuestion?: boolean;\n  inputKey?: string;\n}\n\n/**\n * Class for conducting conversational question-answering tasks with a\n * retrieval component. Extends the BaseChain class and implements the\n * ConversationalRetrievalQAChainInput interface.\n * @example\n * ```typescript\n * import { ChatAnthropic } from \"@langchain/anthropic\";\n * import {\n *   ChatPromptTemplate,\n *   MessagesPlaceholder,\n * } from \"@langchain/core/prompts\";\n * import { BaseMessage } from \"@langchain/core/messages\";\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n *\n * const retriever = ...your retriever;\n * const llm = new ChatAnthropic();\n *\n * // Contextualize question\n * const contextualizeQSystemPrompt = `\n * Given a chat history and the latest user question\n * which might reference context in the chat history,\n * formulate a standalone question which can be understood\n * without the chat history. Do NOT answer the question, just\n * reformulate it if needed and otherwise return it as is.`;\n * const contextualizeQPrompt = ChatPromptTemplate.fromMessages([\n *   [\"system\", contextualizeQSystemPrompt],\n *   new MessagesPlaceholder(\"chat_history\"),\n *   [\"human\", \"{input}\"],\n * ]);\n * const historyAwareRetriever = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt: contextualizeQPrompt,\n * });\n *\n * // Answer question\n * const qaSystemPrompt = `\n * You are an assistant for question-answering tasks. Use\n * the following pieces of retrieved context to answer the\n * question. If you don't know the answer, just say that you\n * don't know. Use three sentences maximum and keep the answer\n * concise.\n * \\n\\n\n * {context}`;\n * const qaPrompt = ChatPromptTemplate.fromMessages([\n *   [\"system\", qaSystemPrompt],\n *   new MessagesPlaceholder(\"chat_history\"),\n *   [\"human\", \"{input}\"],\n * ]);\n *\n * // Below we use createStuffDocuments_chain to feed all retrieved context\n * // into the LLM. Note that we can also use StuffDocumentsChain and other\n * // instances of BaseCombineDocumentsChain.\n * const questionAnswerChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt: qaPrompt,\n * });\n *\n * const ragChain = await createRetrievalChain({\n *   retriever: historyAwareRetriever,\n *   combineDocsChain: questionAnswerChain,\n * });\n *\n * // Usage:\n * const chat_history: BaseMessage[] = [];\n * const response = await ragChain.invoke({\n *   chat_history,\n *   input: \"...\",\n * });\n * ```\n */\nexport class ConversationalRetrievalQAChain\n  extends BaseChain\n  implements ConversationalRetrievalQAChainInput\n{\n  static lc_name() {\n    return \"ConversationalRetrievalQAChain\";\n  }\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  returnGeneratedQuestion = false;\n\n  constructor(fields: ConversationalRetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n    this.returnGeneratedQuestion =\n      fields.returnGeneratedQuestion ?? this.returnGeneratedQuestion;\n  }\n\n  /**\n   * Static method to convert the chat history input into a formatted\n   * string.\n   * @param chatHistory Chat history input which can be a string, an array of BaseMessage instances, or an array of string arrays.\n   * @returns A formatted string representing the chat history.\n   */\n  static getChatHistoryString(\n    chatHistory: string | BaseMessage[] | string[][]\n  ) {\n    let historyMessages: BaseMessage[];\n    if (Array.isArray(chatHistory)) {\n      // TODO: Deprecate on a breaking release\n      if (\n        Array.isArray(chatHistory[0]) &&\n        typeof chatHistory[0][0] === \"string\"\n      ) {\n        console.warn(\n          \"Passing chat history as an array of strings is deprecated.\\nPlease see https://js.langchain.com/docs/modules/chains/popular/chat_vector_db#externally-managed-memory for more information.\"\n        );\n        historyMessages = chatHistory.flat().map((stringMessage, i) => {\n          if (i % 2 === 0) {\n            return new HumanMessage(stringMessage);\n          } else {\n            return new AIMessage(stringMessage);\n          }\n        });\n      } else {\n        historyMessages = chatHistory as BaseMessage[];\n      }\n      return historyMessages\n        .map((chatMessage) => {\n          if (chatMessage._getType() === \"human\") {\n            return `Human: ${chatMessage.content}`;\n          } else if (chatMessage._getType() === \"ai\") {\n            return `Assistant: ${chatMessage.content}`;\n          } else {\n            return `${chatMessage.content}`;\n          }\n        })\n        .join(\"\\n\");\n    }\n    return chatHistory;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string =\n      ConversationalRetrievalQAChain.getChatHistoryString(\n        values[this.chatHistoryKey]\n      );\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.retriever.invoke(\n      newQuestion,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    let result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      result = {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    if (this.returnGeneratedQuestion) {\n      result = {\n        ...result,\n        generatedQuestion: newQuestion,\n      };\n    }\n    return result;\n  }\n\n  _chainType(): string {\n    return \"conversational_retrieval_chain\";\n  }\n\n  static async deserialize(\n    _data: SerializedChatVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<ConversationalRetrievalQAChain> {\n    throw new Error(\"Not implemented.\");\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    throw new Error(\"Not implemented.\");\n  }\n\n  /**\n   * Static method to create a new ConversationalRetrievalQAChain from a\n   * BaseLanguageModel and a BaseRetriever.\n   * @param llm {@link BaseLanguageModelInterface} instance used to generate a new question.\n   * @param retriever {@link BaseRetrieverInterface} instance used to retrieve relevant documents.\n   * @param options.returnSourceDocuments Whether to return source documents in the final output\n   * @param options.questionGeneratorChainOptions Options to initialize the standalone question generation chain used as the first internal step\n   * @param options.qaChainOptions {@link QAChainParams} used to initialize the QA chain used as the second internal step\n   * @returns A new instance of ConversationalRetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options: {\n      outputKey?: string; // not used\n      returnSourceDocuments?: boolean;\n      /** @deprecated Pass in questionGeneratorChainOptions.template instead */\n      questionGeneratorTemplate?: string;\n      /** @deprecated Pass in qaChainOptions.prompt instead */\n      qaTemplate?: string;\n      questionGeneratorChainOptions?: {\n        llm?: BaseLanguageModelInterface;\n        template?: string;\n      };\n      qaChainOptions?: QAChainParams;\n    } & Omit<\n      ConversationalRetrievalQAChainInput,\n      \"retriever\" | \"combineDocumentsChain\" | \"questionGeneratorChain\"\n    > = {}\n  ): ConversationalRetrievalQAChain {\n    const {\n      questionGeneratorTemplate,\n      qaTemplate,\n      qaChainOptions = {\n        type: \"stuff\",\n        prompt: qaTemplate\n          ? PromptTemplate.fromTemplate(qaTemplate)\n          : undefined,\n      },\n      questionGeneratorChainOptions,\n      verbose,\n      ...rest\n    } = options;\n\n    const qaChain = loadQAChain(llm, qaChainOptions);\n\n    const questionGeneratorChainPrompt = PromptTemplate.fromTemplate(\n      questionGeneratorChainOptions?.template ??\n        questionGeneratorTemplate ??\n        question_generator_template\n    );\n    const questionGeneratorChain = new LLMChain({\n      prompt: questionGeneratorChainPrompt,\n      llm: questionGeneratorChainOptions?.llm ?? llm,\n      verbose,\n    });\n    const instance = new this({\n      retriever,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      verbose,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;;AAcA,MAAM,8BAA8B,CAAC;;;;;oBAKjB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwFrB,IAAa,iCAAb,MAAa,uCACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,cAAe;CAC5C;CAED,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,iBAAkB,IAAG,CAAE,EACtD;CACF;CAED;CAEA;CAEA;CAEA,wBAAwB;CAExB,0BAA0B;CAE1B,YAAYA,QAA6C;EACvD,MAAM,OAAO;EACb,KAAK,YAAY,OAAO;EACxB,KAAK,wBAAwB,OAAO;EACpC,KAAK,yBAAyB,OAAO;EACrC,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,wBACH,OAAO,yBAAyB,KAAK;EACvC,KAAK,0BACH,OAAO,2BAA2B,KAAK;CAC1C;;;;;;;CAQD,OAAO,qBACLC,aACA;EACA,IAAIC;AACJ,MAAI,MAAM,QAAQ,YAAY,EAAE;AAE9B,OACE,MAAM,QAAQ,YAAY,GAAG,IAC7B,OAAO,YAAY,GAAG,OAAO,UAC7B;IACA,QAAQ,KACN,6LACD;IACD,kBAAkB,YAAY,MAAM,CAAC,IAAI,CAAC,eAAe,MAAM;AAC7D,SAAI,IAAI,MAAM,EACZ,QAAO,IAAI,aAAa;SAExB,QAAO,IAAI,UAAU;IAExB,EAAC;GACH,OACC,kBAAkB;AAEpB,UAAO,gBACJ,IAAI,CAAC,gBAAgB;AACpB,QAAI,YAAY,UAAU,KAAK,QAC7B,QAAO,CAAC,OAAO,EAAE,YAAY,SAAS;aAC7B,YAAY,UAAU,KAAK,KACpC,QAAO,CAAC,WAAW,EAAE,YAAY,SAAS;QAE1C,QAAO,GAAG,YAAY,SAAS;GAElC,EAAC,CACD,KAAK,KAAK;EACd;AACD,SAAO;CACR;;CAGD,MAAM,MACJC,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,CAAC,aAAa,EAAE,KAAK,SAAS,WAAW,CAAC;AAE5D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,CAAC,iBAAiB,EAAE,KAAK,eAAe,WAAW,CAAC;EAEtE,MAAMC,WAAmB,OAAO,KAAK;EACrC,MAAMC,cACJ,+BAA+B,qBAC7B,OAAO,KAAK,gBACb;EACH,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAMC,WAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;GACf,GACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAKA,SAAO;AAChC,OAAI,KAAK,WAAW,GAClB,cAAcA,SAAO,KAAK;OAE1B,OAAM,IAAI,MACR;EAGL;EACD,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,aACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;EACf;EACD,IAAI,SAAS,MAAM,KAAK,sBAAsB,KAC5C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,uBACP,SAAS;GACP,GAAG;GACH,iBAAiB;EAClB;AAEH,MAAI,KAAK,yBACP,SAAS;GACP,GAAG;GACH,mBAAmB;EACpB;AAEH,SAAO;CACR;CAED,aAAqB;AACnB,SAAO;CACR;CAED,aAAa,YACXC,OACAC,SACyC;AACzC,QAAM,IAAI,MAAM;CACjB;CAED,YAA2C;AACzC,QAAM,IAAI,MAAM;CACjB;;;;;;;;;;;CAYD,OAAO,QACLC,KACAC,WACAC,UAeI,CAAE,GAC0B;EAChC,MAAM,EACJ,2BACA,YACA,iBAAiB;GACf,MAAM;GACN,QAAQ,aACJ,eAAe,aAAa,WAAW,GACvC;EACL,GACD,+BACA,QACA,GAAG,MACJ,GAAG;EAEJ,MAAM,UAAU,YAAY,KAAK,eAAe;EAEhD,MAAM,+BAA+B,eAAe,aAClD,+BAA+B,YAC7B,6BACA,4BACH;EACD,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QAAQ;GACR,KAAK,+BAA+B,OAAO;GAC3C;EACD;EACD,MAAM,WAAW,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA;GACA,GAAG;EACJ;AACD,SAAO;CACR;AACF"}
+{"version":3,"file":"conversational_retrieval_chain.js","names":[],"sources":["../../src/chains/conversational_retrieval_chain.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { BaseMessage, HumanMessage, AIMessage } from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { SerializedChatVectorDBQAChain } from \"./serde.js\";\nimport { BaseChain, ChainInputs } from \"./base.js\";\nimport { LLMChain } from \"./llm_chain.js\";\nimport { QAChainParams, loadQAChain } from \"./question_answering/load.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type LoadValues = Record<string, any>;\n\nconst question_generator_template = `Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.\n\nChat History:\n{chat_history}\nFollow Up Input: {question}\nStandalone question:`;\n\n/**\n * Interface for the input parameters of the\n * ConversationalRetrievalQAChain class.\n */\nexport interface ConversationalRetrievalQAChainInput extends ChainInputs {\n  retriever: BaseRetrieverInterface;\n  combineDocumentsChain: BaseChain;\n  questionGeneratorChain: LLMChain;\n  returnSourceDocuments?: boolean;\n  returnGeneratedQuestion?: boolean;\n  inputKey?: string;\n}\n\n/**\n * Class for conducting conversational question-answering tasks with a\n * retrieval component. Extends the BaseChain class and implements the\n * ConversationalRetrievalQAChainInput interface.\n * @example\n * ```typescript\n * import { ChatAnthropic } from \"@langchain/anthropic\";\n * import {\n *   ChatPromptTemplate,\n *   MessagesPlaceholder,\n * } from \"@langchain/core/prompts\";\n * import { BaseMessage } from \"@langchain/core/messages\";\n * import { createStuffDocumentsChain } from \"@langchain/classic/chains/combine_documents\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n * import { createRetrievalChain } from \"@langchain/classic/chains/retrieval\";\n *\n * const retriever = ...your retriever;\n * const llm = new ChatAnthropic();\n *\n * // Contextualize question\n * const contextualizeQSystemPrompt = `\n * Given a chat history and the latest user question\n * which might reference context in the chat history,\n * formulate a standalone question which can be understood\n * without the chat history. Do NOT answer the question, just\n * reformulate it if needed and otherwise return it as is.`;\n * const contextualizeQPrompt = ChatPromptTemplate.fromMessages([\n *   [\"system\", contextualizeQSystemPrompt],\n *   new MessagesPlaceholder(\"chat_history\"),\n *   [\"human\", \"{input}\"],\n * ]);\n * const historyAwareRetriever = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt: contextualizeQPrompt,\n * });\n *\n * // Answer question\n * const qaSystemPrompt = `\n * You are an assistant for question-answering tasks. Use\n * the following pieces of retrieved context to answer the\n * question. If you don't know the answer, just say that you\n * don't know. Use three sentences maximum and keep the answer\n * concise.\n * \\n\\n\n * {context}`;\n * const qaPrompt = ChatPromptTemplate.fromMessages([\n *   [\"system\", qaSystemPrompt],\n *   new MessagesPlaceholder(\"chat_history\"),\n *   [\"human\", \"{input}\"],\n * ]);\n *\n * // Below we use createStuffDocuments_chain to feed all retrieved context\n * // into the LLM. Note that we can also use StuffDocumentsChain and other\n * // instances of BaseCombineDocumentsChain.\n * const questionAnswerChain = await createStuffDocumentsChain({\n *   llm,\n *   prompt: qaPrompt,\n * });\n *\n * const ragChain = await createRetrievalChain({\n *   retriever: historyAwareRetriever,\n *   combineDocsChain: questionAnswerChain,\n * });\n *\n * // Usage:\n * const chat_history: BaseMessage[] = [];\n * const response = await ragChain.invoke({\n *   chat_history,\n *   input: \"...\",\n * });\n * ```\n */\nexport class ConversationalRetrievalQAChain\n  extends BaseChain\n  implements ConversationalRetrievalQAChainInput\n{\n  static lc_name() {\n    return \"ConversationalRetrievalQAChain\";\n  }\n\n  inputKey = \"question\";\n\n  chatHistoryKey = \"chat_history\";\n\n  get inputKeys() {\n    return [this.inputKey, this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return this.combineDocumentsChain.outputKeys.concat(\n      this.returnSourceDocuments ? [\"sourceDocuments\"] : []\n    );\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  combineDocumentsChain: BaseChain;\n\n  questionGeneratorChain: LLMChain;\n\n  returnSourceDocuments = false;\n\n  returnGeneratedQuestion = false;\n\n  constructor(fields: ConversationalRetrievalQAChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.combineDocumentsChain = fields.combineDocumentsChain;\n    this.questionGeneratorChain = fields.questionGeneratorChain;\n    this.inputKey = fields.inputKey ?? this.inputKey;\n    this.returnSourceDocuments =\n      fields.returnSourceDocuments ?? this.returnSourceDocuments;\n    this.returnGeneratedQuestion =\n      fields.returnGeneratedQuestion ?? this.returnGeneratedQuestion;\n  }\n\n  /**\n   * Static method to convert the chat history input into a formatted\n   * string.\n   * @param chatHistory Chat history input which can be a string, an array of BaseMessage instances, or an array of string arrays.\n   * @returns A formatted string representing the chat history.\n   */\n  static getChatHistoryString(\n    chatHistory: string | BaseMessage[] | string[][]\n  ) {\n    let historyMessages: BaseMessage[];\n    if (Array.isArray(chatHistory)) {\n      // TODO: Deprecate on a breaking release\n      if (\n        Array.isArray(chatHistory[0]) &&\n        typeof chatHistory[0][0] === \"string\"\n      ) {\n        console.warn(\n          \"Passing chat history as an array of strings is deprecated.\\nPlease see https://js.langchain.com/docs/modules/chains/popular/chat_vector_db#externally-managed-memory for more information.\"\n        );\n        historyMessages = chatHistory.flat().map((stringMessage, i) => {\n          if (i % 2 === 0) {\n            return new HumanMessage(stringMessage);\n          } else {\n            return new AIMessage(stringMessage);\n          }\n        });\n      } else {\n        historyMessages = chatHistory as BaseMessage[];\n      }\n      return historyMessages\n        .map((chatMessage) => {\n          if (chatMessage._getType() === \"human\") {\n            return `Human: ${chatMessage.content}`;\n          } else if (chatMessage._getType() === \"ai\") {\n            return `Assistant: ${chatMessage.content}`;\n          } else {\n            return `${chatMessage.content}`;\n          }\n        })\n        .join(\"\\n\");\n    }\n    return chatHistory;\n  }\n\n  /** @ignore */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.inputKey in values)) {\n      throw new Error(`Question key ${this.inputKey} not found.`);\n    }\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found.`);\n    }\n    const question: string = values[this.inputKey];\n    const chatHistory: string =\n      ConversationalRetrievalQAChain.getChatHistoryString(\n        values[this.chatHistoryKey]\n      );\n    let newQuestion = question;\n    if (chatHistory.length > 0) {\n      const result = await this.questionGeneratorChain.call(\n        {\n          question,\n          chat_history: chatHistory,\n        },\n        runManager?.getChild(\"question_generator\")\n      );\n      const keys = Object.keys(result);\n      if (keys.length === 1) {\n        newQuestion = result[keys[0]];\n      } else {\n        throw new Error(\n          \"Return from llm chain has multiple values, only single values supported.\"\n        );\n      }\n    }\n    const docs = await this.retriever.invoke(\n      newQuestion,\n      runManager?.getChild(\"retriever\")\n    );\n    const inputs = {\n      question: newQuestion,\n      input_documents: docs,\n      chat_history: chatHistory,\n    };\n    let result = await this.combineDocumentsChain.call(\n      inputs,\n      runManager?.getChild(\"combine_documents\")\n    );\n    if (this.returnSourceDocuments) {\n      result = {\n        ...result,\n        sourceDocuments: docs,\n      };\n    }\n    if (this.returnGeneratedQuestion) {\n      result = {\n        ...result,\n        generatedQuestion: newQuestion,\n      };\n    }\n    return result;\n  }\n\n  _chainType(): string {\n    return \"conversational_retrieval_chain\";\n  }\n\n  static async deserialize(\n    _data: SerializedChatVectorDBQAChain,\n    _values: LoadValues\n  ): Promise<ConversationalRetrievalQAChain> {\n    throw new Error(\"Not implemented.\");\n  }\n\n  serialize(): SerializedChatVectorDBQAChain {\n    throw new Error(\"Not implemented.\");\n  }\n\n  /**\n   * Static method to create a new ConversationalRetrievalQAChain from a\n   * BaseLanguageModel and a BaseRetriever.\n   * @param llm {@link BaseLanguageModelInterface} instance used to generate a new question.\n   * @param retriever {@link BaseRetrieverInterface} instance used to retrieve relevant documents.\n   * @param options.returnSourceDocuments Whether to return source documents in the final output\n   * @param options.questionGeneratorChainOptions Options to initialize the standalone question generation chain used as the first internal step\n   * @param options.qaChainOptions {@link QAChainParams} used to initialize the QA chain used as the second internal step\n   * @returns A new instance of ConversationalRetrievalQAChain.\n   */\n  static fromLLM(\n    llm: BaseLanguageModelInterface,\n    retriever: BaseRetrieverInterface,\n    options: {\n      outputKey?: string; // not used\n      returnSourceDocuments?: boolean;\n      /** @deprecated Pass in questionGeneratorChainOptions.template instead */\n      questionGeneratorTemplate?: string;\n      /** @deprecated Pass in qaChainOptions.prompt instead */\n      qaTemplate?: string;\n      questionGeneratorChainOptions?: {\n        llm?: BaseLanguageModelInterface;\n        template?: string;\n      };\n      qaChainOptions?: QAChainParams;\n    } & Omit<\n      ConversationalRetrievalQAChainInput,\n      \"retriever\" | \"combineDocumentsChain\" | \"questionGeneratorChain\"\n    > = {}\n  ): ConversationalRetrievalQAChain {\n    const {\n      questionGeneratorTemplate,\n      qaTemplate,\n      qaChainOptions = {\n        type: \"stuff\",\n        prompt: qaTemplate\n          ? PromptTemplate.fromTemplate(qaTemplate)\n          : undefined,\n      },\n      questionGeneratorChainOptions,\n      verbose,\n      ...rest\n    } = options;\n\n    const qaChain = loadQAChain(llm, qaChainOptions);\n\n    const questionGeneratorChainPrompt = PromptTemplate.fromTemplate(\n      questionGeneratorChainOptions?.template ??\n        questionGeneratorTemplate ??\n        question_generator_template\n    );\n    const questionGeneratorChain = new LLMChain({\n      prompt: questionGeneratorChainPrompt,\n      llm: questionGeneratorChainOptions?.llm ?? llm,\n      verbose,\n    });\n    const instance = new this({\n      retriever,\n      combineDocumentsChain: qaChain,\n      questionGeneratorChain,\n      verbose,\n      ...rest,\n    });\n    return instance;\n  }\n}\n"],"mappings":";;;;;;;AAcA,MAAM,8BAA8B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6FpC,IAAa,iCAAb,MAAa,uCACH,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,WAAW;CAEX,iBAAiB;CAEjB,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,UAAU,KAAK,eAAe;;CAG7C,IAAI,aAAa;AACf,SAAO,KAAK,sBAAsB,WAAW,OAC3C,KAAK,wBAAwB,CAAC,kBAAkB,GAAG,EAAE,CACtD;;CAGH;CAEA;CAEA;CAEA,wBAAwB;CAExB,0BAA0B;CAE1B,YAAY,QAA6C;AACvD,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,wBAAwB,OAAO;AACpC,OAAK,yBAAyB,OAAO;AACrC,OAAK,WAAW,OAAO,YAAY,KAAK;AACxC,OAAK,wBACH,OAAO,yBAAyB,KAAK;AACvC,OAAK,0BACH,OAAO,2BAA2B,KAAK;;;;;;;;CAS3C,OAAO,qBACL,aACA;EACA,IAAI;AACJ,MAAI,MAAM,QAAQ,YAAY,EAAE;AAE9B,OACE,MAAM,QAAQ,YAAY,GAAG,IAC7B,OAAO,YAAY,GAAG,OAAO,UAC7B;AACA,YAAQ,KACN,6LACD;AACD,sBAAkB,YAAY,MAAM,CAAC,KAAK,eAAe,MAAM;AAC7D,SAAI,IAAI,MAAM,EACZ,QAAO,IAAI,aAAa,cAAc;SAEtC,QAAO,IAAI,UAAU,cAAc;MAErC;SAEF,mBAAkB;AAEpB,UAAO,gBACJ,KAAK,gBAAgB;AACpB,QAAI,YAAY,UAAU,KAAK,QAC7B,QAAO,UAAU,YAAY;aACpB,YAAY,UAAU,KAAK,KACpC,QAAO,cAAc,YAAY;QAEjC,QAAO,GAAG,YAAY;KAExB,CACD,KAAK,KAAK;;AAEf,SAAO;;;CAIT,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,YAAY,QACrB,OAAM,IAAI,MAAM,gBAAgB,KAAK,SAAS,aAAa;AAE7D,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,eAAe,aAAa;EAEvE,MAAM,WAAmB,OAAO,KAAK;EACrC,MAAM,cACJ,+BAA+B,qBAC7B,OAAO,KAAK,gBACb;EACH,IAAI,cAAc;AAClB,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,SAAS,MAAM,KAAK,uBAAuB,KAC/C;IACE;IACA,cAAc;IACf,EACD,YAAY,SAAS,qBAAqB,CAC3C;GACD,MAAM,OAAO,OAAO,KAAK,OAAO;AAChC,OAAI,KAAK,WAAW,EAClB,eAAc,OAAO,KAAK;OAE1B,OAAM,IAAI,MACR,2EACD;;EAGL,MAAM,OAAO,MAAM,KAAK,UAAU,OAChC,aACA,YAAY,SAAS,YAAY,CAClC;EACD,MAAM,SAAS;GACb,UAAU;GACV,iBAAiB;GACjB,cAAc;GACf;EACD,IAAI,SAAS,MAAM,KAAK,sBAAsB,KAC5C,QACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,MAAI,KAAK,sBACP,UAAS;GACP,GAAG;GACH,iBAAiB;GAClB;AAEH,MAAI,KAAK,wBACP,UAAS;GACP,GAAG;GACH,mBAAmB;GACpB;AAEH,SAAO;;CAGT,aAAqB;AACnB,SAAO;;CAGT,aAAa,YACX,OACA,SACyC;AACzC,QAAM,IAAI,MAAM,mBAAmB;;CAGrC,YAA2C;AACzC,QAAM,IAAI,MAAM,mBAAmB;;;;;;;;;;;;CAarC,OAAO,QACL,KACA,WACA,UAeI,EAAE,EAC0B;EAChC,MAAM,EACJ,2BACA,YACA,iBAAiB;GACf,MAAM;GACN,QAAQ,aACJ,eAAe,aAAa,WAAW,GACvC;GACL,EACD,+BACA,SACA,GAAG,SACD;EAEJ,MAAM,UAAU,YAAY,KAAK,eAAe;EAOhD,MAAM,yBAAyB,IAAI,SAAS;GAC1C,QANmC,eAAe,aAClD,+BAA+B,YAC7B,6BACA,4BACH;GAGC,KAAK,+BAA+B,OAAO;GAC3C;GACD,CAAC;AAQF,SAPiB,IAAI,KAAK;GACxB;GACA,uBAAuB;GACvB;GACA;GACA,GAAG;GACJ,CAAC"}

package/dist/chains/graph_qa/cypher.cjs

@@ -1,12 +1,12 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
 const require_base = require('../base.cjs');
 const require_llm_chain = require('../llm_chain.cjs');
 const require_prompts = require('./prompts.cjs');
 require('../../util/entrypoint_deprecation.cjs');
 
 //#region src/chains/graph_qa/cypher.ts
-var cypher_exports = {};
-require_rolldown_runtime.__export(cypher_exports, {
+var cypher_exports = /* @__PURE__ */ require_runtime.__exportAll({
 	GraphCypherQAChain: () => GraphCypherQAChain,
 	INTERMEDIATE_STEPS_KEY: () => INTERMEDIATE_STEPS_KEY
 });
@@ -74,12 +74,11 @@ var GraphCypherQAChain = class GraphCypherQAChain extends require_base.BaseChain
 			llm: qaLLM || llm,
 			prompt: qaPrompt
 		});
-		const cypherGenerationChain = new require_llm_chain.LLMChain({
-			llm: cypherLLM || llm,
-			prompt: cypherPrompt
-		});
 		return new GraphCypherQAChain({
-			cypherGenerationChain,
+			cypherGenerationChain: new require_llm_chain.LLMChain({
+				llm: cypherLLM || llm,
+				prompt: cypherPrompt
+			}),
 			qaChain,
 			graph,
 			returnIntermediateSteps,
@@ -87,8 +86,7 @@ var GraphCypherQAChain = class GraphCypherQAChain extends require_base.BaseChain
 		});
 	}
 	extractCypher(text) {
-		const pattern = /```(.*?)```/s;
-		const matches = text.match(pattern);
+		const matches = text.match(/```(.*?)```/s);
 		return matches ? matches[1] : text;
 	}
 	async _call(values, runManager) {

package/dist/chains/graph_qa/cypher.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"cypher.cjs","names":["BaseChain","props: GraphCypherQAChainInput","props: FromLLMInput","CYPHER_QA_PROMPT","CYPHER_GENERATION_PROMPT","LLMChain","text: string","values: ChainValues","runManager?: CallbackManagerForChainRun","chainResult: ChainValues"],"sources":["../../../src/chains/graph_qa/cypher.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT } from \"./prompts.js\";\nimport { logVersion020MigrationWarning } from \"../../util/entrypoint_deprecation.js\";\n\n/* #__PURE__ */ logVersion020MigrationWarning({\n  oldEntrypointName: \"chains/graph_qa/cypher\",\n  newPackageName: \"@langchain/community\",\n});\n\nexport const INTERMEDIATE_STEPS_KEY = \"intermediateSteps\";\n\nexport interface GraphCypherQAChainInput extends ChainInputs {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  cypherGenerationChain: LLMChain;\n  qaChain: LLMChain;\n  inputKey?: string;\n  outputKey?: string;\n  topK?: number;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\nexport interface FromLLMInput {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  llm?: BaseLanguageModelInterface;\n  cypherLLM?: BaseLanguageModelInterface;\n  qaLLM?: BaseLanguageModelInterface;\n  qaPrompt?: BasePromptTemplate;\n  cypherPrompt?: BasePromptTemplate;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\n/**\n * Chain for question-answering against a graph by generating Cypher statements.\n *\n * @example\n * ```typescript\n * const chain = new GraphCypherQAChain({\n *   llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   graph: new Neo4jGraph(),\n * });\n * const res = await chain.invoke(\"Who played in Pulp Fiction?\");\n * ```\n *\n * @security\n * This chain will execute Cypher statements against the provided database.\n * Make sure that the database connection uses credentials\n * that are narrowly-scoped to only include necessary permissions.\n * Failure to do so may result in data corruption or loss, since the calling code\n * may attempt commands that would result in deletion, mutation of data\n * if appropriately prompted or reading sensitive data if such data is present in the database.\n * The best way to guard against such negative outcomes is to (as appropriate) limit the\n * permissions granted to the credentials used with this tool.\n *\n * See https://js.langchain.com/docs/security for more information.\n */\nexport class GraphCypherQAChain extends BaseChain {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  private graph: any;\n\n  private cypherGenerationChain: LLMChain;\n\n  private qaChain: LLMChain;\n\n  private inputKey = \"query\";\n\n  private outputKey = \"result\";\n\n  private topK = 10;\n\n  private returnDirect = false;\n\n  private returnIntermediateSteps = false;\n\n  constructor(props: GraphCypherQAChainInput) {\n    super(props);\n    const {\n      graph,\n      cypherGenerationChain,\n      qaChain,\n      inputKey,\n      outputKey,\n      topK,\n      returnIntermediateSteps,\n      returnDirect,\n    } = props;\n\n    this.graph = graph;\n    this.cypherGenerationChain = cypherGenerationChain;\n    this.qaChain = qaChain;\n\n    if (inputKey) {\n      this.inputKey = inputKey;\n    }\n    if (outputKey) {\n      this.outputKey = outputKey;\n    }\n    if (topK) {\n      this.topK = topK;\n    }\n    if (returnIntermediateSteps) {\n      this.returnIntermediateSteps = returnIntermediateSteps;\n    }\n    if (returnDirect) {\n      this.returnDirect = returnDirect;\n    }\n  }\n\n  _chainType() {\n    return \"graph_cypher_chain\" as const;\n  }\n\n  get inputKeys(): string[] {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  static fromLLM(props: FromLLMInput): GraphCypherQAChain {\n    const {\n      graph,\n      qaPrompt = CYPHER_QA_PROMPT,\n      cypherPrompt = CYPHER_GENERATION_PROMPT,\n      llm,\n      cypherLLM,\n      qaLLM,\n      returnIntermediateSteps = false,\n      returnDirect = false,\n    } = props;\n\n    if (!cypherLLM && !llm) {\n      throw new Error(\n        \"Either 'llm' or 'cypherLLM' parameters must be provided\"\n      );\n    }\n\n    if (!qaLLM && !llm) {\n      throw new Error(\"Either 'llm' or 'qaLLM' parameters must be provided\");\n    }\n\n    if (cypherLLM && qaLLM && llm) {\n      throw new Error(\n        \"You can specify up to two of 'cypherLLM', 'qaLLM', and 'llm', but not all three simultaneously.\"\n      );\n    }\n\n    const qaChain = new LLMChain({\n      llm: (qaLLM || llm) as BaseLanguageModelInterface,\n      prompt: qaPrompt,\n    });\n\n    const cypherGenerationChain = new LLMChain({\n      llm: (cypherLLM || llm) as BaseLanguageModelInterface,\n      prompt: cypherPrompt,\n    });\n\n    return new GraphCypherQAChain({\n      cypherGenerationChain,\n      qaChain,\n      graph,\n      returnIntermediateSteps,\n      returnDirect,\n    });\n  }\n\n  private extractCypher(text: string): string {\n    const pattern = /```(.*?)```/s;\n    const matches = text.match(pattern);\n    return matches ? matches[1] : text;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const callbacks = runManager?.getChild();\n    const question = values[this.inputKey];\n\n    const intermediateSteps = [];\n\n    const generatedCypher = await this.cypherGenerationChain.call(\n      { question, schema: this.graph.getSchema() },\n      callbacks\n    );\n\n    const extractedCypher = this.extractCypher(generatedCypher.text);\n\n    await runManager?.handleText(`Generated Cypher:\\n`);\n    await runManager?.handleText(`${extractedCypher} green\\n`);\n\n    intermediateSteps.push({ query: extractedCypher });\n\n    let chainResult: ChainValues;\n    const context = await this.graph.query(extractedCypher, {\n      topK: this.topK,\n    });\n\n    if (this.returnDirect) {\n      chainResult = { [this.outputKey]: context };\n    } else {\n      await runManager?.handleText(\"Full Context:\\n\");\n      await runManager?.handleText(`${context} green\\n`);\n\n      intermediateSteps.push({ context });\n\n      const result = await this.qaChain.call(\n        { question, context: JSON.stringify(context) },\n        callbacks\n      );\n\n      chainResult = {\n        [this.outputKey]: result[this.qaChain.outputKey],\n      };\n    }\n\n    if (this.returnIntermediateSteps) {\n      chainResult[INTERMEDIATE_STEPS_KEY] = intermediateSteps;\n    }\n\n    return chainResult;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAcA,MAAa,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;AAkDtC,IAAa,qBAAb,MAAa,2BAA2BA,uBAAU;CAEhD,AAAQ;CAER,AAAQ;CAER,AAAQ;CAER,AAAQ,WAAW;CAEnB,AAAQ,YAAY;CAEpB,AAAQ,OAAO;CAEf,AAAQ,eAAe;CAEvB,AAAQ,0BAA0B;CAElC,YAAYC,OAAgC;EAC1C,MAAM,MAAM;EACZ,MAAM,EACJ,OACA,uBACA,SACA,UACA,WACA,MACA,yBACA,cACD,GAAG;EAEJ,KAAK,QAAQ;EACb,KAAK,wBAAwB;EAC7B,KAAK,UAAU;AAEf,MAAI,UACF,KAAK,WAAW;AAElB,MAAI,WACF,KAAK,YAAY;AAEnB,MAAI,MACF,KAAK,OAAO;AAEd,MAAI,yBACF,KAAK,0BAA0B;AAEjC,MAAI,cACF,KAAK,eAAe;CAEvB;CAED,aAAa;AACX,SAAO;CACR;CAED,IAAI,YAAsB;AACxB,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,OAAO,QAAQC,OAAyC;EACtD,MAAM,EACJ,OACA,WAAWC,kCACX,eAAeC,0CACf,KACA,WACA,OACA,0BAA0B,OAC1B,eAAe,OAChB,GAAG;AAEJ,MAAI,CAAC,aAAa,CAAC,IACjB,OAAM,IAAI,MACR;AAIJ,MAAI,CAAC,SAAS,CAAC,IACb,OAAM,IAAI,MAAM;AAGlB,MAAI,aAAa,SAAS,IACxB,OAAM,IAAI,MACR;EAIJ,MAAM,UAAU,IAAIC,2BAAS;GAC3B,KAAM,SAAS;GACf,QAAQ;EACT;EAED,MAAM,wBAAwB,IAAIA,2BAAS;GACzC,KAAM,aAAa;GACnB,QAAQ;EACT;AAED,SAAO,IAAI,mBAAmB;GAC5B;GACA;GACA;GACA;GACA;EACD;CACF;CAED,AAAQ,cAAcC,MAAsB;EAC1C,MAAM,UAAU;EAChB,MAAM,UAAU,KAAK,MAAM,QAAQ;AACnC,SAAO,UAAU,QAAQ,KAAK;CAC/B;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,MAAM,YAAY,YAAY,UAAU;EACxC,MAAM,WAAW,OAAO,KAAK;EAE7B,MAAM,oBAAoB,CAAE;EAE5B,MAAM,kBAAkB,MAAM,KAAK,sBAAsB,KACvD;GAAE;GAAU,QAAQ,KAAK,MAAM,WAAW;EAAE,GAC5C,UACD;EAED,MAAM,kBAAkB,KAAK,cAAc,gBAAgB,KAAK;EAEhE,MAAM,YAAY,WAAW,CAAC,mBAAmB,CAAC,CAAC;EACnD,MAAM,YAAY,WAAW,GAAG,gBAAgB,QAAQ,CAAC,CAAC;EAE1D,kBAAkB,KAAK,EAAE,OAAO,gBAAiB,EAAC;EAElD,IAAIC;EACJ,MAAM,UAAU,MAAM,KAAK,MAAM,MAAM,iBAAiB,EACtD,MAAM,KAAK,KACZ,EAAC;AAEF,MAAI,KAAK,cACP,cAAc,GAAG,KAAK,YAAY,QAAS;OACtC;GACL,MAAM,YAAY,WAAW,kBAAkB;GAC/C,MAAM,YAAY,WAAW,GAAG,QAAQ,QAAQ,CAAC,CAAC;GAElD,kBAAkB,KAAK,EAAE,QAAS,EAAC;GAEnC,MAAM,SAAS,MAAM,KAAK,QAAQ,KAChC;IAAE;IAAU,SAAS,KAAK,UAAU,QAAQ;GAAE,GAC9C,UACD;GAED,cAAc,GACX,KAAK,YAAY,OAAO,KAAK,QAAQ,WACvC;EACF;AAED,MAAI,KAAK,yBACP,YAAY,0BAA0B;AAGxC,SAAO;CACR;AACF"}
+{"version":3,"file":"cypher.cjs","names":["BaseChain","CYPHER_QA_PROMPT","CYPHER_GENERATION_PROMPT","LLMChain"],"sources":["../../../src/chains/graph_qa/cypher.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT } from \"./prompts.js\";\nimport { logVersion020MigrationWarning } from \"../../util/entrypoint_deprecation.js\";\n\n/* #__PURE__ */ logVersion020MigrationWarning({\n  oldEntrypointName: \"chains/graph_qa/cypher\",\n  newPackageName: \"@langchain/community\",\n});\n\nexport const INTERMEDIATE_STEPS_KEY = \"intermediateSteps\";\n\nexport interface GraphCypherQAChainInput extends ChainInputs {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  cypherGenerationChain: LLMChain;\n  qaChain: LLMChain;\n  inputKey?: string;\n  outputKey?: string;\n  topK?: number;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\nexport interface FromLLMInput {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  llm?: BaseLanguageModelInterface;\n  cypherLLM?: BaseLanguageModelInterface;\n  qaLLM?: BaseLanguageModelInterface;\n  qaPrompt?: BasePromptTemplate;\n  cypherPrompt?: BasePromptTemplate;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\n/**\n * Chain for question-answering against a graph by generating Cypher statements.\n *\n * @example\n * ```typescript\n * const chain = new GraphCypherQAChain({\n *   llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   graph: new Neo4jGraph(),\n * });\n * const res = await chain.invoke(\"Who played in Pulp Fiction?\");\n * ```\n *\n * @security\n * This chain will execute Cypher statements against the provided database.\n * Make sure that the database connection uses credentials\n * that are narrowly-scoped to only include necessary permissions.\n * Failure to do so may result in data corruption or loss, since the calling code\n * may attempt commands that would result in deletion, mutation of data\n * if appropriately prompted or reading sensitive data if such data is present in the database.\n * The best way to guard against such negative outcomes is to (as appropriate) limit the\n * permissions granted to the credentials used with this tool.\n *\n * See https://js.langchain.com/docs/security for more information.\n */\nexport class GraphCypherQAChain extends BaseChain {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  private graph: any;\n\n  private cypherGenerationChain: LLMChain;\n\n  private qaChain: LLMChain;\n\n  private inputKey = \"query\";\n\n  private outputKey = \"result\";\n\n  private topK = 10;\n\n  private returnDirect = false;\n\n  private returnIntermediateSteps = false;\n\n  constructor(props: GraphCypherQAChainInput) {\n    super(props);\n    const {\n      graph,\n      cypherGenerationChain,\n      qaChain,\n      inputKey,\n      outputKey,\n      topK,\n      returnIntermediateSteps,\n      returnDirect,\n    } = props;\n\n    this.graph = graph;\n    this.cypherGenerationChain = cypherGenerationChain;\n    this.qaChain = qaChain;\n\n    if (inputKey) {\n      this.inputKey = inputKey;\n    }\n    if (outputKey) {\n      this.outputKey = outputKey;\n    }\n    if (topK) {\n      this.topK = topK;\n    }\n    if (returnIntermediateSteps) {\n      this.returnIntermediateSteps = returnIntermediateSteps;\n    }\n    if (returnDirect) {\n      this.returnDirect = returnDirect;\n    }\n  }\n\n  _chainType() {\n    return \"graph_cypher_chain\" as const;\n  }\n\n  get inputKeys(): string[] {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  static fromLLM(props: FromLLMInput): GraphCypherQAChain {\n    const {\n      graph,\n      qaPrompt = CYPHER_QA_PROMPT,\n      cypherPrompt = CYPHER_GENERATION_PROMPT,\n      llm,\n      cypherLLM,\n      qaLLM,\n      returnIntermediateSteps = false,\n      returnDirect = false,\n    } = props;\n\n    if (!cypherLLM && !llm) {\n      throw new Error(\n        \"Either 'llm' or 'cypherLLM' parameters must be provided\"\n      );\n    }\n\n    if (!qaLLM && !llm) {\n      throw new Error(\"Either 'llm' or 'qaLLM' parameters must be provided\");\n    }\n\n    if (cypherLLM && qaLLM && llm) {\n      throw new Error(\n        \"You can specify up to two of 'cypherLLM', 'qaLLM', and 'llm', but not all three simultaneously.\"\n      );\n    }\n\n    const qaChain = new LLMChain({\n      llm: (qaLLM || llm) as BaseLanguageModelInterface,\n      prompt: qaPrompt,\n    });\n\n    const cypherGenerationChain = new LLMChain({\n      llm: (cypherLLM || llm) as BaseLanguageModelInterface,\n      prompt: cypherPrompt,\n    });\n\n    return new GraphCypherQAChain({\n      cypherGenerationChain,\n      qaChain,\n      graph,\n      returnIntermediateSteps,\n      returnDirect,\n    });\n  }\n\n  private extractCypher(text: string): string {\n    const pattern = /```(.*?)```/s;\n    const matches = text.match(pattern);\n    return matches ? matches[1] : text;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const callbacks = runManager?.getChild();\n    const question = values[this.inputKey];\n\n    const intermediateSteps = [];\n\n    const generatedCypher = await this.cypherGenerationChain.call(\n      { question, schema: this.graph.getSchema() },\n      callbacks\n    );\n\n    const extractedCypher = this.extractCypher(generatedCypher.text);\n\n    await runManager?.handleText(`Generated Cypher:\\n`);\n    await runManager?.handleText(`${extractedCypher} green\\n`);\n\n    intermediateSteps.push({ query: extractedCypher });\n\n    let chainResult: ChainValues;\n    const context = await this.graph.query(extractedCypher, {\n      topK: this.topK,\n    });\n\n    if (this.returnDirect) {\n      chainResult = { [this.outputKey]: context };\n    } else {\n      await runManager?.handleText(\"Full Context:\\n\");\n      await runManager?.handleText(`${context} green\\n`);\n\n      intermediateSteps.push({ context });\n\n      const result = await this.qaChain.call(\n        { question, context: JSON.stringify(context) },\n        callbacks\n      );\n\n      chainResult = {\n        [this.outputKey]: result[this.qaChain.outputKey],\n      };\n    }\n\n    if (this.returnIntermediateSteps) {\n      chainResult[INTERMEDIATE_STEPS_KEY] = intermediateSteps;\n    }\n\n    return chainResult;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAcA,MAAa,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;AAkDtC,IAAa,qBAAb,MAAa,2BAA2BA,uBAAU;CAEhD,AAAQ;CAER,AAAQ;CAER,AAAQ;CAER,AAAQ,WAAW;CAEnB,AAAQ,YAAY;CAEpB,AAAQ,OAAO;CAEf,AAAQ,eAAe;CAEvB,AAAQ,0BAA0B;CAElC,YAAY,OAAgC;AAC1C,QAAM,MAAM;EACZ,MAAM,EACJ,OACA,uBACA,SACA,UACA,WACA,MACA,yBACA,iBACE;AAEJ,OAAK,QAAQ;AACb,OAAK,wBAAwB;AAC7B,OAAK,UAAU;AAEf,MAAI,SACF,MAAK,WAAW;AAElB,MAAI,UACF,MAAK,YAAY;AAEnB,MAAI,KACF,MAAK,OAAO;AAEd,MAAI,wBACF,MAAK,0BAA0B;AAEjC,MAAI,aACF,MAAK,eAAe;;CAIxB,aAAa;AACX,SAAO;;CAGT,IAAI,YAAsB;AACxB,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,UAAU;;CAGzB,OAAO,QAAQ,OAAyC;EACtD,MAAM,EACJ,OACA,WAAWC,kCACX,eAAeC,0CACf,KACA,WACA,OACA,0BAA0B,OAC1B,eAAe,UACb;AAEJ,MAAI,CAAC,aAAa,CAAC,IACjB,OAAM,IAAI,MACR,0DACD;AAGH,MAAI,CAAC,SAAS,CAAC,IACb,OAAM,IAAI,MAAM,sDAAsD;AAGxE,MAAI,aAAa,SAAS,IACxB,OAAM,IAAI,MACR,kGACD;EAGH,MAAM,UAAU,IAAIC,2BAAS;GAC3B,KAAM,SAAS;GACf,QAAQ;GACT,CAAC;AAOF,SAAO,IAAI,mBAAmB;GAC5B,uBAN4B,IAAIA,2BAAS;IACzC,KAAM,aAAa;IACnB,QAAQ;IACT,CAAC;GAIA;GACA;GACA;GACA;GACD,CAAC;;CAGJ,AAAQ,cAAc,MAAsB;EAE1C,MAAM,UAAU,KAAK,MADL,eACmB;AACnC,SAAO,UAAU,QAAQ,KAAK;;CAGhC,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,YAAY,YAAY,UAAU;EACxC,MAAM,WAAW,OAAO,KAAK;EAE7B,MAAM,oBAAoB,EAAE;EAE5B,MAAM,kBAAkB,MAAM,KAAK,sBAAsB,KACvD;GAAE;GAAU,QAAQ,KAAK,MAAM,WAAW;GAAE,EAC5C,UACD;EAED,MAAM,kBAAkB,KAAK,cAAc,gBAAgB,KAAK;AAEhE,QAAM,YAAY,WAAW,sBAAsB;AACnD,QAAM,YAAY,WAAW,GAAG,gBAAgB,UAAU;AAE1D,oBAAkB,KAAK,EAAE,OAAO,iBAAiB,CAAC;EAElD,IAAI;EACJ,MAAM,UAAU,MAAM,KAAK,MAAM,MAAM,iBAAiB,EACtD,MAAM,KAAK,MACZ,CAAC;AAEF,MAAI,KAAK,aACP,eAAc,GAAG,KAAK,YAAY,SAAS;OACtC;AACL,SAAM,YAAY,WAAW,kBAAkB;AAC/C,SAAM,YAAY,WAAW,GAAG,QAAQ,UAAU;AAElD,qBAAkB,KAAK,EAAE,SAAS,CAAC;GAEnC,MAAM,SAAS,MAAM,KAAK,QAAQ,KAChC;IAAE;IAAU,SAAS,KAAK,UAAU,QAAQ;IAAE,EAC9C,UACD;AAED,iBAAc,GACX,KAAK,YAAY,OAAO,KAAK,QAAQ,YACvC;;AAGH,MAAI,KAAK,wBACP,aAAY,0BAA0B;AAGxC,SAAO"}

package/dist/chains/graph_qa/cypher.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"cypher.d.cts","names":["BaseLanguageModelInterface","ChainValues","BasePromptTemplate","CallbackManagerForChainRun","LLMChain","BaseChain","ChainInputs","INTERMEDIATE_STEPS_KEY","GraphCypherQAChainInput","FromLLMInput","GraphCypherQAChain","Promise"],"sources":["../../../src/chains/graph_qa/cypher.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nexport declare const INTERMEDIATE_STEPS_KEY = \"intermediateSteps\";\nexport interface GraphCypherQAChainInput extends ChainInputs {\n    graph: any;\n    cypherGenerationChain: LLMChain;\n    qaChain: LLMChain;\n    inputKey?: string;\n    outputKey?: string;\n    topK?: number;\n    returnIntermediateSteps?: boolean;\n    returnDirect?: boolean;\n}\nexport interface FromLLMInput {\n    graph: any;\n    llm?: BaseLanguageModelInterface;\n    cypherLLM?: BaseLanguageModelInterface;\n    qaLLM?: BaseLanguageModelInterface;\n    qaPrompt?: BasePromptTemplate;\n    cypherPrompt?: BasePromptTemplate;\n    returnIntermediateSteps?: boolean;\n    returnDirect?: boolean;\n}\n/**\n * Chain for question-answering against a graph by generating Cypher statements.\n *\n * @example\n * ```typescript\n * const chain = new GraphCypherQAChain({\n *   llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   graph: new Neo4jGraph(),\n * });\n * const res = await chain.invoke(\"Who played in Pulp Fiction?\");\n * ```\n *\n * @security\n * This chain will execute Cypher statements against the provided database.\n * Make sure that the database connection uses credentials\n * that are narrowly-scoped to only include necessary permissions.\n * Failure to do so may result in data corruption or loss, since the calling code\n * may attempt commands that would result in deletion, mutation of data\n * if appropriately prompted or reading sensitive data if such data is present in the database.\n * The best way to guard against such negative outcomes is to (as appropriate) limit the\n * permissions granted to the credentials used with this tool.\n *\n * See https://js.langchain.com/docs/security for more information.\n */\nexport declare class GraphCypherQAChain extends BaseChain {\n    private graph;\n    private cypherGenerationChain;\n    private qaChain;\n    private inputKey;\n    private outputKey;\n    private topK;\n    private returnDirect;\n    private returnIntermediateSteps;\n    constructor(props: GraphCypherQAChainInput);\n    _chainType(): \"graph_cypher_chain\";\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    static fromLLM(props: FromLLMInput): GraphCypherQAChain;\n    private extractCypher;\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n}\n//# sourceMappingURL=cypher.d.ts.map"],"mappings":";;;;;;;;cAMqBO,sBAAAA;UACJC,uBAAAA,SAAgCF;EAD5BC,KAAAA,EAAAA,GAAAA;EACJC,qBAAAA,EAEUJ,QAFa;EAEbA,OAAAA,EACdA,QADcA;EACdA,QAAAA,CAAAA,EAAAA,MAAAA;EAHoCE,SAAAA,CAAAA,EAAAA,MAAAA;EAAW,IAAA,CAAA,EAAA,MAAA;EAU3CG,uBAAY,CAAA,EAAA,OAAA;EAEnBT,YAAAA,CAAAA,EAAAA,OAAAA;;AAEEA,UAJKS,YAAAA,CAILT;EACGE,KAAAA,EAAAA,GAAAA;EACIA,GAAAA,CAAAA,EAJTF,0BAISE;EAAkB,SAAA,CAAA,EAHrBF,0BAGqB;EA4BhBU,KAAAA,CAAAA,EA9BTV,0BA8B2B;EAShBQ,QAAAA,CAAAA,EAtCRN,kBAsCQM;EAIGC,YAAAA,CAAAA,EAzCPP,kBAyCOO;EAAeC,uBAAAA,CAAAA,EAAAA,OAAAA;EAEvBT,YAAAA,CAAAA,EAAAA,OAAAA;;;;;AAfuC;;;;;;;;;;;;;;;;;;;;;cAApCS,kBAAAA,SAA2BL,SAAAA;;;;;;;;;qBASzBG;;;;wBAIGC,eAAeC;;gBAEvBT,0BAA0BE,6BAA6BQ,QAAQV"}
+{"version":3,"file":"cypher.d.cts","names":[],"sources":["../../../src/chains/graph_qa/cypher.ts"],"mappings":";;;;;;;;cAca,sBAAA;AAAA,UAEI,uBAAA,SAAgC,WAAA;EAE/C,KAAA;EACA,qBAAA,EAAuB,QAAA;EACvB,OAAA,EAAS,QAAA;EACT,QAAA;EACA,SAAA;EACA,IAAA;EACA,uBAAA;EACA,YAAA;AAAA;AAAA,UAGe,YAAA;EAEf,KAAA;EACA,GAAA,GAAM,0BAAA;EACN,SAAA,GAAY,0BAAA;EACZ,KAAA,GAAQ,0BAAA;EACR,QAAA,GAAW,kBAAA;EACX,YAAA,GAAe,kBAAA;EACf,uBAAA;EACA,YAAA;AAAA;;;;;;;;AATF;;;;;;;;;;;;;;;;;cAoCa,kBAAA,SAA2B,SAAA;EAAA,QAE9B,KAAA;EAAA,QAEA,qBAAA;EAAA,QAEA,OAAA;EAAA,QAEA,QAAA;EAAA,QAEA,SAAA;EAAA,QAEA,IAAA;EAAA,QAEA,YAAA;EAAA,QAEA,uBAAA;EAER,WAAA,CAAY,KAAA,EAAO,uBAAA;EAkCnB,UAAA,CAAA;EAAA,IAII,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAAA,OAIG,OAAA,CAAQ,KAAA,EAAO,YAAA,GAAe,kBAAA;EAAA,QA+C7B,aAAA;EAMF,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;AAAA"}

package/dist/chains/graph_qa/cypher.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"cypher.d.ts","names":["BaseLanguageModelInterface","ChainValues","BasePromptTemplate","CallbackManagerForChainRun","LLMChain","BaseChain","ChainInputs","INTERMEDIATE_STEPS_KEY","GraphCypherQAChainInput","FromLLMInput","GraphCypherQAChain","Promise"],"sources":["../../../src/chains/graph_qa/cypher.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nexport declare const INTERMEDIATE_STEPS_KEY = \"intermediateSteps\";\nexport interface GraphCypherQAChainInput extends ChainInputs {\n    graph: any;\n    cypherGenerationChain: LLMChain;\n    qaChain: LLMChain;\n    inputKey?: string;\n    outputKey?: string;\n    topK?: number;\n    returnIntermediateSteps?: boolean;\n    returnDirect?: boolean;\n}\nexport interface FromLLMInput {\n    graph: any;\n    llm?: BaseLanguageModelInterface;\n    cypherLLM?: BaseLanguageModelInterface;\n    qaLLM?: BaseLanguageModelInterface;\n    qaPrompt?: BasePromptTemplate;\n    cypherPrompt?: BasePromptTemplate;\n    returnIntermediateSteps?: boolean;\n    returnDirect?: boolean;\n}\n/**\n * Chain for question-answering against a graph by generating Cypher statements.\n *\n * @example\n * ```typescript\n * const chain = new GraphCypherQAChain({\n *   llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   graph: new Neo4jGraph(),\n * });\n * const res = await chain.invoke(\"Who played in Pulp Fiction?\");\n * ```\n *\n * @security\n * This chain will execute Cypher statements against the provided database.\n * Make sure that the database connection uses credentials\n * that are narrowly-scoped to only include necessary permissions.\n * Failure to do so may result in data corruption or loss, since the calling code\n * may attempt commands that would result in deletion, mutation of data\n * if appropriately prompted or reading sensitive data if such data is present in the database.\n * The best way to guard against such negative outcomes is to (as appropriate) limit the\n * permissions granted to the credentials used with this tool.\n *\n * See https://js.langchain.com/docs/security for more information.\n */\nexport declare class GraphCypherQAChain extends BaseChain {\n    private graph;\n    private cypherGenerationChain;\n    private qaChain;\n    private inputKey;\n    private outputKey;\n    private topK;\n    private returnDirect;\n    private returnIntermediateSteps;\n    constructor(props: GraphCypherQAChainInput);\n    _chainType(): \"graph_cypher_chain\";\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    static fromLLM(props: FromLLMInput): GraphCypherQAChain;\n    private extractCypher;\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n}\n//# sourceMappingURL=cypher.d.ts.map"],"mappings":";;;;;;;;cAMqBO,sBAAAA;UACJC,uBAAAA,SAAgCF;EAD5BC,KAAAA,EAAAA,GAAAA;EACJC,qBAAAA,EAEUJ,QAFa;EAEbA,OAAAA,EACdA,QADcA;EACdA,QAAAA,CAAAA,EAAAA,MAAAA;EAHoCE,SAAAA,CAAAA,EAAAA,MAAAA;EAAW,IAAA,CAAA,EAAA,MAAA;EAU3CG,uBAAY,CAAA,EAAA,OAAA;EAEnBT,YAAAA,CAAAA,EAAAA,OAAAA;;AAEEA,UAJKS,YAAAA,CAILT;EACGE,KAAAA,EAAAA,GAAAA;EACIA,GAAAA,CAAAA,EAJTF,0BAISE;EAAkB,SAAA,CAAA,EAHrBF,0BAGqB;EA4BhBU,KAAAA,CAAAA,EA9BTV,0BA8B2B;EAShBQ,QAAAA,CAAAA,EAtCRN,kBAsCQM;EAIGC,YAAAA,CAAAA,EAzCPP,kBAyCOO;EAAeC,uBAAAA,CAAAA,EAAAA,OAAAA;EAEvBT,YAAAA,CAAAA,EAAAA,OAAAA;;;;;AAfuC;;;;;;;;;;;;;;;;;;;;;cAApCS,kBAAAA,SAA2BL,SAAAA;;;;;;;;;qBASzBG;;;;wBAIGC,eAAeC;;gBAEvBT,0BAA0BE,6BAA6BQ,QAAQV"}
+{"version":3,"file":"cypher.d.ts","names":[],"sources":["../../../src/chains/graph_qa/cypher.ts"],"mappings":";;;;;;;;cAca,sBAAA;AAAA,UAEI,uBAAA,SAAgC,WAAA;EAE/C,KAAA;EACA,qBAAA,EAAuB,QAAA;EACvB,OAAA,EAAS,QAAA;EACT,QAAA;EACA,SAAA;EACA,IAAA;EACA,uBAAA;EACA,YAAA;AAAA;AAAA,UAGe,YAAA;EAEf,KAAA;EACA,GAAA,GAAM,0BAAA;EACN,SAAA,GAAY,0BAAA;EACZ,KAAA,GAAQ,0BAAA;EACR,QAAA,GAAW,kBAAA;EACX,YAAA,GAAe,kBAAA;EACf,uBAAA;EACA,YAAA;AAAA;;;;;;;;AATF;;;;;;;;;;;;;;;;;cAoCa,kBAAA,SAA2B,SAAA;EAAA,QAE9B,KAAA;EAAA,QAEA,qBAAA;EAAA,QAEA,OAAA;EAAA,QAEA,QAAA;EAAA,QAEA,SAAA;EAAA,QAEA,IAAA;EAAA,QAEA,YAAA;EAAA,QAEA,uBAAA;EAER,WAAA,CAAY,KAAA,EAAO,uBAAA;EAkCnB,UAAA,CAAA;EAAA,IAII,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAAA,OAIG,OAAA,CAAQ,KAAA,EAAO,YAAA,GAAe,kBAAA;EAAA,QA+C7B,aAAA;EAMF,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;AAAA"}

package/dist/chains/graph_qa/cypher.js

@@ -1,12 +1,11 @@
-import { __export } from "../../_virtual/rolldown_runtime.js";
+import { __exportAll } from "../../_virtual/_rolldown/runtime.js";
 import { BaseChain } from "../base.js";
 import { LLMChain } from "../llm_chain.js";
 import { CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT } from "./prompts.js";
 import "../../util/entrypoint_deprecation.js";
 
 //#region src/chains/graph_qa/cypher.ts
-var cypher_exports = {};
-__export(cypher_exports, {
+var cypher_exports = /* @__PURE__ */ __exportAll({
 	GraphCypherQAChain: () => GraphCypherQAChain,
 	INTERMEDIATE_STEPS_KEY: () => INTERMEDIATE_STEPS_KEY
 });
@@ -74,12 +73,11 @@ var GraphCypherQAChain = class GraphCypherQAChain extends BaseChain {
 			llm: qaLLM || llm,
 			prompt: qaPrompt
 		});
-		const cypherGenerationChain = new LLMChain({
-			llm: cypherLLM || llm,
-			prompt: cypherPrompt
-		});
 		return new GraphCypherQAChain({
-			cypherGenerationChain,
+			cypherGenerationChain: new LLMChain({
+				llm: cypherLLM || llm,
+				prompt: cypherPrompt
+			}),
 			qaChain,
 			graph,
 			returnIntermediateSteps,
@@ -87,8 +85,7 @@ var GraphCypherQAChain = class GraphCypherQAChain extends BaseChain {
 		});
 	}
 	extractCypher(text) {
-		const pattern = /```(.*?)```/s;
-		const matches = text.match(pattern);
+		const matches = text.match(/```(.*?)```/s);
 		return matches ? matches[1] : text;
 	}
 	async _call(values, runManager) {

package/dist/chains/graph_qa/cypher.js.map

@@ -1 +1 @@
-{"version":3,"file":"cypher.js","names":["props: GraphCypherQAChainInput","props: FromLLMInput","text: string","values: ChainValues","runManager?: CallbackManagerForChainRun","chainResult: ChainValues"],"sources":["../../../src/chains/graph_qa/cypher.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT } from \"./prompts.js\";\nimport { logVersion020MigrationWarning } from \"../../util/entrypoint_deprecation.js\";\n\n/* #__PURE__ */ logVersion020MigrationWarning({\n  oldEntrypointName: \"chains/graph_qa/cypher\",\n  newPackageName: \"@langchain/community\",\n});\n\nexport const INTERMEDIATE_STEPS_KEY = \"intermediateSteps\";\n\nexport interface GraphCypherQAChainInput extends ChainInputs {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  cypherGenerationChain: LLMChain;\n  qaChain: LLMChain;\n  inputKey?: string;\n  outputKey?: string;\n  topK?: number;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\nexport interface FromLLMInput {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  llm?: BaseLanguageModelInterface;\n  cypherLLM?: BaseLanguageModelInterface;\n  qaLLM?: BaseLanguageModelInterface;\n  qaPrompt?: BasePromptTemplate;\n  cypherPrompt?: BasePromptTemplate;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\n/**\n * Chain for question-answering against a graph by generating Cypher statements.\n *\n * @example\n * ```typescript\n * const chain = new GraphCypherQAChain({\n *   llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   graph: new Neo4jGraph(),\n * });\n * const res = await chain.invoke(\"Who played in Pulp Fiction?\");\n * ```\n *\n * @security\n * This chain will execute Cypher statements against the provided database.\n * Make sure that the database connection uses credentials\n * that are narrowly-scoped to only include necessary permissions.\n * Failure to do so may result in data corruption or loss, since the calling code\n * may attempt commands that would result in deletion, mutation of data\n * if appropriately prompted or reading sensitive data if such data is present in the database.\n * The best way to guard against such negative outcomes is to (as appropriate) limit the\n * permissions granted to the credentials used with this tool.\n *\n * See https://js.langchain.com/docs/security for more information.\n */\nexport class GraphCypherQAChain extends BaseChain {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  private graph: any;\n\n  private cypherGenerationChain: LLMChain;\n\n  private qaChain: LLMChain;\n\n  private inputKey = \"query\";\n\n  private outputKey = \"result\";\n\n  private topK = 10;\n\n  private returnDirect = false;\n\n  private returnIntermediateSteps = false;\n\n  constructor(props: GraphCypherQAChainInput) {\n    super(props);\n    const {\n      graph,\n      cypherGenerationChain,\n      qaChain,\n      inputKey,\n      outputKey,\n      topK,\n      returnIntermediateSteps,\n      returnDirect,\n    } = props;\n\n    this.graph = graph;\n    this.cypherGenerationChain = cypherGenerationChain;\n    this.qaChain = qaChain;\n\n    if (inputKey) {\n      this.inputKey = inputKey;\n    }\n    if (outputKey) {\n      this.outputKey = outputKey;\n    }\n    if (topK) {\n      this.topK = topK;\n    }\n    if (returnIntermediateSteps) {\n      this.returnIntermediateSteps = returnIntermediateSteps;\n    }\n    if (returnDirect) {\n      this.returnDirect = returnDirect;\n    }\n  }\n\n  _chainType() {\n    return \"graph_cypher_chain\" as const;\n  }\n\n  get inputKeys(): string[] {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  static fromLLM(props: FromLLMInput): GraphCypherQAChain {\n    const {\n      graph,\n      qaPrompt = CYPHER_QA_PROMPT,\n      cypherPrompt = CYPHER_GENERATION_PROMPT,\n      llm,\n      cypherLLM,\n      qaLLM,\n      returnIntermediateSteps = false,\n      returnDirect = false,\n    } = props;\n\n    if (!cypherLLM && !llm) {\n      throw new Error(\n        \"Either 'llm' or 'cypherLLM' parameters must be provided\"\n      );\n    }\n\n    if (!qaLLM && !llm) {\n      throw new Error(\"Either 'llm' or 'qaLLM' parameters must be provided\");\n    }\n\n    if (cypherLLM && qaLLM && llm) {\n      throw new Error(\n        \"You can specify up to two of 'cypherLLM', 'qaLLM', and 'llm', but not all three simultaneously.\"\n      );\n    }\n\n    const qaChain = new LLMChain({\n      llm: (qaLLM || llm) as BaseLanguageModelInterface,\n      prompt: qaPrompt,\n    });\n\n    const cypherGenerationChain = new LLMChain({\n      llm: (cypherLLM || llm) as BaseLanguageModelInterface,\n      prompt: cypherPrompt,\n    });\n\n    return new GraphCypherQAChain({\n      cypherGenerationChain,\n      qaChain,\n      graph,\n      returnIntermediateSteps,\n      returnDirect,\n    });\n  }\n\n  private extractCypher(text: string): string {\n    const pattern = /```(.*?)```/s;\n    const matches = text.match(pattern);\n    return matches ? matches[1] : text;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const callbacks = runManager?.getChild();\n    const question = values[this.inputKey];\n\n    const intermediateSteps = [];\n\n    const generatedCypher = await this.cypherGenerationChain.call(\n      { question, schema: this.graph.getSchema() },\n      callbacks\n    );\n\n    const extractedCypher = this.extractCypher(generatedCypher.text);\n\n    await runManager?.handleText(`Generated Cypher:\\n`);\n    await runManager?.handleText(`${extractedCypher} green\\n`);\n\n    intermediateSteps.push({ query: extractedCypher });\n\n    let chainResult: ChainValues;\n    const context = await this.graph.query(extractedCypher, {\n      topK: this.topK,\n    });\n\n    if (this.returnDirect) {\n      chainResult = { [this.outputKey]: context };\n    } else {\n      await runManager?.handleText(\"Full Context:\\n\");\n      await runManager?.handleText(`${context} green\\n`);\n\n      intermediateSteps.push({ context });\n\n      const result = await this.qaChain.call(\n        { question, context: JSON.stringify(context) },\n        callbacks\n      );\n\n      chainResult = {\n        [this.outputKey]: result[this.qaChain.outputKey],\n      };\n    }\n\n    if (this.returnIntermediateSteps) {\n      chainResult[INTERMEDIATE_STEPS_KEY] = intermediateSteps;\n    }\n\n    return chainResult;\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAcA,MAAa,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;AAkDtC,IAAa,qBAAb,MAAa,2BAA2B,UAAU;CAEhD,AAAQ;CAER,AAAQ;CAER,AAAQ;CAER,AAAQ,WAAW;CAEnB,AAAQ,YAAY;CAEpB,AAAQ,OAAO;CAEf,AAAQ,eAAe;CAEvB,AAAQ,0BAA0B;CAElC,YAAYA,OAAgC;EAC1C,MAAM,MAAM;EACZ,MAAM,EACJ,OACA,uBACA,SACA,UACA,WACA,MACA,yBACA,cACD,GAAG;EAEJ,KAAK,QAAQ;EACb,KAAK,wBAAwB;EAC7B,KAAK,UAAU;AAEf,MAAI,UACF,KAAK,WAAW;AAElB,MAAI,WACF,KAAK,YAAY;AAEnB,MAAI,MACF,KAAK,OAAO;AAEd,MAAI,yBACF,KAAK,0BAA0B;AAEjC,MAAI,cACF,KAAK,eAAe;CAEvB;CAED,aAAa;AACX,SAAO;CACR;CAED,IAAI,YAAsB;AACxB,SAAO,CAAC,KAAK,QAAS;CACvB;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,SAAU;CACxB;CAED,OAAO,QAAQC,OAAyC;EACtD,MAAM,EACJ,OACA,WAAW,kBACX,eAAe,0BACf,KACA,WACA,OACA,0BAA0B,OAC1B,eAAe,OAChB,GAAG;AAEJ,MAAI,CAAC,aAAa,CAAC,IACjB,OAAM,IAAI,MACR;AAIJ,MAAI,CAAC,SAAS,CAAC,IACb,OAAM,IAAI,MAAM;AAGlB,MAAI,aAAa,SAAS,IACxB,OAAM,IAAI,MACR;EAIJ,MAAM,UAAU,IAAI,SAAS;GAC3B,KAAM,SAAS;GACf,QAAQ;EACT;EAED,MAAM,wBAAwB,IAAI,SAAS;GACzC,KAAM,aAAa;GACnB,QAAQ;EACT;AAED,SAAO,IAAI,mBAAmB;GAC5B;GACA;GACA;GACA;GACA;EACD;CACF;CAED,AAAQ,cAAcC,MAAsB;EAC1C,MAAM,UAAU;EAChB,MAAM,UAAU,KAAK,MAAM,QAAQ;AACnC,SAAO,UAAU,QAAQ,KAAK;CAC/B;CAED,MAAM,MACJC,QACAC,YACsB;EACtB,MAAM,YAAY,YAAY,UAAU;EACxC,MAAM,WAAW,OAAO,KAAK;EAE7B,MAAM,oBAAoB,CAAE;EAE5B,MAAM,kBAAkB,MAAM,KAAK,sBAAsB,KACvD;GAAE;GAAU,QAAQ,KAAK,MAAM,WAAW;EAAE,GAC5C,UACD;EAED,MAAM,kBAAkB,KAAK,cAAc,gBAAgB,KAAK;EAEhE,MAAM,YAAY,WAAW,CAAC,mBAAmB,CAAC,CAAC;EACnD,MAAM,YAAY,WAAW,GAAG,gBAAgB,QAAQ,CAAC,CAAC;EAE1D,kBAAkB,KAAK,EAAE,OAAO,gBAAiB,EAAC;EAElD,IAAIC;EACJ,MAAM,UAAU,MAAM,KAAK,MAAM,MAAM,iBAAiB,EACtD,MAAM,KAAK,KACZ,EAAC;AAEF,MAAI,KAAK,cACP,cAAc,GAAG,KAAK,YAAY,QAAS;OACtC;GACL,MAAM,YAAY,WAAW,kBAAkB;GAC/C,MAAM,YAAY,WAAW,GAAG,QAAQ,QAAQ,CAAC,CAAC;GAElD,kBAAkB,KAAK,EAAE,QAAS,EAAC;GAEnC,MAAM,SAAS,MAAM,KAAK,QAAQ,KAChC;IAAE;IAAU,SAAS,KAAK,UAAU,QAAQ;GAAE,GAC9C,UACD;GAED,cAAc,GACX,KAAK,YAAY,OAAO,KAAK,QAAQ,WACvC;EACF;AAED,MAAI,KAAK,yBACP,YAAY,0BAA0B;AAGxC,SAAO;CACR;AACF"}
+{"version":3,"file":"cypher.js","names":[],"sources":["../../../src/chains/graph_qa/cypher.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../llm_chain.js\";\nimport { BaseChain, ChainInputs } from \"../base.js\";\nimport { CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT } from \"./prompts.js\";\nimport { logVersion020MigrationWarning } from \"../../util/entrypoint_deprecation.js\";\n\n/* #__PURE__ */ logVersion020MigrationWarning({\n  oldEntrypointName: \"chains/graph_qa/cypher\",\n  newPackageName: \"@langchain/community\",\n});\n\nexport const INTERMEDIATE_STEPS_KEY = \"intermediateSteps\";\n\nexport interface GraphCypherQAChainInput extends ChainInputs {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  cypherGenerationChain: LLMChain;\n  qaChain: LLMChain;\n  inputKey?: string;\n  outputKey?: string;\n  topK?: number;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\nexport interface FromLLMInput {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  graph: any;\n  llm?: BaseLanguageModelInterface;\n  cypherLLM?: BaseLanguageModelInterface;\n  qaLLM?: BaseLanguageModelInterface;\n  qaPrompt?: BasePromptTemplate;\n  cypherPrompt?: BasePromptTemplate;\n  returnIntermediateSteps?: boolean;\n  returnDirect?: boolean;\n}\n\n/**\n * Chain for question-answering against a graph by generating Cypher statements.\n *\n * @example\n * ```typescript\n * const chain = new GraphCypherQAChain({\n *   llm: new ChatOpenAI({ model: \"gpt-4o-mini\", temperature: 0 }),\n *   graph: new Neo4jGraph(),\n * });\n * const res = await chain.invoke(\"Who played in Pulp Fiction?\");\n * ```\n *\n * @security\n * This chain will execute Cypher statements against the provided database.\n * Make sure that the database connection uses credentials\n * that are narrowly-scoped to only include necessary permissions.\n * Failure to do so may result in data corruption or loss, since the calling code\n * may attempt commands that would result in deletion, mutation of data\n * if appropriately prompted or reading sensitive data if such data is present in the database.\n * The best way to guard against such negative outcomes is to (as appropriate) limit the\n * permissions granted to the credentials used with this tool.\n *\n * See https://js.langchain.com/docs/security for more information.\n */\nexport class GraphCypherQAChain extends BaseChain {\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  private graph: any;\n\n  private cypherGenerationChain: LLMChain;\n\n  private qaChain: LLMChain;\n\n  private inputKey = \"query\";\n\n  private outputKey = \"result\";\n\n  private topK = 10;\n\n  private returnDirect = false;\n\n  private returnIntermediateSteps = false;\n\n  constructor(props: GraphCypherQAChainInput) {\n    super(props);\n    const {\n      graph,\n      cypherGenerationChain,\n      qaChain,\n      inputKey,\n      outputKey,\n      topK,\n      returnIntermediateSteps,\n      returnDirect,\n    } = props;\n\n    this.graph = graph;\n    this.cypherGenerationChain = cypherGenerationChain;\n    this.qaChain = qaChain;\n\n    if (inputKey) {\n      this.inputKey = inputKey;\n    }\n    if (outputKey) {\n      this.outputKey = outputKey;\n    }\n    if (topK) {\n      this.topK = topK;\n    }\n    if (returnIntermediateSteps) {\n      this.returnIntermediateSteps = returnIntermediateSteps;\n    }\n    if (returnDirect) {\n      this.returnDirect = returnDirect;\n    }\n  }\n\n  _chainType() {\n    return \"graph_cypher_chain\" as const;\n  }\n\n  get inputKeys(): string[] {\n    return [this.inputKey];\n  }\n\n  get outputKeys(): string[] {\n    return [this.outputKey];\n  }\n\n  static fromLLM(props: FromLLMInput): GraphCypherQAChain {\n    const {\n      graph,\n      qaPrompt = CYPHER_QA_PROMPT,\n      cypherPrompt = CYPHER_GENERATION_PROMPT,\n      llm,\n      cypherLLM,\n      qaLLM,\n      returnIntermediateSteps = false,\n      returnDirect = false,\n    } = props;\n\n    if (!cypherLLM && !llm) {\n      throw new Error(\n        \"Either 'llm' or 'cypherLLM' parameters must be provided\"\n      );\n    }\n\n    if (!qaLLM && !llm) {\n      throw new Error(\"Either 'llm' or 'qaLLM' parameters must be provided\");\n    }\n\n    if (cypherLLM && qaLLM && llm) {\n      throw new Error(\n        \"You can specify up to two of 'cypherLLM', 'qaLLM', and 'llm', but not all three simultaneously.\"\n      );\n    }\n\n    const qaChain = new LLMChain({\n      llm: (qaLLM || llm) as BaseLanguageModelInterface,\n      prompt: qaPrompt,\n    });\n\n    const cypherGenerationChain = new LLMChain({\n      llm: (cypherLLM || llm) as BaseLanguageModelInterface,\n      prompt: cypherPrompt,\n    });\n\n    return new GraphCypherQAChain({\n      cypherGenerationChain,\n      qaChain,\n      graph,\n      returnIntermediateSteps,\n      returnDirect,\n    });\n  }\n\n  private extractCypher(text: string): string {\n    const pattern = /```(.*?)```/s;\n    const matches = text.match(pattern);\n    return matches ? matches[1] : text;\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const callbacks = runManager?.getChild();\n    const question = values[this.inputKey];\n\n    const intermediateSteps = [];\n\n    const generatedCypher = await this.cypherGenerationChain.call(\n      { question, schema: this.graph.getSchema() },\n      callbacks\n    );\n\n    const extractedCypher = this.extractCypher(generatedCypher.text);\n\n    await runManager?.handleText(`Generated Cypher:\\n`);\n    await runManager?.handleText(`${extractedCypher} green\\n`);\n\n    intermediateSteps.push({ query: extractedCypher });\n\n    let chainResult: ChainValues;\n    const context = await this.graph.query(extractedCypher, {\n      topK: this.topK,\n    });\n\n    if (this.returnDirect) {\n      chainResult = { [this.outputKey]: context };\n    } else {\n      await runManager?.handleText(\"Full Context:\\n\");\n      await runManager?.handleText(`${context} green\\n`);\n\n      intermediateSteps.push({ context });\n\n      const result = await this.qaChain.call(\n        { question, context: JSON.stringify(context) },\n        callbacks\n      );\n\n      chainResult = {\n        [this.outputKey]: result[this.qaChain.outputKey],\n      };\n    }\n\n    if (this.returnIntermediateSteps) {\n      chainResult[INTERMEDIATE_STEPS_KEY] = intermediateSteps;\n    }\n\n    return chainResult;\n  }\n}\n"],"mappings":";;;;;;;;;;;AAcA,MAAa,yBAAyB;;;;;;;;;;;;;;;;;;;;;;;;;AAkDtC,IAAa,qBAAb,MAAa,2BAA2B,UAAU;CAEhD,AAAQ;CAER,AAAQ;CAER,AAAQ;CAER,AAAQ,WAAW;CAEnB,AAAQ,YAAY;CAEpB,AAAQ,OAAO;CAEf,AAAQ,eAAe;CAEvB,AAAQ,0BAA0B;CAElC,YAAY,OAAgC;AAC1C,QAAM,MAAM;EACZ,MAAM,EACJ,OACA,uBACA,SACA,UACA,WACA,MACA,yBACA,iBACE;AAEJ,OAAK,QAAQ;AACb,OAAK,wBAAwB;AAC7B,OAAK,UAAU;AAEf,MAAI,SACF,MAAK,WAAW;AAElB,MAAI,UACF,MAAK,YAAY;AAEnB,MAAI,KACF,MAAK,OAAO;AAEd,MAAI,wBACF,MAAK,0BAA0B;AAEjC,MAAI,aACF,MAAK,eAAe;;CAIxB,aAAa;AACX,SAAO;;CAGT,IAAI,YAAsB;AACxB,SAAO,CAAC,KAAK,SAAS;;CAGxB,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,UAAU;;CAGzB,OAAO,QAAQ,OAAyC;EACtD,MAAM,EACJ,OACA,WAAW,kBACX,eAAe,0BACf,KACA,WACA,OACA,0BAA0B,OAC1B,eAAe,UACb;AAEJ,MAAI,CAAC,aAAa,CAAC,IACjB,OAAM,IAAI,MACR,0DACD;AAGH,MAAI,CAAC,SAAS,CAAC,IACb,OAAM,IAAI,MAAM,sDAAsD;AAGxE,MAAI,aAAa,SAAS,IACxB,OAAM,IAAI,MACR,kGACD;EAGH,MAAM,UAAU,IAAI,SAAS;GAC3B,KAAM,SAAS;GACf,QAAQ;GACT,CAAC;AAOF,SAAO,IAAI,mBAAmB;GAC5B,uBAN4B,IAAI,SAAS;IACzC,KAAM,aAAa;IACnB,QAAQ;IACT,CAAC;GAIA;GACA;GACA;GACA;GACD,CAAC;;CAGJ,AAAQ,cAAc,MAAsB;EAE1C,MAAM,UAAU,KAAK,MADL,eACmB;AACnC,SAAO,UAAU,QAAQ,KAAK;;CAGhC,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,YAAY,YAAY,UAAU;EACxC,MAAM,WAAW,OAAO,KAAK;EAE7B,MAAM,oBAAoB,EAAE;EAE5B,MAAM,kBAAkB,MAAM,KAAK,sBAAsB,KACvD;GAAE;GAAU,QAAQ,KAAK,MAAM,WAAW;GAAE,EAC5C,UACD;EAED,MAAM,kBAAkB,KAAK,cAAc,gBAAgB,KAAK;AAEhE,QAAM,YAAY,WAAW,sBAAsB;AACnD,QAAM,YAAY,WAAW,GAAG,gBAAgB,UAAU;AAE1D,oBAAkB,KAAK,EAAE,OAAO,iBAAiB,CAAC;EAElD,IAAI;EACJ,MAAM,UAAU,MAAM,KAAK,MAAM,MAAM,iBAAiB,EACtD,MAAM,KAAK,MACZ,CAAC;AAEF,MAAI,KAAK,aACP,eAAc,GAAG,KAAK,YAAY,SAAS;OACtC;AACL,SAAM,YAAY,WAAW,kBAAkB;AAC/C,SAAM,YAAY,WAAW,GAAG,QAAQ,UAAU;AAElD,qBAAkB,KAAK,EAAE,SAAS,CAAC;GAEnC,MAAM,SAAS,MAAM,KAAK,QAAQ,KAChC;IAAE;IAAU,SAAS,KAAK,UAAU,QAAQ;IAAE,EAC9C,UACD;AAED,iBAAc,GACX,KAAK,YAAY,OAAO,KAAK,QAAQ,YACvC;;AAGH,MAAI,KAAK,wBACP,aAAY,0BAA0B;AAGxC,SAAO"}

package/dist/chains/graph_qa/prompts.cjs

@@ -1,5 +1,5 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+let _langchain_core_prompts = require("@langchain/core/prompts");
 
 //#region src/chains/graph_qa/prompts.ts
 const CYPHER_GENERATION_TEMPLATE = `Task:Generate Cypher statement to query a graph database.
@@ -14,7 +14,7 @@ Do not include any text except the generated Cypher statement.
 
 The question is:
 {question}`;
-const CYPHER_GENERATION_PROMPT = /* @__PURE__ */ new __langchain_core_prompts.PromptTemplate({
+const CYPHER_GENERATION_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
 	template: CYPHER_GENERATION_TEMPLATE,
 	inputVariables: ["schema", "question"]
 });
@@ -35,7 +35,7 @@ Information:
 
 Question: {question}
 Helpful Answer:`;
-const CYPHER_QA_PROMPT = /* @__PURE__ */ new __langchain_core_prompts.PromptTemplate({
+const CYPHER_QA_PROMPT = /* @__PURE__ */ new _langchain_core_prompts.PromptTemplate({
 	template: CYPHER_QA_TEMPLATE,
 	inputVariables: ["context", "question"]
 });

package/dist/chains/graph_qa/prompts.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/graph_qa/prompts.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nconst CYPHER_GENERATION_TEMPLATE = `Task:Generate Cypher statement to query a graph database.\nInstructions:\nUse only the provided relationship types and properties in the schema.\nDo not use any other relationship types or properties that are not provided.\nSchema:\n{schema}\nNote: Do not include any explanations or apologies in your responses.\nDo not respond to any questions that might ask anything else than for you to construct a Cypher statement.\nDo not include any text except the generated Cypher statement.\n\nThe question is:\n{question}`;\nexport const CYPHER_GENERATION_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_GENERATION_TEMPLATE,\n  inputVariables: [\"schema\", \"question\"],\n});\n\nconst CYPHER_QA_TEMPLATE = `You are an assistant that helps to form nice and human understandable answers.\nThe information part contains the provided information that you must use to construct an answer.\nThe provided information is authoritative, you must never doubt it or try to use your internal knowledge to correct it.\nMake the answer sound as a response to the question. Do not mention that you based the result on the given information.\nHere is an example:\n\nQuestion: Which managers own Neo4j stocks?\nContext:[manager:CTL LLC, manager:JANE STREET GROUP LLC]\nHelpful Answer: CTL LLC, JANE STREET GROUP LLC owns Neo4j stocks.\n\nFollow this example when generating answers.\nIf the provided information is empty, say that you don't know the answer.\nInformation:\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\nexport const CYPHER_QA_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_QA_TEMPLATE,\n  inputVariables: [\"context\", \"question\"],\n});\n"],"mappings":";;;;AAEA,MAAM,6BAA6B,CAAC;;;;;;;;;;;UAW1B,CAAC;AACX,MAAa,2CAA2C,IAAIA,wCAAe;CACzE,UAAU;CACV,gBAAgB,CAAC,UAAU,UAAW;AACvC;AAED,MAAM,qBAAqB,CAAC;;;;;;;;;;;;;;;;eAgBb,CAAC;AAChB,MAAa,mCAAmC,IAAIA,wCAAe;CACjE,UAAU;CACV,gBAAgB,CAAC,WAAW,UAAW;AACxC"}
+{"version":3,"file":"prompts.cjs","names":["PromptTemplate"],"sources":["../../../src/chains/graph_qa/prompts.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nconst CYPHER_GENERATION_TEMPLATE = `Task:Generate Cypher statement to query a graph database.\nInstructions:\nUse only the provided relationship types and properties in the schema.\nDo not use any other relationship types or properties that are not provided.\nSchema:\n{schema}\nNote: Do not include any explanations or apologies in your responses.\nDo not respond to any questions that might ask anything else than for you to construct a Cypher statement.\nDo not include any text except the generated Cypher statement.\n\nThe question is:\n{question}`;\nexport const CYPHER_GENERATION_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_GENERATION_TEMPLATE,\n  inputVariables: [\"schema\", \"question\"],\n});\n\nconst CYPHER_QA_TEMPLATE = `You are an assistant that helps to form nice and human understandable answers.\nThe information part contains the provided information that you must use to construct an answer.\nThe provided information is authoritative, you must never doubt it or try to use your internal knowledge to correct it.\nMake the answer sound as a response to the question. Do not mention that you based the result on the given information.\nHere is an example:\n\nQuestion: Which managers own Neo4j stocks?\nContext:[manager:CTL LLC, manager:JANE STREET GROUP LLC]\nHelpful Answer: CTL LLC, JANE STREET GROUP LLC owns Neo4j stocks.\n\nFollow this example when generating answers.\nIf the provided information is empty, say that you don't know the answer.\nInformation:\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\nexport const CYPHER_QA_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_QA_TEMPLATE,\n  inputVariables: [\"context\", \"question\"],\n});\n"],"mappings":";;;;AAEA,MAAM,6BAA6B;;;;;;;;;;;;AAYnC,MAAa,2CAA2C,IAAIA,uCAAe;CACzE,UAAU;CACV,gBAAgB,CAAC,UAAU,WAAW;CACvC,CAAC;AAEF,MAAM,qBAAqB;;;;;;;;;;;;;;;;;AAiB3B,MAAa,mCAAmC,IAAIA,uCAAe;CACjE,UAAU;CACV,gBAAgB,CAAC,WAAW,WAAW;CACxC,CAAC"}

package/dist/chains/graph_qa/prompts.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.js","names":[],"sources":["../../../src/chains/graph_qa/prompts.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nconst CYPHER_GENERATION_TEMPLATE = `Task:Generate Cypher statement to query a graph database.\nInstructions:\nUse only the provided relationship types and properties in the schema.\nDo not use any other relationship types or properties that are not provided.\nSchema:\n{schema}\nNote: Do not include any explanations or apologies in your responses.\nDo not respond to any questions that might ask anything else than for you to construct a Cypher statement.\nDo not include any text except the generated Cypher statement.\n\nThe question is:\n{question}`;\nexport const CYPHER_GENERATION_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_GENERATION_TEMPLATE,\n  inputVariables: [\"schema\", \"question\"],\n});\n\nconst CYPHER_QA_TEMPLATE = `You are an assistant that helps to form nice and human understandable answers.\nThe information part contains the provided information that you must use to construct an answer.\nThe provided information is authoritative, you must never doubt it or try to use your internal knowledge to correct it.\nMake the answer sound as a response to the question. Do not mention that you based the result on the given information.\nHere is an example:\n\nQuestion: Which managers own Neo4j stocks?\nContext:[manager:CTL LLC, manager:JANE STREET GROUP LLC]\nHelpful Answer: CTL LLC, JANE STREET GROUP LLC owns Neo4j stocks.\n\nFollow this example when generating answers.\nIf the provided information is empty, say that you don't know the answer.\nInformation:\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\nexport const CYPHER_QA_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_QA_TEMPLATE,\n  inputVariables: [\"context\", \"question\"],\n});\n"],"mappings":";;;AAEA,MAAM,6BAA6B,CAAC;;;;;;;;;;;UAW1B,CAAC;AACX,MAAa,2CAA2C,IAAI,eAAe;CACzE,UAAU;CACV,gBAAgB,CAAC,UAAU,UAAW;AACvC;AAED,MAAM,qBAAqB,CAAC;;;;;;;;;;;;;;;;eAgBb,CAAC;AAChB,MAAa,mCAAmC,IAAI,eAAe;CACjE,UAAU;CACV,gBAAgB,CAAC,WAAW,UAAW;AACxC"}
+{"version":3,"file":"prompts.js","names":[],"sources":["../../../src/chains/graph_qa/prompts.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nconst CYPHER_GENERATION_TEMPLATE = `Task:Generate Cypher statement to query a graph database.\nInstructions:\nUse only the provided relationship types and properties in the schema.\nDo not use any other relationship types or properties that are not provided.\nSchema:\n{schema}\nNote: Do not include any explanations or apologies in your responses.\nDo not respond to any questions that might ask anything else than for you to construct a Cypher statement.\nDo not include any text except the generated Cypher statement.\n\nThe question is:\n{question}`;\nexport const CYPHER_GENERATION_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_GENERATION_TEMPLATE,\n  inputVariables: [\"schema\", \"question\"],\n});\n\nconst CYPHER_QA_TEMPLATE = `You are an assistant that helps to form nice and human understandable answers.\nThe information part contains the provided information that you must use to construct an answer.\nThe provided information is authoritative, you must never doubt it or try to use your internal knowledge to correct it.\nMake the answer sound as a response to the question. Do not mention that you based the result on the given information.\nHere is an example:\n\nQuestion: Which managers own Neo4j stocks?\nContext:[manager:CTL LLC, manager:JANE STREET GROUP LLC]\nHelpful Answer: CTL LLC, JANE STREET GROUP LLC owns Neo4j stocks.\n\nFollow this example when generating answers.\nIf the provided information is empty, say that you don't know the answer.\nInformation:\n{context}\n\nQuestion: {question}\nHelpful Answer:`;\nexport const CYPHER_QA_PROMPT = /* #__PURE__ */ new PromptTemplate({\n  template: CYPHER_QA_TEMPLATE,\n  inputVariables: [\"context\", \"question\"],\n});\n"],"mappings":";;;AAEA,MAAM,6BAA6B;;;;;;;;;;;;AAYnC,MAAa,2CAA2C,IAAI,eAAe;CACzE,UAAU;CACV,gBAAgB,CAAC,UAAU,WAAW;CACvC,CAAC;AAEF,MAAM,qBAAqB;;;;;;;;;;;;;;;;;AAiB3B,MAAa,mCAAmC,IAAI,eAAe;CACjE,UAAU;CACV,gBAAgB,CAAC,WAAW,WAAW;CACxC,CAAC"}

package/dist/chains/history_aware_retriever.cjs

@@ -1,10 +1,10 @@
-const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
-const __langchain_core_runnables = require_rolldown_runtime.__toESM(require("@langchain/core/runnables"));
-const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
+let _langchain_core_runnables = require("@langchain/core/runnables");
+let _langchain_core_output_parsers = require("@langchain/core/output_parsers");
 
 //#region src/chains/history_aware_retriever.ts
-var history_aware_retriever_exports = {};
-require_rolldown_runtime.__export(history_aware_retriever_exports, { createHistoryAwareRetriever: () => createHistoryAwareRetriever });
+var history_aware_retriever_exports = /* @__PURE__ */ require_runtime.__exportAll({ createHistoryAwareRetriever: () => createHistoryAwareRetriever });
 /**
 * Create a chain that takes conversation history and returns documents.
 * If there is no `chat_history`, then the `input` is just passed directly to the
@@ -35,13 +35,12 @@ require_rolldown_runtime.__export(history_aware_retriever_exports, { createHisto
 */
 async function createHistoryAwareRetriever({ llm, retriever, rephrasePrompt }) {
 	if (!rephrasePrompt.inputVariables.includes("input")) throw new Error(`Expected "input" to be a prompt variable, but got ${JSON.stringify(rephrasePrompt.inputVariables)}`);
-	const retrieveDocuments = __langchain_core_runnables.RunnableBranch.from([[(input) => !input.chat_history || input.chat_history.length === 0, __langchain_core_runnables.RunnableSequence.from([(input) => input.input, retriever])], __langchain_core_runnables.RunnableSequence.from([
+	return _langchain_core_runnables.RunnableBranch.from([[(input) => !input.chat_history || input.chat_history.length === 0, _langchain_core_runnables.RunnableSequence.from([(input) => input.input, retriever])], _langchain_core_runnables.RunnableSequence.from([
 		rephrasePrompt,
 		llm,
-		new __langchain_core_output_parsers.StringOutputParser(),
+		new _langchain_core_output_parsers.StringOutputParser(),
 		retriever
 	])]).withConfig({ runName: "history_aware_retriever" });
-	return retrieveDocuments;
 }
 
 //#endregion

package/dist/chains/history_aware_retriever.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"history_aware_retriever.cjs","names":["RunnableBranch","RunnableSequence","StringOutputParser"],"sources":["../../src/chains/history_aware_retriever.ts"],"sourcesContent":["import type { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport {\n  type Runnable,\n  type RunnableInterface,\n  RunnableSequence,\n  RunnableBranch,\n} from \"@langchain/core/runnables\";\nimport { type BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\n\n/**\n * Params for the createHistoryAwareRetriever method.\n */\nexport type CreateHistoryAwareRetrieverParams = {\n  /**\n   * Language model to use for generating a search term given chat history.\n   */\n  llm: LanguageModelLike;\n  /**\n   * RetrieverLike object that takes a string as input and outputs a list of Documents.\n   */\n  retriever: RunnableInterface<string, DocumentInterface[]>;\n  /**\n   * The prompt used to generate the search query for the retriever.\n   */\n  rephrasePrompt: BasePromptTemplate;\n};\n\n/**\n * Create a chain that takes conversation history and returns documents.\n * If there is no `chat_history`, then the `input` is just passed directly to the\n * retriever. If there is `chat_history`, then the prompt and LLM will be used\n * to generate a search query. That search query is then passed to the retriever.\n * @param {CreateHistoryAwareRetriever} params\n * @returns An LCEL Runnable. The runnable input must take in `input`, and if there\n * is chat history should take it in the form of `chat_history`.\n * The Runnable output is a list of Documents\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n *\n * const rephrasePrompt = await pull(\"langchain-ai/chat-langchain-rephrase\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const chain = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt,\n * });\n * const result = await chain.invoke({\"input\": \"...\", \"chat_history\": [] })\n * ```\n */\nexport async function createHistoryAwareRetriever({\n  llm,\n  retriever,\n  rephrasePrompt,\n}: CreateHistoryAwareRetrieverParams): Promise<\n  Runnable<\n    { input: string; chat_history: string | BaseMessage[] },\n    DocumentInterface[]\n  >\n> {\n  if (!rephrasePrompt.inputVariables.includes(\"input\")) {\n    throw new Error(\n      `Expected \"input\" to be a prompt variable, but got ${JSON.stringify(\n        rephrasePrompt.inputVariables\n      )}`\n    );\n  }\n  const retrieveDocuments = RunnableBranch.from([\n    [\n      (input) => !input.chat_history || input.chat_history.length === 0,\n      RunnableSequence.from([(input) => input.input, retriever]),\n    ],\n    RunnableSequence.from([\n      rephrasePrompt,\n      llm,\n      new StringOutputParser(),\n      retriever,\n    ]),\n  ]).withConfig({\n    runName: \"history_aware_retriever\",\n  });\n  return retrieveDocuments;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0DA,eAAsB,4BAA4B,EAChD,KACA,WACA,gBACkC,EAKlC;AACA,KAAI,CAAC,eAAe,eAAe,SAAS,QAAQ,CAClD,OAAM,IAAI,MACR,CAAC,kDAAkD,EAAE,KAAK,UACxD,eAAe,eAChB,EAAE;CAGP,MAAM,oBAAoBA,0CAAe,KAAK,CAC5C,CACE,CAAC,UAAU,CAAC,MAAM,gBAAgB,MAAM,aAAa,WAAW,GAChEC,4CAAiB,KAAK,CAAC,CAAC,UAAU,MAAM,OAAO,SAAU,EAAC,AAC3D,GACDA,4CAAiB,KAAK;EACpB;EACA;EACA,IAAIC;EACJ;CACD,EAAC,AACH,EAAC,CAAC,WAAW,EACZ,SAAS,0BACV,EAAC;AACF,QAAO;AACR"}
+{"version":3,"file":"history_aware_retriever.cjs","names":["RunnableBranch","RunnableSequence","StringOutputParser"],"sources":["../../src/chains/history_aware_retriever.ts"],"sourcesContent":["import type { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport {\n  type Runnable,\n  type RunnableInterface,\n  RunnableSequence,\n  RunnableBranch,\n} from \"@langchain/core/runnables\";\nimport { type BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\n\n/**\n * Params for the createHistoryAwareRetriever method.\n */\nexport type CreateHistoryAwareRetrieverParams = {\n  /**\n   * Language model to use for generating a search term given chat history.\n   */\n  llm: LanguageModelLike;\n  /**\n   * RetrieverLike object that takes a string as input and outputs a list of Documents.\n   */\n  retriever: RunnableInterface<string, DocumentInterface[]>;\n  /**\n   * The prompt used to generate the search query for the retriever.\n   */\n  rephrasePrompt: BasePromptTemplate;\n};\n\n/**\n * Create a chain that takes conversation history and returns documents.\n * If there is no `chat_history`, then the `input` is just passed directly to the\n * retriever. If there is `chat_history`, then the prompt and LLM will be used\n * to generate a search query. That search query is then passed to the retriever.\n * @param {CreateHistoryAwareRetriever} params\n * @returns An LCEL Runnable. The runnable input must take in `input`, and if there\n * is chat history should take it in the form of `chat_history`.\n * The Runnable output is a list of Documents\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n *\n * const rephrasePrompt = await pull(\"langchain-ai/chat-langchain-rephrase\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const chain = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt,\n * });\n * const result = await chain.invoke({\"input\": \"...\", \"chat_history\": [] })\n * ```\n */\nexport async function createHistoryAwareRetriever({\n  llm,\n  retriever,\n  rephrasePrompt,\n}: CreateHistoryAwareRetrieverParams): Promise<\n  Runnable<\n    { input: string; chat_history: string | BaseMessage[] },\n    DocumentInterface[]\n  >\n> {\n  if (!rephrasePrompt.inputVariables.includes(\"input\")) {\n    throw new Error(\n      `Expected \"input\" to be a prompt variable, but got ${JSON.stringify(\n        rephrasePrompt.inputVariables\n      )}`\n    );\n  }\n  const retrieveDocuments = RunnableBranch.from([\n    [\n      (input) => !input.chat_history || input.chat_history.length === 0,\n      RunnableSequence.from([(input) => input.input, retriever]),\n    ],\n    RunnableSequence.from([\n      rephrasePrompt,\n      llm,\n      new StringOutputParser(),\n      retriever,\n    ]),\n  ]).withConfig({\n    runName: \"history_aware_retriever\",\n  });\n  return retrieveDocuments;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0DA,eAAsB,4BAA4B,EAChD,KACA,WACA,kBAMA;AACA,KAAI,CAAC,eAAe,eAAe,SAAS,QAAQ,CAClD,OAAM,IAAI,MACR,qDAAqD,KAAK,UACxD,eAAe,eAChB,GACF;AAgBH,QAd0BA,yCAAe,KAAK,CAC5C,EACG,UAAU,CAAC,MAAM,gBAAgB,MAAM,aAAa,WAAW,GAChEC,2CAAiB,KAAK,EAAE,UAAU,MAAM,OAAO,UAAU,CAAC,CAC3D,EACDA,2CAAiB,KAAK;EACpB;EACA;EACA,IAAIC,mDAAoB;EACxB;EACD,CAAC,CACH,CAAC,CAAC,WAAW,EACZ,SAAS,2BACV,CAAC"}

package/dist/chains/history_aware_retriever.d.cts

@@ -5,7 +5,6 @@ import { LanguageModelLike } from "@langchain/core/language_models/base";
 import { DocumentInterface } from "@langchain/core/documents";
 
 //#region src/chains/history_aware_retriever.d.ts
-
 /**
  * Params for the createHistoryAwareRetriever method.
  */

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

@@ -1 +1 @@
-{"version":3,"file":"history_aware_retriever.d.cts","names":["LanguageModelLike","Runnable","RunnableInterface","BasePromptTemplate","DocumentInterface","BaseMessage","CreateHistoryAwareRetrieverParams","createHistoryAwareRetriever","llm","retriever","rephrasePrompt","Promise"],"sources":["../../src/chains/history_aware_retriever.d.ts"],"sourcesContent":["import type { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport { type Runnable, type RunnableInterface } from \"@langchain/core/runnables\";\nimport { type BasePromptTemplate } from \"@langchain/core/prompts\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\n/**\n * Params for the createHistoryAwareRetriever method.\n */\nexport type CreateHistoryAwareRetrieverParams = {\n    /**\n     * Language model to use for generating a search term given chat history.\n     */\n    llm: LanguageModelLike;\n    /**\n     * RetrieverLike object that takes a string as input and outputs a list of Documents.\n     */\n    retriever: RunnableInterface<string, DocumentInterface[]>;\n    /**\n     * The prompt used to generate the search query for the retriever.\n     */\n    rephrasePrompt: BasePromptTemplate;\n};\n/**\n * Create a chain that takes conversation history and returns documents.\n * If there is no `chat_history`, then the `input` is just passed directly to the\n * retriever. If there is `chat_history`, then the prompt and LLM will be used\n * to generate a search query. That search query is then passed to the retriever.\n * @param {CreateHistoryAwareRetriever} params\n * @returns An LCEL Runnable. The runnable input must take in `input`, and if there\n * is chat history should take it in the form of `chat_history`.\n * The Runnable output is a list of Documents\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n *\n * const rephrasePrompt = await pull(\"langchain-ai/chat-langchain-rephrase\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const chain = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt,\n * });\n * const result = await chain.invoke({\"input\": \"...\", \"chat_history\": [] })\n * ```\n */\nexport declare function createHistoryAwareRetriever({ llm, retriever, rephrasePrompt }: CreateHistoryAwareRetrieverParams): Promise<Runnable<{\n    input: string;\n    chat_history: string | BaseMessage[];\n}, DocumentInterface[]>>;\n//# sourceMappingURL=history_aware_retriever.d.ts.map"],"mappings":";;;;;;;;;;AAQA;AAISA,KAJGM,iCAAAA,GAIHN;EAIgCI;;;EAIH,GAAA,EAR7BJ,iBAQ6B;EA8BdO;;;EAA8CG,SAAAA,EAlCvDR,iBAkCuDQ,CAAAA,MAAAA,EAlC7BN,iBAkC6BM,EAAAA,CAAAA;EAAkBJ;;;EAA4CL,cAAAA,EA9BhHE,kBA8BgHF;CAARU;AAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAA3GJ,2BAAAA;;;;GAAgED,oCAAoCK,QAAQV;;yBAEzGI;GACxBD"}
+{"version":3,"file":"history_aware_retriever.d.cts","names":[],"sources":["../../src/chains/history_aware_retriever.ts"],"mappings":";;;;;;;;;AAeA;KAAY,iCAAA;;;;EAIV,GAAA,EAAK,iBAAA;EAQW;;;EAJhB,SAAA,EAAW,iBAAA,SAA0B,iBAAA;EAJhC;;;EAQL,cAAA,EAAgB,kBAAA;AAAA;;;;AA+BlB;;;;;;;;;;;;;;;;;;;;;;;;;iBAAsB,2BAAA,CAAA;EACpB,GAAA;EACA,SAAA;EACA;AAAA,GACC,iCAAA,GAAoC,OAAA,CACrC,QAAA;EACI,KAAA;EAAe,YAAA,WAAuB,WAAA;AAAA,GACxC,iBAAA"}

package/dist/chains/history_aware_retriever.d.ts

@@ -1,11 +1,10 @@
-import { DocumentInterface } from "@langchain/core/documents";
+import { BaseMessage } from "@langchain/core/messages";
 import { BasePromptTemplate } from "@langchain/core/prompts";
 import { Runnable, RunnableInterface } from "@langchain/core/runnables";
 import { LanguageModelLike } from "@langchain/core/language_models/base";
-import { BaseMessage } from "@langchain/core/messages";
+import { DocumentInterface } from "@langchain/core/documents";
 
 //#region src/chains/history_aware_retriever.d.ts
-
 /**
  * Params for the createHistoryAwareRetriever method.
  */

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

@@ -1 +1 @@
-{"version":3,"file":"history_aware_retriever.d.ts","names":["LanguageModelLike","Runnable","RunnableInterface","BasePromptTemplate","DocumentInterface","BaseMessage","CreateHistoryAwareRetrieverParams","createHistoryAwareRetriever","llm","retriever","rephrasePrompt","Promise"],"sources":["../../src/chains/history_aware_retriever.d.ts"],"sourcesContent":["import type { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport { type Runnable, type RunnableInterface } from \"@langchain/core/runnables\";\nimport { type BasePromptTemplate } from \"@langchain/core/prompts\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\n/**\n * Params for the createHistoryAwareRetriever method.\n */\nexport type CreateHistoryAwareRetrieverParams = {\n    /**\n     * Language model to use for generating a search term given chat history.\n     */\n    llm: LanguageModelLike;\n    /**\n     * RetrieverLike object that takes a string as input and outputs a list of Documents.\n     */\n    retriever: RunnableInterface<string, DocumentInterface[]>;\n    /**\n     * The prompt used to generate the search query for the retriever.\n     */\n    rephrasePrompt: BasePromptTemplate;\n};\n/**\n * Create a chain that takes conversation history and returns documents.\n * If there is no `chat_history`, then the `input` is just passed directly to the\n * retriever. If there is `chat_history`, then the prompt and LLM will be used\n * to generate a search query. That search query is then passed to the retriever.\n * @param {CreateHistoryAwareRetriever} params\n * @returns An LCEL Runnable. The runnable input must take in `input`, and if there\n * is chat history should take it in the form of `chat_history`.\n * The Runnable output is a list of Documents\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n *\n * const rephrasePrompt = await pull(\"langchain-ai/chat-langchain-rephrase\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const chain = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt,\n * });\n * const result = await chain.invoke({\"input\": \"...\", \"chat_history\": [] })\n * ```\n */\nexport declare function createHistoryAwareRetriever({ llm, retriever, rephrasePrompt }: CreateHistoryAwareRetrieverParams): Promise<Runnable<{\n    input: string;\n    chat_history: string | BaseMessage[];\n}, DocumentInterface[]>>;\n//# sourceMappingURL=history_aware_retriever.d.ts.map"],"mappings":";;;;;;;;;;AAQA;AAISA,KAJGM,iCAAAA,GAIHN;EAIgCI;;;EAIH,GAAA,EAR7BJ,iBAQ6B;EA8BdO;;;EAA8CG,SAAAA,EAlCvDR,iBAkCuDQ,CAAAA,MAAAA,EAlC7BN,iBAkC6BM,EAAAA,CAAAA;EAAkBJ;;;EAA4CL,cAAAA,EA9BhHE,kBA8BgHF;CAARU;AAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAA3GJ,2BAAAA;;;;GAAgED,oCAAoCK,QAAQV;;yBAEzGI;GACxBD"}
+{"version":3,"file":"history_aware_retriever.d.ts","names":[],"sources":["../../src/chains/history_aware_retriever.ts"],"mappings":";;;;;;;;;AAeA;KAAY,iCAAA;;;;EAIV,GAAA,EAAK,iBAAA;EAQW;;;EAJhB,SAAA,EAAW,iBAAA,SAA0B,iBAAA;EAJhC;;;EAQL,cAAA,EAAgB,kBAAA;AAAA;;;;AA+BlB;;;;;;;;;;;;;;;;;;;;;;;;;iBAAsB,2BAAA,CAAA;EACpB,GAAA;EACA,SAAA;EACA;AAAA,GACC,iCAAA,GAAoC,OAAA,CACrC,QAAA;EACI,KAAA;EAAe,YAAA,WAAuB,WAAA;AAAA,GACxC,iBAAA"}

package/dist/chains/history_aware_retriever.js

@@ -1,10 +1,9 @@
-import { __export } from "../_virtual/rolldown_runtime.js";
+import { __exportAll } from "../_virtual/_rolldown/runtime.js";
 import { RunnableBranch, RunnableSequence } from "@langchain/core/runnables";
 import { StringOutputParser } from "@langchain/core/output_parsers";
 
 //#region src/chains/history_aware_retriever.ts
-var history_aware_retriever_exports = {};
-__export(history_aware_retriever_exports, { createHistoryAwareRetriever: () => createHistoryAwareRetriever });
+var history_aware_retriever_exports = /* @__PURE__ */ __exportAll({ createHistoryAwareRetriever: () => createHistoryAwareRetriever });
 /**
 * Create a chain that takes conversation history and returns documents.
 * If there is no `chat_history`, then the `input` is just passed directly to the
@@ -35,13 +34,12 @@ __export(history_aware_retriever_exports, { createHistoryAwareRetriever: () => c
 */
 async function createHistoryAwareRetriever({ llm, retriever, rephrasePrompt }) {
 	if (!rephrasePrompt.inputVariables.includes("input")) throw new Error(`Expected "input" to be a prompt variable, but got ${JSON.stringify(rephrasePrompt.inputVariables)}`);
-	const retrieveDocuments = RunnableBranch.from([[(input) => !input.chat_history || input.chat_history.length === 0, RunnableSequence.from([(input) => input.input, retriever])], RunnableSequence.from([
+	return RunnableBranch.from([[(input) => !input.chat_history || input.chat_history.length === 0, RunnableSequence.from([(input) => input.input, retriever])], RunnableSequence.from([
 		rephrasePrompt,
 		llm,
 		new StringOutputParser(),
 		retriever
 	])]).withConfig({ runName: "history_aware_retriever" });
-	return retrieveDocuments;
 }
 
 //#endregion

package/dist/chains/history_aware_retriever.js.map

@@ -1 +1 @@
-{"version":3,"file":"history_aware_retriever.js","names":[],"sources":["../../src/chains/history_aware_retriever.ts"],"sourcesContent":["import type { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport {\n  type Runnable,\n  type RunnableInterface,\n  RunnableSequence,\n  RunnableBranch,\n} from \"@langchain/core/runnables\";\nimport { type BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\n\n/**\n * Params for the createHistoryAwareRetriever method.\n */\nexport type CreateHistoryAwareRetrieverParams = {\n  /**\n   * Language model to use for generating a search term given chat history.\n   */\n  llm: LanguageModelLike;\n  /**\n   * RetrieverLike object that takes a string as input and outputs a list of Documents.\n   */\n  retriever: RunnableInterface<string, DocumentInterface[]>;\n  /**\n   * The prompt used to generate the search query for the retriever.\n   */\n  rephrasePrompt: BasePromptTemplate;\n};\n\n/**\n * Create a chain that takes conversation history and returns documents.\n * If there is no `chat_history`, then the `input` is just passed directly to the\n * retriever. If there is `chat_history`, then the prompt and LLM will be used\n * to generate a search query. That search query is then passed to the retriever.\n * @param {CreateHistoryAwareRetriever} params\n * @returns An LCEL Runnable. The runnable input must take in `input`, and if there\n * is chat history should take it in the form of `chat_history`.\n * The Runnable output is a list of Documents\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n *\n * const rephrasePrompt = await pull(\"langchain-ai/chat-langchain-rephrase\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const chain = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt,\n * });\n * const result = await chain.invoke({\"input\": \"...\", \"chat_history\": [] })\n * ```\n */\nexport async function createHistoryAwareRetriever({\n  llm,\n  retriever,\n  rephrasePrompt,\n}: CreateHistoryAwareRetrieverParams): Promise<\n  Runnable<\n    { input: string; chat_history: string | BaseMessage[] },\n    DocumentInterface[]\n  >\n> {\n  if (!rephrasePrompt.inputVariables.includes(\"input\")) {\n    throw new Error(\n      `Expected \"input\" to be a prompt variable, but got ${JSON.stringify(\n        rephrasePrompt.inputVariables\n      )}`\n    );\n  }\n  const retrieveDocuments = RunnableBranch.from([\n    [\n      (input) => !input.chat_history || input.chat_history.length === 0,\n      RunnableSequence.from([(input) => input.input, retriever]),\n    ],\n    RunnableSequence.from([\n      rephrasePrompt,\n      llm,\n      new StringOutputParser(),\n      retriever,\n    ]),\n  ]).withConfig({\n    runName: \"history_aware_retriever\",\n  });\n  return retrieveDocuments;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0DA,eAAsB,4BAA4B,EAChD,KACA,WACA,gBACkC,EAKlC;AACA,KAAI,CAAC,eAAe,eAAe,SAAS,QAAQ,CAClD,OAAM,IAAI,MACR,CAAC,kDAAkD,EAAE,KAAK,UACxD,eAAe,eAChB,EAAE;CAGP,MAAM,oBAAoB,eAAe,KAAK,CAC5C,CACE,CAAC,UAAU,CAAC,MAAM,gBAAgB,MAAM,aAAa,WAAW,GAChE,iBAAiB,KAAK,CAAC,CAAC,UAAU,MAAM,OAAO,SAAU,EAAC,AAC3D,GACD,iBAAiB,KAAK;EACpB;EACA;EACA,IAAI;EACJ;CACD,EAAC,AACH,EAAC,CAAC,WAAW,EACZ,SAAS,0BACV,EAAC;AACF,QAAO;AACR"}
+{"version":3,"file":"history_aware_retriever.js","names":[],"sources":["../../src/chains/history_aware_retriever.ts"],"sourcesContent":["import type { LanguageModelLike } from \"@langchain/core/language_models/base\";\nimport {\n  type Runnable,\n  type RunnableInterface,\n  RunnableSequence,\n  RunnableBranch,\n} from \"@langchain/core/runnables\";\nimport { type BasePromptTemplate } from \"@langchain/core/prompts\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport type { DocumentInterface } from \"@langchain/core/documents\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\n\n/**\n * Params for the createHistoryAwareRetriever method.\n */\nexport type CreateHistoryAwareRetrieverParams = {\n  /**\n   * Language model to use for generating a search term given chat history.\n   */\n  llm: LanguageModelLike;\n  /**\n   * RetrieverLike object that takes a string as input and outputs a list of Documents.\n   */\n  retriever: RunnableInterface<string, DocumentInterface[]>;\n  /**\n   * The prompt used to generate the search query for the retriever.\n   */\n  rephrasePrompt: BasePromptTemplate;\n};\n\n/**\n * Create a chain that takes conversation history and returns documents.\n * If there is no `chat_history`, then the `input` is just passed directly to the\n * retriever. If there is `chat_history`, then the prompt and LLM will be used\n * to generate a search query. That search query is then passed to the retriever.\n * @param {CreateHistoryAwareRetriever} params\n * @returns An LCEL Runnable. The runnable input must take in `input`, and if there\n * is chat history should take it in the form of `chat_history`.\n * The Runnable output is a list of Documents\n * @example\n * ```typescript\n * // pnpm add langchain @langchain/openai\n *\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { pull } from \"langchain/hub\";\n * import { createHistoryAwareRetriever } from \"@langchain/classic/chains/history_aware_retriever\";\n *\n * const rephrasePrompt = await pull(\"langchain-ai/chat-langchain-rephrase\");\n * const llm = new ChatOpenAI({ model: \"gpt-4o-mini\" });\n * const retriever = ...\n * const chain = await createHistoryAwareRetriever({\n *   llm,\n *   retriever,\n *   rephrasePrompt,\n * });\n * const result = await chain.invoke({\"input\": \"...\", \"chat_history\": [] })\n * ```\n */\nexport async function createHistoryAwareRetriever({\n  llm,\n  retriever,\n  rephrasePrompt,\n}: CreateHistoryAwareRetrieverParams): Promise<\n  Runnable<\n    { input: string; chat_history: string | BaseMessage[] },\n    DocumentInterface[]\n  >\n> {\n  if (!rephrasePrompt.inputVariables.includes(\"input\")) {\n    throw new Error(\n      `Expected \"input\" to be a prompt variable, but got ${JSON.stringify(\n        rephrasePrompt.inputVariables\n      )}`\n    );\n  }\n  const retrieveDocuments = RunnableBranch.from([\n    [\n      (input) => !input.chat_history || input.chat_history.length === 0,\n      RunnableSequence.from([(input) => input.input, retriever]),\n    ],\n    RunnableSequence.from([\n      rephrasePrompt,\n      llm,\n      new StringOutputParser(),\n      retriever,\n    ]),\n  ]).withConfig({\n    runName: \"history_aware_retriever\",\n  });\n  return retrieveDocuments;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0DA,eAAsB,4BAA4B,EAChD,KACA,WACA,kBAMA;AACA,KAAI,CAAC,eAAe,eAAe,SAAS,QAAQ,CAClD,OAAM,IAAI,MACR,qDAAqD,KAAK,UACxD,eAAe,eAChB,GACF;AAgBH,QAd0B,eAAe,KAAK,CAC5C,EACG,UAAU,CAAC,MAAM,gBAAgB,MAAM,aAAa,WAAW,GAChE,iBAAiB,KAAK,EAAE,UAAU,MAAM,OAAO,UAAU,CAAC,CAC3D,EACD,iBAAiB,KAAK;EACpB;EACA;EACA,IAAI,oBAAoB;EACxB;EACD,CAAC,CACH,CAAC,CAAC,WAAW,EACZ,SAAS,2BACV,CAAC"}