@langchain/classic 1.0.0-alpha.1

package/dist/agents/tool_calling/output_parser.js

@@ -0,0 +1,64 @@
+import { AgentMultiActionOutputParser } from "../types.js";
+import { OutputParserException } from "@langchain/core/output_parsers";
+import { isBaseMessage } from "@langchain/core/messages";
+
+//#region src/agents/tool_calling/output_parser.ts
+function parseAIMessageToToolAction(message) {
+	const stringifiedMessageContent = typeof message.content === "string" ? message.content : JSON.stringify(message.content);
+	let toolCalls = [];
+	if (message.tool_calls !== void 0 && message.tool_calls.length > 0) toolCalls = message.tool_calls;
+	else {
+		if (!message.additional_kwargs.tool_calls || message.additional_kwargs.tool_calls.length === 0) return {
+			returnValues: { output: message.content },
+			log: stringifiedMessageContent
+		};
+		for (const toolCall of message.additional_kwargs.tool_calls ?? []) {
+			const functionName = toolCall.function?.name;
+			try {
+				const args = JSON.parse(toolCall.function.arguments);
+				toolCalls.push({
+					name: functionName,
+					args,
+					id: toolCall.id
+				});
+			} catch (e) {
+				throw new OutputParserException(`Failed to parse tool arguments from chat model response. Text: "${JSON.stringify(toolCalls)}". ${e}`);
+			}
+		}
+	}
+	return toolCalls.map((toolCall, i) => {
+		const messageLog = i === 0 ? [message] : [];
+		const log = `Invoking "${toolCall.name}" with ${JSON.stringify(toolCall.args ?? {})}\n${stringifiedMessageContent}`;
+		return {
+			tool: toolCall.name,
+			toolInput: toolCall.args,
+			toolCallId: toolCall.id ?? "",
+			log,
+			messageLog
+		};
+	});
+}
+var ToolCallingAgentOutputParser = class extends AgentMultiActionOutputParser {
+	lc_namespace = [
+		"langchain",
+		"agents",
+		"tool_calling"
+	];
+	static lc_name() {
+		return "ToolCallingAgentOutputParser";
+	}
+	async parse(text) {
+		throw new Error(`ToolCallingAgentOutputParser can only parse messages.\nPassed input: ${text}`);
+	}
+	async parseResult(generations) {
+		if ("message" in generations[0] && isBaseMessage(generations[0].message)) return parseAIMessageToToolAction(generations[0].message);
+		throw new Error("parseResult on ToolCallingAgentOutputParser only works on ChatGeneration output");
+	}
+	getFormatInstructions() {
+		throw new Error("getFormatInstructions not implemented inside ToolCallingAgentOutputParser.");
+	}
+};
+
+//#endregion
+export { ToolCallingAgentOutputParser };
+//# sourceMappingURL=output_parser.js.map

package/dist/agents/tool_calling/output_parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"output_parser.js","names":["message: AIMessage","toolCalls: ToolCall[]","e: any","text: string","generations: ChatGeneration[]"],"sources":["../../../src/agents/tool_calling/output_parser.ts"],"sourcesContent":["import { AgentAction, AgentFinish, AgentStep } from \"@langchain/core/agents\";\nimport {\n  AIMessage,\n  BaseMessage,\n  isBaseMessage,\n} from \"@langchain/core/messages\";\nimport { OutputParserException } from \"@langchain/core/output_parsers\";\nimport { ChatGeneration } from \"@langchain/core/outputs\";\nimport { ToolCall } from \"@langchain/core/messages/tool\";\nimport { AgentMultiActionOutputParser } from \"../types.js\";\n\n/**\n * Type that represents an agent action with an optional message log.\n */\nexport type ToolsAgentAction = AgentAction & {\n  toolCallId: string;\n  messageLog?: BaseMessage[];\n};\n\nexport type ToolsAgentStep = AgentStep & {\n  action: ToolsAgentAction;\n};\n\nexport function parseAIMessageToToolAction(\n  message: AIMessage\n): ToolsAgentAction[] | AgentFinish {\n  const stringifiedMessageContent =\n    typeof message.content === \"string\"\n      ? message.content\n      : JSON.stringify(message.content);\n  let toolCalls: ToolCall[] = [];\n  if (message.tool_calls !== undefined && message.tool_calls.length > 0) {\n    toolCalls = message.tool_calls;\n  } else {\n    if (\n      !message.additional_kwargs.tool_calls ||\n      message.additional_kwargs.tool_calls.length === 0\n    ) {\n      return {\n        returnValues: { output: message.content },\n        log: stringifiedMessageContent,\n      };\n    }\n    // Best effort parsing\n    for (const toolCall of message.additional_kwargs.tool_calls ?? []) {\n      const functionName = toolCall.function?.name;\n      try {\n        const args = JSON.parse(toolCall.function.arguments);\n        toolCalls.push({ name: functionName, args, id: toolCall.id });\n        // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      } catch (e: any) {\n        throw new OutputParserException(\n          `Failed to parse tool arguments from chat model response. Text: \"${JSON.stringify(\n            toolCalls\n          )}\". ${e}`\n        );\n      }\n    }\n  }\n  return toolCalls.map((toolCall, i) => {\n    const messageLog = i === 0 ? [message] : [];\n    const log = `Invoking \"${toolCall.name}\" with ${JSON.stringify(\n      toolCall.args ?? {}\n    )}\\n${stringifiedMessageContent}`;\n    return {\n      tool: toolCall.name as string,\n      toolInput: toolCall.args,\n      toolCallId: toolCall.id ?? \"\",\n      log,\n      messageLog,\n    };\n  });\n}\n\nexport class ToolCallingAgentOutputParser extends AgentMultiActionOutputParser {\n  lc_namespace = [\"langchain\", \"agents\", \"tool_calling\"];\n\n  static lc_name() {\n    return \"ToolCallingAgentOutputParser\";\n  }\n\n  async parse(text: string): Promise<AgentAction[] | AgentFinish> {\n    throw new Error(\n      `ToolCallingAgentOutputParser can only parse messages.\\nPassed input: ${text}`\n    );\n  }\n\n  async parseResult(generations: ChatGeneration[]) {\n    if (\"message\" in generations[0] && isBaseMessage(generations[0].message)) {\n      return parseAIMessageToToolAction(generations[0].message);\n    }\n    throw new Error(\n      \"parseResult on ToolCallingAgentOutputParser only works on ChatGeneration output\"\n    );\n  }\n\n  getFormatInstructions(): string {\n    throw new Error(\n      \"getFormatInstructions not implemented inside ToolCallingAgentOutputParser.\"\n    );\n  }\n}\n"],"mappings":";;;;;AAuBA,SAAgB,2BACdA,SACkC;CAClC,MAAM,4BACJ,OAAO,QAAQ,YAAY,WACvB,QAAQ,UACR,KAAK,UAAU,QAAQ,QAAQ;CACrC,IAAIC,YAAwB,CAAE;AAC9B,KAAI,QAAQ,eAAe,UAAa,QAAQ,WAAW,SAAS,GAClE,YAAY,QAAQ;MACf;AACL,MACE,CAAC,QAAQ,kBAAkB,cAC3B,QAAQ,kBAAkB,WAAW,WAAW,EAEhD,QAAO;GACL,cAAc,EAAE,QAAQ,QAAQ,QAAS;GACzC,KAAK;EACN;AAGH,OAAK,MAAM,YAAY,QAAQ,kBAAkB,cAAc,CAAE,GAAE;GACjE,MAAM,eAAe,SAAS,UAAU;AACxC,OAAI;IACF,MAAM,OAAO,KAAK,MAAM,SAAS,SAAS,UAAU;IACpD,UAAU,KAAK;KAAE,MAAM;KAAc;KAAM,IAAI,SAAS;IAAI,EAAC;GAE9D,SAAQC,GAAQ;AACf,UAAM,IAAI,sBACR,CAAC,gEAAgE,EAAE,KAAK,UACtE,UACD,CAAC,GAAG,EAAE,GAAG;GAEb;EACF;CACF;AACD,QAAO,UAAU,IAAI,CAAC,UAAU,MAAM;EACpC,MAAM,aAAa,MAAM,IAAI,CAAC,OAAQ,IAAG,CAAE;EAC3C,MAAM,MAAM,CAAC,UAAU,EAAE,SAAS,KAAK,OAAO,EAAE,KAAK,UACnD,SAAS,QAAQ,CAAE,EACpB,CAAC,EAAE,EAAE,2BAA2B;AACjC,SAAO;GACL,MAAM,SAAS;GACf,WAAW,SAAS;GACpB,YAAY,SAAS,MAAM;GAC3B;GACA;EACD;CACF,EAAC;AACH;AAED,IAAa,+BAAb,cAAkD,6BAA6B;CAC7E,eAAe;EAAC;EAAa;EAAU;CAAe;CAEtD,OAAO,UAAU;AACf,SAAO;CACR;CAED,MAAM,MAAMC,MAAoD;AAC9D,QAAM,IAAI,MACR,CAAC,qEAAqE,EAAE,MAAM;CAEjF;CAED,MAAM,YAAYC,aAA+B;AAC/C,MAAI,aAAa,YAAY,MAAM,cAAc,YAAY,GAAG,QAAQ,CACtE,QAAO,2BAA2B,YAAY,GAAG,QAAQ;AAE3D,QAAM,IAAI,MACR;CAEH;CAED,wBAAgC;AAC9B,QAAM,IAAI,MACR;CAEH;AACF"}

package/dist/agents/toolkits/base.cjs

@@ -0,0 +1,2 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const __langchain_core_tools = require_rolldown_runtime.__toESM(require("@langchain/core/tools"));

package/dist/agents/toolkits/base.d.cts

@@ -0,0 +1,2 @@
+import { BaseToolkit as Toolkit } from "@langchain/core/tools";
+export { Toolkit };

package/dist/agents/toolkits/base.d.ts

@@ -0,0 +1,2 @@
+import { BaseToolkit as Toolkit } from "@langchain/core/tools";
+export { Toolkit };

package/dist/agents/toolkits/base.js

@@ -0,0 +1,3 @@
+import { BaseToolkit as Toolkit } from "@langchain/core/tools";
+
+export { Toolkit };

package/dist/agents/toolkits/conversational_retrieval/openai_functions.cjs

@@ -0,0 +1,45 @@
+const require_summary_buffer = require('../../../memory/summary_buffer.cjs');
+const require_initialize = require('../../initialize.cjs');
+const require_token_buffer_memory = require('./token_buffer_memory.cjs');
+
+//#region src/agents/toolkits/conversational_retrieval/openai_functions.ts
+/**
+* Asynchronous function that creates a conversational retrieval agent
+* using a language model, tools, and options. It initializes the buffer
+* memory based on the provided options and initializes the AgentExecutor
+* with the tools, language model, and memory.
+* @param llm Instance of ChatOpenAI used as the language model for the agent.
+* @param tools Array of StructuredTool instances used by the agent.
+* @param options Optional ConversationalRetrievalAgentOptions to customize the agent.
+* @returns A Promise that resolves to an initialized AgentExecutor.
+*/
+async function createConversationalRetrievalAgent(llm, tools, options) {
+	const { rememberIntermediateSteps = true, memoryKey = "chat_history", outputKey = "output", inputKey = "input", prefix, verbose } = options ?? {};
+	let memory;
+	if (rememberIntermediateSteps) memory = new require_token_buffer_memory.OpenAIAgentTokenBufferMemory({
+		memoryKey,
+		llm,
+		outputKey,
+		inputKey
+	});
+	else memory = new require_summary_buffer.ConversationSummaryBufferMemory({
+		memoryKey,
+		llm,
+		maxTokenLimit: 12e3,
+		returnMessages: true,
+		outputKey,
+		inputKey
+	});
+	const executor = await require_initialize.initializeAgentExecutorWithOptions(tools, llm, {
+		agentType: "openai-functions",
+		memory,
+		verbose,
+		returnIntermediateSteps: rememberIntermediateSteps,
+		agentArgs: { prefix: prefix ?? `Do your best to answer the questions. Feel free to use any tools available to look up relevant information, only if necessary.` }
+	});
+	return executor;
+}
+
+//#endregion
+exports.createConversationalRetrievalAgent = createConversationalRetrievalAgent;
+//# sourceMappingURL=openai_functions.cjs.map

package/dist/agents/toolkits/conversational_retrieval/openai_functions.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai_functions.cjs","names":["llm: ChatOpenAI","tools: StructuredToolInterface[]","options?: ConversationalRetrievalAgentOptions","OpenAIAgentTokenBufferMemory","ConversationSummaryBufferMemory","initializeAgentExecutorWithOptions"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/openai_functions.ts"],"sourcesContent":["import { StructuredToolInterface } from \"@langchain/core/tools\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport { ConversationSummaryBufferMemory } from \"../../../memory/summary_buffer.js\";\nimport { initializeAgentExecutorWithOptions } from \"../../initialize.js\";\nimport { OpenAIAgentTokenBufferMemory } from \"./token_buffer_memory.js\";\n\nexport type ConversationalRetrievalAgentOptions = {\n  rememberIntermediateSteps?: boolean;\n  memoryKey?: string;\n  outputKey?: string;\n  inputKey?: string;\n  prefix?: string;\n  verbose?: boolean;\n};\n\n/**\n * Asynchronous function that creates a conversational retrieval agent\n * using a language model, tools, and options. It initializes the buffer\n * memory based on the provided options and initializes the AgentExecutor\n * with the tools, language model, and memory.\n * @param llm Instance of ChatOpenAI used as the language model for the agent.\n * @param tools Array of StructuredTool instances used by the agent.\n * @param options Optional ConversationalRetrievalAgentOptions to customize the agent.\n * @returns A Promise that resolves to an initialized AgentExecutor.\n */\nexport async function createConversationalRetrievalAgent(\n  llm: ChatOpenAI,\n  tools: StructuredToolInterface[],\n  options?: ConversationalRetrievalAgentOptions\n) {\n  const {\n    rememberIntermediateSteps = true,\n    memoryKey = \"chat_history\",\n    outputKey = \"output\",\n    inputKey = \"input\",\n    prefix,\n    verbose,\n  } = options ?? {};\n  let memory;\n  if (rememberIntermediateSteps) {\n    memory = new OpenAIAgentTokenBufferMemory({\n      memoryKey,\n      llm,\n      outputKey,\n      inputKey,\n    });\n  } else {\n    memory = new ConversationSummaryBufferMemory({\n      memoryKey,\n      llm,\n      maxTokenLimit: 12000,\n      returnMessages: true,\n      outputKey,\n      inputKey,\n    });\n  }\n  const executor = await initializeAgentExecutorWithOptions(tools, llm, {\n    agentType: \"openai-functions\",\n    memory,\n    verbose,\n    returnIntermediateSteps: rememberIntermediateSteps,\n    agentArgs: {\n      prefix:\n        prefix ??\n        `Do your best to answer the questions. Feel free to use any tools available to look up relevant information, only if necessary.`,\n    },\n  });\n  return executor;\n}\n"],"mappings":";;;;;;;;;;;;;;;AAyBA,eAAsB,mCACpBA,KACAC,OACAC,SACA;CACA,MAAM,EACJ,4BAA4B,MAC5B,YAAY,gBACZ,YAAY,UACZ,WAAW,SACX,QACA,SACD,GAAG,WAAW,CAAE;CACjB,IAAI;AACJ,KAAI,2BACF,SAAS,IAAIC,yDAA6B;EACxC;EACA;EACA;EACA;CACD;MAED,SAAS,IAAIC,uDAAgC;EAC3C;EACA;EACA,eAAe;EACf,gBAAgB;EAChB;EACA;CACD;CAEH,MAAM,WAAW,MAAMC,sDAAmC,OAAO,KAAK;EACpE,WAAW;EACX;EACA;EACA,yBAAyB;EACzB,WAAW,EACT,QACE,UACA,CAAC,8HAA8H,CAAC,CACnI;CACF,EAAC;AACF,QAAO;AACR"}

package/dist/agents/toolkits/conversational_retrieval/openai_functions.d.cts

@@ -0,0 +1,27 @@
+import { AgentExecutor } from "../../executor.cjs";
+import { StructuredToolInterface } from "@langchain/core/tools";
+import { ChatOpenAI } from "@langchain/openai";
+
+//#region src/agents/toolkits/conversational_retrieval/openai_functions.d.ts
+type ConversationalRetrievalAgentOptions = {
+  rememberIntermediateSteps?: boolean;
+  memoryKey?: string;
+  outputKey?: string;
+  inputKey?: string;
+  prefix?: string;
+  verbose?: boolean;
+};
+/**
+ * Asynchronous function that creates a conversational retrieval agent
+ * using a language model, tools, and options. It initializes the buffer
+ * memory based on the provided options and initializes the AgentExecutor
+ * with the tools, language model, and memory.
+ * @param llm Instance of ChatOpenAI used as the language model for the agent.
+ * @param tools Array of StructuredTool instances used by the agent.
+ * @param options Optional ConversationalRetrievalAgentOptions to customize the agent.
+ * @returns A Promise that resolves to an initialized AgentExecutor.
+ */
+declare function createConversationalRetrievalAgent(llm: ChatOpenAI, tools: StructuredToolInterface[], options?: ConversationalRetrievalAgentOptions): Promise<AgentExecutor>;
+//#endregion
+export { ConversationalRetrievalAgentOptions, createConversationalRetrievalAgent };
+//# sourceMappingURL=openai_functions.d.cts.map

package/dist/agents/toolkits/conversational_retrieval/openai_functions.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai_functions.d.cts","names":["StructuredToolInterface","ChatOpenAI","ConversationalRetrievalAgentOptions","createConversationalRetrievalAgent","______executor_js0","AgentExecutor","Promise"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/openai_functions.d.ts"],"sourcesContent":["import { StructuredToolInterface } from \"@langchain/core/tools\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nexport type ConversationalRetrievalAgentOptions = {\n    rememberIntermediateSteps?: boolean;\n    memoryKey?: string;\n    outputKey?: string;\n    inputKey?: string;\n    prefix?: string;\n    verbose?: boolean;\n};\n/**\n * Asynchronous function that creates a conversational retrieval agent\n * using a language model, tools, and options. It initializes the buffer\n * memory based on the provided options and initializes the AgentExecutor\n * with the tools, language model, and memory.\n * @param llm Instance of ChatOpenAI used as the language model for the agent.\n * @param tools Array of StructuredTool instances used by the agent.\n * @param options Optional ConversationalRetrievalAgentOptions to customize the agent.\n * @returns A Promise that resolves to an initialized AgentExecutor.\n */\nexport declare function createConversationalRetrievalAgent(llm: ChatOpenAI, tools: StructuredToolInterface[], options?: ConversationalRetrievalAgentOptions): Promise<import(\"../../executor.js\").AgentExecutor>;\n"],"mappings":";;;;;KAEYE,mCAAAA;;;EAAAA,SAAAA,CAAAA,EAAAA,MAAAA;EAkBYC,QAAAA,CAAAA,EAAAA,MAAAA;EAAkC,MAAA,CAAA,EAAA,MAAA;EAAA,OAAMF,CAAAA,EAAAA,OAAAA;CAAU;;;;AAA2F;;;;;;;iBAA7IE,kCAAAA,MAAwCF,mBAAmBD,qCAAqCE,sCAAsCI,QAAH,aAAA"}

package/dist/agents/toolkits/conversational_retrieval/openai_functions.d.ts

@@ -0,0 +1,27 @@
+import { AgentExecutor } from "../../executor.js";
+import { StructuredToolInterface } from "@langchain/core/tools";
+import { ChatOpenAI } from "@langchain/openai";
+
+//#region src/agents/toolkits/conversational_retrieval/openai_functions.d.ts
+type ConversationalRetrievalAgentOptions = {
+  rememberIntermediateSteps?: boolean;
+  memoryKey?: string;
+  outputKey?: string;
+  inputKey?: string;
+  prefix?: string;
+  verbose?: boolean;
+};
+/**
+ * Asynchronous function that creates a conversational retrieval agent
+ * using a language model, tools, and options. It initializes the buffer
+ * memory based on the provided options and initializes the AgentExecutor
+ * with the tools, language model, and memory.
+ * @param llm Instance of ChatOpenAI used as the language model for the agent.
+ * @param tools Array of StructuredTool instances used by the agent.
+ * @param options Optional ConversationalRetrievalAgentOptions to customize the agent.
+ * @returns A Promise that resolves to an initialized AgentExecutor.
+ */
+declare function createConversationalRetrievalAgent(llm: ChatOpenAI, tools: StructuredToolInterface[], options?: ConversationalRetrievalAgentOptions): Promise<AgentExecutor>;
+//#endregion
+export { ConversationalRetrievalAgentOptions, createConversationalRetrievalAgent };
+//# sourceMappingURL=openai_functions.d.ts.map

package/dist/agents/toolkits/conversational_retrieval/openai_functions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai_functions.d.ts","names":["StructuredToolInterface","ChatOpenAI","ConversationalRetrievalAgentOptions","createConversationalRetrievalAgent","______executor_js0","AgentExecutor","Promise"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/openai_functions.d.ts"],"sourcesContent":["import { StructuredToolInterface } from \"@langchain/core/tools\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nexport type ConversationalRetrievalAgentOptions = {\n    rememberIntermediateSteps?: boolean;\n    memoryKey?: string;\n    outputKey?: string;\n    inputKey?: string;\n    prefix?: string;\n    verbose?: boolean;\n};\n/**\n * Asynchronous function that creates a conversational retrieval agent\n * using a language model, tools, and options. It initializes the buffer\n * memory based on the provided options and initializes the AgentExecutor\n * with the tools, language model, and memory.\n * @param llm Instance of ChatOpenAI used as the language model for the agent.\n * @param tools Array of StructuredTool instances used by the agent.\n * @param options Optional ConversationalRetrievalAgentOptions to customize the agent.\n * @returns A Promise that resolves to an initialized AgentExecutor.\n */\nexport declare function createConversationalRetrievalAgent(llm: ChatOpenAI, tools: StructuredToolInterface[], options?: ConversationalRetrievalAgentOptions): Promise<import(\"../../executor.js\").AgentExecutor>;\n"],"mappings":";;;;;KAEYE,mCAAAA;;;EAAAA,SAAAA,CAAAA,EAAAA,MAAAA;EAkBYC,QAAAA,CAAAA,EAAAA,MAAAA;EAAkC,MAAA,CAAA,EAAA,MAAA;EAAA,OAAMF,CAAAA,EAAAA,OAAAA;CAAU;;;;AAA2F;;;;;;;iBAA7IE,kCAAAA,MAAwCF,mBAAmBD,qCAAqCE,sCAAsCI,QAAH,aAAA"}

package/dist/agents/toolkits/conversational_retrieval/openai_functions.js

@@ -0,0 +1,45 @@
+import { ConversationSummaryBufferMemory } from "../../../memory/summary_buffer.js";
+import { initializeAgentExecutorWithOptions } from "../../initialize.js";
+import { OpenAIAgentTokenBufferMemory } from "./token_buffer_memory.js";
+
+//#region src/agents/toolkits/conversational_retrieval/openai_functions.ts
+/**
+* Asynchronous function that creates a conversational retrieval agent
+* using a language model, tools, and options. It initializes the buffer
+* memory based on the provided options and initializes the AgentExecutor
+* with the tools, language model, and memory.
+* @param llm Instance of ChatOpenAI used as the language model for the agent.
+* @param tools Array of StructuredTool instances used by the agent.
+* @param options Optional ConversationalRetrievalAgentOptions to customize the agent.
+* @returns A Promise that resolves to an initialized AgentExecutor.
+*/
+async function createConversationalRetrievalAgent(llm, tools, options) {
+	const { rememberIntermediateSteps = true, memoryKey = "chat_history", outputKey = "output", inputKey = "input", prefix, verbose } = options ?? {};
+	let memory;
+	if (rememberIntermediateSteps) memory = new OpenAIAgentTokenBufferMemory({
+		memoryKey,
+		llm,
+		outputKey,
+		inputKey
+	});
+	else memory = new ConversationSummaryBufferMemory({
+		memoryKey,
+		llm,
+		maxTokenLimit: 12e3,
+		returnMessages: true,
+		outputKey,
+		inputKey
+	});
+	const executor = await initializeAgentExecutorWithOptions(tools, llm, {
+		agentType: "openai-functions",
+		memory,
+		verbose,
+		returnIntermediateSteps: rememberIntermediateSteps,
+		agentArgs: { prefix: prefix ?? `Do your best to answer the questions. Feel free to use any tools available to look up relevant information, only if necessary.` }
+	});
+	return executor;
+}
+
+//#endregion
+export { createConversationalRetrievalAgent };
+//# sourceMappingURL=openai_functions.js.map

package/dist/agents/toolkits/conversational_retrieval/openai_functions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai_functions.js","names":["llm: ChatOpenAI","tools: StructuredToolInterface[]","options?: ConversationalRetrievalAgentOptions"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/openai_functions.ts"],"sourcesContent":["import { StructuredToolInterface } from \"@langchain/core/tools\";\nimport { ChatOpenAI } from \"@langchain/openai\";\nimport { ConversationSummaryBufferMemory } from \"../../../memory/summary_buffer.js\";\nimport { initializeAgentExecutorWithOptions } from \"../../initialize.js\";\nimport { OpenAIAgentTokenBufferMemory } from \"./token_buffer_memory.js\";\n\nexport type ConversationalRetrievalAgentOptions = {\n  rememberIntermediateSteps?: boolean;\n  memoryKey?: string;\n  outputKey?: string;\n  inputKey?: string;\n  prefix?: string;\n  verbose?: boolean;\n};\n\n/**\n * Asynchronous function that creates a conversational retrieval agent\n * using a language model, tools, and options. It initializes the buffer\n * memory based on the provided options and initializes the AgentExecutor\n * with the tools, language model, and memory.\n * @param llm Instance of ChatOpenAI used as the language model for the agent.\n * @param tools Array of StructuredTool instances used by the agent.\n * @param options Optional ConversationalRetrievalAgentOptions to customize the agent.\n * @returns A Promise that resolves to an initialized AgentExecutor.\n */\nexport async function createConversationalRetrievalAgent(\n  llm: ChatOpenAI,\n  tools: StructuredToolInterface[],\n  options?: ConversationalRetrievalAgentOptions\n) {\n  const {\n    rememberIntermediateSteps = true,\n    memoryKey = \"chat_history\",\n    outputKey = \"output\",\n    inputKey = \"input\",\n    prefix,\n    verbose,\n  } = options ?? {};\n  let memory;\n  if (rememberIntermediateSteps) {\n    memory = new OpenAIAgentTokenBufferMemory({\n      memoryKey,\n      llm,\n      outputKey,\n      inputKey,\n    });\n  } else {\n    memory = new ConversationSummaryBufferMemory({\n      memoryKey,\n      llm,\n      maxTokenLimit: 12000,\n      returnMessages: true,\n      outputKey,\n      inputKey,\n    });\n  }\n  const executor = await initializeAgentExecutorWithOptions(tools, llm, {\n    agentType: \"openai-functions\",\n    memory,\n    verbose,\n    returnIntermediateSteps: rememberIntermediateSteps,\n    agentArgs: {\n      prefix:\n        prefix ??\n        `Do your best to answer the questions. Feel free to use any tools available to look up relevant information, only if necessary.`,\n    },\n  });\n  return executor;\n}\n"],"mappings":";;;;;;;;;;;;;;;AAyBA,eAAsB,mCACpBA,KACAC,OACAC,SACA;CACA,MAAM,EACJ,4BAA4B,MAC5B,YAAY,gBACZ,YAAY,UACZ,WAAW,SACX,QACA,SACD,GAAG,WAAW,CAAE;CACjB,IAAI;AACJ,KAAI,2BACF,SAAS,IAAI,6BAA6B;EACxC;EACA;EACA;EACA;CACD;MAED,SAAS,IAAI,gCAAgC;EAC3C;EACA;EACA,eAAe;EACf,gBAAgB;EAChB;EACA;CACD;CAEH,MAAM,WAAW,MAAM,mCAAmC,OAAO,KAAK;EACpE,WAAW;EACX;EACA;EACA,yBAAyB;EACzB,WAAW,EACT,QACE,UACA,CAAC,8HAA8H,CAAC,CACnI;CACF,EAAC;AACF,QAAO;AACR"}

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.cjs

@@ -0,0 +1,90 @@
+const require_rolldown_runtime = require('../../../_virtual/rolldown_runtime.cjs');
+const require_memory_chat_memory = require('../../../memory/chat_memory.cjs');
+const require_index = require('../../openai_functions/index.cjs');
+const __langchain_core_messages = require_rolldown_runtime.__toESM(require("@langchain/core/messages"));
+const __langchain_core_memory = require_rolldown_runtime.__toESM(require("@langchain/core/memory"));
+
+//#region src/agents/toolkits/conversational_retrieval/token_buffer_memory.ts
+/**
+* Memory used to save agent output and intermediate steps.
+*/
+var OpenAIAgentTokenBufferMemory = class extends require_memory_chat_memory.BaseChatMemory {
+	humanPrefix = "Human";
+	aiPrefix = "AI";
+	llm;
+	memoryKey = "history";
+	maxTokenLimit = 12e3;
+	returnMessages = true;
+	outputKey = "output";
+	intermediateStepsKey = "intermediateSteps";
+	constructor(fields) {
+		super(fields);
+		this.humanPrefix = fields.humanPrefix ?? this.humanPrefix;
+		this.aiPrefix = fields.aiPrefix ?? this.aiPrefix;
+		this.llm = fields.llm;
+		this.memoryKey = fields.memoryKey ?? this.memoryKey;
+		this.maxTokenLimit = fields.maxTokenLimit ?? this.maxTokenLimit;
+		this.returnMessages = fields.returnMessages ?? this.returnMessages;
+		this.outputKey = fields.outputKey ?? this.outputKey;
+		this.intermediateStepsKey = fields.intermediateStepsKey ?? this.intermediateStepsKey;
+	}
+	get memoryKeys() {
+		return [this.memoryKey];
+	}
+	/**
+	* Retrieves the messages from the chat history.
+	* @returns Promise that resolves with the messages from the chat history.
+	*/
+	async getMessages() {
+		return this.chatHistory.getMessages();
+	}
+	/**
+	* Loads memory variables from the input values.
+	* @param _values Input values.
+	* @returns Promise that resolves with the loaded memory variables.
+	*/
+	async loadMemoryVariables(_values) {
+		const buffer = await this.getMessages();
+		if (this.returnMessages) return { [this.memoryKey]: buffer };
+		else {
+			const bufferString = (0, __langchain_core_messages.getBufferString)(buffer, this.humanPrefix, this.aiPrefix);
+			return { [this.memoryKey]: bufferString };
+		}
+	}
+	/**
+	* Saves the context of the chat, including user input, AI output, and
+	* intermediate steps. Prunes the chat history if the total token count
+	* exceeds the maximum limit.
+	* @param inputValues Input values.
+	* @param outputValues Output values.
+	* @returns Promise that resolves when the context has been saved.
+	*/
+	async saveContext(inputValues, outputValues) {
+		const inputValue = (0, __langchain_core_memory.getInputValue)(inputValues, this.inputKey);
+		const outputValue = (0, __langchain_core_memory.getOutputValue)(outputValues, this.outputKey);
+		await this.chatHistory.addUserMessage(inputValue);
+		const intermediateStepMessages = require_index._formatIntermediateSteps(outputValues[this.intermediateStepsKey]);
+		for (const message of intermediateStepMessages) await this.chatHistory.addMessage(message);
+		await this.chatHistory.addAIMessage(outputValue);
+		const currentMessages = await this.chatHistory.getMessages();
+		let tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);
+		if (tokenInfo.totalCount > this.maxTokenLimit) {
+			const prunedMemory = [];
+			while (tokenInfo.totalCount > this.maxTokenLimit) {
+				const retainedMessage = currentMessages.pop();
+				if (!retainedMessage) {
+					console.warn(`Could not prune enough messages from chat history to stay under ${this.maxTokenLimit} tokens.`);
+					break;
+				}
+				prunedMemory.push(retainedMessage);
+				tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);
+			}
+			await this.chatHistory.clear();
+			for (const message of prunedMemory) await this.chatHistory.addMessage(message);
+		}
+	}
+};
+
+//#endregion
+exports.OpenAIAgentTokenBufferMemory = OpenAIAgentTokenBufferMemory;
+//# sourceMappingURL=token_buffer_memory.cjs.map

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"token_buffer_memory.cjs","names":["BaseChatMemory","fields: OpenAIAgentTokenBufferMemoryFields","_values: InputValues","inputValues: InputValues","outputValues: OutputValues","_formatIntermediateSteps"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/token_buffer_memory.ts"],"sourcesContent":["import { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  InputValues,\n  MemoryVariables,\n  OutputValues,\n  getInputValue,\n  getOutputValue,\n} from \"@langchain/core/memory\";\nimport { getBufferString } from \"@langchain/core/messages\";\nimport {\n  BaseChatMemory,\n  BaseChatMemoryInput,\n} from \"../../../memory/chat_memory.js\";\nimport { _formatIntermediateSteps } from \"../../openai_functions/index.js\";\n\n/**\n * Type definition for the fields required to initialize an instance of\n * OpenAIAgentTokenBufferMemory.\n */\nexport type OpenAIAgentTokenBufferMemoryFields = BaseChatMemoryInput & {\n  llm: ChatOpenAI;\n  humanPrefix?: string;\n  aiPrefix?: string;\n  memoryKey?: string;\n  maxTokenLimit?: number;\n  returnMessages?: boolean;\n  outputKey?: string;\n  intermediateStepsKey?: string;\n};\n\n/**\n * Memory used to save agent output and intermediate steps.\n */\nexport class OpenAIAgentTokenBufferMemory extends BaseChatMemory {\n  humanPrefix = \"Human\";\n\n  aiPrefix = \"AI\";\n\n  llm: ChatOpenAI;\n\n  memoryKey = \"history\";\n\n  maxTokenLimit = 12000;\n\n  returnMessages = true;\n\n  outputKey = \"output\";\n\n  intermediateStepsKey = \"intermediateSteps\";\n\n  constructor(fields: OpenAIAgentTokenBufferMemoryFields) {\n    super(fields);\n    this.humanPrefix = fields.humanPrefix ?? this.humanPrefix;\n    this.aiPrefix = fields.aiPrefix ?? this.aiPrefix;\n    this.llm = fields.llm;\n    this.memoryKey = fields.memoryKey ?? this.memoryKey;\n    this.maxTokenLimit = fields.maxTokenLimit ?? this.maxTokenLimit;\n    this.returnMessages = fields.returnMessages ?? this.returnMessages;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.intermediateStepsKey =\n      fields.intermediateStepsKey ?? this.intermediateStepsKey;\n  }\n\n  get memoryKeys(): string[] {\n    return [this.memoryKey];\n  }\n\n  /**\n   * Retrieves the messages from the chat history.\n   * @returns Promise that resolves with the messages from the chat history.\n   */\n  async getMessages() {\n    return this.chatHistory.getMessages();\n  }\n\n  /**\n   * Loads memory variables from the input values.\n   * @param _values Input values.\n   * @returns Promise that resolves with the loaded memory variables.\n   */\n  async loadMemoryVariables(_values: InputValues): Promise<MemoryVariables> {\n    const buffer = await this.getMessages();\n    if (this.returnMessages) {\n      return { [this.memoryKey]: buffer };\n    } else {\n      const bufferString = getBufferString(\n        buffer,\n        this.humanPrefix,\n        this.aiPrefix\n      );\n      return { [this.memoryKey]: bufferString };\n    }\n  }\n\n  /**\n   * Saves the context of the chat, including user input, AI output, and\n   * intermediate steps. Prunes the chat history if the total token count\n   * exceeds the maximum limit.\n   * @param inputValues Input values.\n   * @param outputValues Output values.\n   * @returns Promise that resolves when the context has been saved.\n   */\n  async saveContext(\n    inputValues: InputValues,\n    outputValues: OutputValues\n  ): Promise<void> {\n    const inputValue = getInputValue(inputValues, this.inputKey);\n    const outputValue = getOutputValue(outputValues, this.outputKey);\n    await this.chatHistory.addUserMessage(inputValue);\n    const intermediateStepMessages = _formatIntermediateSteps(\n      outputValues[this.intermediateStepsKey]\n    );\n    for (const message of intermediateStepMessages) {\n      await this.chatHistory.addMessage(message);\n    }\n    await this.chatHistory.addAIMessage(outputValue);\n    const currentMessages = await this.chatHistory.getMessages();\n    let tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);\n    if (tokenInfo.totalCount > this.maxTokenLimit) {\n      const prunedMemory = [];\n      while (tokenInfo.totalCount > this.maxTokenLimit) {\n        const retainedMessage = currentMessages.pop();\n        if (!retainedMessage) {\n          console.warn(\n            `Could not prune enough messages from chat history to stay under ${this.maxTokenLimit} tokens.`\n          );\n          break;\n        }\n        prunedMemory.push(retainedMessage);\n        tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);\n      }\n      await this.chatHistory.clear();\n      for (const message of prunedMemory) {\n        await this.chatHistory.addMessage(message);\n      }\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;AAiCA,IAAa,+BAAb,cAAkDA,0CAAe;CAC/D,cAAc;CAEd,WAAW;CAEX;CAEA,YAAY;CAEZ,gBAAgB;CAEhB,iBAAiB;CAEjB,YAAY;CAEZ,uBAAuB;CAEvB,YAAYC,QAA4C;EACtD,MAAM,OAAO;EACb,KAAK,cAAc,OAAO,eAAe,KAAK;EAC9C,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,MAAM,OAAO;EAClB,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,gBAAgB,OAAO,iBAAiB,KAAK;EAClD,KAAK,iBAAiB,OAAO,kBAAkB,KAAK;EACpD,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,uBACH,OAAO,wBAAwB,KAAK;CACvC;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,SAAU;CACxB;;;;;CAMD,MAAM,cAAc;AAClB,SAAO,KAAK,YAAY,aAAa;CACtC;;;;;;CAOD,MAAM,oBAAoBC,SAAgD;EACxE,MAAM,SAAS,MAAM,KAAK,aAAa;AACvC,MAAI,KAAK,eACP,QAAO,GAAG,KAAK,YAAY,OAAQ;OAC9B;GACL,MAAM,8DACJ,QACA,KAAK,aACL,KAAK,SACN;AACD,UAAO,GAAG,KAAK,YAAY,aAAc;EAC1C;CACF;;;;;;;;;CAUD,MAAM,YACJC,aACAC,cACe;EACf,MAAM,wDAA2B,aAAa,KAAK,SAAS;EAC5D,MAAM,0DAA6B,cAAc,KAAK,UAAU;EAChE,MAAM,KAAK,YAAY,eAAe,WAAW;EACjD,MAAM,2BAA2BC,uCAC/B,aAAa,KAAK,sBACnB;AACD,OAAK,MAAM,WAAW,0BACpB,MAAM,KAAK,YAAY,WAAW,QAAQ;EAE5C,MAAM,KAAK,YAAY,aAAa,YAAY;EAChD,MAAM,kBAAkB,MAAM,KAAK,YAAY,aAAa;EAC5D,IAAI,YAAY,MAAM,KAAK,IAAI,yBAAyB,gBAAgB;AACxE,MAAI,UAAU,aAAa,KAAK,eAAe;GAC7C,MAAM,eAAe,CAAE;AACvB,UAAO,UAAU,aAAa,KAAK,eAAe;IAChD,MAAM,kBAAkB,gBAAgB,KAAK;AAC7C,QAAI,CAAC,iBAAiB;KACpB,QAAQ,KACN,CAAC,gEAAgE,EAAE,KAAK,cAAc,QAAQ,CAAC,CAChG;AACD;IACD;IACD,aAAa,KAAK,gBAAgB;IAClC,YAAY,MAAM,KAAK,IAAI,yBAAyB,gBAAgB;GACrE;GACD,MAAM,KAAK,YAAY,OAAO;AAC9B,QAAK,MAAM,WAAW,cACpB,MAAM,KAAK,YAAY,WAAW,QAAQ;EAE7C;CACF;AACF"}

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.d.cts

@@ -0,0 +1,59 @@
+import { BaseChatMemory, BaseChatMemoryInput } from "../../../memory/chat_memory.cjs";
+import * as _langchain_core_messages0 from "@langchain/core/messages";
+import { InputValues, MemoryVariables, OutputValues } from "@langchain/core/memory";
+import { ChatOpenAI } from "@langchain/openai";
+
+//#region src/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts
+
+/**
+ * Type definition for the fields required to initialize an instance of
+ * OpenAIAgentTokenBufferMemory.
+ */
+type OpenAIAgentTokenBufferMemoryFields = BaseChatMemoryInput & {
+  llm: ChatOpenAI;
+  humanPrefix?: string;
+  aiPrefix?: string;
+  memoryKey?: string;
+  maxTokenLimit?: number;
+  returnMessages?: boolean;
+  outputKey?: string;
+  intermediateStepsKey?: string;
+};
+/**
+ * Memory used to save agent output and intermediate steps.
+ */
+declare class OpenAIAgentTokenBufferMemory extends BaseChatMemory {
+  humanPrefix: string;
+  aiPrefix: string;
+  llm: ChatOpenAI;
+  memoryKey: string;
+  maxTokenLimit: number;
+  returnMessages: boolean;
+  outputKey: string;
+  intermediateStepsKey: string;
+  constructor(fields: OpenAIAgentTokenBufferMemoryFields);
+  get memoryKeys(): string[];
+  /**
+   * Retrieves the messages from the chat history.
+   * @returns Promise that resolves with the messages from the chat history.
+   */
+  getMessages(): Promise<_langchain_core_messages0.BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[]>;
+  /**
+   * Loads memory variables from the input values.
+   * @param _values Input values.
+   * @returns Promise that resolves with the loaded memory variables.
+   */
+  loadMemoryVariables(_values: InputValues): Promise<MemoryVariables>;
+  /**
+   * Saves the context of the chat, including user input, AI output, and
+   * intermediate steps. Prunes the chat history if the total token count
+   * exceeds the maximum limit.
+   * @param inputValues Input values.
+   * @param outputValues Output values.
+   * @returns Promise that resolves when the context has been saved.
+   */
+  saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;
+}
+//#endregion
+export { OpenAIAgentTokenBufferMemory };
+//# sourceMappingURL=token_buffer_memory.d.cts.map

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token_buffer_memory.d.cts","names":["ChatOpenAI","InputValues","MemoryVariables","OutputValues","BaseChatMemory","BaseChatMemoryInput","OpenAIAgentTokenBufferMemoryFields","OpenAIAgentTokenBufferMemory","_langchain_core_messages0","MessageStructure","MessageType","BaseMessage","Promise"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts"],"sourcesContent":["import { ChatOpenAI } from \"@langchain/openai\";\nimport { InputValues, MemoryVariables, OutputValues } from \"@langchain/core/memory\";\nimport { BaseChatMemory, BaseChatMemoryInput } from \"../../../memory/chat_memory.js\";\n/**\n * Type definition for the fields required to initialize an instance of\n * OpenAIAgentTokenBufferMemory.\n */\nexport type OpenAIAgentTokenBufferMemoryFields = BaseChatMemoryInput & {\n    llm: ChatOpenAI;\n    humanPrefix?: string;\n    aiPrefix?: string;\n    memoryKey?: string;\n    maxTokenLimit?: number;\n    returnMessages?: boolean;\n    outputKey?: string;\n    intermediateStepsKey?: string;\n};\n/**\n * Memory used to save agent output and intermediate steps.\n */\nexport declare class OpenAIAgentTokenBufferMemory extends BaseChatMemory {\n    humanPrefix: string;\n    aiPrefix: string;\n    llm: ChatOpenAI;\n    memoryKey: string;\n    maxTokenLimit: number;\n    returnMessages: boolean;\n    outputKey: string;\n    intermediateStepsKey: string;\n    constructor(fields: OpenAIAgentTokenBufferMemoryFields);\n    get memoryKeys(): string[];\n    /**\n     * Retrieves the messages from the chat history.\n     * @returns Promise that resolves with the messages from the chat history.\n     */\n    getMessages(): Promise<import(\"@langchain/core/messages\").BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[]>;\n    /**\n     * Loads memory variables from the input values.\n     * @param _values Input values.\n     * @returns Promise that resolves with the loaded memory variables.\n     */\n    loadMemoryVariables(_values: InputValues): Promise<MemoryVariables>;\n    /**\n     * Saves the context of the chat, including user input, AI output, and\n     * intermediate steps. Prunes the chat history if the total token count\n     * exceeds the maximum limit.\n     * @param inputValues Input values.\n     * @param outputValues Output values.\n     * @returns Promise that resolves when the context has been saved.\n     */\n    saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;\n}\n"],"mappings":";;;;;;;;;;AAOA;AAA8C,KAAlCM,kCAAAA,GAAqCD,mBAAH,GAAA;EAAA,GAAGA,EACxCL,UADwCK;EAAmB,WAC3DL,CAAAA,EAAAA,MAAAA;EAAU,QAAA,CAAA,EAAA,MAAA;EAYEO,SAAAA,CAAAA,EAAAA,MAAAA;EAA4B,aAAA,CAAA,EAAA,MAAA;EAAA,cAGxCP,CAAAA,EAAAA,OAAAA;EAAU,SAMKM,CAAAA,EAAAA,MAAAA;EAAkC,oBAAAE,CAAAA,EAAAA,MAMmDC;CAAgB;;;;AAMtEP,cArBlCK,4BAAAA,SAAqCH,cAAAA,CAqBHF;EAAe,WAAvBU,EAAAA,MAAAA;EAAO,QASzBX,EAAAA,MAAAA;EAAW,GAAgBE,EA3B/CH,UA2B+CG;EAAY,SAAGS,EAAAA,MAAAA;EAAO,aA9BpBR,EAAAA,MAAAA;EAAc,cAAA,EAAA,OAAA;;;sBAShDE;;;;;;iBAMLM,QAA0JJ,yBAAAA,CAA/GG,YANJH,yBAAAA,CAMmDC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA;;;;;;+BAMjIT,cAAcW,QAAQV;;;;;;;;;2BAS1BD,2BAA2BE,eAAeS"}

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts

@@ -0,0 +1,59 @@
+import { BaseChatMemory, BaseChatMemoryInput } from "../../../memory/chat_memory.js";
+import * as _langchain_core_messages1 from "@langchain/core/messages";
+import { InputValues, MemoryVariables, OutputValues } from "@langchain/core/memory";
+import { ChatOpenAI } from "@langchain/openai";
+
+//#region src/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts
+
+/**
+ * Type definition for the fields required to initialize an instance of
+ * OpenAIAgentTokenBufferMemory.
+ */
+type OpenAIAgentTokenBufferMemoryFields = BaseChatMemoryInput & {
+  llm: ChatOpenAI;
+  humanPrefix?: string;
+  aiPrefix?: string;
+  memoryKey?: string;
+  maxTokenLimit?: number;
+  returnMessages?: boolean;
+  outputKey?: string;
+  intermediateStepsKey?: string;
+};
+/**
+ * Memory used to save agent output and intermediate steps.
+ */
+declare class OpenAIAgentTokenBufferMemory extends BaseChatMemory {
+  humanPrefix: string;
+  aiPrefix: string;
+  llm: ChatOpenAI;
+  memoryKey: string;
+  maxTokenLimit: number;
+  returnMessages: boolean;
+  outputKey: string;
+  intermediateStepsKey: string;
+  constructor(fields: OpenAIAgentTokenBufferMemoryFields);
+  get memoryKeys(): string[];
+  /**
+   * Retrieves the messages from the chat history.
+   * @returns Promise that resolves with the messages from the chat history.
+   */
+  getMessages(): Promise<_langchain_core_messages1.BaseMessage<_langchain_core_messages1.MessageStructure, _langchain_core_messages1.MessageType>[]>;
+  /**
+   * Loads memory variables from the input values.
+   * @param _values Input values.
+   * @returns Promise that resolves with the loaded memory variables.
+   */
+  loadMemoryVariables(_values: InputValues): Promise<MemoryVariables>;
+  /**
+   * Saves the context of the chat, including user input, AI output, and
+   * intermediate steps. Prunes the chat history if the total token count
+   * exceeds the maximum limit.
+   * @param inputValues Input values.
+   * @param outputValues Output values.
+   * @returns Promise that resolves when the context has been saved.
+   */
+  saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;
+}
+//#endregion
+export { OpenAIAgentTokenBufferMemory };
+//# sourceMappingURL=token_buffer_memory.d.ts.map

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token_buffer_memory.d.ts","names":["ChatOpenAI","InputValues","MemoryVariables","OutputValues","BaseChatMemory","BaseChatMemoryInput","OpenAIAgentTokenBufferMemoryFields","OpenAIAgentTokenBufferMemory","_langchain_core_messages1","MessageStructure","MessageType","BaseMessage","Promise"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/token_buffer_memory.d.ts"],"sourcesContent":["import { ChatOpenAI } from \"@langchain/openai\";\nimport { InputValues, MemoryVariables, OutputValues } from \"@langchain/core/memory\";\nimport { BaseChatMemory, BaseChatMemoryInput } from \"../../../memory/chat_memory.js\";\n/**\n * Type definition for the fields required to initialize an instance of\n * OpenAIAgentTokenBufferMemory.\n */\nexport type OpenAIAgentTokenBufferMemoryFields = BaseChatMemoryInput & {\n    llm: ChatOpenAI;\n    humanPrefix?: string;\n    aiPrefix?: string;\n    memoryKey?: string;\n    maxTokenLimit?: number;\n    returnMessages?: boolean;\n    outputKey?: string;\n    intermediateStepsKey?: string;\n};\n/**\n * Memory used to save agent output and intermediate steps.\n */\nexport declare class OpenAIAgentTokenBufferMemory extends BaseChatMemory {\n    humanPrefix: string;\n    aiPrefix: string;\n    llm: ChatOpenAI;\n    memoryKey: string;\n    maxTokenLimit: number;\n    returnMessages: boolean;\n    outputKey: string;\n    intermediateStepsKey: string;\n    constructor(fields: OpenAIAgentTokenBufferMemoryFields);\n    get memoryKeys(): string[];\n    /**\n     * Retrieves the messages from the chat history.\n     * @returns Promise that resolves with the messages from the chat history.\n     */\n    getMessages(): Promise<import(\"@langchain/core/messages\").BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[]>;\n    /**\n     * Loads memory variables from the input values.\n     * @param _values Input values.\n     * @returns Promise that resolves with the loaded memory variables.\n     */\n    loadMemoryVariables(_values: InputValues): Promise<MemoryVariables>;\n    /**\n     * Saves the context of the chat, including user input, AI output, and\n     * intermediate steps. Prunes the chat history if the total token count\n     * exceeds the maximum limit.\n     * @param inputValues Input values.\n     * @param outputValues Output values.\n     * @returns Promise that resolves when the context has been saved.\n     */\n    saveContext(inputValues: InputValues, outputValues: OutputValues): Promise<void>;\n}\n"],"mappings":";;;;;;;;;;AAOA;AAA8C,KAAlCM,kCAAAA,GAAqCD,mBAAH,GAAA;EAAA,GAAGA,EACxCL,UADwCK;EAAmB,WAC3DL,CAAAA,EAAAA,MAAAA;EAAU,QAAA,CAAA,EAAA,MAAA;EAYEO,SAAAA,CAAAA,EAAAA,MAAAA;EAA4B,aAAA,CAAA,EAAA,MAAA;EAAA,cAGxCP,CAAAA,EAAAA,OAAAA;EAAU,SAMKM,CAAAA,EAAAA,MAAAA;EAAkC,oBAAAE,CAAAA,EAAAA,MAMmDC;CAAgB;;;;AAMtEP,cArBlCK,4BAAAA,SAAqCH,cAAAA,CAqBHF;EAAe,WAAvBU,EAAAA,MAAAA;EAAO,QASzBX,EAAAA,MAAAA;EAAW,GAAgBE,EA3B/CH,UA2B+CG;EAAY,SAAGS,EAAAA,MAAAA;EAAO,aA9BpBR,EAAAA,MAAAA;EAAc,cAAA,EAAA,OAAA;;;sBAShDE;;;;;;iBAMLM,QAA0JJ,yBAAAA,CAA/GG,YANJH,yBAAAA,CAMmDC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA;;;;;;+BAMjIT,cAAcW,QAAQV;;;;;;;;;2BAS1BD,2BAA2BE,eAAeS"}

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.js

@@ -0,0 +1,89 @@
+import { BaseChatMemory } from "../../../memory/chat_memory.js";
+import { _formatIntermediateSteps } from "../../openai_functions/index.js";
+import { getBufferString } from "@langchain/core/messages";
+import { getInputValue, getOutputValue } from "@langchain/core/memory";
+
+//#region src/agents/toolkits/conversational_retrieval/token_buffer_memory.ts
+/**
+* Memory used to save agent output and intermediate steps.
+*/
+var OpenAIAgentTokenBufferMemory = class extends BaseChatMemory {
+	humanPrefix = "Human";
+	aiPrefix = "AI";
+	llm;
+	memoryKey = "history";
+	maxTokenLimit = 12e3;
+	returnMessages = true;
+	outputKey = "output";
+	intermediateStepsKey = "intermediateSteps";
+	constructor(fields) {
+		super(fields);
+		this.humanPrefix = fields.humanPrefix ?? this.humanPrefix;
+		this.aiPrefix = fields.aiPrefix ?? this.aiPrefix;
+		this.llm = fields.llm;
+		this.memoryKey = fields.memoryKey ?? this.memoryKey;
+		this.maxTokenLimit = fields.maxTokenLimit ?? this.maxTokenLimit;
+		this.returnMessages = fields.returnMessages ?? this.returnMessages;
+		this.outputKey = fields.outputKey ?? this.outputKey;
+		this.intermediateStepsKey = fields.intermediateStepsKey ?? this.intermediateStepsKey;
+	}
+	get memoryKeys() {
+		return [this.memoryKey];
+	}
+	/**
+	* Retrieves the messages from the chat history.
+	* @returns Promise that resolves with the messages from the chat history.
+	*/
+	async getMessages() {
+		return this.chatHistory.getMessages();
+	}
+	/**
+	* Loads memory variables from the input values.
+	* @param _values Input values.
+	* @returns Promise that resolves with the loaded memory variables.
+	*/
+	async loadMemoryVariables(_values) {
+		const buffer = await this.getMessages();
+		if (this.returnMessages) return { [this.memoryKey]: buffer };
+		else {
+			const bufferString = getBufferString(buffer, this.humanPrefix, this.aiPrefix);
+			return { [this.memoryKey]: bufferString };
+		}
+	}
+	/**
+	* Saves the context of the chat, including user input, AI output, and
+	* intermediate steps. Prunes the chat history if the total token count
+	* exceeds the maximum limit.
+	* @param inputValues Input values.
+	* @param outputValues Output values.
+	* @returns Promise that resolves when the context has been saved.
+	*/
+	async saveContext(inputValues, outputValues) {
+		const inputValue = getInputValue(inputValues, this.inputKey);
+		const outputValue = getOutputValue(outputValues, this.outputKey);
+		await this.chatHistory.addUserMessage(inputValue);
+		const intermediateStepMessages = _formatIntermediateSteps(outputValues[this.intermediateStepsKey]);
+		for (const message of intermediateStepMessages) await this.chatHistory.addMessage(message);
+		await this.chatHistory.addAIMessage(outputValue);
+		const currentMessages = await this.chatHistory.getMessages();
+		let tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);
+		if (tokenInfo.totalCount > this.maxTokenLimit) {
+			const prunedMemory = [];
+			while (tokenInfo.totalCount > this.maxTokenLimit) {
+				const retainedMessage = currentMessages.pop();
+				if (!retainedMessage) {
+					console.warn(`Could not prune enough messages from chat history to stay under ${this.maxTokenLimit} tokens.`);
+					break;
+				}
+				prunedMemory.push(retainedMessage);
+				tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);
+			}
+			await this.chatHistory.clear();
+			for (const message of prunedMemory) await this.chatHistory.addMessage(message);
+		}
+	}
+};
+
+//#endregion
+export { OpenAIAgentTokenBufferMemory };
+//# sourceMappingURL=token_buffer_memory.js.map

package/dist/agents/toolkits/conversational_retrieval/token_buffer_memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token_buffer_memory.js","names":["fields: OpenAIAgentTokenBufferMemoryFields","_values: InputValues","inputValues: InputValues","outputValues: OutputValues"],"sources":["../../../../src/agents/toolkits/conversational_retrieval/token_buffer_memory.ts"],"sourcesContent":["import { ChatOpenAI } from \"@langchain/openai\";\nimport {\n  InputValues,\n  MemoryVariables,\n  OutputValues,\n  getInputValue,\n  getOutputValue,\n} from \"@langchain/core/memory\";\nimport { getBufferString } from \"@langchain/core/messages\";\nimport {\n  BaseChatMemory,\n  BaseChatMemoryInput,\n} from \"../../../memory/chat_memory.js\";\nimport { _formatIntermediateSteps } from \"../../openai_functions/index.js\";\n\n/**\n * Type definition for the fields required to initialize an instance of\n * OpenAIAgentTokenBufferMemory.\n */\nexport type OpenAIAgentTokenBufferMemoryFields = BaseChatMemoryInput & {\n  llm: ChatOpenAI;\n  humanPrefix?: string;\n  aiPrefix?: string;\n  memoryKey?: string;\n  maxTokenLimit?: number;\n  returnMessages?: boolean;\n  outputKey?: string;\n  intermediateStepsKey?: string;\n};\n\n/**\n * Memory used to save agent output and intermediate steps.\n */\nexport class OpenAIAgentTokenBufferMemory extends BaseChatMemory {\n  humanPrefix = \"Human\";\n\n  aiPrefix = \"AI\";\n\n  llm: ChatOpenAI;\n\n  memoryKey = \"history\";\n\n  maxTokenLimit = 12000;\n\n  returnMessages = true;\n\n  outputKey = \"output\";\n\n  intermediateStepsKey = \"intermediateSteps\";\n\n  constructor(fields: OpenAIAgentTokenBufferMemoryFields) {\n    super(fields);\n    this.humanPrefix = fields.humanPrefix ?? this.humanPrefix;\n    this.aiPrefix = fields.aiPrefix ?? this.aiPrefix;\n    this.llm = fields.llm;\n    this.memoryKey = fields.memoryKey ?? this.memoryKey;\n    this.maxTokenLimit = fields.maxTokenLimit ?? this.maxTokenLimit;\n    this.returnMessages = fields.returnMessages ?? this.returnMessages;\n    this.outputKey = fields.outputKey ?? this.outputKey;\n    this.intermediateStepsKey =\n      fields.intermediateStepsKey ?? this.intermediateStepsKey;\n  }\n\n  get memoryKeys(): string[] {\n    return [this.memoryKey];\n  }\n\n  /**\n   * Retrieves the messages from the chat history.\n   * @returns Promise that resolves with the messages from the chat history.\n   */\n  async getMessages() {\n    return this.chatHistory.getMessages();\n  }\n\n  /**\n   * Loads memory variables from the input values.\n   * @param _values Input values.\n   * @returns Promise that resolves with the loaded memory variables.\n   */\n  async loadMemoryVariables(_values: InputValues): Promise<MemoryVariables> {\n    const buffer = await this.getMessages();\n    if (this.returnMessages) {\n      return { [this.memoryKey]: buffer };\n    } else {\n      const bufferString = getBufferString(\n        buffer,\n        this.humanPrefix,\n        this.aiPrefix\n      );\n      return { [this.memoryKey]: bufferString };\n    }\n  }\n\n  /**\n   * Saves the context of the chat, including user input, AI output, and\n   * intermediate steps. Prunes the chat history if the total token count\n   * exceeds the maximum limit.\n   * @param inputValues Input values.\n   * @param outputValues Output values.\n   * @returns Promise that resolves when the context has been saved.\n   */\n  async saveContext(\n    inputValues: InputValues,\n    outputValues: OutputValues\n  ): Promise<void> {\n    const inputValue = getInputValue(inputValues, this.inputKey);\n    const outputValue = getOutputValue(outputValues, this.outputKey);\n    await this.chatHistory.addUserMessage(inputValue);\n    const intermediateStepMessages = _formatIntermediateSteps(\n      outputValues[this.intermediateStepsKey]\n    );\n    for (const message of intermediateStepMessages) {\n      await this.chatHistory.addMessage(message);\n    }\n    await this.chatHistory.addAIMessage(outputValue);\n    const currentMessages = await this.chatHistory.getMessages();\n    let tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);\n    if (tokenInfo.totalCount > this.maxTokenLimit) {\n      const prunedMemory = [];\n      while (tokenInfo.totalCount > this.maxTokenLimit) {\n        const retainedMessage = currentMessages.pop();\n        if (!retainedMessage) {\n          console.warn(\n            `Could not prune enough messages from chat history to stay under ${this.maxTokenLimit} tokens.`\n          );\n          break;\n        }\n        prunedMemory.push(retainedMessage);\n        tokenInfo = await this.llm.getNumTokensFromMessages(currentMessages);\n      }\n      await this.chatHistory.clear();\n      for (const message of prunedMemory) {\n        await this.chatHistory.addMessage(message);\n      }\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AAiCA,IAAa,+BAAb,cAAkD,eAAe;CAC/D,cAAc;CAEd,WAAW;CAEX;CAEA,YAAY;CAEZ,gBAAgB;CAEhB,iBAAiB;CAEjB,YAAY;CAEZ,uBAAuB;CAEvB,YAAYA,QAA4C;EACtD,MAAM,OAAO;EACb,KAAK,cAAc,OAAO,eAAe,KAAK;EAC9C,KAAK,WAAW,OAAO,YAAY,KAAK;EACxC,KAAK,MAAM,OAAO;EAClB,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,gBAAgB,OAAO,iBAAiB,KAAK;EAClD,KAAK,iBAAiB,OAAO,kBAAkB,KAAK;EACpD,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,uBACH,OAAO,wBAAwB,KAAK;CACvC;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,KAAK,SAAU;CACxB;;;;;CAMD,MAAM,cAAc;AAClB,SAAO,KAAK,YAAY,aAAa;CACtC;;;;;;CAOD,MAAM,oBAAoBC,SAAgD;EACxE,MAAM,SAAS,MAAM,KAAK,aAAa;AACvC,MAAI,KAAK,eACP,QAAO,GAAG,KAAK,YAAY,OAAQ;OAC9B;GACL,MAAM,eAAe,gBACnB,QACA,KAAK,aACL,KAAK,SACN;AACD,UAAO,GAAG,KAAK,YAAY,aAAc;EAC1C;CACF;;;;;;;;;CAUD,MAAM,YACJC,aACAC,cACe;EACf,MAAM,aAAa,cAAc,aAAa,KAAK,SAAS;EAC5D,MAAM,cAAc,eAAe,cAAc,KAAK,UAAU;EAChE,MAAM,KAAK,YAAY,eAAe,WAAW;EACjD,MAAM,2BAA2B,yBAC/B,aAAa,KAAK,sBACnB;AACD,OAAK,MAAM,WAAW,0BACpB,MAAM,KAAK,YAAY,WAAW,QAAQ;EAE5C,MAAM,KAAK,YAAY,aAAa,YAAY;EAChD,MAAM,kBAAkB,MAAM,KAAK,YAAY,aAAa;EAC5D,IAAI,YAAY,MAAM,KAAK,IAAI,yBAAyB,gBAAgB;AACxE,MAAI,UAAU,aAAa,KAAK,eAAe;GAC7C,MAAM,eAAe,CAAE;AACvB,UAAO,UAAU,aAAa,KAAK,eAAe;IAChD,MAAM,kBAAkB,gBAAgB,KAAK;AAC7C,QAAI,CAAC,iBAAiB;KACpB,QAAQ,KACN,CAAC,gEAAgE,EAAE,KAAK,cAAc,QAAQ,CAAC,CAChG;AACD;IACD;IACD,aAAa,KAAK,gBAAgB;IAClC,YAAY,MAAM,KAAK,IAAI,yBAAyB,gBAAgB;GACrE;GACD,MAAM,KAAK,YAAY,OAAO;AAC9B,QAAK,MAAM,WAAW,cACpB,MAAM,KAAK,YAAY,WAAW,QAAQ;EAE7C;CACF;AACF"}