add enable_longterm_memory config

This commit is contained in:
seehi 2024-09-12 11:21:59 +08:00
parent cf3b80d5f1
commit d9ab81bbae

View file

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