From 047f1e429dcbd7d48c57cbac58f6039b55c456f3 Mon Sep 17 00:00:00 2001 From: seehi <6580@pm.me> Date: Mon, 9 Sep 2024 14:02:26 +0800 Subject: [PATCH] Keep user and AI messages are paired --- metagpt/roles/di/role_zero.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/metagpt/roles/di/role_zero.py b/metagpt/roles/di/role_zero.py index 5e1d239c0..b4ee43228 100644 --- a/metagpt/roles/di/role_zero.py +++ b/metagpt/roles/di/role_zero.py @@ -608,8 +608,9 @@ class RoleZero(Role): related_memories = self.longterm_memory.fetch(memories[-1].content) logger.info(f"Fetched {len(related_memories)} long-term memories.") - if related_memories and self._is_first_message_from_ai(memories): - memories = memories[1:] + # Keep user and AI messages are paired. + if self._is_first_message_from_ai(memories): + memories.append(self.rc.memory.get_by_position(-(k + 1))) final_memories = related_memories + memories