From 912773c25a113f49c5df63cd3a8086d38c75103e Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 24 Sep 2024 07:42:24 +0800
Subject: refactor: embeddings/rerank fn accept ref data (#878)

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

(limited to 'src/rag')

diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 62a8ff8..96fb9d5 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -483,7 +483,7 @@ impl Rag {
                     }
                 }
                 let data = RerankData::new(query.to_string(), documents, top_k);
-                let list = client.rerank(data).await?;
+                let list = client.rerank(&data).await?;
                 let ids: Vec<_> = list
                     .into_iter()
                     .take(top_k)
@@ -588,7 +588,7 @@ impl Rag {
                 query,
             };
             let chunk_output = embedding_client
-                .embeddings(chunk_data)
+                .embeddings(&chunk_data)
                 .await
                 .context("Failed to create embedding")?;
             output.extend(chunk_output);
-- 
cgit v1.2.3