From 3454761f950d49db49588eb35518708cf9d5b0e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Fri, 1 Sep 2023 21:16:56 +0800
Subject: [PATCH] fixbug: exceed length

---
 metagpt/memory/brain_memory.py | 5 +++--
 metagpt/roles/assistant.py     | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/metagpt/memory/brain_memory.py b/metagpt/memory/brain_memory.py
index b3445a1f2..23b50afb3 100644
--- a/metagpt/memory/brain_memory.py
+++ b/metagpt/memory/brain_memory.py
@@ -8,7 +8,7 @@
 """
 
 from enum import Enum
-from typing import List, Dict
+from typing import Dict, List
 
 import pydantic
 
@@ -48,7 +48,7 @@ class BrainMemory(pydantic.BaseModel):
         texts = [Message(**m).content for m in self.history[:-1]]
         return "\n".join(texts)
 
-    def move_to_solution(self):
+    def move_to_solution(self, history_summary):
         if len(self.history) < 2:
             return
         msgs = self.history[:-1]
@@ -58,6 +58,7 @@ class BrainMemory(pydantic.BaseModel):
             self.history = []
         else:
             self.history = self.history[-1:]
+        self.history.insert(0, Message(content=history_summary))
 
     @property
     def last_talk(self):
diff --git a/metagpt/roles/assistant.py b/metagpt/roles/assistant.py
index c681da65b..719dfc29b 100644
--- a/metagpt/roles/assistant.py
+++ b/metagpt/roles/assistant.py
@@ -125,7 +125,7 @@ class Assistant(Role):
             last_talk = await self._llm.rewrite(sentence=last_talk, context=history_text)
             return last_talk
 
-        self.memory.move_to_solution()  # Promptly clear memory after the issue is resolved.
+        self.memory.move_to_solution(history_summary)  # Promptly clear memory after the issue is resolved.
         return last_talk
 
     @staticmethod