diff options
Diffstat (limited to 'src/client')
| -rw-r--r-- | src/client/common.rs | 6 | ||||
| -rw-r--r-- | src/client/model.rs | 10 |
2 files changed, 8 insertions, 8 deletions
diff --git a/src/client/common.rs b/src/client/common.rs index 6c86acd..de2fa50 100644 --- a/src/client/common.rs +++ b/src/client/common.rs @@ -145,15 +145,15 @@ macro_rules! register_client { } pub fn list_chat_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { - list_models(config).into_iter().filter(|v| v.mode() == "chat").collect() + list_models(config).into_iter().filter(|v| v.model_type() == "chat").collect() } pub fn list_embedding_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { - list_models(config).into_iter().filter(|v| v.mode() == "embedding").collect() + list_models(config).into_iter().filter(|v| v.model_type() == "embedding").collect() } pub fn list_rerank_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { - list_models(config).into_iter().filter(|v| v.mode() == "rerank").collect() + list_models(config).into_iter().filter(|v| v.model_type() == "rerank").collect() } }; } diff --git a/src/client/model.rs b/src/client/model.rs index 06f5577..0584f82 100644 --- a/src/client/model.rs +++ b/src/client/model.rs @@ -111,8 +111,8 @@ impl Model { &self.data.name } - pub fn mode(&self) -> &str { - &self.data.mode + pub fn model_type(&self) -> &str { + &self.data.model_type } pub fn data(&self) -> &ModelData { @@ -245,8 +245,8 @@ impl Model { #[derive(Debug, Clone, Default, Serialize, Deserialize)] pub struct ModelData { pub name: String, - #[serde(default = "default_model_mode")] - pub mode: String, + #[serde(default = "default_model_type", rename = "type")] + pub model_type: String, pub max_input_tokens: Option<usize>, pub input_price: Option<f64>, pub output_price: Option<f64>, @@ -280,6 +280,6 @@ pub struct BuiltinModels { pub models: Vec<ModelData>, } -fn default_model_mode() -> String { +fn default_model_type() -> String { "chat".into() } |