From ac89b7ad6716a8af811089b130c2039627d47b48 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 20 Oct 2024 14:28:18 +0800
Subject: refactor: RAG bm25 (#939)

---
 src/rag/mod.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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

diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 4cee391..87f7a69 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -10,7 +10,7 @@ mod serde_vectors;
 mod splitter;
 
 use anyhow::{anyhow, bail, Context, Result};
-use bm25::{LanguageMode, SearchEngine, SearchEngineBuilder};
+use bm25::{Language, SearchEngine, SearchEngineBuilder};
 use hnsw_rs::prelude::*;
 use indexmap::{IndexMap, IndexSet};
 use inquire::{required, validator::Validation, Confirm, Select, Text};
@@ -724,7 +724,7 @@ impl RagData {
                 documents.push(bm25::Document::new(id, &document.page_content))
             }
         }
-        SearchEngineBuilder::<DocumentId>::with_documents(LanguageMode::Detect, documents)
+        SearchEngineBuilder::<DocumentId>::with_documents(Language::English, documents)
             .k1(1.5)
             .b(0.75)
             .build()
-- 
cgit v1.2.3