From 1254f93467ca8cd9cad34e3c6791ce9ffef3d633 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, 8 Sep 2023 10:22:31 +0800
Subject: [PATCH] refactor: brain memory

---
 metagpt/memory/brain_memory.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/metagpt/memory/brain_memory.py b/metagpt/memory/brain_memory.py
index 9878fa750..b8f9a2a15 100644
--- a/metagpt/memory/brain_memory.py
+++ b/metagpt/memory/brain_memory.py
@@ -42,6 +42,7 @@ class BrainMemory(pydantic.BaseModel):
     is_dirty: bool = False
     last_talk: str = None
     llm_type: Optional[str] = None
+    cacheable: bool = True
 
     def add_talk(self, msg: Message):
         msg.add_tag(MessageType.Talk.value)
@@ -78,8 +79,9 @@ class BrainMemory(pydantic.BaseModel):
         if not redis.is_valid() or not redis_key:
             return False
         v = self.json()
-        await redis.set(key=redis_key, data=v, timeout_sec=timeout_sec)
-        logger.debug(f"REDIS SET {redis_key} {v}")
+        if self.cacheable:
+            await redis.set(key=redis_key, data=v, timeout_sec=timeout_sec)
+            logger.debug(f"REDIS SET {redis_key} {v}")
         self.is_dirty = False
 
     @staticmethod