diff options
| author | sigoden <sigoden@gmail.com> | 2025-02-06 19:20:48 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-02-06 19:20:48 +0800 |
| commit | a491820ab5ff9bb58e62fd25d88f31ab53a81db9 (patch) | |
| tree | 86a84d7176a5fb16bce9b195808e952d181a8146 /src/client/gemini.rs | |
| parent | 4812e446ee30467bd3ecce983a826f7649de7110 (diff) | |
| download | aichat-a491820ab5ff9bb58e62fd25d88f31ab53a81db9.tar.gz | |
feat: support model alias (#1150)
Diffstat (limited to 'src/client/gemini.rs')
| -rw-r--r-- | src/client/gemini.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/client/gemini.rs b/src/client/gemini.rs index 85917c3..77304e7 100644 --- a/src/client/gemini.rs +++ b/src/client/gemini.rs @@ -54,7 +54,7 @@ fn prepare_chat_completions( let url = format!( "{}/models/{}:{}?key={}", api_base.trim_end_matches('/'), - self_.model.name(), + self_.model.real_name(), func, api_key ); @@ -75,11 +75,11 @@ fn prepare_embeddings(self_: &GeminiClient, data: &EmbeddingsData) -> Result<Req let url = format!( "{}/models/{}:batchEmbedContents?key={}", api_base.trim_end_matches('/'), - self_.model.name(), + self_.model.real_name(), api_key ); - let model_id = format!("models/{}", self_.model.name()); + let model_id = format!("models/{}", self_.model.real_name()); let requests: Vec<_> = data .texts |