@langchain/classic 1.0.17 → 1.0.18

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.cjs

@@ -1,10 +1,10 @@
-const require_rolldown_runtime = require('../../../_virtual/rolldown_runtime.cjs');
+const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
 const require_base = require('../../../chains/base.cjs');
 const require_openai_functions = require('../../../output_parsers/openai_functions.cjs');
 const require_types = require('./types.cjs');
 const require_violation_of_expectations_prompt = require('./violation_of_expectations_prompt.cjs');
-const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
-const __langchain_core_messages = require_rolldown_runtime.__toESM(require("@langchain/core/messages"));
+let _langchain_core_messages = require("@langchain/core/messages");
+let _langchain_core_output_parsers = require("@langchain/core/output_parsers");
 
 //#region src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts
 /**
@@ -35,7 +35,7 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 		this.retriever = fields.retriever;
 		this.llm = fields.llm;
 		this.jsonOutputParser = new require_openai_functions.JsonOutputFunctionsParser();
-		this.stringOutputParser = new __langchain_core_output_parsers.StringOutputParser();
+		this.stringOutputParser = new _langchain_core_output_parsers.StringOutputParser();
 	}
 	getChatHistoryString(chatHistory) {
 		return chatHistory.map((chatMessage) => {
@@ -73,7 +73,7 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 				if (!userResponse || userResponse._getType() !== "human") userResponse = void 0;
 				newArray.push({
 					chunkedMessages: tempArray,
-					userResponse: userResponse ? new __langchain_core_messages.HumanMessage(userResponse) : void 0
+					userResponse: userResponse ? new _langchain_core_messages.HumanMessage(userResponse) : void 0
 				});
 			}
 		});
@@ -101,8 +101,7 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 	async _call(values, runManager) {
 		if (!(this.chatHistoryKey in values)) throw new Error(`Chat history key ${this.chatHistoryKey} not found`);
 		const chatHistory = values[this.chatHistoryKey];
-		const isEveryMessageBaseMessage = chatHistory.every((message) => (0, __langchain_core_messages.isBaseMessage)(message));
-		if (!isEveryMessageBaseMessage) throw new Error("Chat history must be an array of BaseMessages");
+		if (!chatHistory.every((message) => (0, _langchain_core_messages.isBaseMessage)(message))) throw new Error("Chat history must be an array of BaseMessages");
 		const messageChunks = this.chunkMessagesByAIResponse(chatHistory);
 		const userPredictions = await Promise.all(messageChunks.map(async (chatHistoryChunk) => ({
 			userPredictions: await this.predictNextUserMessage(chatHistoryChunk.chunkedMessages),
@@ -114,14 +113,13 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 			userResponse: prediction.userResponse,
 			runManager
 		})));
-		const insights = await Promise.all(predictionViolations.map((violation) => this.generateFacts({
+		return { insights: await Promise.all(predictionViolations.map((violation) => this.generateFacts({
 			userResponse: violation.userResponse,
 			predictions: {
 				revisedPrediction: violation.revisedPrediction,
 				explainedPredictionErrors: violation.explainedPredictionErrors
 			}
-		})));
-		return { insights };
+		}))) };
 	}
 	/**
 	* This method predicts the next user message based on the chat history.
@@ -136,8 +134,7 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 	async predictNextUserMessage(chatHistory, runManager) {
 		const messageString = this.getChatHistoryString(chatHistory);
 		const llmWithFunctions = this.llm.bindTools([require_types.PREDICT_NEXT_USER_MESSAGE_FUNCTION]).withConfig({ function_call: { name: require_types.PREDICT_NEXT_USER_MESSAGE_FUNCTION.name } });
-		const chain = require_violation_of_expectations_prompt.PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(this.jsonOutputParser);
-		const res = await chain.invoke({ chat_history: messageString }, runManager?.getChild("prediction"));
+		const res = await require_violation_of_expectations_prompt.PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(this.jsonOutputParser).invoke({ chat_history: messageString }, runManager?.getChild("prediction"));
 		if (!("userState" in res && "predictedUserMessage" in res && "insights" in res)) throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);
 		const predictionResponse = res;
 		const retrievedDocs = await this.retrieveRelevantInsights(predictionResponse.insights);
@@ -155,12 +152,9 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 	* @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.
 	*/
 	async retrieveRelevantInsights(insights) {
-		const relevantInsightsDocuments = await Promise.all(insights.map(async (insight) => {
-			const relevantInsight = await this.retriever.invoke(insight);
-			return relevantInsight[0];
-		}));
-		const relevantInsightsContent = relevantInsightsDocuments.map((document) => document.pageContent);
-		return relevantInsightsContent;
+		return (await Promise.all(insights.map(async (insight) => {
+			return (await this.retriever.invoke(insight))[0];
+		}))).map((document) => document.pageContent);
 	}
 	/**
 	* This method generates prediction violations based on the predicted and actual user responses.
@@ -184,15 +178,14 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 			actual_output: userResponse?.content ?? "",
 			user_insights: userPredictions.insights.join("\n")
 		}, runManager?.getChild("prediction_violations"));
-		const revisedPrediction = await this.generateRevisedPrediction({
-			originalPrediction: userPredictions.predictedUserMessage,
-			explainedPredictionErrors: res.explainedPredictionErrors,
-			userInsights: userPredictions.insights,
-			runManager
-		});
 		return {
 			userResponse,
-			revisedPrediction,
+			revisedPrediction: await this.generateRevisedPrediction({
+				originalPrediction: userPredictions.predictedUserMessage,
+				explainedPredictionErrors: res.explainedPredictionErrors,
+				userInsights: userPredictions.insights,
+				runManager
+			}),
 			explainedPredictionErrors: res.explainedPredictionErrors
 		};
 	}
@@ -208,13 +201,11 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 	* @returns {Promise<string>} A promise that resolves to a revised prediction.
 	*/
 	async generateRevisedPrediction({ originalPrediction, explainedPredictionErrors, userInsights, runManager }) {
-		const revisedPredictionChain = require_violation_of_expectations_prompt.GENERATE_REVISED_PREDICTION_PROMPT.pipe(this.llm).pipe(this.stringOutputParser);
-		const revisedPredictionRes = await revisedPredictionChain.invoke({
+		return await require_violation_of_expectations_prompt.GENERATE_REVISED_PREDICTION_PROMPT.pipe(this.llm).pipe(this.stringOutputParser).invoke({
 			prediction: originalPrediction,
 			explained_prediction_errors: explainedPredictionErrors.join("\n"),
 			user_insights: userInsights.join("\n")
 		}, runManager?.getChild("prediction_revision"));
-		return revisedPredictionRes;
 	}
 	/**
 	* This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.
@@ -231,12 +222,11 @@ var ViolationOfExpectationsChain = class extends require_base.BaseChain {
 	async generateFacts({ userResponse, predictions, runManager }) {
 		const chain = require_violation_of_expectations_prompt.GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(this.stringOutputParser);
 		if (typeof userResponse?.content !== "string") throw new Error("This chain does not support non-string model output.");
-		const res = await chain.invoke({
+		return await chain.invoke({
 			prediction_violations: predictions.explainedPredictionErrors.join("\n"),
 			prediction: predictions.revisedPrediction,
 			user_message: userResponse?.content ?? ""
 		}, runManager?.getChild("generate_facts"));
-		return res;
 	}
 	/**
 	* Static method that creates a ViolationOfExpectationsChain instance from a

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"violation_of_expectations_chain.cjs","names":["BaseChain","fields: ViolationOfExpectationsChainInput","JsonOutputFunctionsParser","StringOutputParser","chatHistory: BaseMessage[]","strings: Array<string>","newArray: MessageChunkResult[]","tempArray: BaseMessage[]","userResponse: BaseMessage | undefined","HumanMessage","values: ChainValues","runManager?: CallbackManagerForChainRun","chatHistory: unknown[]","PREDICT_NEXT_USER_MESSAGE_FUNCTION","PREDICT_NEXT_USER_MESSAGE_PROMPT","insights: Array<string>","PREDICTION_VIOLATIONS_FUNCTION","PREDICTION_VIOLATIONS_PROMPT","GENERATE_REVISED_PREDICTION_PROMPT","GENERATE_FACTS_PROMPT","llm: ChatOpenAI","retriever: BaseRetrieverInterface","options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport {\n  GetPredictionViolationsResponse,\n  MessageChunkResult,\n  PREDICTION_VIOLATIONS_FUNCTION,\n  PREDICT_NEXT_USER_MESSAGE_FUNCTION,\n  PredictNextUserMessageResponse,\n} from \"./types.js\";\nimport {\n  GENERATE_FACTS_PROMPT,\n  GENERATE_REVISED_PREDICTION_PROMPT,\n  PREDICTION_VIOLATIONS_PROMPT,\n  PREDICT_NEXT_USER_MESSAGE_PROMPT,\n} from \"./violation_of_expectations_prompt.js\";\n\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n  /**\n   * The retriever to use for retrieving stored\n   * thoughts and insights.\n   */\n  retriever: BaseRetrieverInterface;\n  /**\n   * The LLM to use\n   */\n  llm: ChatOpenAI;\n}\n\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport class ViolationOfExpectationsChain\n  extends BaseChain\n  implements ViolationOfExpectationsChainInput\n{\n  static lc_name() {\n    return \"ViolationOfExpectationsChain\";\n  }\n\n  _chainType(): string {\n    return \"violation_of_expectation_chain\";\n  }\n\n  chatHistoryKey = \"chat_history\";\n\n  thoughtsKey = \"thoughts\";\n\n  get inputKeys() {\n    return [this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return [this.thoughtsKey];\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  llm: ChatOpenAI;\n\n  jsonOutputParser: JsonOutputFunctionsParser;\n\n  stringOutputParser: StringOutputParser;\n\n  constructor(fields: ViolationOfExpectationsChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llm = fields.llm;\n    this.jsonOutputParser = new JsonOutputFunctionsParser();\n    this.stringOutputParser = new StringOutputParser();\n  }\n\n  getChatHistoryString(chatHistory: BaseMessage[]): string {\n    return chatHistory\n      .map((chatMessage) => {\n        if (chatMessage._getType() === \"human\") {\n          return `Human: ${chatMessage.content}`;\n        } else if (chatMessage._getType() === \"ai\") {\n          return `AI: ${chatMessage.content}`;\n        } else {\n          return `${chatMessage.content}`;\n        }\n      })\n      .join(\"\\n\");\n  }\n\n  removeDuplicateStrings(strings: Array<string>): Array<string> {\n    return [...new Set(strings)];\n  }\n\n  /**\n   * This method breaks down the chat history into chunks of messages.\n   * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n   *\n   * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n   *\n   * @description\n   * The method iterates over the chat history and pushes each message into a temporary array.\n   * When it encounters an AI message, it checks for a subsequent user message.\n   * If a user message is found, it is considered as the user response to the AI message.\n   * If no user message is found after the AI message, the user response is undefined.\n   * The method then pushes the chunk (sequence of messages and user response) into the result array.\n   * This process continues until all messages in the chat history have been processed.\n   */\n  chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[] {\n    const newArray: MessageChunkResult[] = [];\n    const tempArray: BaseMessage[] = [];\n\n    chatHistory.forEach((item, index) => {\n      tempArray.push(item);\n      if (item._getType() === \"ai\") {\n        let userResponse: BaseMessage | undefined = chatHistory[index + 1];\n        if (!userResponse || userResponse._getType() !== \"human\") {\n          userResponse = undefined;\n        }\n\n        newArray.push({\n          chunkedMessages: tempArray,\n          userResponse: userResponse\n            ? new HumanMessage(userResponse)\n            : undefined,\n        });\n      }\n    });\n\n    return newArray;\n  }\n\n  /**\n   * This method processes a chat history to generate insights about the user.\n   *\n   * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n   *\n   * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n   *\n   * @description\n   * The method performs the following steps:\n   * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n   * 2. Breaks the chat history into chunks of messages.\n   * 3. For each chunk, it generates an initial prediction for the user's next message.\n   * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n   * 5. For each set of messages, it generates a fact/insight about the user.\n   * The method returns a list of these insights.\n   */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found`);\n    }\n\n    const chatHistory: unknown[] = values[this.chatHistoryKey];\n\n    const isEveryMessageBaseMessage = chatHistory.every((message) =>\n      isBaseMessage(message)\n    );\n    if (!isEveryMessageBaseMessage) {\n      throw new Error(\"Chat history must be an array of BaseMessages\");\n    }\n\n    const messageChunks = this.chunkMessagesByAIResponse(\n      chatHistory as BaseMessage[]\n    );\n\n    // Generate the initial prediction for every user message.\n    const userPredictions = await Promise.all(\n      messageChunks.map(async (chatHistoryChunk) => ({\n        userPredictions: await this.predictNextUserMessage(\n          chatHistoryChunk.chunkedMessages\n        ),\n        userResponse: chatHistoryChunk.userResponse,\n        runManager,\n      }))\n    );\n\n    // Generate insights, and prediction violations for every user message.\n    // This call also regenerates the prediction based on the violations.\n    const predictionViolations = await Promise.all(\n      userPredictions.map((prediction) =>\n        this.getPredictionViolations({\n          userPredictions: prediction.userPredictions,\n          userResponse: prediction.userResponse,\n          runManager,\n        })\n      )\n    );\n\n    // Generate a fact/insight about the user for every set of messages.\n    const insights = await Promise.all(\n      predictionViolations.map((violation) =>\n        this.generateFacts({\n          userResponse: violation.userResponse,\n          predictions: {\n            revisedPrediction: violation.revisedPrediction,\n            explainedPredictionErrors: violation.explainedPredictionErrors,\n          },\n        })\n      )\n    );\n\n    return {\n      insights,\n    };\n  }\n\n  /**\n   * This method predicts the next user message based on the chat history.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history based on which the next user message is predicted.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<PredictNextUserMessageResponse>} A promise that resolves to the predicted next user message, the user state, and any insights.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'userState', 'predictedUserMessage', and 'insights'.\n   */\n  private async predictNextUserMessage(\n    chatHistory: BaseMessage[],\n    runManager?: CallbackManagerForChainRun\n  ): Promise<PredictNextUserMessageResponse> {\n    const messageString = this.getChatHistoryString(chatHistory);\n\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name },\n      });\n\n    const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    const res = await chain.invoke(\n      {\n        chat_history: messageString,\n      },\n      runManager?.getChild(\"prediction\")\n    );\n\n    if (\n      !(\n        \"userState\" in res &&\n        \"predictedUserMessage\" in res &&\n        \"insights\" in res\n      )\n    ) {\n      throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);\n    }\n\n    const predictionResponse = res as PredictNextUserMessageResponse;\n\n    // Query the retriever for relevant insights. Use the generates insights as a query.\n    const retrievedDocs = await this.retrieveRelevantInsights(\n      predictionResponse.insights\n    );\n    const relevantDocs = this.removeDuplicateStrings([\n      ...predictionResponse.insights,\n      ...retrievedDocs,\n    ]);\n\n    return {\n      ...predictionResponse,\n      insights: relevantDocs,\n    };\n  }\n\n  /**\n   * Retrieves relevant insights based on the provided insights.\n   *\n   * @param {Array<string>} insights - An array of insights to be used for retrieving relevant documents.\n   *\n   * @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.\n   */\n  private async retrieveRelevantInsights(\n    insights: Array<string>\n  ): Promise<Array<string>> {\n    // Only extract the first relevant doc from the retriever. We don't need more than one.\n    const relevantInsightsDocuments = await Promise.all(\n      insights.map(async (insight) => {\n        const relevantInsight = await this.retriever.invoke(insight);\n        return relevantInsight[0];\n      })\n    );\n\n    const relevantInsightsContent = relevantInsightsDocuments.map(\n      (document) => document.pageContent\n    );\n\n    return relevantInsightsContent;\n  }\n\n  /**\n   * This method generates prediction violations based on the predicted and actual user responses.\n   * It also generates a revised prediction based on the identified violations.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {PredictNextUserMessageResponse} params.userPredictions - The predicted user message, user state, and insights.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<{ userResponse: BaseMessage | undefined; revisedPrediction: string; explainedPredictionErrors: Array<string>; }>} A promise that resolves to an object containing the actual user response, the revised prediction, and the explained prediction errors.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'violationExplanation', 'explainedPredictionErrors', and 'accuratePrediction'.\n   */\n  private async getPredictionViolations({\n    userPredictions,\n    userResponse,\n    runManager,\n  }: {\n    userPredictions: PredictNextUserMessageResponse;\n    userResponse?: BaseMessage;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<GetPredictionViolationsResponse> {\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICTION_VIOLATIONS_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },\n      });\n\n    const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = (await chain.invoke(\n      {\n        predicted_output: userPredictions.predictedUserMessage,\n        actual_output: userResponse?.content ?? \"\",\n        user_insights: userPredictions.insights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_violations\")\n    )) as Awaited<{\n      violationExplanation: string;\n      explainedPredictionErrors: Array<string>;\n      accuratePrediction: boolean;\n    }>;\n\n    // Generate a revised prediction based on violations.\n    const revisedPrediction = await this.generateRevisedPrediction({\n      originalPrediction: userPredictions.predictedUserMessage,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n      userInsights: userPredictions.insights,\n      runManager,\n    });\n\n    return {\n      userResponse,\n      revisedPrediction,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n    };\n  }\n\n  /**\n   * This method generates a revised prediction based on the original prediction, explained prediction errors, and user insights.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {string} params.originalPrediction - The original prediction made by the model.\n   * @param {Array<string>} params.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {Array<string>} params.userInsights - An array of insights about the user.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a revised prediction.\n   */\n  private async generateRevisedPrediction({\n    originalPrediction,\n    explainedPredictionErrors,\n    userInsights,\n    runManager,\n  }: {\n    originalPrediction: string;\n    explainedPredictionErrors: Array<string>;\n    userInsights: Array<string>;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(\n      this.llm\n    ).pipe(this.stringOutputParser);\n\n    const revisedPredictionRes = await revisedPredictionChain.invoke(\n      {\n        prediction: originalPrediction,\n        explained_prediction_errors: explainedPredictionErrors.join(\"\\n\"),\n        user_insights: userInsights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_revision\")\n    );\n\n    return revisedPredictionRes;\n  }\n\n  /**\n   * This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {Object} params.predictions - The revised prediction and explained prediction errors.\n   * @param {string} params.predictions.revisedPrediction - The revised prediction made by the model.\n   * @param {Array<string>} params.predictions.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a string containing the generated facts or insights about the user.\n   */\n  private async generateFacts({\n    userResponse,\n    predictions,\n    runManager,\n  }: {\n    userResponse?: BaseMessage;\n    /**\n     * Optional if the prediction was accurate.\n     */\n    predictions: {\n      revisedPrediction: string;\n      explainedPredictionErrors: Array<string>;\n    };\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(\n      this.stringOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = await chain.invoke(\n      {\n        prediction_violations: predictions.explainedPredictionErrors.join(\"\\n\"),\n        prediction: predictions.revisedPrediction,\n        user_message: userResponse?.content ?? \"\",\n      },\n      runManager?.getChild(\"generate_facts\")\n    );\n\n    return res;\n  }\n\n  /**\n   * Static method that creates a ViolationOfExpectationsChain instance from a\n   * ChatOpenAI and retriever. It also accepts optional options\n   * to customize the chain.\n   *\n   * @param llm The ChatOpenAI instance.\n   * @param retriever The retriever used for similarity search.\n   * @param options Optional options to customize the chain.\n   *\n   * @returns A new instance of ViolationOfExpectationsChain.\n   */\n  static fromLLM(\n    llm: ChatOpenAI,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >\n  ): ViolationOfExpectationsChain {\n    return new this({\n      retriever,\n      llm,\n      ...options,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AA6CA,IAAa,+BAAb,cACUA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED,aAAqB;AACnB,SAAO;CACR;CAED,iBAAiB;CAEjB,cAAc;CAEd,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,cAAe;CAC7B;CAED,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,WAAY;CAC1B;CAED;CAEA;CAEA;CAEA;CAEA,YAAYC,QAA2C;EACrD,MAAM,OAAO;EACb,KAAK,YAAY,OAAO;EACxB,KAAK,MAAM,OAAO;EAClB,KAAK,mBAAmB,IAAIC;EAC5B,KAAK,qBAAqB,IAAIC;CAC/B;CAED,qBAAqBC,aAAoC;AACvD,SAAO,YACJ,IAAI,CAAC,gBAAgB;AACpB,OAAI,YAAY,UAAU,KAAK,QAC7B,QAAO,CAAC,OAAO,EAAE,YAAY,SAAS;YAC7B,YAAY,UAAU,KAAK,KACpC,QAAO,CAAC,IAAI,EAAE,YAAY,SAAS;OAEnC,QAAO,GAAG,YAAY,SAAS;EAElC,EAAC,CACD,KAAK,KAAK;CACd;CAED,uBAAuBC,SAAuC;AAC5D,SAAO,CAAC,GAAG,IAAI,IAAI,QAAS;CAC7B;;;;;;;;;;;;;;;;;CAkBD,0BAA0BD,aAAkD;EAC1E,MAAME,WAAiC,CAAE;EACzC,MAAMC,YAA2B,CAAE;EAEnC,YAAY,QAAQ,CAAC,MAAM,UAAU;GACnC,UAAU,KAAK,KAAK;AACpB,OAAI,KAAK,UAAU,KAAK,MAAM;IAC5B,IAAIC,eAAwC,YAAY,QAAQ;AAChE,QAAI,CAAC,gBAAgB,aAAa,UAAU,KAAK,SAC/C,eAAe;IAGjB,SAAS,KAAK;KACZ,iBAAiB;KACjB,cAAc,eACV,IAAIC,uCAAa,gBACjB;IACL,EAAC;GACH;EACF,EAAC;AAEF,SAAO;CACR;;;;;;;;;;;;;;;;;;;;CAqBD,MAAM,MACJC,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,CAAC,iBAAiB,EAAE,KAAK,eAAe,UAAU,CAAC;EAGrE,MAAMC,cAAyB,OAAO,KAAK;EAE3C,MAAM,4BAA4B,YAAY,MAAM,CAAC,yDACrC,QAAQ,CACvB;AACD,MAAI,CAAC,0BACH,OAAM,IAAI,MAAM;EAGlB,MAAM,gBAAgB,KAAK,0BACzB,YACD;EAGD,MAAM,kBAAkB,MAAM,QAAQ,IACpC,cAAc,IAAI,OAAO,sBAAsB;GAC7C,iBAAiB,MAAM,KAAK,uBAC1B,iBAAiB,gBAClB;GACD,cAAc,iBAAiB;GAC/B;EACD,GAAE,CACJ;EAID,MAAM,uBAAuB,MAAM,QAAQ,IACzC,gBAAgB,IAAI,CAAC,eACnB,KAAK,wBAAwB;GAC3B,iBAAiB,WAAW;GAC5B,cAAc,WAAW;GACzB;EACD,EAAC,CACH,CACF;EAGD,MAAM,WAAW,MAAM,QAAQ,IAC7B,qBAAqB,IAAI,CAAC,cACxB,KAAK,cAAc;GACjB,cAAc,UAAU;GACxB,aAAa;IACX,mBAAmB,UAAU;IAC7B,2BAA2B,UAAU;GACtC;EACF,EAAC,CACH,CACF;AAED,SAAO,EACL,SACD;CACF;;;;;;;;;;;CAYD,MAAc,uBACZR,aACAO,YACyC;EACzC,MAAM,gBAAgB,KAAK,qBAAqB,YAAY;EAE5D,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACE,gDAAmC,EAAC,CAC/C,WAAW,EACV,eAAe,EAAE,MAAMA,iDAAmC,KAAM,EACjE,EAAC;EAEJ,MAAM,QAAQC,0EAAiC,KAAK,iBAAiB,CAAC,KACpE,KAAK,iBACN;EAED,MAAM,MAAM,MAAM,MAAM,OACtB,EACE,cAAc,cACf,GACD,YAAY,SAAS,aAAa,CACnC;AAED,MACE,EACE,eAAe,OACf,0BAA0B,OAC1B,cAAc,KAGhB,OAAM,IAAI,MAAM,CAAC,2BAA2B,EAAE,KAAK,UAAU,IAAI,EAAE;EAGrE,MAAM,qBAAqB;EAG3B,MAAM,gBAAgB,MAAM,KAAK,yBAC/B,mBAAmB,SACpB;EACD,MAAM,eAAe,KAAK,uBAAuB,CAC/C,GAAG,mBAAmB,UACtB,GAAG,aACJ,EAAC;AAEF,SAAO;GACL,GAAG;GACH,UAAU;EACX;CACF;;;;;;;;CASD,MAAc,yBACZC,UACwB;EAExB,MAAM,4BAA4B,MAAM,QAAQ,IAC9C,SAAS,IAAI,OAAO,YAAY;GAC9B,MAAM,kBAAkB,MAAM,KAAK,UAAU,OAAO,QAAQ;AAC5D,UAAO,gBAAgB;EACxB,EAAC,CACH;EAED,MAAM,0BAA0B,0BAA0B,IACxD,CAAC,aAAa,SAAS,YACxB;AAED,SAAO;CACR;;;;;;;;;;;;;;CAeD,MAAc,wBAAwB,EACpC,iBACA,cACA,YAKD,EAA4C;EAC3C,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,4CAA+B,EAAC,CAC3C,WAAW,EACV,eAAe,EAAE,MAAMA,6CAA+B,KAAM,EAC7D,EAAC;EAEJ,MAAM,QAAQC,sEAA6B,KAAK,iBAAiB,CAAC,KAChE,KAAK,iBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM;EAElB,MAAM,MAAO,MAAM,MAAM,OACvB;GACE,kBAAkB,gBAAgB;GAClC,eAAe,cAAc,WAAW;GACxC,eAAe,gBAAgB,SAAS,KAAK,KAAK;EACnD,GACD,YAAY,SAAS,wBAAwB,CAC9C;EAOD,MAAM,oBAAoB,MAAM,KAAK,0BAA0B;GAC7D,oBAAoB,gBAAgB;GACpC,2BAA2B,IAAI;GAC/B,cAAc,gBAAgB;GAC9B;EACD,EAAC;AAEF,SAAO;GACL;GACA;GACA,2BAA2B,IAAI;EAChC;CACF;;;;;;;;;;;;CAaD,MAAc,0BAA0B,EACtC,oBACA,2BACA,cACA,YAMD,EAAmB;EAClB,MAAM,yBAAyBC,4EAAmC,KAChE,KAAK,IACN,CAAC,KAAK,KAAK,mBAAmB;EAE/B,MAAM,uBAAuB,MAAM,uBAAuB,OACxD;GACE,YAAY;GACZ,6BAA6B,0BAA0B,KAAK,KAAK;GACjE,eAAe,aAAa,KAAK,KAAK;EACvC,GACD,YAAY,SAAS,sBAAsB,CAC5C;AAED,SAAO;CACR;;;;;;;;;;;;;CAcD,MAAc,cAAc,EAC1B,cACA,aACA,YAWD,EAAmB;EAClB,MAAM,QAAQC,+DAAsB,KAAK,KAAK,IAAI,CAAC,KACjD,KAAK,mBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM;EAElB,MAAM,MAAM,MAAM,MAAM,OACtB;GACE,uBAAuB,YAAY,0BAA0B,KAAK,KAAK;GACvE,YAAY,YAAY;GACxB,cAAc,cAAc,WAAW;EACxC,GACD,YAAY,SAAS,iBAAiB,CACvC;AAED,SAAO;CACR;;;;;;;;;;;;CAaD,OAAO,QACLC,KACAC,WACAC,SAG8B;AAC9B,SAAO,IAAI,KAAK;GACd;GACA;GACA,GAAG;EACJ;CACF;AACF"}
+{"version":3,"file":"violation_of_expectations_chain.cjs","names":["BaseChain","JsonOutputFunctionsParser","StringOutputParser","HumanMessage","PREDICT_NEXT_USER_MESSAGE_FUNCTION","PREDICT_NEXT_USER_MESSAGE_PROMPT","PREDICTION_VIOLATIONS_FUNCTION","PREDICTION_VIOLATIONS_PROMPT","GENERATE_REVISED_PREDICTION_PROMPT","GENERATE_FACTS_PROMPT"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport {\n  GetPredictionViolationsResponse,\n  MessageChunkResult,\n  PREDICTION_VIOLATIONS_FUNCTION,\n  PREDICT_NEXT_USER_MESSAGE_FUNCTION,\n  PredictNextUserMessageResponse,\n} from \"./types.js\";\nimport {\n  GENERATE_FACTS_PROMPT,\n  GENERATE_REVISED_PREDICTION_PROMPT,\n  PREDICTION_VIOLATIONS_PROMPT,\n  PREDICT_NEXT_USER_MESSAGE_PROMPT,\n} from \"./violation_of_expectations_prompt.js\";\n\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n  /**\n   * The retriever to use for retrieving stored\n   * thoughts and insights.\n   */\n  retriever: BaseRetrieverInterface;\n  /**\n   * The LLM to use\n   */\n  llm: ChatOpenAI;\n}\n\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport class ViolationOfExpectationsChain\n  extends BaseChain\n  implements ViolationOfExpectationsChainInput\n{\n  static lc_name() {\n    return \"ViolationOfExpectationsChain\";\n  }\n\n  _chainType(): string {\n    return \"violation_of_expectation_chain\";\n  }\n\n  chatHistoryKey = \"chat_history\";\n\n  thoughtsKey = \"thoughts\";\n\n  get inputKeys() {\n    return [this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return [this.thoughtsKey];\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  llm: ChatOpenAI;\n\n  jsonOutputParser: JsonOutputFunctionsParser;\n\n  stringOutputParser: StringOutputParser;\n\n  constructor(fields: ViolationOfExpectationsChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llm = fields.llm;\n    this.jsonOutputParser = new JsonOutputFunctionsParser();\n    this.stringOutputParser = new StringOutputParser();\n  }\n\n  getChatHistoryString(chatHistory: BaseMessage[]): string {\n    return chatHistory\n      .map((chatMessage) => {\n        if (chatMessage._getType() === \"human\") {\n          return `Human: ${chatMessage.content}`;\n        } else if (chatMessage._getType() === \"ai\") {\n          return `AI: ${chatMessage.content}`;\n        } else {\n          return `${chatMessage.content}`;\n        }\n      })\n      .join(\"\\n\");\n  }\n\n  removeDuplicateStrings(strings: Array<string>): Array<string> {\n    return [...new Set(strings)];\n  }\n\n  /**\n   * This method breaks down the chat history into chunks of messages.\n   * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n   *\n   * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n   *\n   * @description\n   * The method iterates over the chat history and pushes each message into a temporary array.\n   * When it encounters an AI message, it checks for a subsequent user message.\n   * If a user message is found, it is considered as the user response to the AI message.\n   * If no user message is found after the AI message, the user response is undefined.\n   * The method then pushes the chunk (sequence of messages and user response) into the result array.\n   * This process continues until all messages in the chat history have been processed.\n   */\n  chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[] {\n    const newArray: MessageChunkResult[] = [];\n    const tempArray: BaseMessage[] = [];\n\n    chatHistory.forEach((item, index) => {\n      tempArray.push(item);\n      if (item._getType() === \"ai\") {\n        let userResponse: BaseMessage | undefined = chatHistory[index + 1];\n        if (!userResponse || userResponse._getType() !== \"human\") {\n          userResponse = undefined;\n        }\n\n        newArray.push({\n          chunkedMessages: tempArray,\n          userResponse: userResponse\n            ? new HumanMessage(userResponse)\n            : undefined,\n        });\n      }\n    });\n\n    return newArray;\n  }\n\n  /**\n   * This method processes a chat history to generate insights about the user.\n   *\n   * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n   *\n   * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n   *\n   * @description\n   * The method performs the following steps:\n   * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n   * 2. Breaks the chat history into chunks of messages.\n   * 3. For each chunk, it generates an initial prediction for the user's next message.\n   * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n   * 5. For each set of messages, it generates a fact/insight about the user.\n   * The method returns a list of these insights.\n   */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found`);\n    }\n\n    const chatHistory: unknown[] = values[this.chatHistoryKey];\n\n    const isEveryMessageBaseMessage = chatHistory.every((message) =>\n      isBaseMessage(message)\n    );\n    if (!isEveryMessageBaseMessage) {\n      throw new Error(\"Chat history must be an array of BaseMessages\");\n    }\n\n    const messageChunks = this.chunkMessagesByAIResponse(\n      chatHistory as BaseMessage[]\n    );\n\n    // Generate the initial prediction for every user message.\n    const userPredictions = await Promise.all(\n      messageChunks.map(async (chatHistoryChunk) => ({\n        userPredictions: await this.predictNextUserMessage(\n          chatHistoryChunk.chunkedMessages\n        ),\n        userResponse: chatHistoryChunk.userResponse,\n        runManager,\n      }))\n    );\n\n    // Generate insights, and prediction violations for every user message.\n    // This call also regenerates the prediction based on the violations.\n    const predictionViolations = await Promise.all(\n      userPredictions.map((prediction) =>\n        this.getPredictionViolations({\n          userPredictions: prediction.userPredictions,\n          userResponse: prediction.userResponse,\n          runManager,\n        })\n      )\n    );\n\n    // Generate a fact/insight about the user for every set of messages.\n    const insights = await Promise.all(\n      predictionViolations.map((violation) =>\n        this.generateFacts({\n          userResponse: violation.userResponse,\n          predictions: {\n            revisedPrediction: violation.revisedPrediction,\n            explainedPredictionErrors: violation.explainedPredictionErrors,\n          },\n        })\n      )\n    );\n\n    return {\n      insights,\n    };\n  }\n\n  /**\n   * This method predicts the next user message based on the chat history.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history based on which the next user message is predicted.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<PredictNextUserMessageResponse>} A promise that resolves to the predicted next user message, the user state, and any insights.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'userState', 'predictedUserMessage', and 'insights'.\n   */\n  private async predictNextUserMessage(\n    chatHistory: BaseMessage[],\n    runManager?: CallbackManagerForChainRun\n  ): Promise<PredictNextUserMessageResponse> {\n    const messageString = this.getChatHistoryString(chatHistory);\n\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name },\n      });\n\n    const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    const res = await chain.invoke(\n      {\n        chat_history: messageString,\n      },\n      runManager?.getChild(\"prediction\")\n    );\n\n    if (\n      !(\n        \"userState\" in res &&\n        \"predictedUserMessage\" in res &&\n        \"insights\" in res\n      )\n    ) {\n      throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);\n    }\n\n    const predictionResponse = res as PredictNextUserMessageResponse;\n\n    // Query the retriever for relevant insights. Use the generates insights as a query.\n    const retrievedDocs = await this.retrieveRelevantInsights(\n      predictionResponse.insights\n    );\n    const relevantDocs = this.removeDuplicateStrings([\n      ...predictionResponse.insights,\n      ...retrievedDocs,\n    ]);\n\n    return {\n      ...predictionResponse,\n      insights: relevantDocs,\n    };\n  }\n\n  /**\n   * Retrieves relevant insights based on the provided insights.\n   *\n   * @param {Array<string>} insights - An array of insights to be used for retrieving relevant documents.\n   *\n   * @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.\n   */\n  private async retrieveRelevantInsights(\n    insights: Array<string>\n  ): Promise<Array<string>> {\n    // Only extract the first relevant doc from the retriever. We don't need more than one.\n    const relevantInsightsDocuments = await Promise.all(\n      insights.map(async (insight) => {\n        const relevantInsight = await this.retriever.invoke(insight);\n        return relevantInsight[0];\n      })\n    );\n\n    const relevantInsightsContent = relevantInsightsDocuments.map(\n      (document) => document.pageContent\n    );\n\n    return relevantInsightsContent;\n  }\n\n  /**\n   * This method generates prediction violations based on the predicted and actual user responses.\n   * It also generates a revised prediction based on the identified violations.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {PredictNextUserMessageResponse} params.userPredictions - The predicted user message, user state, and insights.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<{ userResponse: BaseMessage | undefined; revisedPrediction: string; explainedPredictionErrors: Array<string>; }>} A promise that resolves to an object containing the actual user response, the revised prediction, and the explained prediction errors.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'violationExplanation', 'explainedPredictionErrors', and 'accuratePrediction'.\n   */\n  private async getPredictionViolations({\n    userPredictions,\n    userResponse,\n    runManager,\n  }: {\n    userPredictions: PredictNextUserMessageResponse;\n    userResponse?: BaseMessage;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<GetPredictionViolationsResponse> {\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICTION_VIOLATIONS_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },\n      });\n\n    const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = (await chain.invoke(\n      {\n        predicted_output: userPredictions.predictedUserMessage,\n        actual_output: userResponse?.content ?? \"\",\n        user_insights: userPredictions.insights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_violations\")\n    )) as Awaited<{\n      violationExplanation: string;\n      explainedPredictionErrors: Array<string>;\n      accuratePrediction: boolean;\n    }>;\n\n    // Generate a revised prediction based on violations.\n    const revisedPrediction = await this.generateRevisedPrediction({\n      originalPrediction: userPredictions.predictedUserMessage,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n      userInsights: userPredictions.insights,\n      runManager,\n    });\n\n    return {\n      userResponse,\n      revisedPrediction,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n    };\n  }\n\n  /**\n   * This method generates a revised prediction based on the original prediction, explained prediction errors, and user insights.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {string} params.originalPrediction - The original prediction made by the model.\n   * @param {Array<string>} params.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {Array<string>} params.userInsights - An array of insights about the user.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a revised prediction.\n   */\n  private async generateRevisedPrediction({\n    originalPrediction,\n    explainedPredictionErrors,\n    userInsights,\n    runManager,\n  }: {\n    originalPrediction: string;\n    explainedPredictionErrors: Array<string>;\n    userInsights: Array<string>;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(\n      this.llm\n    ).pipe(this.stringOutputParser);\n\n    const revisedPredictionRes = await revisedPredictionChain.invoke(\n      {\n        prediction: originalPrediction,\n        explained_prediction_errors: explainedPredictionErrors.join(\"\\n\"),\n        user_insights: userInsights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_revision\")\n    );\n\n    return revisedPredictionRes;\n  }\n\n  /**\n   * This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {Object} params.predictions - The revised prediction and explained prediction errors.\n   * @param {string} params.predictions.revisedPrediction - The revised prediction made by the model.\n   * @param {Array<string>} params.predictions.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a string containing the generated facts or insights about the user.\n   */\n  private async generateFacts({\n    userResponse,\n    predictions,\n    runManager,\n  }: {\n    userResponse?: BaseMessage;\n    /**\n     * Optional if the prediction was accurate.\n     */\n    predictions: {\n      revisedPrediction: string;\n      explainedPredictionErrors: Array<string>;\n    };\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(\n      this.stringOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = await chain.invoke(\n      {\n        prediction_violations: predictions.explainedPredictionErrors.join(\"\\n\"),\n        prediction: predictions.revisedPrediction,\n        user_message: userResponse?.content ?? \"\",\n      },\n      runManager?.getChild(\"generate_facts\")\n    );\n\n    return res;\n  }\n\n  /**\n   * Static method that creates a ViolationOfExpectationsChain instance from a\n   * ChatOpenAI and retriever. It also accepts optional options\n   * to customize the chain.\n   *\n   * @param llm The ChatOpenAI instance.\n   * @param retriever The retriever used for similarity search.\n   * @param options Optional options to customize the chain.\n   *\n   * @returns A new instance of ViolationOfExpectationsChain.\n   */\n  static fromLLM(\n    llm: ChatOpenAI,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >\n  ): ViolationOfExpectationsChain {\n    return new this({\n      retriever,\n      llm,\n      ...options,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AA6CA,IAAa,+BAAb,cACUA,uBAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,aAAqB;AACnB,SAAO;;CAGT,iBAAiB;CAEjB,cAAc;CAEd,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,eAAe;;CAG9B,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,YAAY;;CAG3B;CAEA;CAEA;CAEA;CAEA,YAAY,QAA2C;AACrD,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,MAAM,OAAO;AAClB,OAAK,mBAAmB,IAAIC,oDAA2B;AACvD,OAAK,qBAAqB,IAAIC,mDAAoB;;CAGpD,qBAAqB,aAAoC;AACvD,SAAO,YACJ,KAAK,gBAAgB;AACpB,OAAI,YAAY,UAAU,KAAK,QAC7B,QAAO,UAAU,YAAY;YACpB,YAAY,UAAU,KAAK,KACpC,QAAO,OAAO,YAAY;OAE1B,QAAO,GAAG,YAAY;IAExB,CACD,KAAK,KAAK;;CAGf,uBAAuB,SAAuC;AAC5D,SAAO,CAAC,GAAG,IAAI,IAAI,QAAQ,CAAC;;;;;;;;;;;;;;;;;;CAmB9B,0BAA0B,aAAkD;EAC1E,MAAM,WAAiC,EAAE;EACzC,MAAM,YAA2B,EAAE;AAEnC,cAAY,SAAS,MAAM,UAAU;AACnC,aAAU,KAAK,KAAK;AACpB,OAAI,KAAK,UAAU,KAAK,MAAM;IAC5B,IAAI,eAAwC,YAAY,QAAQ;AAChE,QAAI,CAAC,gBAAgB,aAAa,UAAU,KAAK,QAC/C,gBAAe;AAGjB,aAAS,KAAK;KACZ,iBAAiB;KACjB,cAAc,eACV,IAAIC,sCAAa,aAAa,GAC9B;KACL,CAAC;;IAEJ;AAEF,SAAO;;;;;;;;;;;;;;;;;;;;;CAsBT,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,eAAe,YAAY;EAGtE,MAAM,cAAyB,OAAO,KAAK;AAK3C,MAAI,CAH8B,YAAY,OAAO,wDACrC,QAAQ,CACvB,CAEC,OAAM,IAAI,MAAM,gDAAgD;EAGlE,MAAM,gBAAgB,KAAK,0BACzB,YACD;EAGD,MAAM,kBAAkB,MAAM,QAAQ,IACpC,cAAc,IAAI,OAAO,sBAAsB;GAC7C,iBAAiB,MAAM,KAAK,uBAC1B,iBAAiB,gBAClB;GACD,cAAc,iBAAiB;GAC/B;GACD,EAAE,CACJ;EAID,MAAM,uBAAuB,MAAM,QAAQ,IACzC,gBAAgB,KAAK,eACnB,KAAK,wBAAwB;GAC3B,iBAAiB,WAAW;GAC5B,cAAc,WAAW;GACzB;GACD,CAAC,CACH,CACF;AAeD,SAAO,EACL,UAbe,MAAM,QAAQ,IAC7B,qBAAqB,KAAK,cACxB,KAAK,cAAc;GACjB,cAAc,UAAU;GACxB,aAAa;IACX,mBAAmB,UAAU;IAC7B,2BAA2B,UAAU;IACtC;GACF,CAAC,CACH,CACF,EAIA;;;;;;;;;;;;CAaH,MAAc,uBACZ,aACA,YACyC;EACzC,MAAM,gBAAgB,KAAK,qBAAqB,YAAY;EAE5D,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,iDAAmC,CAAC,CAC/C,WAAW,EACV,eAAe,EAAE,MAAMA,iDAAmC,MAAM,EACjE,CAAC;EAMJ,MAAM,MAAM,MAJEC,0EAAiC,KAAK,iBAAiB,CAAC,KACpE,KAAK,iBACN,CAEuB,OACtB,EACE,cAAc,eACf,EACD,YAAY,SAAS,aAAa,CACnC;AAED,MACE,EACE,eAAe,OACf,0BAA0B,OAC1B,cAAc,KAGhB,OAAM,IAAI,MAAM,8BAA8B,KAAK,UAAU,IAAI,GAAG;EAGtE,MAAM,qBAAqB;EAG3B,MAAM,gBAAgB,MAAM,KAAK,yBAC/B,mBAAmB,SACpB;EACD,MAAM,eAAe,KAAK,uBAAuB,CAC/C,GAAG,mBAAmB,UACtB,GAAG,cACJ,CAAC;AAEF,SAAO;GACL,GAAG;GACH,UAAU;GACX;;;;;;;;;CAUH,MAAc,yBACZ,UACwB;AAaxB,UAXkC,MAAM,QAAQ,IAC9C,SAAS,IAAI,OAAO,YAAY;AAE9B,WADwB,MAAM,KAAK,UAAU,OAAO,QAAQ,EACrC;IACvB,CACH,EAEyD,KACvD,aAAa,SAAS,YACxB;;;;;;;;;;;;;;;CAkBH,MAAc,wBAAwB,EACpC,iBACA,cACA,cAK2C;EAC3C,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAACC,6CAA+B,CAAC,CAC3C,WAAW,EACV,eAAe,EAAE,MAAMA,6CAA+B,MAAM,EAC7D,CAAC;EAEJ,MAAM,QAAQC,sEAA6B,KAAK,iBAAiB,CAAC,KAChE,KAAK,iBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM,uDAAuD;EAEzE,MAAM,MAAO,MAAM,MAAM,OACvB;GACE,kBAAkB,gBAAgB;GAClC,eAAe,cAAc,WAAW;GACxC,eAAe,gBAAgB,SAAS,KAAK,KAAK;GACnD,EACD,YAAY,SAAS,wBAAwB,CAC9C;AAcD,SAAO;GACL;GACA,mBATwB,MAAM,KAAK,0BAA0B;IAC7D,oBAAoB,gBAAgB;IACpC,2BAA2B,IAAI;IAC/B,cAAc,gBAAgB;IAC9B;IACD,CAAC;GAKA,2BAA2B,IAAI;GAChC;;;;;;;;;;;;;CAcH,MAAc,0BAA0B,EACtC,oBACA,2BACA,cACA,cAMkB;AAclB,SAT6B,MAJEC,4EAAmC,KAChE,KAAK,IACN,CAAC,KAAK,KAAK,mBAAmB,CAE2B,OACxD;GACE,YAAY;GACZ,6BAA6B,0BAA0B,KAAK,KAAK;GACjE,eAAe,aAAa,KAAK,KAAK;GACvC,EACD,YAAY,SAAS,sBAAsB,CAC5C;;;;;;;;;;;;;;CAiBH,MAAc,cAAc,EAC1B,cACA,aACA,cAWkB;EAClB,MAAM,QAAQC,+DAAsB,KAAK,KAAK,IAAI,CAAC,KACjD,KAAK,mBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM,uDAAuD;AAWzE,SATY,MAAM,MAAM,OACtB;GACE,uBAAuB,YAAY,0BAA0B,KAAK,KAAK;GACvE,YAAY,YAAY;GACxB,cAAc,cAAc,WAAW;GACxC,EACD,YAAY,SAAS,iBAAiB,CACvC;;;;;;;;;;;;;CAgBH,OAAO,QACL,KACA,WACA,SAG8B;AAC9B,SAAO,IAAI,KAAK;GACd;GACA;GACA,GAAG;GACJ,CAAC"}

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.cts

@@ -9,7 +9,6 @@ import { BaseRetrieverInterface } from "@langchain/core/retrievers";
 import { ChatOpenAI } from "@langchain/openai";
 
 //#region src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.ts
-
 /**
  * Interface for the input parameters of the ViolationOfExpectationsChain class.
  */

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"violation_of_expectations_chain.d.cts","names":["BaseRetrieverInterface","ChatOpenAI","BaseMessage","ChainValues","StringOutputParser","CallbackManagerForChainRun","JsonOutputFunctionsParser","BaseChain","ChainInputs","MessageChunkResult","ViolationOfExpectationsChainInput","ViolationOfExpectationsChain","Array","Promise","Omit","Partial"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport { MessageChunkResult } from \"./types.js\";\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n    /**\n     * The retriever to use for retrieving stored\n     * thoughts and insights.\n     */\n    retriever: BaseRetrieverInterface;\n    /**\n     * The LLM to use\n     */\n    llm: ChatOpenAI;\n}\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport declare class ViolationOfExpectationsChain extends BaseChain implements ViolationOfExpectationsChainInput {\n    static lc_name(): string;\n    _chainType(): string;\n    chatHistoryKey: string;\n    thoughtsKey: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    retriever: BaseRetrieverInterface;\n    llm: ChatOpenAI;\n    jsonOutputParser: JsonOutputFunctionsParser;\n    stringOutputParser: StringOutputParser;\n    constructor(fields: ViolationOfExpectationsChainInput);\n    getChatHistoryString(chatHistory: BaseMessage[]): string;\n    removeDuplicateStrings(strings: Array<string>): Array<string>;\n    /**\n     * This method breaks down the chat history into chunks of messages.\n     * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n     *\n     * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n     *\n     * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n     *\n     * @description\n     * The method iterates over the chat history and pushes each message into a temporary array.\n     * When it encounters an AI message, it checks for a subsequent user message.\n     * If a user message is found, it is considered as the user response to the AI message.\n     * If no user message is found after the AI message, the user response is undefined.\n     * The method then pushes the chunk (sequence of messages and user response) into the result array.\n     * This process continues until all messages in the chat history have been processed.\n     */\n    chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[];\n    /**\n     * This method processes a chat history to generate insights about the user.\n     *\n     * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n     * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n     *\n     * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n     *\n     * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n     *\n     * @description\n     * The method performs the following steps:\n     * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n     * 2. Breaks the chat history into chunks of messages.\n     * 3. For each chunk, it generates an initial prediction for the user's next message.\n     * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n     * 5. For each set of messages, it generates a fact/insight about the user.\n     * The method returns a list of these insights.\n     */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    private predictNextUserMessage;\n    private retrieveRelevantInsights;\n    private getPredictionViolations;\n    private generateRevisedPrediction;\n    private generateFacts;\n    /**\n     * Static method that creates a ViolationOfExpectationsChain instance from a\n     * ChatOpenAI and retriever. It also accepts optional options\n     * to customize the chain.\n     *\n     * @param llm The ChatOpenAI instance.\n     * @param retriever The retriever used for similarity search.\n     * @param options Optional options to customize the chain.\n     *\n     * @returns A new instance of ViolationOfExpectationsChain.\n     */\n    static fromLLM(llm: ChatOpenAI, retriever: BaseRetrieverInterface, options?: Partial<Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">>): ViolationOfExpectationsChain;\n}\n//# sourceMappingURL=violation_of_expectations_chain.d.ts.map"],"mappings":";;;;;;;;;;;;;;AAYA;AAKeA,UALEU,iCAAAA,SAA0CF,WAK5CR,CAAAA;EAINC;;AAT6D;AAetE;EAOeD,SAAAA,EAjBAA,sBAiBAA;EACNC;;;EAGeS,GAAAA,EAjBfT,UAiBeS;;;;;;AAuCNP,cAlDGQ,4BAAAA,SAAqCJ,SAAAA,YAAqBG,iCAkD7DP,CAAAA;EAA0BE,OAAAA,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAqCF,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAARU,cAAAA,EAAAA,MAAAA;EAiBjDZ,WAAAA,EAAAA,MAAAA;EAAuBD,IAAAA,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAA+CU,IAAAA,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAALI,SAAAA,EA5D1Ed,sBA4D0Ec;EAARC,GAAAA,EA3DxEd,UA2DwEc;EAAwEJ,gBAAAA,EA1DnIL,yBA0DmIK;EAnE/FJ,kBAAAA,EAUlCH,kBAVkCG;EAAqBG,WAAAA,CAAAA,MAAAA,EAWvDA,iCAXuDA;EAAiC,oBAAA,CAAA,WAAA,EAY1ER,WAZ0E,EAAA,CAAA,EAAA,MAAA;kCAa5EU,gBAAgBA;;;;;;;;;;;;;;;;;yCAiBTV,gBAAgBO;;;;;;;;;;;;;;;;;;;;gBAoBzCN,0BAA0BE,6BAA6BQ,QAAQV;;;;;;;;;;;;;;;;;sBAiBzDF,uBAAuBD,kCAAkCe,QAAQD,KAAKJ,2DAA2DC"}
+{"version":3,"file":"violation_of_expectations_chain.d.cts","names":[],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"mappings":";;;;;;;;;;;;;AA6BA;UAAiB,iCAAA,SAA0C,WAAA;;;;;EAKzD,SAAA,EAAW,sBAAA;EALyD;;;EASpE,GAAA,EAAK,UAAA;AAAA;;;;AAOP;cAAa,4BAAA,SACH,SAAA,YACG,iCAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,UAAA,CAAA;EAIA,cAAA;EAEA,WAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,SAAA,EAAW,sBAAA;EAEX,GAAA,EAAK,UAAA;EAEL,gBAAA,EAAkB,yBAAA;EAElB,kBAAA,EAAoB,kBAAA;EAEpB,WAAA,CAAY,MAAA,EAAQ,iCAAA;EAQpB,oBAAA,CAAqB,WAAA,EAAa,WAAA;EAclC,sBAAA,CAAuB,OAAA,EAAS,KAAA,WAAgB,KAAA;EAkErC;;;;;;;;;;;;;;;;EA9CX,yBAAA,CAA0B,WAAA,EAAa,WAAA,KAAgB,kBAAA;EA5DvD;;;;;;;;;;;;;;;;;;;EAuGM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAAA,QAqEG,sBAAA;EAAA,QAyDA,wBAAA;EAAA,QA+BA,uBAAA;EAAA,QA6DA,yBAAA;EAAA,QAuCA,aAAA;EApQR;;;;;;;;;;;EAAA,OAiTC,OAAA,CACL,GAAA,EAAK,UAAA,EACL,SAAA,EAAW,sBAAA,EACX,OAAA,GAAU,OAAA,CACR,IAAA,CAAK,iCAAA,0BAEN,4BAAA;AAAA"}

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.ts

@@ -1,15 +1,14 @@
 import { BaseChain, ChainInputs } from "../../../chains/base.js";
 import { JsonOutputFunctionsParser } from "../../../output_parsers/openai_functions.js";
 import { MessageChunkResult } from "./types.js";
+import { BaseMessage } from "@langchain/core/messages";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
 import { StringOutputParser } from "@langchain/core/output_parsers";
-import { BaseMessage } from "@langchain/core/messages";
 import { ChainValues } from "@langchain/core/utils/types";
 import { ChatOpenAI } from "@langchain/openai";
 import { BaseRetrieverInterface } from "@langchain/core/retrievers";
 
 //#region src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.ts
-
 /**
  * Interface for the input parameters of the ViolationOfExpectationsChain class.
  */

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"violation_of_expectations_chain.d.ts","names":["BaseRetrieverInterface","ChatOpenAI","BaseMessage","ChainValues","StringOutputParser","CallbackManagerForChainRun","JsonOutputFunctionsParser","BaseChain","ChainInputs","MessageChunkResult","ViolationOfExpectationsChainInput","ViolationOfExpectationsChain","Array","Promise","Omit","Partial"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.d.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport { MessageChunkResult } from \"./types.js\";\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n    /**\n     * The retriever to use for retrieving stored\n     * thoughts and insights.\n     */\n    retriever: BaseRetrieverInterface;\n    /**\n     * The LLM to use\n     */\n    llm: ChatOpenAI;\n}\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport declare class ViolationOfExpectationsChain extends BaseChain implements ViolationOfExpectationsChainInput {\n    static lc_name(): string;\n    _chainType(): string;\n    chatHistoryKey: string;\n    thoughtsKey: string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    retriever: BaseRetrieverInterface;\n    llm: ChatOpenAI;\n    jsonOutputParser: JsonOutputFunctionsParser;\n    stringOutputParser: StringOutputParser;\n    constructor(fields: ViolationOfExpectationsChainInput);\n    getChatHistoryString(chatHistory: BaseMessage[]): string;\n    removeDuplicateStrings(strings: Array<string>): Array<string>;\n    /**\n     * This method breaks down the chat history into chunks of messages.\n     * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n     *\n     * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n     *\n     * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n     *\n     * @description\n     * The method iterates over the chat history and pushes each message into a temporary array.\n     * When it encounters an AI message, it checks for a subsequent user message.\n     * If a user message is found, it is considered as the user response to the AI message.\n     * If no user message is found after the AI message, the user response is undefined.\n     * The method then pushes the chunk (sequence of messages and user response) into the result array.\n     * This process continues until all messages in the chat history have been processed.\n     */\n    chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[];\n    /**\n     * This method processes a chat history to generate insights about the user.\n     *\n     * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n     * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n     *\n     * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n     *\n     * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n     *\n     * @description\n     * The method performs the following steps:\n     * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n     * 2. Breaks the chat history into chunks of messages.\n     * 3. For each chunk, it generates an initial prediction for the user's next message.\n     * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n     * 5. For each set of messages, it generates a fact/insight about the user.\n     * The method returns a list of these insights.\n     */\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<ChainValues>;\n    private predictNextUserMessage;\n    private retrieveRelevantInsights;\n    private getPredictionViolations;\n    private generateRevisedPrediction;\n    private generateFacts;\n    /**\n     * Static method that creates a ViolationOfExpectationsChain instance from a\n     * ChatOpenAI and retriever. It also accepts optional options\n     * to customize the chain.\n     *\n     * @param llm The ChatOpenAI instance.\n     * @param retriever The retriever used for similarity search.\n     * @param options Optional options to customize the chain.\n     *\n     * @returns A new instance of ViolationOfExpectationsChain.\n     */\n    static fromLLM(llm: ChatOpenAI, retriever: BaseRetrieverInterface, options?: Partial<Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">>): ViolationOfExpectationsChain;\n}\n//# sourceMappingURL=violation_of_expectations_chain.d.ts.map"],"mappings":";;;;;;;;;;;;;;AAYA;AAKeA,UALEU,iCAAAA,SAA0CF,WAK5CR,CAAAA;EAINC;;AAT6D;AAetE;EAOeD,SAAAA,EAjBAA,sBAiBAA;EACNC;;;EAGeS,GAAAA,EAjBfT,UAiBeS;;;;;;AAuCNP,cAlDGQ,4BAAAA,SAAqCJ,SAAAA,YAAqBG,iCAkD7DP,CAAAA;EAA0BE,OAAAA,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAqCF,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAARU,cAAAA,EAAAA,MAAAA;EAiBjDZ,WAAAA,EAAAA,MAAAA;EAAuBD,IAAAA,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAA+CU,IAAAA,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAALI,SAAAA,EA5D1Ed,sBA4D0Ec;EAARC,GAAAA,EA3DxEd,UA2DwEc;EAAwEJ,gBAAAA,EA1DnIL,yBA0DmIK;EAnE/FJ,kBAAAA,EAUlCH,kBAVkCG;EAAqBG,WAAAA,CAAAA,MAAAA,EAWvDA,iCAXuDA;EAAiC,oBAAA,CAAA,WAAA,EAY1ER,WAZ0E,EAAA,CAAA,EAAA,MAAA;kCAa5EU,gBAAgBA;;;;;;;;;;;;;;;;;yCAiBTV,gBAAgBO;;;;;;;;;;;;;;;;;;;;gBAoBzCN,0BAA0BE,6BAA6BQ,QAAQV;;;;;;;;;;;;;;;;;sBAiBzDF,uBAAuBD,kCAAkCe,QAAQD,KAAKJ,2DAA2DC"}
+{"version":3,"file":"violation_of_expectations_chain.d.ts","names":[],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"mappings":";;;;;;;;;;;;;AA6BA;UAAiB,iCAAA,SAA0C,WAAA;;;;;EAKzD,SAAA,EAAW,sBAAA;EALyD;;;EASpE,GAAA,EAAK,UAAA;AAAA;;;;AAOP;cAAa,4BAAA,SACH,SAAA,YACG,iCAAA;EAAA,OAEJ,OAAA,CAAA;EAIP,UAAA,CAAA;EAIA,cAAA;EAEA,WAAA;EAAA,IAEI,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EAIJ,SAAA,EAAW,sBAAA;EAEX,GAAA,EAAK,UAAA;EAEL,gBAAA,EAAkB,yBAAA;EAElB,kBAAA,EAAoB,kBAAA;EAEpB,WAAA,CAAY,MAAA,EAAQ,iCAAA;EAQpB,oBAAA,CAAqB,WAAA,EAAa,WAAA;EAclC,sBAAA,CAAuB,OAAA,EAAS,KAAA,WAAgB,KAAA;EAkErC;;;;;;;;;;;;;;;;EA9CX,yBAAA,CAA0B,WAAA,EAAa,WAAA,KAAgB,kBAAA;EA5DvD;;;;;;;;;;;;;;;;;;;EAuGM,KAAA,CACJ,MAAA,EAAQ,WAAA,EACR,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,WAAA;EAAA,QAqEG,sBAAA;EAAA,QAyDA,wBAAA;EAAA,QA+BA,uBAAA;EAAA,QA6DA,yBAAA;EAAA,QAuCA,aAAA;EApQR;;;;;;;;;;;EAAA,OAiTC,OAAA,CACL,GAAA,EAAK,UAAA,EACL,SAAA,EAAW,sBAAA,EACX,OAAA,GAAU,OAAA,CACR,IAAA,CAAK,iCAAA,0BAEN,4BAAA;AAAA"}

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.js

@@ -2,8 +2,8 @@ import { BaseChain } from "../../../chains/base.js";
 import { JsonOutputFunctionsParser } from "../../../output_parsers/openai_functions.js";
 import { PREDICTION_VIOLATIONS_FUNCTION, PREDICT_NEXT_USER_MESSAGE_FUNCTION } from "./types.js";
 import { GENERATE_FACTS_PROMPT, GENERATE_REVISED_PREDICTION_PROMPT, PREDICTION_VIOLATIONS_PROMPT, PREDICT_NEXT_USER_MESSAGE_PROMPT } from "./violation_of_expectations_prompt.js";
-import { StringOutputParser } from "@langchain/core/output_parsers";
 import { HumanMessage, isBaseMessage } from "@langchain/core/messages";
+import { StringOutputParser } from "@langchain/core/output_parsers";
 
 //#region src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts
 /**
@@ -100,8 +100,7 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 	async _call(values, runManager) {
 		if (!(this.chatHistoryKey in values)) throw new Error(`Chat history key ${this.chatHistoryKey} not found`);
 		const chatHistory = values[this.chatHistoryKey];
-		const isEveryMessageBaseMessage = chatHistory.every((message) => isBaseMessage(message));
-		if (!isEveryMessageBaseMessage) throw new Error("Chat history must be an array of BaseMessages");
+		if (!chatHistory.every((message) => isBaseMessage(message))) throw new Error("Chat history must be an array of BaseMessages");
 		const messageChunks = this.chunkMessagesByAIResponse(chatHistory);
 		const userPredictions = await Promise.all(messageChunks.map(async (chatHistoryChunk) => ({
 			userPredictions: await this.predictNextUserMessage(chatHistoryChunk.chunkedMessages),
@@ -113,14 +112,13 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 			userResponse: prediction.userResponse,
 			runManager
 		})));
-		const insights = await Promise.all(predictionViolations.map((violation) => this.generateFacts({
+		return { insights: await Promise.all(predictionViolations.map((violation) => this.generateFacts({
 			userResponse: violation.userResponse,
 			predictions: {
 				revisedPrediction: violation.revisedPrediction,
 				explainedPredictionErrors: violation.explainedPredictionErrors
 			}
-		})));
-		return { insights };
+		}))) };
 	}
 	/**
 	* This method predicts the next user message based on the chat history.
@@ -135,8 +133,7 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 	async predictNextUserMessage(chatHistory, runManager) {
 		const messageString = this.getChatHistoryString(chatHistory);
 		const llmWithFunctions = this.llm.bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION]).withConfig({ function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name } });
-		const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(this.jsonOutputParser);
-		const res = await chain.invoke({ chat_history: messageString }, runManager?.getChild("prediction"));
+		const res = await PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(this.jsonOutputParser).invoke({ chat_history: messageString }, runManager?.getChild("prediction"));
 		if (!("userState" in res && "predictedUserMessage" in res && "insights" in res)) throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);
 		const predictionResponse = res;
 		const retrievedDocs = await this.retrieveRelevantInsights(predictionResponse.insights);
@@ -154,12 +151,9 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 	* @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.
 	*/
 	async retrieveRelevantInsights(insights) {
-		const relevantInsightsDocuments = await Promise.all(insights.map(async (insight) => {
-			const relevantInsight = await this.retriever.invoke(insight);
-			return relevantInsight[0];
-		}));
-		const relevantInsightsContent = relevantInsightsDocuments.map((document) => document.pageContent);
-		return relevantInsightsContent;
+		return (await Promise.all(insights.map(async (insight) => {
+			return (await this.retriever.invoke(insight))[0];
+		}))).map((document) => document.pageContent);
 	}
 	/**
 	* This method generates prediction violations based on the predicted and actual user responses.
@@ -183,15 +177,14 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 			actual_output: userResponse?.content ?? "",
 			user_insights: userPredictions.insights.join("\n")
 		}, runManager?.getChild("prediction_violations"));
-		const revisedPrediction = await this.generateRevisedPrediction({
-			originalPrediction: userPredictions.predictedUserMessage,
-			explainedPredictionErrors: res.explainedPredictionErrors,
-			userInsights: userPredictions.insights,
-			runManager
-		});
 		return {
 			userResponse,
-			revisedPrediction,
+			revisedPrediction: await this.generateRevisedPrediction({
+				originalPrediction: userPredictions.predictedUserMessage,
+				explainedPredictionErrors: res.explainedPredictionErrors,
+				userInsights: userPredictions.insights,
+				runManager
+			}),
 			explainedPredictionErrors: res.explainedPredictionErrors
 		};
 	}
@@ -207,13 +200,11 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 	* @returns {Promise<string>} A promise that resolves to a revised prediction.
 	*/
 	async generateRevisedPrediction({ originalPrediction, explainedPredictionErrors, userInsights, runManager }) {
-		const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(this.llm).pipe(this.stringOutputParser);
-		const revisedPredictionRes = await revisedPredictionChain.invoke({
+		return await GENERATE_REVISED_PREDICTION_PROMPT.pipe(this.llm).pipe(this.stringOutputParser).invoke({
 			prediction: originalPrediction,
 			explained_prediction_errors: explainedPredictionErrors.join("\n"),
 			user_insights: userInsights.join("\n")
 		}, runManager?.getChild("prediction_revision"));
-		return revisedPredictionRes;
 	}
 	/**
 	* This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.
@@ -230,12 +221,11 @@ var ViolationOfExpectationsChain = class extends BaseChain {
 	async generateFacts({ userResponse, predictions, runManager }) {
 		const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(this.stringOutputParser);
 		if (typeof userResponse?.content !== "string") throw new Error("This chain does not support non-string model output.");
-		const res = await chain.invoke({
+		return await chain.invoke({
 			prediction_violations: predictions.explainedPredictionErrors.join("\n"),
 			prediction: predictions.revisedPrediction,
 			user_message: userResponse?.content ?? ""
 		}, runManager?.getChild("generate_facts"));
-		return res;
 	}
 	/**
 	* Static method that creates a ViolationOfExpectationsChain instance from a

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"violation_of_expectations_chain.js","names":["fields: ViolationOfExpectationsChainInput","chatHistory: BaseMessage[]","strings: Array<string>","newArray: MessageChunkResult[]","tempArray: BaseMessage[]","userResponse: BaseMessage | undefined","values: ChainValues","runManager?: CallbackManagerForChainRun","chatHistory: unknown[]","insights: Array<string>","llm: ChatOpenAI","retriever: BaseRetrieverInterface","options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport {\n  GetPredictionViolationsResponse,\n  MessageChunkResult,\n  PREDICTION_VIOLATIONS_FUNCTION,\n  PREDICT_NEXT_USER_MESSAGE_FUNCTION,\n  PredictNextUserMessageResponse,\n} from \"./types.js\";\nimport {\n  GENERATE_FACTS_PROMPT,\n  GENERATE_REVISED_PREDICTION_PROMPT,\n  PREDICTION_VIOLATIONS_PROMPT,\n  PREDICT_NEXT_USER_MESSAGE_PROMPT,\n} from \"./violation_of_expectations_prompt.js\";\n\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n  /**\n   * The retriever to use for retrieving stored\n   * thoughts and insights.\n   */\n  retriever: BaseRetrieverInterface;\n  /**\n   * The LLM to use\n   */\n  llm: ChatOpenAI;\n}\n\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport class ViolationOfExpectationsChain\n  extends BaseChain\n  implements ViolationOfExpectationsChainInput\n{\n  static lc_name() {\n    return \"ViolationOfExpectationsChain\";\n  }\n\n  _chainType(): string {\n    return \"violation_of_expectation_chain\";\n  }\n\n  chatHistoryKey = \"chat_history\";\n\n  thoughtsKey = \"thoughts\";\n\n  get inputKeys() {\n    return [this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return [this.thoughtsKey];\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  llm: ChatOpenAI;\n\n  jsonOutputParser: JsonOutputFunctionsParser;\n\n  stringOutputParser: StringOutputParser;\n\n  constructor(fields: ViolationOfExpectationsChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llm = fields.llm;\n    this.jsonOutputParser = new JsonOutputFunctionsParser();\n    this.stringOutputParser = new StringOutputParser();\n  }\n\n  getChatHistoryString(chatHistory: BaseMessage[]): string {\n    return chatHistory\n      .map((chatMessage) => {\n        if (chatMessage._getType() === \"human\") {\n          return `Human: ${chatMessage.content}`;\n        } else if (chatMessage._getType() === \"ai\") {\n          return `AI: ${chatMessage.content}`;\n        } else {\n          return `${chatMessage.content}`;\n        }\n      })\n      .join(\"\\n\");\n  }\n\n  removeDuplicateStrings(strings: Array<string>): Array<string> {\n    return [...new Set(strings)];\n  }\n\n  /**\n   * This method breaks down the chat history into chunks of messages.\n   * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n   *\n   * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n   *\n   * @description\n   * The method iterates over the chat history and pushes each message into a temporary array.\n   * When it encounters an AI message, it checks for a subsequent user message.\n   * If a user message is found, it is considered as the user response to the AI message.\n   * If no user message is found after the AI message, the user response is undefined.\n   * The method then pushes the chunk (sequence of messages and user response) into the result array.\n   * This process continues until all messages in the chat history have been processed.\n   */\n  chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[] {\n    const newArray: MessageChunkResult[] = [];\n    const tempArray: BaseMessage[] = [];\n\n    chatHistory.forEach((item, index) => {\n      tempArray.push(item);\n      if (item._getType() === \"ai\") {\n        let userResponse: BaseMessage | undefined = chatHistory[index + 1];\n        if (!userResponse || userResponse._getType() !== \"human\") {\n          userResponse = undefined;\n        }\n\n        newArray.push({\n          chunkedMessages: tempArray,\n          userResponse: userResponse\n            ? new HumanMessage(userResponse)\n            : undefined,\n        });\n      }\n    });\n\n    return newArray;\n  }\n\n  /**\n   * This method processes a chat history to generate insights about the user.\n   *\n   * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n   *\n   * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n   *\n   * @description\n   * The method performs the following steps:\n   * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n   * 2. Breaks the chat history into chunks of messages.\n   * 3. For each chunk, it generates an initial prediction for the user's next message.\n   * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n   * 5. For each set of messages, it generates a fact/insight about the user.\n   * The method returns a list of these insights.\n   */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found`);\n    }\n\n    const chatHistory: unknown[] = values[this.chatHistoryKey];\n\n    const isEveryMessageBaseMessage = chatHistory.every((message) =>\n      isBaseMessage(message)\n    );\n    if (!isEveryMessageBaseMessage) {\n      throw new Error(\"Chat history must be an array of BaseMessages\");\n    }\n\n    const messageChunks = this.chunkMessagesByAIResponse(\n      chatHistory as BaseMessage[]\n    );\n\n    // Generate the initial prediction for every user message.\n    const userPredictions = await Promise.all(\n      messageChunks.map(async (chatHistoryChunk) => ({\n        userPredictions: await this.predictNextUserMessage(\n          chatHistoryChunk.chunkedMessages\n        ),\n        userResponse: chatHistoryChunk.userResponse,\n        runManager,\n      }))\n    );\n\n    // Generate insights, and prediction violations for every user message.\n    // This call also regenerates the prediction based on the violations.\n    const predictionViolations = await Promise.all(\n      userPredictions.map((prediction) =>\n        this.getPredictionViolations({\n          userPredictions: prediction.userPredictions,\n          userResponse: prediction.userResponse,\n          runManager,\n        })\n      )\n    );\n\n    // Generate a fact/insight about the user for every set of messages.\n    const insights = await Promise.all(\n      predictionViolations.map((violation) =>\n        this.generateFacts({\n          userResponse: violation.userResponse,\n          predictions: {\n            revisedPrediction: violation.revisedPrediction,\n            explainedPredictionErrors: violation.explainedPredictionErrors,\n          },\n        })\n      )\n    );\n\n    return {\n      insights,\n    };\n  }\n\n  /**\n   * This method predicts the next user message based on the chat history.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history based on which the next user message is predicted.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<PredictNextUserMessageResponse>} A promise that resolves to the predicted next user message, the user state, and any insights.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'userState', 'predictedUserMessage', and 'insights'.\n   */\n  private async predictNextUserMessage(\n    chatHistory: BaseMessage[],\n    runManager?: CallbackManagerForChainRun\n  ): Promise<PredictNextUserMessageResponse> {\n    const messageString = this.getChatHistoryString(chatHistory);\n\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name },\n      });\n\n    const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    const res = await chain.invoke(\n      {\n        chat_history: messageString,\n      },\n      runManager?.getChild(\"prediction\")\n    );\n\n    if (\n      !(\n        \"userState\" in res &&\n        \"predictedUserMessage\" in res &&\n        \"insights\" in res\n      )\n    ) {\n      throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);\n    }\n\n    const predictionResponse = res as PredictNextUserMessageResponse;\n\n    // Query the retriever for relevant insights. Use the generates insights as a query.\n    const retrievedDocs = await this.retrieveRelevantInsights(\n      predictionResponse.insights\n    );\n    const relevantDocs = this.removeDuplicateStrings([\n      ...predictionResponse.insights,\n      ...retrievedDocs,\n    ]);\n\n    return {\n      ...predictionResponse,\n      insights: relevantDocs,\n    };\n  }\n\n  /**\n   * Retrieves relevant insights based on the provided insights.\n   *\n   * @param {Array<string>} insights - An array of insights to be used for retrieving relevant documents.\n   *\n   * @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.\n   */\n  private async retrieveRelevantInsights(\n    insights: Array<string>\n  ): Promise<Array<string>> {\n    // Only extract the first relevant doc from the retriever. We don't need more than one.\n    const relevantInsightsDocuments = await Promise.all(\n      insights.map(async (insight) => {\n        const relevantInsight = await this.retriever.invoke(insight);\n        return relevantInsight[0];\n      })\n    );\n\n    const relevantInsightsContent = relevantInsightsDocuments.map(\n      (document) => document.pageContent\n    );\n\n    return relevantInsightsContent;\n  }\n\n  /**\n   * This method generates prediction violations based on the predicted and actual user responses.\n   * It also generates a revised prediction based on the identified violations.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {PredictNextUserMessageResponse} params.userPredictions - The predicted user message, user state, and insights.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<{ userResponse: BaseMessage | undefined; revisedPrediction: string; explainedPredictionErrors: Array<string>; }>} A promise that resolves to an object containing the actual user response, the revised prediction, and the explained prediction errors.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'violationExplanation', 'explainedPredictionErrors', and 'accuratePrediction'.\n   */\n  private async getPredictionViolations({\n    userPredictions,\n    userResponse,\n    runManager,\n  }: {\n    userPredictions: PredictNextUserMessageResponse;\n    userResponse?: BaseMessage;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<GetPredictionViolationsResponse> {\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICTION_VIOLATIONS_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },\n      });\n\n    const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = (await chain.invoke(\n      {\n        predicted_output: userPredictions.predictedUserMessage,\n        actual_output: userResponse?.content ?? \"\",\n        user_insights: userPredictions.insights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_violations\")\n    )) as Awaited<{\n      violationExplanation: string;\n      explainedPredictionErrors: Array<string>;\n      accuratePrediction: boolean;\n    }>;\n\n    // Generate a revised prediction based on violations.\n    const revisedPrediction = await this.generateRevisedPrediction({\n      originalPrediction: userPredictions.predictedUserMessage,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n      userInsights: userPredictions.insights,\n      runManager,\n    });\n\n    return {\n      userResponse,\n      revisedPrediction,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n    };\n  }\n\n  /**\n   * This method generates a revised prediction based on the original prediction, explained prediction errors, and user insights.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {string} params.originalPrediction - The original prediction made by the model.\n   * @param {Array<string>} params.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {Array<string>} params.userInsights - An array of insights about the user.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a revised prediction.\n   */\n  private async generateRevisedPrediction({\n    originalPrediction,\n    explainedPredictionErrors,\n    userInsights,\n    runManager,\n  }: {\n    originalPrediction: string;\n    explainedPredictionErrors: Array<string>;\n    userInsights: Array<string>;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(\n      this.llm\n    ).pipe(this.stringOutputParser);\n\n    const revisedPredictionRes = await revisedPredictionChain.invoke(\n      {\n        prediction: originalPrediction,\n        explained_prediction_errors: explainedPredictionErrors.join(\"\\n\"),\n        user_insights: userInsights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_revision\")\n    );\n\n    return revisedPredictionRes;\n  }\n\n  /**\n   * This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {Object} params.predictions - The revised prediction and explained prediction errors.\n   * @param {string} params.predictions.revisedPrediction - The revised prediction made by the model.\n   * @param {Array<string>} params.predictions.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a string containing the generated facts or insights about the user.\n   */\n  private async generateFacts({\n    userResponse,\n    predictions,\n    runManager,\n  }: {\n    userResponse?: BaseMessage;\n    /**\n     * Optional if the prediction was accurate.\n     */\n    predictions: {\n      revisedPrediction: string;\n      explainedPredictionErrors: Array<string>;\n    };\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(\n      this.stringOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = await chain.invoke(\n      {\n        prediction_violations: predictions.explainedPredictionErrors.join(\"\\n\"),\n        prediction: predictions.revisedPrediction,\n        user_message: userResponse?.content ?? \"\",\n      },\n      runManager?.getChild(\"generate_facts\")\n    );\n\n    return res;\n  }\n\n  /**\n   * Static method that creates a ViolationOfExpectationsChain instance from a\n   * ChatOpenAI and retriever. It also accepts optional options\n   * to customize the chain.\n   *\n   * @param llm The ChatOpenAI instance.\n   * @param retriever The retriever used for similarity search.\n   * @param options Optional options to customize the chain.\n   *\n   * @returns A new instance of ViolationOfExpectationsChain.\n   */\n  static fromLLM(\n    llm: ChatOpenAI,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >\n  ): ViolationOfExpectationsChain {\n    return new this({\n      retriever,\n      llm,\n      ...options,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;AA6CA,IAAa,+BAAb,cACU,UAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED,aAAqB;AACnB,SAAO;CACR;CAED,iBAAiB;CAEjB,cAAc;CAEd,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,cAAe;CAC7B;CAED,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,WAAY;CAC1B;CAED;CAEA;CAEA;CAEA;CAEA,YAAYA,QAA2C;EACrD,MAAM,OAAO;EACb,KAAK,YAAY,OAAO;EACxB,KAAK,MAAM,OAAO;EAClB,KAAK,mBAAmB,IAAI;EAC5B,KAAK,qBAAqB,IAAI;CAC/B;CAED,qBAAqBC,aAAoC;AACvD,SAAO,YACJ,IAAI,CAAC,gBAAgB;AACpB,OAAI,YAAY,UAAU,KAAK,QAC7B,QAAO,CAAC,OAAO,EAAE,YAAY,SAAS;YAC7B,YAAY,UAAU,KAAK,KACpC,QAAO,CAAC,IAAI,EAAE,YAAY,SAAS;OAEnC,QAAO,GAAG,YAAY,SAAS;EAElC,EAAC,CACD,KAAK,KAAK;CACd;CAED,uBAAuBC,SAAuC;AAC5D,SAAO,CAAC,GAAG,IAAI,IAAI,QAAS;CAC7B;;;;;;;;;;;;;;;;;CAkBD,0BAA0BD,aAAkD;EAC1E,MAAME,WAAiC,CAAE;EACzC,MAAMC,YAA2B,CAAE;EAEnC,YAAY,QAAQ,CAAC,MAAM,UAAU;GACnC,UAAU,KAAK,KAAK;AACpB,OAAI,KAAK,UAAU,KAAK,MAAM;IAC5B,IAAIC,eAAwC,YAAY,QAAQ;AAChE,QAAI,CAAC,gBAAgB,aAAa,UAAU,KAAK,SAC/C,eAAe;IAGjB,SAAS,KAAK;KACZ,iBAAiB;KACjB,cAAc,eACV,IAAI,aAAa,gBACjB;IACL,EAAC;GACH;EACF,EAAC;AAEF,SAAO;CACR;;;;;;;;;;;;;;;;;;;;CAqBD,MAAM,MACJC,QACAC,YACsB;AACtB,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,CAAC,iBAAiB,EAAE,KAAK,eAAe,UAAU,CAAC;EAGrE,MAAMC,cAAyB,OAAO,KAAK;EAE3C,MAAM,4BAA4B,YAAY,MAAM,CAAC,YACnD,cAAc,QAAQ,CACvB;AACD,MAAI,CAAC,0BACH,OAAM,IAAI,MAAM;EAGlB,MAAM,gBAAgB,KAAK,0BACzB,YACD;EAGD,MAAM,kBAAkB,MAAM,QAAQ,IACpC,cAAc,IAAI,OAAO,sBAAsB;GAC7C,iBAAiB,MAAM,KAAK,uBAC1B,iBAAiB,gBAClB;GACD,cAAc,iBAAiB;GAC/B;EACD,GAAE,CACJ;EAID,MAAM,uBAAuB,MAAM,QAAQ,IACzC,gBAAgB,IAAI,CAAC,eACnB,KAAK,wBAAwB;GAC3B,iBAAiB,WAAW;GAC5B,cAAc,WAAW;GACzB;EACD,EAAC,CACH,CACF;EAGD,MAAM,WAAW,MAAM,QAAQ,IAC7B,qBAAqB,IAAI,CAAC,cACxB,KAAK,cAAc;GACjB,cAAc,UAAU;GACxB,aAAa;IACX,mBAAmB,UAAU;IAC7B,2BAA2B,UAAU;GACtC;EACF,EAAC,CACH,CACF;AAED,SAAO,EACL,SACD;CACF;;;;;;;;;;;CAYD,MAAc,uBACZP,aACAM,YACyC;EACzC,MAAM,gBAAgB,KAAK,qBAAqB,YAAY;EAE5D,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAAC,kCAAmC,EAAC,CAC/C,WAAW,EACV,eAAe,EAAE,MAAM,mCAAmC,KAAM,EACjE,EAAC;EAEJ,MAAM,QAAQ,iCAAiC,KAAK,iBAAiB,CAAC,KACpE,KAAK,iBACN;EAED,MAAM,MAAM,MAAM,MAAM,OACtB,EACE,cAAc,cACf,GACD,YAAY,SAAS,aAAa,CACnC;AAED,MACE,EACE,eAAe,OACf,0BAA0B,OAC1B,cAAc,KAGhB,OAAM,IAAI,MAAM,CAAC,2BAA2B,EAAE,KAAK,UAAU,IAAI,EAAE;EAGrE,MAAM,qBAAqB;EAG3B,MAAM,gBAAgB,MAAM,KAAK,yBAC/B,mBAAmB,SACpB;EACD,MAAM,eAAe,KAAK,uBAAuB,CAC/C,GAAG,mBAAmB,UACtB,GAAG,aACJ,EAAC;AAEF,SAAO;GACL,GAAG;GACH,UAAU;EACX;CACF;;;;;;;;CASD,MAAc,yBACZE,UACwB;EAExB,MAAM,4BAA4B,MAAM,QAAQ,IAC9C,SAAS,IAAI,OAAO,YAAY;GAC9B,MAAM,kBAAkB,MAAM,KAAK,UAAU,OAAO,QAAQ;AAC5D,UAAO,gBAAgB;EACxB,EAAC,CACH;EAED,MAAM,0BAA0B,0BAA0B,IACxD,CAAC,aAAa,SAAS,YACxB;AAED,SAAO;CACR;;;;;;;;;;;;;;CAeD,MAAc,wBAAwB,EACpC,iBACA,cACA,YAKD,EAA4C;EAC3C,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAAC,8BAA+B,EAAC,CAC3C,WAAW,EACV,eAAe,EAAE,MAAM,+BAA+B,KAAM,EAC7D,EAAC;EAEJ,MAAM,QAAQ,6BAA6B,KAAK,iBAAiB,CAAC,KAChE,KAAK,iBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM;EAElB,MAAM,MAAO,MAAM,MAAM,OACvB;GACE,kBAAkB,gBAAgB;GAClC,eAAe,cAAc,WAAW;GACxC,eAAe,gBAAgB,SAAS,KAAK,KAAK;EACnD,GACD,YAAY,SAAS,wBAAwB,CAC9C;EAOD,MAAM,oBAAoB,MAAM,KAAK,0BAA0B;GAC7D,oBAAoB,gBAAgB;GACpC,2BAA2B,IAAI;GAC/B,cAAc,gBAAgB;GAC9B;EACD,EAAC;AAEF,SAAO;GACL;GACA;GACA,2BAA2B,IAAI;EAChC;CACF;;;;;;;;;;;;CAaD,MAAc,0BAA0B,EACtC,oBACA,2BACA,cACA,YAMD,EAAmB;EAClB,MAAM,yBAAyB,mCAAmC,KAChE,KAAK,IACN,CAAC,KAAK,KAAK,mBAAmB;EAE/B,MAAM,uBAAuB,MAAM,uBAAuB,OACxD;GACE,YAAY;GACZ,6BAA6B,0BAA0B,KAAK,KAAK;GACjE,eAAe,aAAa,KAAK,KAAK;EACvC,GACD,YAAY,SAAS,sBAAsB,CAC5C;AAED,SAAO;CACR;;;;;;;;;;;;;CAcD,MAAc,cAAc,EAC1B,cACA,aACA,YAWD,EAAmB;EAClB,MAAM,QAAQ,sBAAsB,KAAK,KAAK,IAAI,CAAC,KACjD,KAAK,mBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM;EAElB,MAAM,MAAM,MAAM,MAAM,OACtB;GACE,uBAAuB,YAAY,0BAA0B,KAAK,KAAK;GACvE,YAAY,YAAY;GACxB,cAAc,cAAc,WAAW;EACxC,GACD,YAAY,SAAS,iBAAiB,CACvC;AAED,SAAO;CACR;;;;;;;;;;;;CAaD,OAAO,QACLC,KACAC,WACAC,SAG8B;AAC9B,SAAO,IAAI,KAAK;GACd;GACA;GACA,GAAG;EACJ;CACF;AACF"}
+{"version":3,"file":"violation_of_expectations_chain.js","names":[],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_chain.ts"],"sourcesContent":["import type { BaseRetrieverInterface } from \"@langchain/core/retrievers\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  BaseMessage,\n  HumanMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { StringOutputParser } from \"@langchain/core/output_parsers\";\nimport { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport { JsonOutputFunctionsParser } from \"../../../output_parsers/openai_functions.js\";\nimport { BaseChain, ChainInputs } from \"../../../chains/base.js\";\nimport {\n  GetPredictionViolationsResponse,\n  MessageChunkResult,\n  PREDICTION_VIOLATIONS_FUNCTION,\n  PREDICT_NEXT_USER_MESSAGE_FUNCTION,\n  PredictNextUserMessageResponse,\n} from \"./types.js\";\nimport {\n  GENERATE_FACTS_PROMPT,\n  GENERATE_REVISED_PREDICTION_PROMPT,\n  PREDICTION_VIOLATIONS_PROMPT,\n  PREDICT_NEXT_USER_MESSAGE_PROMPT,\n} from \"./violation_of_expectations_prompt.js\";\n\n/**\n * Interface for the input parameters of the ViolationOfExpectationsChain class.\n */\nexport interface ViolationOfExpectationsChainInput extends ChainInputs {\n  /**\n   * The retriever to use for retrieving stored\n   * thoughts and insights.\n   */\n  retriever: BaseRetrieverInterface;\n  /**\n   * The LLM to use\n   */\n  llm: ChatOpenAI;\n}\n\n/**\n * Chain that generates key insights/facts of a user based on a\n * a chat conversation with an AI.\n */\nexport class ViolationOfExpectationsChain\n  extends BaseChain\n  implements ViolationOfExpectationsChainInput\n{\n  static lc_name() {\n    return \"ViolationOfExpectationsChain\";\n  }\n\n  _chainType(): string {\n    return \"violation_of_expectation_chain\";\n  }\n\n  chatHistoryKey = \"chat_history\";\n\n  thoughtsKey = \"thoughts\";\n\n  get inputKeys() {\n    return [this.chatHistoryKey];\n  }\n\n  get outputKeys() {\n    return [this.thoughtsKey];\n  }\n\n  retriever: BaseRetrieverInterface;\n\n  llm: ChatOpenAI;\n\n  jsonOutputParser: JsonOutputFunctionsParser;\n\n  stringOutputParser: StringOutputParser;\n\n  constructor(fields: ViolationOfExpectationsChainInput) {\n    super(fields);\n    this.retriever = fields.retriever;\n    this.llm = fields.llm;\n    this.jsonOutputParser = new JsonOutputFunctionsParser();\n    this.stringOutputParser = new StringOutputParser();\n  }\n\n  getChatHistoryString(chatHistory: BaseMessage[]): string {\n    return chatHistory\n      .map((chatMessage) => {\n        if (chatMessage._getType() === \"human\") {\n          return `Human: ${chatMessage.content}`;\n        } else if (chatMessage._getType() === \"ai\") {\n          return `AI: ${chatMessage.content}`;\n        } else {\n          return `${chatMessage.content}`;\n        }\n      })\n      .join(\"\\n\");\n  }\n\n  removeDuplicateStrings(strings: Array<string>): Array<string> {\n    return [...new Set(strings)];\n  }\n\n  /**\n   * This method breaks down the chat history into chunks of messages.\n   * Each chunk consists of a sequence of messages ending with an AI message and the subsequent user response, if any.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history to be chunked.\n   *\n   * @returns {MessageChunkResult[]} An array of message chunks. Each chunk includes a sequence of messages and the subsequent user response.\n   *\n   * @description\n   * The method iterates over the chat history and pushes each message into a temporary array.\n   * When it encounters an AI message, it checks for a subsequent user message.\n   * If a user message is found, it is considered as the user response to the AI message.\n   * If no user message is found after the AI message, the user response is undefined.\n   * The method then pushes the chunk (sequence of messages and user response) into the result array.\n   * This process continues until all messages in the chat history have been processed.\n   */\n  chunkMessagesByAIResponse(chatHistory: BaseMessage[]): MessageChunkResult[] {\n    const newArray: MessageChunkResult[] = [];\n    const tempArray: BaseMessage[] = [];\n\n    chatHistory.forEach((item, index) => {\n      tempArray.push(item);\n      if (item._getType() === \"ai\") {\n        let userResponse: BaseMessage | undefined = chatHistory[index + 1];\n        if (!userResponse || userResponse._getType() !== \"human\") {\n          userResponse = undefined;\n        }\n\n        newArray.push({\n          chunkedMessages: tempArray,\n          userResponse: userResponse\n            ? new HumanMessage(userResponse)\n            : undefined,\n        });\n      }\n    });\n\n    return newArray;\n  }\n\n  /**\n   * This method processes a chat history to generate insights about the user.\n   *\n   * @param {ChainValues} values - The input values for the chain. It should contain a key for chat history.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<ChainValues>} A promise that resolves to a list of insights about the user.\n   *\n   * @throws {Error} If the chat history key is not found in the input values or if the chat history is not an array of BaseMessages.\n   *\n   * @description\n   * The method performs the following steps:\n   * 1. Checks if the chat history key is present in the input values and if the chat history is an array of BaseMessages.\n   * 2. Breaks the chat history into chunks of messages.\n   * 3. For each chunk, it generates an initial prediction for the user's next message.\n   * 4. For each prediction, it generates insights and prediction violations, and regenerates the prediction based on the violations.\n   * 5. For each set of messages, it generates a fact/insight about the user.\n   * The method returns a list of these insights.\n   */\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    if (!(this.chatHistoryKey in values)) {\n      throw new Error(`Chat history key ${this.chatHistoryKey} not found`);\n    }\n\n    const chatHistory: unknown[] = values[this.chatHistoryKey];\n\n    const isEveryMessageBaseMessage = chatHistory.every((message) =>\n      isBaseMessage(message)\n    );\n    if (!isEveryMessageBaseMessage) {\n      throw new Error(\"Chat history must be an array of BaseMessages\");\n    }\n\n    const messageChunks = this.chunkMessagesByAIResponse(\n      chatHistory as BaseMessage[]\n    );\n\n    // Generate the initial prediction for every user message.\n    const userPredictions = await Promise.all(\n      messageChunks.map(async (chatHistoryChunk) => ({\n        userPredictions: await this.predictNextUserMessage(\n          chatHistoryChunk.chunkedMessages\n        ),\n        userResponse: chatHistoryChunk.userResponse,\n        runManager,\n      }))\n    );\n\n    // Generate insights, and prediction violations for every user message.\n    // This call also regenerates the prediction based on the violations.\n    const predictionViolations = await Promise.all(\n      userPredictions.map((prediction) =>\n        this.getPredictionViolations({\n          userPredictions: prediction.userPredictions,\n          userResponse: prediction.userResponse,\n          runManager,\n        })\n      )\n    );\n\n    // Generate a fact/insight about the user for every set of messages.\n    const insights = await Promise.all(\n      predictionViolations.map((violation) =>\n        this.generateFacts({\n          userResponse: violation.userResponse,\n          predictions: {\n            revisedPrediction: violation.revisedPrediction,\n            explainedPredictionErrors: violation.explainedPredictionErrors,\n          },\n        })\n      )\n    );\n\n    return {\n      insights,\n    };\n  }\n\n  /**\n   * This method predicts the next user message based on the chat history.\n   *\n   * @param {BaseMessage[]} chatHistory - The chat history based on which the next user message is predicted.\n   * @param {CallbackManagerForChainRun} [runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<PredictNextUserMessageResponse>} A promise that resolves to the predicted next user message, the user state, and any insights.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'userState', 'predictedUserMessage', and 'insights'.\n   */\n  private async predictNextUserMessage(\n    chatHistory: BaseMessage[],\n    runManager?: CallbackManagerForChainRun\n  ): Promise<PredictNextUserMessageResponse> {\n    const messageString = this.getChatHistoryString(chatHistory);\n\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICT_NEXT_USER_MESSAGE_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICT_NEXT_USER_MESSAGE_FUNCTION.name },\n      });\n\n    const chain = PREDICT_NEXT_USER_MESSAGE_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    const res = await chain.invoke(\n      {\n        chat_history: messageString,\n      },\n      runManager?.getChild(\"prediction\")\n    );\n\n    if (\n      !(\n        \"userState\" in res &&\n        \"predictedUserMessage\" in res &&\n        \"insights\" in res\n      )\n    ) {\n      throw new Error(`Invalid response from LLM: ${JSON.stringify(res)}`);\n    }\n\n    const predictionResponse = res as PredictNextUserMessageResponse;\n\n    // Query the retriever for relevant insights. Use the generates insights as a query.\n    const retrievedDocs = await this.retrieveRelevantInsights(\n      predictionResponse.insights\n    );\n    const relevantDocs = this.removeDuplicateStrings([\n      ...predictionResponse.insights,\n      ...retrievedDocs,\n    ]);\n\n    return {\n      ...predictionResponse,\n      insights: relevantDocs,\n    };\n  }\n\n  /**\n   * Retrieves relevant insights based on the provided insights.\n   *\n   * @param {Array<string>} insights - An array of insights to be used for retrieving relevant documents.\n   *\n   * @returns {Promise<Array<string>>} A promise that resolves to an array of relevant insights content.\n   */\n  private async retrieveRelevantInsights(\n    insights: Array<string>\n  ): Promise<Array<string>> {\n    // Only extract the first relevant doc from the retriever. We don't need more than one.\n    const relevantInsightsDocuments = await Promise.all(\n      insights.map(async (insight) => {\n        const relevantInsight = await this.retriever.invoke(insight);\n        return relevantInsight[0];\n      })\n    );\n\n    const relevantInsightsContent = relevantInsightsDocuments.map(\n      (document) => document.pageContent\n    );\n\n    return relevantInsightsContent;\n  }\n\n  /**\n   * This method generates prediction violations based on the predicted and actual user responses.\n   * It also generates a revised prediction based on the identified violations.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {PredictNextUserMessageResponse} params.userPredictions - The predicted user message, user state, and insights.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<{ userResponse: BaseMessage | undefined; revisedPrediction: string; explainedPredictionErrors: Array<string>; }>} A promise that resolves to an object containing the actual user response, the revised prediction, and the explained prediction errors.\n   *\n   * @throws {Error} If the response from the language model does not contain the expected keys: 'violationExplanation', 'explainedPredictionErrors', and 'accuratePrediction'.\n   */\n  private async getPredictionViolations({\n    userPredictions,\n    userResponse,\n    runManager,\n  }: {\n    userPredictions: PredictNextUserMessageResponse;\n    userResponse?: BaseMessage;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<GetPredictionViolationsResponse> {\n    const llmWithFunctions = this.llm\n      .bindTools([PREDICTION_VIOLATIONS_FUNCTION])\n      .withConfig({\n        function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },\n      });\n\n    const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(\n      this.jsonOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = (await chain.invoke(\n      {\n        predicted_output: userPredictions.predictedUserMessage,\n        actual_output: userResponse?.content ?? \"\",\n        user_insights: userPredictions.insights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_violations\")\n    )) as Awaited<{\n      violationExplanation: string;\n      explainedPredictionErrors: Array<string>;\n      accuratePrediction: boolean;\n    }>;\n\n    // Generate a revised prediction based on violations.\n    const revisedPrediction = await this.generateRevisedPrediction({\n      originalPrediction: userPredictions.predictedUserMessage,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n      userInsights: userPredictions.insights,\n      runManager,\n    });\n\n    return {\n      userResponse,\n      revisedPrediction,\n      explainedPredictionErrors: res.explainedPredictionErrors,\n    };\n  }\n\n  /**\n   * This method generates a revised prediction based on the original prediction, explained prediction errors, and user insights.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {string} params.originalPrediction - The original prediction made by the model.\n   * @param {Array<string>} params.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {Array<string>} params.userInsights - An array of insights about the user.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a revised prediction.\n   */\n  private async generateRevisedPrediction({\n    originalPrediction,\n    explainedPredictionErrors,\n    userInsights,\n    runManager,\n  }: {\n    originalPrediction: string;\n    explainedPredictionErrors: Array<string>;\n    userInsights: Array<string>;\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const revisedPredictionChain = GENERATE_REVISED_PREDICTION_PROMPT.pipe(\n      this.llm\n    ).pipe(this.stringOutputParser);\n\n    const revisedPredictionRes = await revisedPredictionChain.invoke(\n      {\n        prediction: originalPrediction,\n        explained_prediction_errors: explainedPredictionErrors.join(\"\\n\"),\n        user_insights: userInsights.join(\"\\n\"),\n      },\n      runManager?.getChild(\"prediction_revision\")\n    );\n\n    return revisedPredictionRes;\n  }\n\n  /**\n   * This method generates facts or insights about the user based on the revised prediction, explained prediction errors, and the user's response.\n   *\n   * @param {Object} params - The parameters for the method.\n   * @param {BaseMessage} [params.userResponse] - The actual user response.\n   * @param {Object} params.predictions - The revised prediction and explained prediction errors.\n   * @param {string} params.predictions.revisedPrediction - The revised prediction made by the model.\n   * @param {Array<string>} params.predictions.explainedPredictionErrors - An array of explained prediction errors.\n   * @param {CallbackManagerForChainRun} [params.runManager] - Optional callback manager for the chain run.\n   *\n   * @returns {Promise<string>} A promise that resolves to a string containing the generated facts or insights about the user.\n   */\n  private async generateFacts({\n    userResponse,\n    predictions,\n    runManager,\n  }: {\n    userResponse?: BaseMessage;\n    /**\n     * Optional if the prediction was accurate.\n     */\n    predictions: {\n      revisedPrediction: string;\n      explainedPredictionErrors: Array<string>;\n    };\n    runManager?: CallbackManagerForChainRun;\n  }): Promise<string> {\n    const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(\n      this.stringOutputParser\n    );\n\n    if (typeof userResponse?.content !== \"string\") {\n      throw new Error(\"This chain does not support non-string model output.\");\n    }\n    const res = await chain.invoke(\n      {\n        prediction_violations: predictions.explainedPredictionErrors.join(\"\\n\"),\n        prediction: predictions.revisedPrediction,\n        user_message: userResponse?.content ?? \"\",\n      },\n      runManager?.getChild(\"generate_facts\")\n    );\n\n    return res;\n  }\n\n  /**\n   * Static method that creates a ViolationOfExpectationsChain instance from a\n   * ChatOpenAI and retriever. It also accepts optional options\n   * to customize the chain.\n   *\n   * @param llm The ChatOpenAI instance.\n   * @param retriever The retriever used for similarity search.\n   * @param options Optional options to customize the chain.\n   *\n   * @returns A new instance of ViolationOfExpectationsChain.\n   */\n  static fromLLM(\n    llm: ChatOpenAI,\n    retriever: BaseRetrieverInterface,\n    options?: Partial<\n      Omit<ViolationOfExpectationsChainInput, \"llm\" | \"retriever\">\n    >\n  ): ViolationOfExpectationsChain {\n    return new this({\n      retriever,\n      llm,\n      ...options,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;AA6CA,IAAa,+BAAb,cACU,UAEV;CACE,OAAO,UAAU;AACf,SAAO;;CAGT,aAAqB;AACnB,SAAO;;CAGT,iBAAiB;CAEjB,cAAc;CAEd,IAAI,YAAY;AACd,SAAO,CAAC,KAAK,eAAe;;CAG9B,IAAI,aAAa;AACf,SAAO,CAAC,KAAK,YAAY;;CAG3B;CAEA;CAEA;CAEA;CAEA,YAAY,QAA2C;AACrD,QAAM,OAAO;AACb,OAAK,YAAY,OAAO;AACxB,OAAK,MAAM,OAAO;AAClB,OAAK,mBAAmB,IAAI,2BAA2B;AACvD,OAAK,qBAAqB,IAAI,oBAAoB;;CAGpD,qBAAqB,aAAoC;AACvD,SAAO,YACJ,KAAK,gBAAgB;AACpB,OAAI,YAAY,UAAU,KAAK,QAC7B,QAAO,UAAU,YAAY;YACpB,YAAY,UAAU,KAAK,KACpC,QAAO,OAAO,YAAY;OAE1B,QAAO,GAAG,YAAY;IAExB,CACD,KAAK,KAAK;;CAGf,uBAAuB,SAAuC;AAC5D,SAAO,CAAC,GAAG,IAAI,IAAI,QAAQ,CAAC;;;;;;;;;;;;;;;;;;CAmB9B,0BAA0B,aAAkD;EAC1E,MAAM,WAAiC,EAAE;EACzC,MAAM,YAA2B,EAAE;AAEnC,cAAY,SAAS,MAAM,UAAU;AACnC,aAAU,KAAK,KAAK;AACpB,OAAI,KAAK,UAAU,KAAK,MAAM;IAC5B,IAAI,eAAwC,YAAY,QAAQ;AAChE,QAAI,CAAC,gBAAgB,aAAa,UAAU,KAAK,QAC/C,gBAAe;AAGjB,aAAS,KAAK;KACZ,iBAAiB;KACjB,cAAc,eACV,IAAI,aAAa,aAAa,GAC9B;KACL,CAAC;;IAEJ;AAEF,SAAO;;;;;;;;;;;;;;;;;;;;;CAsBT,MAAM,MACJ,QACA,YACsB;AACtB,MAAI,EAAE,KAAK,kBAAkB,QAC3B,OAAM,IAAI,MAAM,oBAAoB,KAAK,eAAe,YAAY;EAGtE,MAAM,cAAyB,OAAO,KAAK;AAK3C,MAAI,CAH8B,YAAY,OAAO,YACnD,cAAc,QAAQ,CACvB,CAEC,OAAM,IAAI,MAAM,gDAAgD;EAGlE,MAAM,gBAAgB,KAAK,0BACzB,YACD;EAGD,MAAM,kBAAkB,MAAM,QAAQ,IACpC,cAAc,IAAI,OAAO,sBAAsB;GAC7C,iBAAiB,MAAM,KAAK,uBAC1B,iBAAiB,gBAClB;GACD,cAAc,iBAAiB;GAC/B;GACD,EAAE,CACJ;EAID,MAAM,uBAAuB,MAAM,QAAQ,IACzC,gBAAgB,KAAK,eACnB,KAAK,wBAAwB;GAC3B,iBAAiB,WAAW;GAC5B,cAAc,WAAW;GACzB;GACD,CAAC,CACH,CACF;AAeD,SAAO,EACL,UAbe,MAAM,QAAQ,IAC7B,qBAAqB,KAAK,cACxB,KAAK,cAAc;GACjB,cAAc,UAAU;GACxB,aAAa;IACX,mBAAmB,UAAU;IAC7B,2BAA2B,UAAU;IACtC;GACF,CAAC,CACH,CACF,EAIA;;;;;;;;;;;;CAaH,MAAc,uBACZ,aACA,YACyC;EACzC,MAAM,gBAAgB,KAAK,qBAAqB,YAAY;EAE5D,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAAC,mCAAmC,CAAC,CAC/C,WAAW,EACV,eAAe,EAAE,MAAM,mCAAmC,MAAM,EACjE,CAAC;EAMJ,MAAM,MAAM,MAJE,iCAAiC,KAAK,iBAAiB,CAAC,KACpE,KAAK,iBACN,CAEuB,OACtB,EACE,cAAc,eACf,EACD,YAAY,SAAS,aAAa,CACnC;AAED,MACE,EACE,eAAe,OACf,0BAA0B,OAC1B,cAAc,KAGhB,OAAM,IAAI,MAAM,8BAA8B,KAAK,UAAU,IAAI,GAAG;EAGtE,MAAM,qBAAqB;EAG3B,MAAM,gBAAgB,MAAM,KAAK,yBAC/B,mBAAmB,SACpB;EACD,MAAM,eAAe,KAAK,uBAAuB,CAC/C,GAAG,mBAAmB,UACtB,GAAG,cACJ,CAAC;AAEF,SAAO;GACL,GAAG;GACH,UAAU;GACX;;;;;;;;;CAUH,MAAc,yBACZ,UACwB;AAaxB,UAXkC,MAAM,QAAQ,IAC9C,SAAS,IAAI,OAAO,YAAY;AAE9B,WADwB,MAAM,KAAK,UAAU,OAAO,QAAQ,EACrC;IACvB,CACH,EAEyD,KACvD,aAAa,SAAS,YACxB;;;;;;;;;;;;;;;CAkBH,MAAc,wBAAwB,EACpC,iBACA,cACA,cAK2C;EAC3C,MAAM,mBAAmB,KAAK,IAC3B,UAAU,CAAC,+BAA+B,CAAC,CAC3C,WAAW,EACV,eAAe,EAAE,MAAM,+BAA+B,MAAM,EAC7D,CAAC;EAEJ,MAAM,QAAQ,6BAA6B,KAAK,iBAAiB,CAAC,KAChE,KAAK,iBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM,uDAAuD;EAEzE,MAAM,MAAO,MAAM,MAAM,OACvB;GACE,kBAAkB,gBAAgB;GAClC,eAAe,cAAc,WAAW;GACxC,eAAe,gBAAgB,SAAS,KAAK,KAAK;GACnD,EACD,YAAY,SAAS,wBAAwB,CAC9C;AAcD,SAAO;GACL;GACA,mBATwB,MAAM,KAAK,0BAA0B;IAC7D,oBAAoB,gBAAgB;IACpC,2BAA2B,IAAI;IAC/B,cAAc,gBAAgB;IAC9B;IACD,CAAC;GAKA,2BAA2B,IAAI;GAChC;;;;;;;;;;;;;CAcH,MAAc,0BAA0B,EACtC,oBACA,2BACA,cACA,cAMkB;AAclB,SAT6B,MAJE,mCAAmC,KAChE,KAAK,IACN,CAAC,KAAK,KAAK,mBAAmB,CAE2B,OACxD;GACE,YAAY;GACZ,6BAA6B,0BAA0B,KAAK,KAAK;GACjE,eAAe,aAAa,KAAK,KAAK;GACvC,EACD,YAAY,SAAS,sBAAsB,CAC5C;;;;;;;;;;;;;;CAiBH,MAAc,cAAc,EAC1B,cACA,aACA,cAWkB;EAClB,MAAM,QAAQ,sBAAsB,KAAK,KAAK,IAAI,CAAC,KACjD,KAAK,mBACN;AAED,MAAI,OAAO,cAAc,YAAY,SACnC,OAAM,IAAI,MAAM,uDAAuD;AAWzE,SATY,MAAM,MAAM,OACtB;GACE,uBAAuB,YAAY,0BAA0B,KAAK,KAAK;GACvE,YAAY,YAAY;GACxB,cAAc,cAAc,WAAW;GACxC,EACD,YAAY,SAAS,iBAAiB,CACvC;;;;;;;;;;;;;CAgBH,OAAO,QACL,KACA,WACA,SAG8B;AAC9B,SAAO,IAAI,KAAK;GACd;GACA;GACA,GAAG;GACJ,CAAC"}

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_prompt.cjs

@@ -1,8 +1,8 @@
-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/experimental/chains/violation_of_expectations/violation_of_expectations_prompt.ts
-const PREDICT_NEXT_USER_MESSAGE_PROMPT = /* @__PURE__ */ __langchain_core_prompts.PromptTemplate.fromTemplate(`
+const PREDICT_NEXT_USER_MESSAGE_PROMPT = /* @__PURE__ */ _langchain_core_prompts.PromptTemplate.fromTemplate(`
 You have been tasked with coming up with insights and data-points based on a chat history between a human and an AI.
 Given the user's chat history provide the following:
 - Concise reasoning about the users internal mental state.
@@ -10,7 +10,7 @@ Given the user's chat history provide the following:
 - A concise list of any additional insights that would be useful to improve prediction.
 --------
 Chat History: {chat_history}`);
-const PREDICTION_VIOLATIONS_PROMPT = /* @__PURE__ */ __langchain_core_prompts.PromptTemplate.fromTemplate(`You have been given a prediction and an actual message from a human and AI conversation.
+const PREDICTION_VIOLATIONS_PROMPT = /* @__PURE__ */ _langchain_core_prompts.PromptTemplate.fromTemplate(`You have been given a prediction and an actual message from a human and AI conversation.
 Using the prediction, actual message, and additional user insights, generate the following:
 - How exactly was the original prediction violated? Which parts were wrong? State the exact differences.
 - If there were errors with the prediction, what were they and why?
@@ -22,7 +22,7 @@ Actual Output: {actual_output}
 User Insights: {user_insights}
 --------
 `);
-const GENERATE_REVISED_PREDICTION_PROMPT = /* @__PURE__ */ __langchain_core_prompts.PromptTemplate.fromTemplate(`
+const GENERATE_REVISED_PREDICTION_PROMPT = /* @__PURE__ */ _langchain_core_prompts.PromptTemplate.fromTemplate(`
 You have been tasked with revising a prediction on what a user might say in a chat conversation.
 --------
 Your previous prediction: {prediction}
@@ -33,7 +33,7 @@ Key insights to the user: {user_insights}
 --------
 Given the above, revise your prediction to be more accurate.
 Revised Prediction:`);
-const GENERATE_FACTS_PROMPT = /* @__PURE__ */ __langchain_core_prompts.PromptTemplate.fromTemplate(`
+const GENERATE_FACTS_PROMPT = /* @__PURE__ */ _langchain_core_prompts.PromptTemplate.fromTemplate(`
 Given a user message, an LLM generated prediction of what that message might be, and a list of violations which the prediction made compared to the actual message, generate a fact about the user, relevant to the users message.
 --------
 Prediction violations: {prediction_violations}

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_prompt.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"violation_of_expectations_prompt.cjs","names":["PromptTemplate"],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_prompt.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const PREDICT_NEXT_USER_MESSAGE_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`\nYou have been tasked with coming up with insights and data-points based on a chat history between a human and an AI.\nGiven the user's chat history provide the following:\n- Concise reasoning about the users internal mental state.\n- Your prediction on how they will respond to the AI's most recent message.\n- A concise list of any additional insights that would be useful to improve prediction.\n--------\nChat History: {chat_history}`);\n\nexport const PREDICTION_VIOLATIONS_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`You have been given a prediction and an actual message from a human and AI conversation.\nUsing the prediction, actual message, and additional user insights, generate the following:\n- How exactly was the original prediction violated? Which parts were wrong? State the exact differences.\n- If there were errors with the prediction, what were they and why?\n--------\nPredicted Output: {predicted_output}\n--------\nActual Output: {actual_output}\n--------\nUser Insights: {user_insights}\n--------\n`);\n\nexport const GENERATE_REVISED_PREDICTION_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`\nYou have been tasked with revising a prediction on what a user might say in a chat conversation.\n--------\nYour previous prediction: {prediction}\n--------\nWays in which your prediction was off: {explained_prediction_errors}\n--------\nKey insights to the user: {user_insights}\n--------\nGiven the above, revise your prediction to be more accurate.\nRevised Prediction:`);\n\nexport const GENERATE_FACTS_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`\nGiven a user message, an LLM generated prediction of what that message might be, and a list of violations which the prediction made compared to the actual message, generate a fact about the user, relevant to the users message.\n--------\nPrediction violations: {prediction_violations}\n--------\nRevised prediction: {prediction}\n--------\nActual user message: {user_message}\n--------\nRelevant fact:`);\n"],"mappings":";;;;AAEA,MAAa,mDACKA,wCAAe,aAAa,CAAC;;;;;;;4BAOnB,CAAC,CAAC;AAE9B,MAAa,+CACKA,wCAAe,aAAa,CAAC;;;;;;;;;;;AAW/C,CAAC,CAAC;AAEF,MAAa,qDACKA,wCAAe,aAAa,CAAC;;;;;;;;;;mBAU5B,CAAC,CAAC;AAErB,MAAa,wCACKA,wCAAe,aAAa,CAAC;;;;;;;;;cASjC,CAAC,CAAC"}
+{"version":3,"file":"violation_of_expectations_prompt.cjs","names":[],"sources":["../../../../src/experimental/chains/violation_of_expectations/violation_of_expectations_prompt.ts"],"sourcesContent":["import { PromptTemplate } from \"@langchain/core/prompts\";\n\nexport const PREDICT_NEXT_USER_MESSAGE_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`\nYou have been tasked with coming up with insights and data-points based on a chat history between a human and an AI.\nGiven the user's chat history provide the following:\n- Concise reasoning about the users internal mental state.\n- Your prediction on how they will respond to the AI's most recent message.\n- A concise list of any additional insights that would be useful to improve prediction.\n--------\nChat History: {chat_history}`);\n\nexport const PREDICTION_VIOLATIONS_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`You have been given a prediction and an actual message from a human and AI conversation.\nUsing the prediction, actual message, and additional user insights, generate the following:\n- How exactly was the original prediction violated? Which parts were wrong? State the exact differences.\n- If there were errors with the prediction, what were they and why?\n--------\nPredicted Output: {predicted_output}\n--------\nActual Output: {actual_output}\n--------\nUser Insights: {user_insights}\n--------\n`);\n\nexport const GENERATE_REVISED_PREDICTION_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`\nYou have been tasked with revising a prediction on what a user might say in a chat conversation.\n--------\nYour previous prediction: {prediction}\n--------\nWays in which your prediction was off: {explained_prediction_errors}\n--------\nKey insights to the user: {user_insights}\n--------\nGiven the above, revise your prediction to be more accurate.\nRevised Prediction:`);\n\nexport const GENERATE_FACTS_PROMPT =\n  /* #__PURE__ */ PromptTemplate.fromTemplate(`\nGiven a user message, an LLM generated prediction of what that message might be, and a list of violations which the prediction made compared to the actual message, generate a fact about the user, relevant to the users message.\n--------\nPrediction violations: {prediction_violations}\n--------\nRevised prediction: {prediction}\n--------\nActual user message: {user_message}\n--------\nRelevant fact:`);\n"],"mappings":";;;;AAEA,MAAa,mCACK,uDAAe,aAAa;;;;;;;8BAOhB;AAE9B,MAAa,+BACK,uDAAe,aAAa;;;;;;;;;;;EAW5C;AAEF,MAAa,qCACK,uDAAe,aAAa;;;;;;;;;;qBAUzB;AAErB,MAAa,wBACK,uDAAe,aAAa;;;;;;;;;gBAS9B"}