summaryrefslogtreecommitdiffstats
path: root/src/rag
diff options
context:
space:
mode:
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?;