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

Merge branch 'feat-memory-opt' into 'mgx_ops'

Browse source 
make memory_k configurable

See merge request pub/MetaGPT!392
This commit is contained in:
张雷 2024-10-18 07:49:32 +00:00
commit 31e0a46e6a
4 changed files with 19 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -181,11 +181,14 @@ class RoleZero(Role):
"""
if self.config.role_zero.enable_longterm_memory:
# Use config.role_zero to initialize long-term memory
self.rc.memory = RoleZeroLongTermMemory(
**self.rc.memory.model_dump(),
persist_path=self.config.role_zero.longterm_memory_persist_path,
collection_name=self.name.replace(" ", ""),
memory_k=self.memory_k,
memory_k=self.config.role_zero.memory_k,
similarity_top_k=self.config.role_zero.similarity_top_k,
use_llm_ranker=self.config.role_zero.use_llm_ranker,
)
logger.info(f"Long-term memory set for role '{self.name}'")