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)]
14pub struct CountTokensRequestBody {
15    pub model: ClaudeModel,
16    pub messages: Vec<MessageParam>,
17    #[serde(skip_serializing_if = "Option::is_none")]
18    pub cache_control: Option<CacheControl>,
19    #[serde(skip_serializing_if = "Option::is_none")]
20    pub context_management: Option<ContextManagementConfig>,
21    #[serde(skip_serializing_if = "Option::is_none")]
22    pub diagnostics: Option<DiagnosticsParam>,
23    #[serde(skip_serializing_if = "Option::is_none")]
24    pub mcp_servers: Option<Vec<McpServer>>,
25    #[serde(skip_serializing_if = "Option::is_none")]
26    pub output_config: Option<OutputConfig>,
27    /// Deprecated. Use `output_config.format` instead.
28    #[deprecated(note = "use output_config.format instead")]
29    #[serde(skip_serializing_if = "Option::is_none")]
30    pub output_format: Option<JsonSchemaFormat>,
31    #[serde(skip_serializing_if = "Option::is_none")]
32    pub service_tier: Option<RequestServiceTier>,
33    #[serde(skip_serializing_if = "Option::is_none")]
34    pub speed: Option<Speed>,
35    #[serde(skip_serializing_if = "Option::is_none")]
36    pub system: Option<SystemPrompt>,
37    #[serde(skip_serializing_if = "Option::is_none")]
38    pub thinking: Option<ThinkingConfig>,
39    #[serde(skip_serializing_if = "Option::is_none")]
40    pub tool_choice: Option<ToolChoice>,
41    #[serde(skip_serializing_if = "Option::is_none")]
42    pub tools: Option<Vec<Tool>>,
43    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
44    pub extra: JsonObject,
45}
46
47#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
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 = "BTreeMap::is_empty")]
53    pub extra: JsonObject,
54}
55
56#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
57pub struct CountTokensContextManagement {
58    #[serde(skip_serializing_if = "Option::is_none")]
59    pub original_input_tokens: Option<u64>,
60    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
61    pub extra: JsonObject,
62}
gproxy_protocol/protocol/claude/count_tokens.rs

gproxy_protocol/protocol/claude/
count_tokens.rs
