update exp_pool manager

This commit is contained in:
seehi 2024-08-19 14:07:13 +08:00
parent 243c7a65d6
commit 665ca6ff97
9 changed files with 31 additions and 22 deletions

View file

@ -5,7 +5,7 @@ from pydantic import Field
from metagpt.utils.yaml_model import YamlModel
class ExperiencePoolStorageType(Enum):
class ExperiencePoolRetrievalType(Enum):
BM25 = "bm25"
CHROMA = "chroma"
@ -18,6 +18,6 @@ class ExperiencePoolConfig(YamlModel):
enable_read: bool = Field(default=False, description="Enable to read from experience pool.")
enable_write: bool = Field(default=False, description="Enable to write to experience pool.")
persist_path: str = Field(default=".chroma_exp_data", description="The persist path for experience pool.")
storage_type: ExperiencePoolStorageType = Field(
default=ExperiencePoolStorageType.BM25, description="The storage type for experience pool."
retrieval_type: ExperiencePoolRetrievalType = Field(
default=ExperiencePoolRetrievalType.BM25, description="The retrieval type for experience pool."
)