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/serve.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/serve.rs')

diff --git a/src/serve.rs b/src/serve.rs
index ab8e654..51cc010 100644
--- a/src/serve.rs
+++ b/src/serve.rs
@@ -471,7 +471,7 @@ impl Server {
         };
         let client = init_client(&config, Some(embedding_model))?;
         let data = client
-            .embeddings(EmbeddingsData {
+            .embeddings(&EmbeddingsData {
                 query: false,
                 texts,
             })
@@ -526,7 +526,7 @@ impl Server {
 
         let client = init_client(&config, Some(reranker_model))?;
         let data = client
-            .rerank(RerankData {
+            .rerank(&RerankData {
                 query,
                 documents: documents.clone(),
                 top_n,
-- 
cgit v1.2.3