use std::fmt;
use serde::Serialize;
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Model {
Sonnet,
Opus,
Haiku,
Inherit,
Custom(String),
}
impl Model {
pub fn as_str(&self) -> &str {
match self {
Self::Sonnet => "sonnet",
Self::Opus => "opus",
Self::Haiku => "haiku",
Self::Inherit => "inherit",
Self::Custom(s) => s,
}
}
}
impl fmt::Display for Model {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.as_str())
}
}
impl From<&str> for Model {
fn from(s: &str) -> Self {
match s {
"sonnet" | "sonnet-4-5" | "claude-sonnet-4-5-20250929" => Self::Sonnet,
"opus" | "opus-4-5" | "claude-opus-4-5-20250929" => Self::Opus,
"haiku" | "haiku-4-5" | "claude-haiku-4-5-20251001" => Self::Haiku,
"inherit" => Self::Inherit,
_ => Self::Custom(s.to_owned()),
}
}
}
impl From<String> for Model {
fn from(s: String) -> Self {
Self::from(s.as_str())
}
}
impl Serialize for Model {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serializer.serialize_str(self.as_str())
}
}