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

add enable_longterm_memory config

Browse source
This commit is contained in:
seehi 2024-09-12 11:20:40 +08:00
parent c0d5c031f8
commit cf3b80d5f1
4 changed files with 18 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
metagpt/roles/di/role_zero.py
View file

@ -92,7 +92,6 @@ class RoleZero(Role):
command_rsp: str = "" # the raw string containing the commands
commands: list[dict] = [] # commands to be executed
memory_k: int = 200 # number of memories (messages) to use as historical context
enable_longterm_memory: bool = True # whether to use longterm memory
use_fixed_sop: bool = False
requirements_constraints: str = "" # the constraints in user requirements
use_summary: bool = True # whether to summarize at the end
@ -168,14 +167,16 @@ class RoleZero(Role):
@model_validator(mode="after")
def set_longterm_memory(self) -> "RoleZero":
"""Set longterm memory.
"""Set up long-term memory for the role if enabled in the configuration.
If enable_longterm_memory is True and longterm_memory is not set, set it.
If `enable_longterm_memory` is True, set up long-term memory.
The role name will be used as the collection name.
"""
if self.enable_longterm_memory:
enable_longterm_memory = bool(self.config.role_zero and self.config.role_zero.enable_longterm_memory)
if enable_longterm_memory:
self.rc.memory = RoleZeroLongTermMemory(collection_name=self.name.replace(" ", ""), memory_k=self.memory_k)
logger.info(f"Long-term memory set for role '{self.name}'")
return self