make similarity_top_k configurable

This commit is contained in:
seehi 2024-10-11 15:37:33 +08:00
parent 2965a22e1d
commit dd955c848d
4 changed files with 9 additions and 1 deletions

View file

@ -7,3 +7,4 @@ class RoleZeroConfig(YamlModel):
enable_longterm_memory: bool = Field(default=False, description="Whether to use long-term memory.")
longterm_memory_persist_path: str = Field(default=".role_memory_data", description="The directory to save data.")
memory_k: int = Field(default=200, description="The capacity of short-term memory.")
similarity_top_k: int = Field(default=5, description="The number of long-term memories to retrieve.")