apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-27 14:25:20 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: brain memory

Browse source
This commit is contained in:
莘权 马 2023-09-07 19:23:09 +08:00
parent 4c873a9158
commit 44706ba141
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/memory/brain_memory.py
View file

@ -80,7 +80,7 @@ class BrainMemory(pydantic.BaseModel):
async def loads(redis_key: str, redis_conf: Dict = None) -> "BrainMemory":
redis = Redis(conf=redis_conf)
if not redis.is_valid() or not redis_key:
return BrainMemory()
return BrainMemory(llm_type=CONFIG.LLM_TYPE)
v = await redis.get(key=redis_key)
logger.debug(f"REDIS GET {redis_key} {v}")
if v:
@ -88,9 +88,11 @@ class BrainMemory(pydantic.BaseModel):
bm = BrainMemory(**data)
bm.is_dirty = False
return bm
return BrainMemory()
return BrainMemory(llm_type=CONFIG.LLM_TYPE)
async def dumps(self, redis_key: str, timeout_sec: int = 30 * 60, redis_conf: Dict = None):
if not self.is_dirty:
return
redis = Redis(conf=redis_conf)
if not redis.is_valid() or not redis_key:
return False