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

exp pool support bm25

Browse source
This commit is contained in:
seehi 2024-08-16 21:13:08 +08:00
parent aa8e2fa8c3
commit a6af03efb8
10 changed files with 231 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

10
metagpt/configs/exp_pool_config.py
View file

@ -1,8 +1,15 @@
from enum import Enum
from pydantic import Field
from metagpt.utils.yaml_model import YamlModel
class ExperiencePoolStorageType(Enum):
BM25 = "bm25"
CHROMA = "chroma"
class ExperiencePoolConfig(YamlModel):
enabled: bool = Field(
default=False,
@ -11,3 +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."
)