feat: rag hybrid search (#618)

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);