Skip to main content

gproxy_protocol/claude/
count_tokens.rs

1use 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. Use `output_config.format` instead.
27    #[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}