apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-25 00:36:55 +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
config/config2.example.yaml
View file

@ -84,7 +84,7 @@ exp_pool:
collection_name: experience_pool # When `retrieval_type` is `chroma`, `collection_name` is the collection name in chromadb.
role_zero:
enable_longterm_memory: false # Whether to use long-term memory.
enable_longterm_memory: false # Whether to use long-term memory. Default is `false`.
azure_tts_subscription_key: "YOUR_SUBSCRIPTION_KEY"
azure_tts_region: "eastus"

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}'")