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, gproxy_protocol_macros::WireBuilder)]
8#[non_exhaustive]
9pub struct OutputConfig {
10    #[serde(skip_serializing_if = "Option::is_none")]
11    pub effort: Option<OutputEffort>,
12    #[serde(skip_serializing_if = "Option::is_none")]
13    pub format: Option<JsonSchemaFormat>,
14    #[serde(skip_serializing_if = "Option::is_none")]
15    pub task_budget: Option<TokenTaskBudget>,
16    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
17    pub extra: JsonObject,
18}
19
20#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
21#[non_exhaustive]
22pub struct TokenTaskBudget {
23    pub total: u64,
24    #[serde(rename = "type")]
25    pub type_: TaskBudgetType,
26    #[serde(skip_serializing_if = "Option::is_none")]
27    pub remaining: Option<u64>,
28    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
29    pub extra: JsonObject,
30}