opt RoleZeroLongTermMemory init

This commit is contained in:
seehi 2024-09-23 15:38:08 +08:00
parent 57b6108318
commit 59d94f7760

View file

@ -175,11 +175,9 @@ class RoleZero(Role):
if self.config.role_zero.enable_longterm_memory:
self.rc.memory = RoleZeroLongTermMemory(
**self.rc.memory.model_dump(),
collection_name=self.name.replace(" ", ""),
memory_k=self.memory_k,
storage=self.rc.memory.storage,
index=self.rc.memory.index,
ignore_id=self.rc.memory.ignore_id,
)
logger.info(f"Long-term memory set for role '{self.name}'")