From 34d568d5e229729031245aee488481c7f623125c Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 21 Jun 2024 07:22:43 +0800 Subject: refactor: improve rag (#622) --- src/config/input.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/config/input.rs') diff --git a/src/config/input.rs b/src/config/input.rs index 03ec999..b9b1746 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -169,12 +169,12 @@ impl Input { if !self.text.is_empty() { let rag = self.config.read().rag.clone(); if let Some(rag) = rag { - let (top_k, min_score_vector_search, min_score_fulltext_search) = { + let (top_k, min_score_vector_search, min_score_keyword_search) = { let config = self.config.read(); ( config.rag_top_k, config.rag_min_score_vector_search, - config.rag_min_score_fulltext_search, + config.rag_min_score_keyword_search, ) }; let rerank = match self.config.read().rag_rerank_model.clone() { @@ -192,7 +192,7 @@ impl Input { &self.text, top_k, min_score_vector_search, - min_score_fulltext_search, + min_score_keyword_search, rerank, abort_signal, ) -- cgit v1.2.3