summaryrefslogtreecommitdiffstats
path: root/src/config/input.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/config/input.rs')
-rw-r--r--src/config/input.rs9
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());