summaryrefslogtreecommitdiffstats
path: root/src/config/input.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-21 07:22:43 +0800
committerGitHub <noreply@github.com>2024-06-21 07:22:43 +0800
commit34d568d5e229729031245aee488481c7f623125c (patch)
treeb54f078561c4ffe9fb7ae445788b0b495b6e0450 /src/config/input.rs
parentba832016f3b272aac561ac9b053fcb9df1cb8d13 (diff)
downloadaichat-34d568d5e229729031245aee488481c7f623125c.tar.gz
refactor: improve rag (#622)
Diffstat (limited to 'src/config/input.rs')
-rw-r--r--src/config/input.rs6
1 files changed, 3 insertions, 3 deletions
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,
)