Skip to main content

gproxy_protocol/protocol/gemini/common/
safety.rs

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