diff options
Diffstat (limited to 'src/client/model.rs')
| -rw-r--r-- | src/client/model.rs | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/client/model.rs b/src/client/model.rs index bcd474f..94332f5 100644 --- a/src/client/model.rs +++ b/src/client/model.rs @@ -131,6 +131,7 @@ impl Model { output_price, supports_vision, supports_function_calling, + supports_reasoning, .. } = &self.data; let max_input_tokens = stringify_option_value(max_input_tokens); @@ -144,6 +145,9 @@ impl Model { if *supports_function_calling { capabilities.push('⚒'); }; + if *supports_reasoning { + capabilities.push('🄡'); + } let capabilities: String = capabilities .into_iter() .map(|v| format!("{v} ")) @@ -178,6 +182,10 @@ impl Model { self.data.max_output_tokens } + pub fn supports_reasoning(&self) -> bool { + self.data.supports_reasoning + } + pub fn no_stream(&self) -> bool { self.data.no_stream } @@ -302,6 +310,8 @@ pub struct ModelData { #[serde(default, skip_serializing_if = "std::ops::Not::not")] pub supports_function_calling: bool, #[serde(default, skip_serializing_if = "std::ops::Not::not")] + supports_reasoning: bool, + #[serde(default, skip_serializing_if = "std::ops::Not::not")] no_stream: bool, #[serde(default, skip_serializing_if = "std::ops::Not::not")] no_system_message: bool, |