From a491820ab5ff9bb58e62fd25d88f31ab53a81db9 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 6 Feb 2025 19:20:48 +0800
Subject: feat: support model alias (#1150)

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

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

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
-- 
cgit v1.2.3