diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-21 07:22:43 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-21 07:22:43 +0800 |
| commit | 34d568d5e229729031245aee488481c7f623125c (patch) | |
| tree | b54f078561c4ffe9fb7ae445788b0b495b6e0450 /src/config | |
| parent | ba832016f3b272aac561ac9b053fcb9df1cb8d13 (diff) | |
| download | aichat-34d568d5e229729031245aee488481c7f623125c.tar.gz | |
refactor: improve rag (#622)
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/input.rs | 6 | ||||
| -rw-r--r-- | src/config/mod.rs | 4 |
2 files changed, 5 insertions, 5 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, ) 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, |