From 3b3d39cef0211b607d51360d0739022f953ba1a7 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 19 Jun 2024 12:15:54 +0800 Subject: refactor: rag add rag_minimum_score config (#617) --- src/config/input.rs | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'src/config/input.rs') 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()); -- cgit v1.2.3