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