summaryrefslogtreecommitdiffstats
path: root/src/client/gemini.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-09-02 08:23:21 +0800
committerGitHub <noreply@github.com>2024-09-02 08:23:21 +0800
commit2cba09c064b366ad91c7eda3acbc2030215a1842 (patch)
tree467d550c12b2cab8172da7cc3af01903a5b5477a /src/client/gemini.rs
parent6462a587428c38c17cbb6bce440cbf10c966eecc (diff)
downloadaichat-2cba09c064b366ad91c7eda3acbc2030215a1842.tar.gz
feat: migrate `cloudflare` client to `openai-compatible` (#821)
Diffstat (limited to 'src/client/gemini.rs')
-rw-r--r--src/client/gemini.rs4
1 files changed, 2 insertions, 2 deletions
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?;