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/model.rs | 6 ++++++
 1 file changed, 6 insertions(+)

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

diff --git a/src/client/model.rs b/src/client/model.rs
index 8b8a290..d80a1d4 100644
--- a/src/client/model.rs
+++ b/src/client/model.rs
@@ -103,6 +103,10 @@ impl Model {
         &self.data.name
     }
 
+    pub fn real_name(&self) -> &str {
+        self.data.real_name.as_deref().unwrap_or(&self.data.name)
+    }
+
     pub fn model_type(&self) -> ModelType {
         if self.data.model_type.starts_with("embed") {
             ModelType::Embedding
@@ -294,6 +298,8 @@ pub struct ModelData {
     #[serde(default = "default_model_type", rename = "type")]
     pub model_type: String,
     #[serde(skip_serializing_if = "Option::is_none")]
+    pub real_name: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     pub max_input_tokens: Option<usize>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub input_price: Option<f64>,
-- 
cgit v1.2.3