From e39046d175c4a2ae0b33a13dbab21bc339adc25f Mon Sep 17 00:00:00 2001
From: zhiyuanren <1131876818@qq.com>
Date: Tue, 18 Mar 2025 11:12:59 +0800
Subject: [PATCH] Added message limit to the function "add_messages" to ensure
 that the number of messages does not exceed the maximum limit.

---
 app/schema.py | 3 +++
 1 file changed, 3 insertions(+)

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"""