Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed k=0 bug on ConversationBufferWindowMemory (#2796)

Browse Source 
Updated the "load_memory_variables" function of the
ConversationBufferWindowMemory to support a window size of 0 (k=0).
Previous behavior would return the full memory instead of an empty
array.
This commit is contained in:
Tim Connors 2023-04-12 17:05:54 -07:00 committed by GitHub
parent a094b7f807
commit e081c62aac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
langchain/memory/buffer_window.py
View File

@ -28,11 +28,10 @@ class ConversationBufferWindowMemory(BaseChatMemory):
def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]:
"""Return history buffer."""
if self.return_messages:
buffer: Any = self.buffer[-self.k * 2 :]
else:
buffer: Any = self.buffer[-self.k * 2 :] if self.k > 0 else []
if not self.return_messages:
buffer = get_buffer_string(
self.buffer[-self.k * 2 :],
buffer,
human_prefix=self.human_prefix,
ai_prefix=self.ai_prefix,
)