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

(limited to 'src/client/bedrock.rs')

diff --git a/src/client/bedrock.rs b/src/client/bedrock.rs
index 25c666c..f7b3019 100644
--- a/src/client/bedrock.rs
+++ b/src/client/bedrock.rs
@@ -98,7 +98,7 @@ impl BedrockClient {
     fn embeddings_builder(
         &self,
         client: &ReqwestClient,
-        data: EmbeddingsData,
+        data: &EmbeddingsData,
     ) -> Result<RequestBuilder> {
         let access_key_id = self.get_access_key_id()?;
         let secret_access_key = self.get_secret_access_key()?;
@@ -173,7 +173,7 @@ impl Client for BedrockClient {
     async fn embeddings_inner(
         &self,
         client: &ReqwestClient,
-        data: EmbeddingsData,
+        data: &EmbeddingsData,
     ) -> Result<EmbeddingsOutput> {
         let builder = self.embeddings_builder(client, data)?;
         embeddings(builder).await
-- 
cgit v1.2.3