apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-26 17:26:22 +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:31:12 +08:00
parent d9ab81bbae
commit 99e5a73fbb
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/config2.py
View file

@ -91,7 +91,7 @@ class Config(CLIParams, YamlModel):
roles: Optional[List[RoleCustomConfig]] = None
# RoleZero's configuration
role_zero: Optional[RoleZeroConfig] = None
role_zero: RoleZeroConfig = Field(default_factory=RoleZeroConfig)
omniparse: Optional[OmniParseConfig] = None

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

@ -173,7 +173,7 @@ class RoleZero(Role):
The role name will be used as the collection name.
"""
if self.config.role_zero and self.config.role_zero.enable_longterm_memory:
if self.config.role_zero.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}'")