diff --git a/app/schema.py b/app/schema.py index de18c4f..5f743f9 100644 --- a/app/schema.py +++ b/app/schema.py @@ -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"""