From 88419224586ec683db92ae83e4b4aad35bfb5d8e 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: Mon, 4 Sep 2023 16:05:01 +0800
Subject: [PATCH] feat: +cache

---
 metagpt/memory/brain_memory.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/metagpt/memory/brain_memory.py b/metagpt/memory/brain_memory.py
index a925474b7..8b1b31aae 100644
--- a/metagpt/memory/brain_memory.py
+++ b/metagpt/memory/brain_memory.py
@@ -103,7 +103,9 @@ class BrainMemory(pydantic.BaseModel):
         v = await redis.get(key=redis_key)
         if v:
             data = json.loads(v)
-            return BrainMemory(**data)
+            bm = BrainMemory(**data)
+            bm.is_dirty = False
+            return bm
         return BrainMemory()
 
     async def dumps(self, redis_key: str, timeout_sec: int = 30 * 60, redis_conf: Dict = None):
@@ -112,6 +114,7 @@ class BrainMemory(pydantic.BaseModel):
             return False
         v = self.json()
         await redis.set(key=redis_key, data=v, timeout_sec=timeout_sec)
+        self.is_dirty = False
 
     @staticmethod
     def to_redis_key(prefix: str, user_id: str, chat_id: str):