From 2eab71a641827e503b14952373aec82661192ba2 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 20 Jun 2024 11:26:45 +0800
Subject: feat: rag hybrid search (#618)

---
 src/config/input.rs | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

(limited to 'src/config/input.rs')

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);
-- 
cgit v1.2.3