diff options
| author | sigoden <sigoden@gmail.com> | 2024-09-02 08:23:21 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-02 08:23:21 +0800 |
| commit | 2cba09c064b366ad91c7eda3acbc2030215a1842 (patch) | |
| tree | 467d550c12b2cab8172da7cc3af01903a5b5477a /src/client/gemini.rs | |
| parent | 6462a587428c38c17cbb6bce440cbf10c966eecc (diff) | |
| download | aichat-2cba09c064b366ad91c7eda3acbc2030215a1842.tar.gz | |
feat: migrate `cloudflare` client to `openai-compatible` (#821)
Diffstat (limited to 'src/client/gemini.rs')
| -rw-r--r-- | src/client/gemini.rs | 4 |
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?; |
