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/model.rs | |
| parent | 4812e446ee30467bd3ecce983a826f7649de7110 (diff) | |
| download | aichat-a491820ab5ff9bb58e62fd25d88f31ab53a81db9.tar.gz | |
feat: support model alias (#1150)
Diffstat (limited to 'src/client/model.rs')
| -rw-r--r-- | src/client/model.rs | 6 |
1 files changed, 6 insertions, 0 deletions
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>, |