gproxy_protocol/protocol/claude/common/
stop.rs

1use serde::{Deserialize, Serialize};
2
3use super::{ClaudeModel, JsonObject, TypedObject};
4
5#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
6#[serde(untagged)]
7pub enum StopDetails {
8    Refusal(RefusalStopDetails),
9    Unknown(TypedObject),
10}
11
12#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
13pub struct RefusalStopDetails {
14    pub category: Option<RefusalCategory>,
15    pub explanation: Option<String>,
16    #[serde(rename = "type")]
17    pub type_: RefusalStopDetailsType,
18    #[serde(skip_serializing_if = "Option::is_none")]
19    pub fallback_credit_token: Option<String>,
20    #[serde(skip_serializing_if = "Option::is_none")]
21    pub fallback_has_prefill_claim: Option<bool>,
22    #[serde(skip_serializing_if = "Option::is_none")]
23    pub recommended_model: Option<ClaudeModel>,
24    #[serde(default, flatten, skip_serializing_if = "JsonObject::is_empty")]
25    pub extra: JsonObject,
26}
27
28#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
29pub enum RefusalStopDetailsType {
30    #[serde(rename = "refusal")]
31    Refusal,
32}
33
34#[derive(Debug, Clone, PartialEq, Eq, Serialize)]
35#[serde(untagged)]
36pub enum RefusalCategory {
37    Known(RefusalCategoryKnown),
38    Unknown(String),
39}
40
41// Manual Deserialize: unknown values fall back without formatting an
42// unknown-variant error (see `protocol::extensible`).
43impl<'de> serde::Deserialize<'de> for RefusalCategory {
44    fn deserialize<D: serde::Deserializer<'de>>(d: D) -> Result<Self, D::Error> {
45        crate::protocol::extensible::deserialize_extensible(d, Self::Known, Self::Unknown)
46    }
47}
48
49#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
50pub enum RefusalCategoryKnown {
51    #[serde(rename = "cyber")]
52    Cyber,
53    #[serde(rename = "bio")]
54    Bio,
55    #[serde(rename = "reasoning_extraction")]
56    ReasoningExtraction,
57    #[serde(rename = "frontier_llm")]
58    FrontierLlm,
59    #[serde(rename = "general_harms")]
60    GeneralHarms,
61}
gproxy_protocol/protocol/claude/common/stop.rs

gproxy_protocol/protocol/claude/common/
stop.rs
