Merge pull request #794 from zyren123/memorylimit

Added message limit to the function "add_messages" to ensure that the…
2025-03-29 20:33:52 +08:00
parent f821ee729c e39046d175
commit 49e106fd2c
1 changed files with 3 additions and 0 deletions
@@ -170,6 +170,9 @@ class Memory(BaseModel):
    def add_messages(self, messages: List[Message]) -> None:
        """Add multiple messages to memory"""
        self.messages.extend(messages)
+        # Optional: Implement message limit
+        if len(self.messages) > self.max_messages:
+            self.messages = self.messages[-self.max_messages :]

    def clear(self) -> None:
        """Clear all messages"""