add collection_name in exp_pool config

This commit is contained in:
seehi 2024-08-23 10:14:12 +08:00
parent cb3541f204
commit 9d327081ba
5 changed files with 7 additions and 10 deletions

View file

@ -22,3 +22,4 @@ class ExperiencePoolConfig(YamlModel):
default=ExperiencePoolRetrievalType.BM25, description="The retrieval type for experience pool."
)
use_llm_ranker: bool = Field(default=True, description="Use LLM Reranker to get better result.")
collection_name: str = Field(default="experience_pool", description="The collection name in chromadb")