summaryrefslogtreecommitdiffstats
path: root/src/rag
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/rag
parent927b73665ffc8025a7d18a162165e38e3af08747 (diff)
downloadaichat-486ee57bf8123116a326d1433bcb1b36ff88ee67.tar.gz
feat: discontinue `rag_min_score_*` config items (#1081)
Diffstat (limited to 'src/rag')
-rw-r--r--src/rag/mod.rs16
1 files changed, 3 insertions, 13 deletions
diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index ceb83f6..8855f4f 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -297,19 +297,11 @@ impl Rag {
&self,
text: &str,
top_k: usize,
- min_score_vector_search: f32,
- min_score_keyword_search: f32,
rerank_model: Option<&str>,
abort_signal: AbortSignal,
) -> Result<(String, Vec<DocumentId>)> {
let ret = abortable_run_with_spinner(
- self.hybird_search(
- text,
- top_k,
- min_score_vector_search,
- min_score_keyword_search,
- rerank_model,
- ),
+ self.hybird_search(text, top_k, rerank_model),
"Searching",
abort_signal,
)
@@ -497,13 +489,11 @@ impl Rag {
&self,
query: &str,
top_k: usize,
- min_score_vector_search: f32,
- min_score_keyword_search: f32,
rerank_model: Option<&str>,
) -> Result<Vec<(DocumentId, String)>> {
let (vector_search_results, keyword_search_results) = tokio::join!(
- self.vector_search(query, top_k, min_score_vector_search),
- self.keyword_search(query, top_k, min_score_keyword_search)
+ self.vector_search(query, top_k, 0.0),
+ self.keyword_search(query, top_k, 0.0),
);
let vector_search_results = vector_search_results?;