diff options
Diffstat (limited to 'src/config/input.rs')
| -rw-r--r-- | src/config/input.rs | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/config/input.rs b/src/config/input.rs index 4ddc06f..935d517 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -169,8 +169,13 @@ impl Input { if !self.text.is_empty() { let rag = self.config.read().rag.clone(); if let Some(rag) = rag { - let top_k = self.config.read().rag_top_k; - let embeddings = rag.search(&self.text, top_k, abort_signal).await?; + let (top_k, minimum_score) = { + let config = self.config.read(); + (config.rag_top_k, config.rag_minimum_score) + }; + let embeddings = rag + .search(&self.text, top_k, minimum_score, abort_signal) + .await?; let text = self.config.read().rag_template(&embeddings, &self.text); self.patched_text = Some(text); self.rag_name = Some(rag.name().to_string()); |
