From db3a7b6ec2dee82df74fa055ab83b5a140cf49f0 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 17 Oct 2024 19:36:00 +0800
Subject: refactor: improve RAG search (#931)

---
 src/rag/serde_vectors.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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

diff --git a/src/rag/serde_vectors.rs b/src/rag/serde_vectors.rs
index 894c22c..bd821fa 100644
--- a/src/rag/serde_vectors.rs
+++ b/src/rag/serde_vectors.rs
@@ -12,8 +12,8 @@ where
 {
     let encoded_map: IndexMap<String, String> = vectors
         .iter()
-        .map(|(key, vec)| {
-            let (h, l) = split_document_id(*key);
+        .map(|(id, vec)| {
+            let (h, l) = id.split();
             let byte_slice = unsafe {
                 std::slice::from_raw_parts(
                     vec.as_ptr() as *const u8,
@@ -41,7 +41,7 @@ where
             .and_then(|(h, l)| {
                 let h = h.parse::<usize>().ok()?;
                 let l = l.parse::<usize>().ok()?;
-                Some(combine_document_id(h, l))
+                Some(DocumentId::new(h, l))
             })
             .ok_or_else(|| de::Error::custom(format!("Invalid key '{key}'")))?;
 
-- 
cgit v1.2.3