summaryrefslogtreecommitdiffstats
path: root/src/rag
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-11-11 09:46:21 +0800
committerGitHub <noreply@github.com>2024-11-11 09:46:21 +0800
commite18aa0169e80b8af9241e57f0ab66bf463ac8a4f (patch)
tree49f5b07831894644e9e3ab3c291a2501190ebc56 /src/rag
parentb2dbd73b8c8938105246f04e6ebcd68221740bd7 (diff)
downloadaichat-e18aa0169e80b8af9241e57f0ab66bf463ac8a4f.tar.gz
refactor: improve RAG (#982)
- set max default_chuck_size to 2000 - set default rag_top_k to 4 - change reciprocal_rank_fusion weights
Diffstat (limited to 'src/rag')
-rw-r--r--src/rag/mod.rs2
1 files changed, 1 insertions, 1 deletions
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:?}");