Skip to main content

gproxy_protocol/protocol/claude/common/
fallback.rs

1use serde::{Deserialize, Serialize};
2
3use super::{ClaudeModel, JsonObject, OutputConfig, Speed, ThinkingConfig};
4
5/// Request-level fallback routing: either Anthropic's category-aware defaults
6/// or an ordered chain of up to three caller-selected models.
7#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
8#[serde(untagged)]
9#[non_exhaustive]
10pub enum FallbacksParam {
11    Default(FallbacksDefault),
12    Models(Vec<FallbackParam>),
13}
14
15#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
16#[non_exhaustive]
17pub enum FallbacksDefault {
18    #[serde(rename = "default")]
19    Default,
20}
21
22#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
23#[serde(untagged)]
24#[non_exhaustive]
25pub enum FallbackCreditTokenParam {
26    Token(String),
27    Config(FallbackCreditTokenConfig),
28}
29
30#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
31#[non_exhaustive]
32pub struct FallbackCreditTokenConfig {
33    pub token: String,
34    #[serde(skip_serializing_if = "Option::is_none")]
35    pub mode: Option<FallbackCreditMode>,
36    #[serde(default, flatten, skip_serializing_if = "JsonObject::is_empty")]
37    pub extra: JsonObject,
38}
39
40#[derive(Debug, Clone, PartialEq, Eq, Serialize)]
41#[serde(untagged)]
42#[non_exhaustive]
43pub enum FallbackCreditMode {
44    Known(FallbackCreditModeKnown),
45    Unknown(String),
46}
47
48// Manual Deserialize: unknown values fall back without formatting an
49// unknown-variant error (see `protocol::extensible`).
50impl<'de> serde::Deserialize<'de> for FallbackCreditMode {
51    fn deserialize<D: serde::Deserializer<'de>>(d: D) -> Result<Self, D::Error> {
52        crate::protocol::extensible::deserialize_extensible(d, Self::Known, Self::Unknown)
53    }
54}
55
56#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
57#[non_exhaustive]
58pub enum FallbackCreditModeKnown {
59    #[serde(rename = "strict")]
60    Strict,
61    #[serde(rename = "best_effort")]
62    BestEffort,
63}
64
65/// One ordered entry of the request-level `fallbacks` chain.
66#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, gproxy_protocol_macros::WireBuilder)]
67#[non_exhaustive]
68pub struct FallbackParam {
69    pub model: ClaudeModel,
70    #[serde(skip_serializing_if = "Option::is_none")]
71    pub max_tokens: Option<u64>,
72    #[serde(skip_serializing_if = "Option::is_none")]
73    pub output_config: Option<OutputConfig>,
74    #[serde(skip_serializing_if = "Option::is_none")]
75    pub speed: Option<Speed>,
76    #[serde(skip_serializing_if = "Option::is_none")]
77    pub thinking: Option<ThinkingConfig>,
78    #[serde(default, flatten, skip_serializing_if = "JsonObject::is_empty")]
79    pub extra: JsonObject,
80}