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

gproxy_protocol/protocol/claude/common/
stop.rs
