From e18aa0169e80b8af9241e57f0ab66bf463ac8a4f Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 11 Nov 2024 09:46:21 +0800
Subject: refactor: improve RAG (#982)

- set max default_chuck_size to 2000
- set default rag_top_k to 4
- change reciprocal_rank_fusion weights
---
 src/rag/mod.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/rag/mod.rs')

diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 4ddb0da..82da7c8 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -488,7 +488,7 @@ impl Rag {
             None => {
                 let ids = reciprocal_rank_fusion(
                     vec![vector_search_ids, keyword_search_ids],
-                    vec![1.0, 1.0],
+                    vec![1.125, 1.0],
                     top_k,
                 );
                 debug!("rrf_ids: {ids:?}");
-- 
cgit v1.2.3