From dcc379afbb53a5820077b59c912648fab0b49d43 Mon Sep 17 00:00:00 2001 From: Morgan McGuire Date: Wed, 22 Jan 2025 19:28:16 +0000 Subject: [PATCH] Return to fetch_k 60 for experiment. --- src/wandbot/configs/chat_config.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/wandbot/configs/chat_config.py b/src/wandbot/configs/chat_config.py index e069371..77a0577 100644 --- a/src/wandbot/configs/chat_config.py +++ b/src/wandbot/configs/chat_config.py @@ -22,9 +22,8 @@ class ChatConfig(BaseSettings): search_type: Literal["mmr", "similarity"] = "mmr" do_web_search: bool = False redundant_similarity_threshold: float = 0.95 # used to remove very similar retrieved documents - similarity_score_threshold: float = 0.0 # Used in similarity score threshold retrieval, similarity search used # Retrieval settings: MMR settings - fetch_k: int = 20 # Used in mmr retrieval. Typically set as top_k * 4 + fetch_k: int = 60 # Used in mmr retrieval. Typically set as top_k * 4 mmr_lambda_mult: float = 0.5 # used in mmr retrieval # Reranker models rereanker_provider: str = "cohere"