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

Merge branch 'feat-exp-pool-bm25' into 'mgx_ops'

Browse source 
Feat exp pool bm25

See merge request pub/MetaGPT!327
This commit is contained in:
王金淋 2024-08-19 06:45:03 +00:00
commit c6cc5e2da3
14 changed files with 244 additions and 54 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 ExperiencePoolRetrievalType(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.")
retrieval_type: ExperiencePoolRetrievalType = Field(
default=ExperiencePoolRetrievalType.BM25, description="The retrieval type for experience pool."
)