summaryrefslogtreecommitdiffstats
path: root/src/config/mod.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2025-01-10 17:31:41 +0800
committerGitHub <noreply@github.com>2025-01-10 17:31:41 +0800
commit486ee57bf8123116a326d1433bcb1b36ff88ee67 (patch)
treeb5993c2b012086a94bdac148000f118d0e6c3756 /src/config/mod.rs
parent927b73665ffc8025a7d18a162165e38e3af08747 (diff)
downloadaichat-486ee57bf8123116a326d1433bcb1b36ff88ee67.tar.gz
feat: discontinue `rag_min_score_*` config items (#1081)
Diffstat (limited to 'src/config/mod.rs')
-rw-r--r--src/config/mod.rs27
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;
}