diff options
Diffstat (limited to 'src/config/input.rs')
| -rw-r--r-- | src/config/input.rs | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/src/config/input.rs b/src/config/input.rs index 935d517..892a41f 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -169,12 +169,22 @@ impl Input { if !self.text.is_empty() { let rag = self.config.read().rag.clone(); if let Some(rag) = rag { - let (top_k, minimum_score) = { + let (top_k, min_score_vector, min_score_text) = { let config = self.config.read(); - (config.rag_top_k, config.rag_minimum_score) + ( + config.rag_top_k, + config.rag_min_score_vector, + config.rag_min_score_text, + ) }; let embeddings = rag - .search(&self.text, top_k, minimum_score, abort_signal) + .search( + &self.text, + top_k, + min_score_vector, + min_score_text, + abort_signal, + ) .await?; let text = self.config.read().rag_template(&embeddings, &self.text); self.patched_text = Some(text); |
