diff options
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?; |
