Skip to main content

gproxy_protocol/protocol/claude/common/
output.rs

1use std::collections::BTreeMap;
2
3use serde::{Deserialize, Serialize};
4
5use super::{JsonObject, JsonSchemaFormat, OutputEffort, TaskBudgetType};
6
7#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
8pub struct OutputConfig {
9    #[serde(skip_serializing_if = "Option::is_none")]
10    pub effort: Option<OutputEffort>,
11    #[serde(skip_serializing_if = "Option::is_none")]
12    pub format: Option<JsonSchemaFormat>,
13    #[serde(skip_serializing_if = "Option::is_none")]
14    pub task_budget: Option<TokenTaskBudget>,
15    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
16    pub extra: JsonObject,
17}
18
19#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
20pub struct TokenTaskBudget {
21    pub total: u64,
22    #[serde(rename = "type")]
23    pub type_: TaskBudgetType,
24    #[serde(skip_serializing_if = "Option::is_none")]
25    pub remaining: Option<u64>,
26    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
27    pub extra: JsonObject,
28}