gproxy_protocol/protocol/claude/
count_tokens.rs

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

gproxy_protocol/protocol/claude/
count_tokens.rs
