use std::collections::BTreeMap;
use serde::{Deserialize, Serialize};
use crate::claude::count_tokens::types::{
AnthropicBeta, AnthropicVersion, BetaCacheControlEphemeral, BetaContextManagementConfig,
BetaMessageParam, BetaOutputConfig, BetaRequestMcpServerUrlDefinition, BetaSystemPrompt,
BetaThinkingConfigParam, BetaToolChoice, BetaToolUnion, HttpMethod, Model,
};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct ClaudeCountTokensRequest {
pub method: HttpMethod,
pub path: PathParameters,
pub query: QueryParameters,
pub headers: RequestHeaders,
pub body: RequestBody,
}
impl Default for ClaudeCountTokensRequest {
fn default() -> Self {
Self {
method: HttpMethod::Post,
path: PathParameters::default(),
query: QueryParameters::default(),
headers: RequestHeaders::default(),
body: RequestBody::default(),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct PathParameters {}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct QueryParameters {}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct RequestHeaders {
#[serde(rename = "anthropic-version")]
pub anthropic_version: AnthropicVersion,
#[serde(
rename = "anthropic-beta",
default,
skip_serializing_if = "Option::is_none"
)]
pub anthropic_beta: Option<Vec<AnthropicBeta>>,
#[serde(flatten, default, skip_serializing_if = "BTreeMap::is_empty")]
pub extra: BTreeMap<String, String>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct RequestBody {
pub messages: Vec<BetaMessageParam>,
pub model: Model,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub context_management: Option<BetaContextManagementConfig>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mcp_servers: Option<Vec<BetaRequestMcpServerUrlDefinition>>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub cache_control: Option<BetaCacheControlEphemeral>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub output_config: Option<BetaOutputConfig>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub speed: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub system: Option<BetaSystemPrompt>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub thinking: Option<BetaThinkingConfigParam>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub tool_choice: Option<BetaToolChoice>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub tools: Option<Vec<BetaToolUnion>>,
}
impl Default for RequestBody {
fn default() -> Self {
Self {
messages: Vec::new(),
model: Model::Custom(String::new()),
context_management: None,
mcp_servers: None,
cache_control: None,
output_config: None,
speed: None,
system: None,
thinking: None,
tool_choice: None,
tools: None,
}
}
}