Skip to main content

gproxy_protocol/gemini/common/
safety.rs

1use serde::{Deserialize, Serialize};
2
3use super::{HarmBlockThreshold, HarmCategory, HarmProbability};
4
5#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
6#[serde(rename_all = "camelCase")]
7#[derive(gproxy_protocol_macros::WireBuilder)]
8#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
9pub struct SafetySetting {
10    #[serde(skip_serializing_if = "Option::is_none")]
11    pub category: Option<HarmCategory>,
12    #[serde(skip_serializing_if = "Option::is_none")]
13    pub threshold: Option<HarmBlockThreshold>,
14    #[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
15    pub rest: serde_json::Map<String, serde_json::Value>,
16}
17
18#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
19#[serde(rename_all = "camelCase")]
20#[derive(gproxy_protocol_macros::WireBuilder)]
21#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
22pub struct SafetyRating {
23    #[serde(skip_serializing_if = "Option::is_none")]
24    pub category: Option<HarmCategory>,
25    #[serde(skip_serializing_if = "Option::is_none")]
26    pub probability: Option<HarmProbability>,
27    #[serde(skip_serializing_if = "Option::is_none")]
28    pub blocked: Option<bool>,
29    #[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
30    pub rest: serde_json::Map<String, serde_json::Value>,
31}