use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct WorkflowAnthropicModel {
#[serde(rename = "provider")]
pub provider: ProviderTrue,
#[serde(rename = "model")]
pub model: ModelTrue,
#[serde(rename = "thinking", skip_serializing_if = "Option::is_none")]
pub thinking: Option<models::AnthropicThinkingConfig>,
#[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
pub temperature: Option<f64>,
#[serde(rename = "maxTokens", skip_serializing_if = "Option::is_none")]
pub max_tokens: Option<f64>,
}
impl WorkflowAnthropicModel {
pub fn new(provider: ProviderTrue, model: ModelTrue) -> WorkflowAnthropicModel {
WorkflowAnthropicModel {
provider,
model,
thinking: None,
temperature: None,
max_tokens: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ProviderTrue {
#[serde(rename = "anthropic")]
Anthropic,
}
impl Default for ProviderTrue {
fn default() -> ProviderTrue {
Self::Anthropic
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ModelTrue {
#[serde(rename = "claude-3-opus-20240229")]
Claude3Opus20240229,
#[serde(rename = "claude-3-sonnet-20240229")]
Claude3Sonnet20240229,
#[serde(rename = "claude-3-haiku-20240307")]
Claude3Haiku20240307,
#[serde(rename = "claude-3-5-sonnet-20240620")]
Claude35Sonnet20240620,
#[serde(rename = "claude-3-5-sonnet-20241022")]
Claude35Sonnet20241022,
#[serde(rename = "claude-3-5-haiku-20241022")]
Claude35Haiku20241022,
#[serde(rename = "claude-3-7-sonnet-20250219")]
Claude37Sonnet20250219,
#[serde(rename = "claude-opus-4-20250514")]
ClaudeOpus420250514,
#[serde(rename = "claude-sonnet-4-20250514")]
ClaudeSonnet420250514,
}
impl Default for ModelTrue {
fn default() -> ModelTrue {
Self::Claude3Opus20240229
}
}