From 2cba09c064b366ad91c7eda3acbc2030215a1842 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 2 Sep 2024 08:23:21 +0800
Subject: feat: migrate `cloudflare` client to `openai-compatible` (#821)

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

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

diff --git a/src/client/gemini.rs b/src/client/gemini.rs
index 572e082..9578e23 100644
--- a/src/client/gemini.rs
+++ b/src/client/gemini.rs
@@ -34,7 +34,7 @@ impl_client_trait!(
         gemini_chat_completions,
         gemini_chat_completions_streaming
     ),
-    (prepare_embeddings, gemini_embeddings),
+    (prepare_embeddings, embeddings),
     (noop_prepare_rerank, noop_rerank),
 );
 
@@ -95,7 +95,7 @@ fn prepare_embeddings(self_: &GeminiClient, data: EmbeddingsData) -> Result<Requ
     Ok(request_data)
 }
 
-async fn gemini_embeddings(builder: RequestBuilder, _model: &Model) -> Result<EmbeddingsOutput> {
+async fn embeddings(builder: RequestBuilder, _model: &Model) -> Result<EmbeddingsOutput> {
     let res = builder.send().await?;
     let status = res.status();
     let data: Value = res.json().await?;
-- 
cgit v1.2.3