Add inputKey and outputKey fields to BufferWindowMemory (langchain-ai…

…#1023)
philrwebb · Apr 28, 2023 · b027521 · b027521
1 parent 11397ab
commit b027521
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/langchain/src/memory/buffer_window_memory.ts b/langchain/src/memory/buffer_window_memory.ts
@@ -25,6 +25,8 @@ export class BufferWindowMemory
     super({
       returnMessages: fields?.returnMessages ?? false,
       chatHistory: fields?.chatHistory,
+      inputKey: fields?.inputKey,
+      outputKey: fields?.outputKey,
     });
     this.humanPrefix = fields?.humanPrefix ?? this.humanPrefix;
     this.aiPrefix = fields?.aiPrefix ?? this.aiPrefix;