987 lines · 39.4 KB · Rust
RawDownload
use serde::{Deserialize, Serialize};use strum::EnumIter; #[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq)]pub enum BedrockModelMode {    #[default]    Default,    Thinking {        budget_tokens: Option<u64>,    },} #[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq)]pub struct BedrockModelCacheConfiguration {    pub max_cache_anchors: usize,    pub min_total_token: u64,} #[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq, EnumIter)]pub enum Model {    // Anthropic models (already included)    #[serde(rename = "claude-sonnet-4", alias = "claude-sonnet-4-latest")]    ClaudeSonnet4,    #[serde(        rename = "claude-sonnet-4-thinking",        alias = "claude-sonnet-4-thinking-latest"    )]    ClaudeSonnet4Thinking,    #[default]    #[serde(rename = "claude-sonnet-4-5", alias = "claude-sonnet-4-5-latest")]    ClaudeSonnet4_5,    #[serde(        rename = "claude-sonnet-4-5-thinking",        alias = "claude-sonnet-4-5-thinking-latest"    )]    ClaudeSonnet4_5Thinking,    #[serde(rename = "claude-opus-4", alias = "claude-opus-4-latest")]    ClaudeOpus4,    #[serde(rename = "claude-opus-4-1", alias = "claude-opus-4-1-latest")]    ClaudeOpus4_1,    #[serde(        rename = "claude-opus-4-thinking",        alias = "claude-opus-4-thinking-latest"    )]    ClaudeOpus4Thinking,    #[serde(        rename = "claude-opus-4-1-thinking",        alias = "claude-opus-4-1-thinking-latest"    )]    ClaudeOpus4_1Thinking,    #[serde(rename = "claude-opus-4-5", alias = "claude-opus-4-5-latest")]    ClaudeOpus4_5,    #[serde(        rename = "claude-opus-4-5-thinking",        alias = "claude-opus-4-5-thinking-latest"    )]    ClaudeOpus4_5Thinking,    #[serde(rename = "claude-3-5-sonnet-v2", alias = "claude-3-5-sonnet-latest")]    Claude3_5SonnetV2,    #[serde(rename = "claude-3-7-sonnet", alias = "claude-3-7-sonnet-latest")]    Claude3_7Sonnet,    #[serde(        rename = "claude-3-7-sonnet-thinking",        alias = "claude-3-7-sonnet-thinking-latest"    )]    Claude3_7SonnetThinking,    #[serde(rename = "claude-3-opus", alias = "claude-3-opus-latest")]    Claude3Opus,    #[serde(rename = "claude-3-sonnet", alias = "claude-3-sonnet-latest")]    Claude3Sonnet,    #[serde(rename = "claude-3-5-haiku", alias = "claude-3-5-haiku-latest")]    Claude3_5Haiku,    #[serde(rename = "claude-haiku-4-5", alias = "claude-haiku-4-5-latest")]    ClaudeHaiku4_5,    Claude3_5Sonnet,    Claude3Haiku,    // Amazon Nova Models    AmazonNovaLite,    AmazonNovaMicro,    AmazonNovaPro,    AmazonNovaPremier,    // AI21 models    AI21J2GrandeInstruct,    AI21J2JumboInstruct,    AI21J2Mid,    AI21J2MidV1,    AI21J2Ultra,    AI21J2UltraV1_8k,    AI21J2UltraV1,    AI21JambaInstructV1,    AI21Jamba15LargeV1,    AI21Jamba15MiniV1,    // Cohere models    CohereCommandTextV14_4k,    CohereCommandRV1,    CohereCommandRPlusV1,    CohereCommandLightTextV14_4k,    // DeepSeek    DeepSeekR1,    // Meta models    MetaLlama38BInstructV1,    MetaLlama370BInstructV1,    MetaLlama318BInstructV1_128k,    MetaLlama318BInstructV1,    MetaLlama3170BInstructV1_128k,    MetaLlama3170BInstructV1,    MetaLlama31405BInstructV1,    MetaLlama321BInstructV1,    MetaLlama323BInstructV1,    MetaLlama3211BInstructV1,    MetaLlama3290BInstructV1,    MetaLlama3370BInstructV1,    #[allow(non_camel_case_types)]    MetaLlama4Scout17BInstructV1,    #[allow(non_camel_case_types)]    MetaLlama4Maverick17BInstructV1,    // Mistral models    MistralMistral7BInstructV0,    MistralMixtral8x7BInstructV0,    MistralMistralLarge2402V1,    MistralMistralSmall2402V1,    MistralPixtralLarge2502V1,    // Writer models    PalmyraWriterX5,    PalmyraWriterX4,    #[serde(rename = "custom")]    Custom {        name: String,        max_tokens: u64,        /// The name displayed in the UI, such as in the assistant panel model dropdown menu.        display_name: Option<String>,        max_output_tokens: Option<u64>,        default_temperature: Option<f32>,        cache_configuration: Option<BedrockModelCacheConfiguration>,    },} impl Model {    pub fn default_fast(region: &str) -> Self {        if region.starts_with("us-") {            Self::Claude3_5Haiku        } else {            Self::Claude3Haiku        }    }     pub fn from_id(id: &str) -> anyhow::Result<Self> {        if id.starts_with("claude-opus-4-5-thinking") {            Ok(Self::ClaudeOpus4_5Thinking)        } else if id.starts_with("claude-opus-4-5") {            Ok(Self::ClaudeOpus4_5)        } else if id.starts_with("claude-opus-4-1-thinking") {            Ok(Self::ClaudeOpus4_1Thinking)        } else if id.starts_with("claude-opus-4-1") {            Ok(Self::ClaudeOpus4_1)        } else if id.starts_with("claude-opus-4-thinking") {            Ok(Self::ClaudeOpus4Thinking)        } else if id.starts_with("claude-opus-4") {            Ok(Self::ClaudeOpus4)        } else if id.starts_with("claude-3-5-sonnet-v2") {            Ok(Self::Claude3_5SonnetV2)        } else if id.starts_with("claude-3-opus") {            Ok(Self::Claude3Opus)        } else if id.starts_with("claude-3-sonnet") {            Ok(Self::Claude3Sonnet)        } else if id.starts_with("claude-3-5-haiku") {            Ok(Self::Claude3_5Haiku)        } else if id.starts_with("claude-haiku-4-5") {            Ok(Self::ClaudeHaiku4_5)        } else if id.starts_with("claude-3-7-sonnet") {            Ok(Self::Claude3_7Sonnet)        } else if id.starts_with("claude-3-7-sonnet-thinking") {            Ok(Self::Claude3_7SonnetThinking)        } else if id.starts_with("claude-sonnet-4-5-thinking") {            Ok(Self::ClaudeSonnet4_5Thinking)        } else if id.starts_with("claude-sonnet-4-5") {            Ok(Self::ClaudeSonnet4_5)        } else if id.starts_with("claude-sonnet-4-thinking") {            Ok(Self::ClaudeSonnet4Thinking)        } else if id.starts_with("claude-sonnet-4") {            Ok(Self::ClaudeSonnet4)        } else {            anyhow::bail!("invalid model id {id}");        }    }     pub fn id(&self) -> &str {        match self {            Model::ClaudeSonnet4 => "claude-sonnet-4",            Model::ClaudeSonnet4Thinking => "claude-sonnet-4-thinking",            Model::ClaudeSonnet4_5 => "claude-sonnet-4-5",            Model::ClaudeSonnet4_5Thinking => "claude-sonnet-4-5-thinking",            Model::ClaudeOpus4 => "claude-opus-4",            Model::ClaudeOpus4_1 => "claude-opus-4-1",            Model::ClaudeOpus4Thinking => "claude-opus-4-thinking",            Model::ClaudeOpus4_1Thinking => "claude-opus-4-1-thinking",            Model::ClaudeOpus4_5 => "claude-opus-4-5",            Model::ClaudeOpus4_5Thinking => "claude-opus-4-5-thinking",            Model::Claude3_5SonnetV2 => "claude-3-5-sonnet-v2",            Model::Claude3_5Sonnet => "claude-3-5-sonnet",            Model::Claude3Opus => "claude-3-opus",            Model::Claude3Sonnet => "claude-3-sonnet",            Model::Claude3Haiku => "claude-3-haiku",            Model::Claude3_5Haiku => "claude-3-5-haiku",            Model::ClaudeHaiku4_5 => "claude-haiku-4-5",            Model::Claude3_7Sonnet => "claude-3-7-sonnet",            Model::Claude3_7SonnetThinking => "claude-3-7-sonnet-thinking",            Model::AmazonNovaLite => "amazon-nova-lite",            Model::AmazonNovaMicro => "amazon-nova-micro",            Model::AmazonNovaPro => "amazon-nova-pro",            Model::AmazonNovaPremier => "amazon-nova-premier",            Model::DeepSeekR1 => "deepseek-r1",            Model::AI21J2GrandeInstruct => "ai21-j2-grande-instruct",            Model::AI21J2JumboInstruct => "ai21-j2-jumbo-instruct",            Model::AI21J2Mid => "ai21-j2-mid",            Model::AI21J2MidV1 => "ai21-j2-mid-v1",            Model::AI21J2Ultra => "ai21-j2-ultra",            Model::AI21J2UltraV1_8k => "ai21-j2-ultra-v1-8k",            Model::AI21J2UltraV1 => "ai21-j2-ultra-v1",            Model::AI21JambaInstructV1 => "ai21-jamba-instruct-v1",            Model::AI21Jamba15LargeV1 => "ai21-jamba-1-5-large-v1",            Model::AI21Jamba15MiniV1 => "ai21-jamba-1-5-mini-v1",            Model::CohereCommandTextV14_4k => "cohere-command-text-v14-4k",            Model::CohereCommandRV1 => "cohere-command-r-v1",            Model::CohereCommandRPlusV1 => "cohere-command-r-plus-v1",            Model::CohereCommandLightTextV14_4k => "cohere-command-light-text-v14-4k",            Model::MetaLlama38BInstructV1 => "meta-llama3-8b-instruct-v1",            Model::MetaLlama370BInstructV1 => "meta-llama3-70b-instruct-v1",            Model::MetaLlama318BInstructV1_128k => "meta-llama3-1-8b-instruct-v1-128k",            Model::MetaLlama318BInstructV1 => "meta-llama3-1-8b-instruct-v1",            Model::MetaLlama3170BInstructV1_128k => "meta-llama3-1-70b-instruct-v1-128k",            Model::MetaLlama3170BInstructV1 => "meta-llama3-1-70b-instruct-v1",            Model::MetaLlama31405BInstructV1 => "meta-llama3-1-405b-instruct-v1",            Model::MetaLlama321BInstructV1 => "meta-llama3-2-1b-instruct-v1",            Model::MetaLlama323BInstructV1 => "meta-llama3-2-3b-instruct-v1",            Model::MetaLlama3211BInstructV1 => "meta-llama3-2-11b-instruct-v1",            Model::MetaLlama3290BInstructV1 => "meta-llama3-2-90b-instruct-v1",            Model::MetaLlama3370BInstructV1 => "meta-llama3-3-70b-instruct-v1",            Model::MetaLlama4Scout17BInstructV1 => "meta-llama4-scout-17b-instruct-v1",            Model::MetaLlama4Maverick17BInstructV1 => "meta-llama4-maverick-17b-instruct-v1",            Model::MistralMistral7BInstructV0 => "mistral-7b-instruct-v0",            Model::MistralMixtral8x7BInstructV0 => "mistral-mixtral-8x7b-instruct-v0",            Model::MistralMistralLarge2402V1 => "mistral-large-2402-v1",            Model::MistralMistralSmall2402V1 => "mistral-small-2402-v1",            Model::MistralPixtralLarge2502V1 => "mistral-pixtral-large-2502-v1",            Model::PalmyraWriterX4 => "palmyra-writer-x4",            Model::PalmyraWriterX5 => "palmyra-writer-x5",            Self::Custom { name, .. } => name,        }    }     pub fn request_id(&self) -> &str {        match self {            Model::ClaudeSonnet4 | Model::ClaudeSonnet4Thinking => {                "anthropic.claude-sonnet-4-20250514-v1:0"            }            Model::ClaudeSonnet4_5 | Model::ClaudeSonnet4_5Thinking => {                "anthropic.claude-sonnet-4-5-20250929-v1:0"            }            Model::ClaudeOpus4 | Model::ClaudeOpus4Thinking => {                "anthropic.claude-opus-4-20250514-v1:0"            }            Model::ClaudeOpus4_1 | Model::ClaudeOpus4_1Thinking => {                "anthropic.claude-opus-4-1-20250805-v1:0"            }            Model::ClaudeOpus4_5 | Model::ClaudeOpus4_5Thinking => {                "anthropic.claude-opus-4-5-20251101-v1:0"            }            Model::Claude3_5SonnetV2 => "anthropic.claude-3-5-sonnet-20241022-v2:0",            Model::Claude3_5Sonnet => "anthropic.claude-3-5-sonnet-20240620-v1:0",            Model::Claude3Opus => "anthropic.claude-3-opus-20240229-v1:0",            Model::Claude3Sonnet => "anthropic.claude-3-sonnet-20240229-v1:0",            Model::Claude3Haiku => "anthropic.claude-3-haiku-20240307-v1:0",            Model::Claude3_5Haiku => "anthropic.claude-3-5-haiku-20241022-v1:0",            Model::ClaudeHaiku4_5 => "anthropic.claude-haiku-4-5-20251001-v1:0",            Model::Claude3_7Sonnet | Model::Claude3_7SonnetThinking => {                "anthropic.claude-3-7-sonnet-20250219-v1:0"            }            Model::AmazonNovaLite => "amazon.nova-lite-v1:0",            Model::AmazonNovaMicro => "amazon.nova-micro-v1:0",            Model::AmazonNovaPro => "amazon.nova-pro-v1:0",            Model::AmazonNovaPremier => "amazon.nova-premier-v1:0",            Model::DeepSeekR1 => "deepseek.r1-v1:0",            Model::AI21J2GrandeInstruct => "ai21.j2-grande-instruct",            Model::AI21J2JumboInstruct => "ai21.j2-jumbo-instruct",            Model::AI21J2Mid => "ai21.j2-mid",            Model::AI21J2MidV1 => "ai21.j2-mid-v1",            Model::AI21J2Ultra => "ai21.j2-ultra",            Model::AI21J2UltraV1_8k => "ai21.j2-ultra-v1:0:8k",            Model::AI21J2UltraV1 => "ai21.j2-ultra-v1",            Model::AI21JambaInstructV1 => "ai21.jamba-instruct-v1:0",            Model::AI21Jamba15LargeV1 => "ai21.jamba-1-5-large-v1:0",            Model::AI21Jamba15MiniV1 => "ai21.jamba-1-5-mini-v1:0",            Model::CohereCommandTextV14_4k => "cohere.command-text-v14:7:4k",            Model::CohereCommandRV1 => "cohere.command-r-v1:0",            Model::CohereCommandRPlusV1 => "cohere.command-r-plus-v1:0",            Model::CohereCommandLightTextV14_4k => "cohere.command-light-text-v14:7:4k",            Model::MetaLlama38BInstructV1 => "meta.llama3-8b-instruct-v1:0",            Model::MetaLlama370BInstructV1 => "meta.llama3-70b-instruct-v1:0",            Model::MetaLlama318BInstructV1_128k => "meta.llama3-1-8b-instruct-v1:0",            Model::MetaLlama318BInstructV1 => "meta.llama3-1-8b-instruct-v1:0",            Model::MetaLlama3170BInstructV1_128k => "meta.llama3-1-70b-instruct-v1:0",            Model::MetaLlama3170BInstructV1 => "meta.llama3-1-70b-instruct-v1:0",            Model::MetaLlama31405BInstructV1 => "meta.llama3-1-405b-instruct-v1:0",            Model::MetaLlama3211BInstructV1 => "meta.llama3-2-11b-instruct-v1:0",            Model::MetaLlama3290BInstructV1 => "meta.llama3-2-90b-instruct-v1:0",            Model::MetaLlama321BInstructV1 => "meta.llama3-2-1b-instruct-v1:0",            Model::MetaLlama323BInstructV1 => "meta.llama3-2-3b-instruct-v1:0",            Model::MetaLlama3370BInstructV1 => "meta.llama3-3-70b-instruct-v1:0",            Model::MetaLlama4Scout17BInstructV1 => "meta.llama4-scout-17b-instruct-v1:0",            Model::MetaLlama4Maverick17BInstructV1 => "meta.llama4-maverick-17b-instruct-v1:0",            Model::MistralMistral7BInstructV0 => "mistral.mistral-7b-instruct-v0:2",            Model::MistralMixtral8x7BInstructV0 => "mistral.mixtral-8x7b-instruct-v0:1",            Model::MistralMistralLarge2402V1 => "mistral.mistral-large-2402-v1:0",            Model::MistralMistralSmall2402V1 => "mistral.mistral-small-2402-v1:0",            Model::MistralPixtralLarge2502V1 => "mistral.pixtral-large-2502-v1:0",            Model::PalmyraWriterX4 => "writer.palmyra-x4-v1:0",            Model::PalmyraWriterX5 => "writer.palmyra-x5-v1:0",            Self::Custom { name, .. } => name,        }    }     pub fn display_name(&self) -> &str {        match self {            Self::ClaudeSonnet4 => "Claude Sonnet 4",            Self::ClaudeSonnet4Thinking => "Claude Sonnet 4 Thinking",            Self::ClaudeSonnet4_5 => "Claude Sonnet 4.5",            Self::ClaudeSonnet4_5Thinking => "Claude Sonnet 4.5 Thinking",            Self::ClaudeOpus4 => "Claude Opus 4",            Self::ClaudeOpus4_1 => "Claude Opus 4.1",            Self::ClaudeOpus4Thinking => "Claude Opus 4 Thinking",            Self::ClaudeOpus4_1Thinking => "Claude Opus 4.1 Thinking",            Self::ClaudeOpus4_5 => "Claude Opus 4.5",            Self::ClaudeOpus4_5Thinking => "Claude Opus 4.5 Thinking",            Self::Claude3_5SonnetV2 => "Claude 3.5 Sonnet v2",            Self::Claude3_5Sonnet => "Claude 3.5 Sonnet",            Self::Claude3Opus => "Claude 3 Opus",            Self::Claude3Sonnet => "Claude 3 Sonnet",            Self::Claude3Haiku => "Claude 3 Haiku",            Self::Claude3_5Haiku => "Claude 3.5 Haiku",            Self::ClaudeHaiku4_5 => "Claude Haiku 4.5",            Self::Claude3_7Sonnet => "Claude 3.7 Sonnet",            Self::Claude3_7SonnetThinking => "Claude 3.7 Sonnet Thinking",            Self::AmazonNovaLite => "Amazon Nova Lite",            Self::AmazonNovaMicro => "Amazon Nova Micro",            Self::AmazonNovaPro => "Amazon Nova Pro",            Self::AmazonNovaPremier => "Amazon Nova Premier",            Self::DeepSeekR1 => "DeepSeek R1",            Self::AI21J2GrandeInstruct => "AI21 Jurassic2 Grande Instruct",            Self::AI21J2JumboInstruct => "AI21 Jurassic2 Jumbo Instruct",            Self::AI21J2Mid => "AI21 Jurassic2 Mid",            Self::AI21J2MidV1 => "AI21 Jurassic2 Mid V1",            Self::AI21J2Ultra => "AI21 Jurassic2 Ultra",            Self::AI21J2UltraV1_8k => "AI21 Jurassic2 Ultra V1 8K",            Self::AI21J2UltraV1 => "AI21 Jurassic2 Ultra V1",            Self::AI21JambaInstructV1 => "AI21 Jamba Instruct",            Self::AI21Jamba15LargeV1 => "AI21 Jamba 1.5 Large",            Self::AI21Jamba15MiniV1 => "AI21 Jamba 1.5 Mini",            Self::CohereCommandTextV14_4k => "Cohere Command Text V14 4K",            Self::CohereCommandRV1 => "Cohere Command R V1",            Self::CohereCommandRPlusV1 => "Cohere Command R Plus V1",            Self::CohereCommandLightTextV14_4k => "Cohere Command Light Text V14 4K",            Self::MetaLlama38BInstructV1 => "Meta Llama 3 8B Instruct",            Self::MetaLlama370BInstructV1 => "Meta Llama 3 70B Instruct",            Self::MetaLlama318BInstructV1_128k => "Meta Llama 3.1 8B Instruct 128K",            Self::MetaLlama318BInstructV1 => "Meta Llama 3.1 8B Instruct",            Self::MetaLlama3170BInstructV1_128k => "Meta Llama 3.1 70B Instruct 128K",            Self::MetaLlama3170BInstructV1 => "Meta Llama 3.1 70B Instruct",            Self::MetaLlama31405BInstructV1 => "Meta Llama 3.1 405B Instruct",            Self::MetaLlama3211BInstructV1 => "Meta Llama 3.2 11B Instruct",            Self::MetaLlama3290BInstructV1 => "Meta Llama 3.2 90B Instruct",            Self::MetaLlama321BInstructV1 => "Meta Llama 3.2 1B Instruct",            Self::MetaLlama323BInstructV1 => "Meta Llama 3.2 3B Instruct",            Self::MetaLlama3370BInstructV1 => "Meta Llama 3.3 70B Instruct",            Self::MetaLlama4Scout17BInstructV1 => "Meta Llama 4 Scout 17B Instruct",            Self::MetaLlama4Maverick17BInstructV1 => "Meta Llama 4 Maverick 17B Instruct",            Self::MistralMistral7BInstructV0 => "Mistral 7B Instruct V0",            Self::MistralMixtral8x7BInstructV0 => "Mistral Mixtral 8x7B Instruct V0",            Self::MistralMistralLarge2402V1 => "Mistral Large 2402 V1",            Self::MistralMistralSmall2402V1 => "Mistral Small 2402 V1",            Self::MistralPixtralLarge2502V1 => "Pixtral Large 25.02 V1",            Self::PalmyraWriterX5 => "Writer Palmyra X5",            Self::PalmyraWriterX4 => "Writer Palmyra X4",            Self::Custom {                display_name, name, ..            } => display_name.as_deref().unwrap_or(name),        }    }     pub fn max_token_count(&self) -> u64 {        match self {            Self::Claude3_5SonnetV2            | Self::Claude3Opus            | Self::Claude3Sonnet            | Self::Claude3_5Haiku            | Self::ClaudeHaiku4_5            | Self::Claude3_7Sonnet            | Self::ClaudeSonnet4            | Self::ClaudeOpus4            | Self::ClaudeOpus4_1            | Self::ClaudeSonnet4Thinking            | Self::ClaudeSonnet4_5            | Self::ClaudeSonnet4_5Thinking            | Self::ClaudeOpus4Thinking            | Self::ClaudeOpus4_1Thinking            | Self::ClaudeOpus4_5            | Self::ClaudeOpus4_5Thinking => 200_000,            Self::AmazonNovaPremier => 1_000_000,            Self::PalmyraWriterX5 => 1_000_000,            Self::PalmyraWriterX4 => 128_000,            Self::Custom { max_tokens, .. } => *max_tokens,            _ => 128_000,        }    }     pub fn max_output_tokens(&self) -> u64 {        match self {            Self::Claude3Opus | Self::Claude3Sonnet | Self::Claude3_5Haiku => 4_096,            Self::Claude3_7Sonnet | Self::Claude3_7SonnetThinking => 128_000,            Self::ClaudeSonnet4 | Self::ClaudeSonnet4Thinking => 64_000,            Self::ClaudeSonnet4_5            | Self::ClaudeSonnet4_5Thinking            | Self::ClaudeHaiku4_5            | Self::ClaudeOpus4_5            | Self::ClaudeOpus4_5Thinking => 64_000,            Self::ClaudeOpus4            | Self::ClaudeOpus4Thinking            | Self::ClaudeOpus4_1            | Self::ClaudeOpus4_1Thinking => 32_000,            Self::Claude3_5SonnetV2 | Self::PalmyraWriterX4 | Self::PalmyraWriterX5 => 8_192,            Self::Custom {                max_output_tokens, ..            } => max_output_tokens.unwrap_or(4_096),            _ => 4_096,        }    }     pub fn default_temperature(&self) -> f32 {        match self {            Self::Claude3_5SonnetV2            | Self::Claude3Opus            | Self::Claude3Sonnet            | Self::Claude3_5Haiku            | Self::ClaudeHaiku4_5            | Self::Claude3_7Sonnet            | Self::ClaudeOpus4            | Self::ClaudeOpus4Thinking            | Self::ClaudeOpus4_1            | Self::ClaudeOpus4_1Thinking            | Self::ClaudeOpus4_5            | Self::ClaudeOpus4_5Thinking            | Self::ClaudeSonnet4            | Self::ClaudeSonnet4Thinking            | Self::ClaudeSonnet4_5            | Self::ClaudeSonnet4_5Thinking => 1.0,            Self::Custom {                default_temperature,                ..            } => default_temperature.unwrap_or(1.0),            _ => 1.0,        }    }     pub fn supports_tool_use(&self) -> bool {        match self {            // Anthropic Claude 3 models (all support tool use)            Self::Claude3Opus            | Self::Claude3Sonnet            | Self::Claude3_5Sonnet            | Self::Claude3_5SonnetV2            | Self::Claude3_7Sonnet            | Self::Claude3_7SonnetThinking            | Self::ClaudeOpus4            | Self::ClaudeOpus4Thinking            | Self::ClaudeOpus4_1            | Self::ClaudeOpus4_1Thinking            | Self::ClaudeOpus4_5            | Self::ClaudeOpus4_5Thinking            | Self::ClaudeSonnet4            | Self::ClaudeSonnet4Thinking            | Self::ClaudeSonnet4_5            | Self::ClaudeSonnet4_5Thinking            | Self::Claude3_5Haiku            | Self::ClaudeHaiku4_5 => true,             // Amazon Nova models (all support tool use)            Self::AmazonNovaPremier            | Self::AmazonNovaPro            | Self::AmazonNovaLite            | Self::AmazonNovaMicro => true,             // AI21 Jamba 1.5 models support tool use            Self::AI21Jamba15LargeV1 | Self::AI21Jamba15MiniV1 => true,             // Cohere Command R models support tool use            Self::CohereCommandRV1 | Self::CohereCommandRPlusV1 => true, 

Lines 1–500 of 987.