diff options
| author | sigoden <sigoden@gmail.com> | 2025-01-10 17:31:41 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-01-10 17:31:41 +0800 |
| commit | 486ee57bf8123116a326d1433bcb1b36ff88ee67 (patch) | |
| tree | b5993c2b012086a94bdac148000f118d0e6c3756 /src/config | |
| parent | 927b73665ffc8025a7d18a162165e38e3af08747 (diff) | |
| download | aichat-486ee57bf8123116a326d1433bcb1b36ff88ee67.tar.gz | |
feat: discontinue `rag_min_score_*` config items (#1081)
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/mod.rs | 27 |
1 files changed, 1 insertions, 26 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs index ca0990a..407d041 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -123,8 +123,6 @@ pub struct Config { pub rag_top_k: usize, pub rag_chunk_size: Option<usize>, pub rag_chunk_overlap: Option<usize>, - pub rag_min_score_vector_search: f32, - pub rag_min_score_keyword_search: f32, pub rag_template: Option<String>, #[serde(default)] @@ -197,8 +195,6 @@ impl Default for Config { rag_top_k: 5, rag_chunk_size: None, rag_chunk_overlap: None, - rag_min_score_vector_search: 0.0, - rag_min_score_keyword_search: 0.0, rag_template: None, document_loaders: Default::default(), @@ -1413,22 +1409,8 @@ impl Config { abort_signal: AbortSignal, ) -> Result<String> { let (reranker_model, top_k) = rag.get_config(); - let (min_score_vector_search, min_score_keyword_search) = { - let config = config.read(); - ( - config.rag_min_score_vector_search, - config.rag_min_score_keyword_search, - ) - }; let (embeddings, ids) = rag - .search( - text, - top_k, - min_score_vector_search, - min_score_keyword_search, - reranker_model.as_deref(), - abort_signal, - ) + .search(text, top_k, reranker_model.as_deref(), abort_signal) .await?; let text = config.read().rag_template(&embeddings, text); rag.set_last_sources(&ids); @@ -2222,13 +2204,6 @@ impl Config { if let Some(v) = read_env_value::<usize>(&get_env_name("rag_chunk_overlap")) { self.rag_chunk_overlap = v; } - if let Some(Some(v)) = read_env_value::<f32>(&get_env_name("rag_min_score_vector_search")) { - self.rag_min_score_vector_search = v; - } - if let Some(Some(v)) = read_env_value::<f32>(&get_env_name("rag_min_score_keyword_search")) - { - self.rag_min_score_keyword_search = v; - } if let Some(v) = read_env_value::<String>(&get_env_name("rag_template")) { self.rag_template = v; } |
