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 SystemMessageOutputConfig {
23    #[serde(skip_serializing_if = "Option::is_none")]
24    pub effort: Option<OutputEffort>,
25    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
26    pub extra: JsonObject,
27}
28
29#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
30#[non_exhaustive]
31pub struct TokenTaskBudget {
32    pub total: u64,
33    #[serde(rename = "type")]
34    pub type_: TaskBudgetType,
35    #[serde(skip_serializing_if = "Option::is_none")]
36    pub remaining: Option<u64>,
37    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
38    pub extra: JsonObject,
39}