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/rag/mod.rs | |
| parent | 927b73665ffc8025a7d18a162165e38e3af08747 (diff) | |
| download | aichat-486ee57bf8123116a326d1433bcb1b36ff88ee67.tar.gz | |
feat: discontinue `rag_min_score_*` config items (#1081)
Diffstat (limited to 'src/rag/mod.rs')
| -rw-r--r-- | src/rag/mod.rs | 16 |
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?; |
