gproxy_protocol/claude/
count_tokens.rs1use serde::{Deserialize, Serialize};
2
3use super::common::{
4 AnthropicBetaHeaders, CacheControl, ClaudeModel, ContextManagementConfig, DiagnosticsParam,
5 JsonSchemaFormat, McpServer, MessageParam, OutputConfig, RequestServiceTier, Speed,
6 SystemPrompt, ThinkingConfig, Tool, ToolChoice,
7};
8
9pub type CountTokensRequestHeaders = AnthropicBetaHeaders;
10
11#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
12#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
13pub struct CountTokensRequestBody {
14 pub model: ClaudeModel,
15 pub messages: Vec<MessageParam>,
16 #[serde(skip_serializing_if = "Option::is_none")]
17 pub cache_control: Option<CacheControl>,
18 #[serde(skip_serializing_if = "Option::is_none")]
19 pub context_management: Option<ContextManagementConfig>,
20 #[serde(skip_serializing_if = "Option::is_none")]
21 pub diagnostics: Option<DiagnosticsParam>,
22 #[serde(skip_serializing_if = "Option::is_none")]
23 pub mcp_servers: Option<Vec<McpServer>>,
24 #[serde(skip_serializing_if = "Option::is_none")]
25 pub output_config: Option<OutputConfig>,
26 #[deprecated(note = "use output_config.format instead")]
28 #[serde(skip_serializing_if = "Option::is_none")]
29 pub output_format: Option<JsonSchemaFormat>,
30 #[serde(skip_serializing_if = "Option::is_none")]
31 pub service_tier: Option<RequestServiceTier>,
32 #[serde(skip_serializing_if = "Option::is_none")]
33 pub speed: Option<Speed>,
34 #[serde(skip_serializing_if = "Option::is_none")]
35 pub system: Option<SystemPrompt>,
36 #[serde(skip_serializing_if = "Option::is_none")]
37 pub thinking: Option<ThinkingConfig>,
38 #[serde(skip_serializing_if = "Option::is_none")]
39 pub tool_choice: Option<ToolChoice>,
40 #[serde(skip_serializing_if = "Option::is_none")]
41 pub tools: Option<Vec<Tool>>,
42 #[serde(default, flatten, skip_serializing_if = "serde_json::Map::is_empty")]
43 pub rest: serde_json::Map<String, serde_json::Value>,
44}
45
46#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
47#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
48pub struct CountTokensResponseBody {
49 pub input_tokens: u64,
50 #[serde(skip_serializing_if = "Option::is_none")]
51 pub context_management: Option<CountTokensContextManagement>,
52 #[serde(default, flatten, skip_serializing_if = "serde_json::Map::is_empty")]
53 pub rest: serde_json::Map<String, serde_json::Value>,
54}
55
56#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
57#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
58pub struct CountTokensContextManagement {
59 #[serde(skip_serializing_if = "Option::is_none")]
60 pub original_input_tokens: Option<u64>,
61 #[serde(default, flatten, skip_serializing_if = "serde_json::Map::is_empty")]
62 pub rest: serde_json::Map<String, serde_json::Value>,
63}