diff options
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>, |