From 34d568d5e229729031245aee488481c7f623125c Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 21 Jun 2024 07:22:43 +0800
Subject: refactor: improve rag (#622)

---
 src/config/input.rs | 6 +++---
 src/config/mod.rs   | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

(limited to 'src/config')

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,
                     )
diff --git a/src/config/mod.rs b/src/config/mod.rs
index e07dca6..903face 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -107,7 +107,7 @@ pub struct Config {
     pub rag_chunk_overlap: Option<usize>,
     pub rag_top_k: usize,
     pub rag_min_score_vector_search: f32,
-    pub rag_min_score_fulltext_search: f32,
+    pub rag_min_score_keyword_search: f32,
     pub rag_min_score_rerank: f32,
     pub rag_template: Option<String>,
     pub compress_threshold: usize,
@@ -163,7 +163,7 @@ impl Default for Config {
             rag_chunk_overlap: None,
             rag_top_k: 4,
             rag_min_score_vector_search: 0.0,
-            rag_min_score_fulltext_search: 0.0,
+            rag_min_score_keyword_search: 0.0,
             rag_min_score_rerank: 0.0,
             rag_template: None,
             compress_threshold: 4000,
-- 
cgit v1.2.3