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")]
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 = "BTreeMap::is_empty", flatten)]
15    pub extra: ExtraFields,
16}
17
18#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
19#[serde(rename_all = "camelCase")]
20pub struct SafetyRating {
21    #[serde(skip_serializing_if = "Option::is_none")]
22    pub category: Option<HarmCategory>,
23    #[serde(skip_serializing_if = "Option::is_none")]
24    pub probability: Option<HarmProbability>,
25    #[serde(skip_serializing_if = "Option::is_none")]
26    pub blocked: Option<bool>,
27    #[serde(default, skip_serializing_if = "BTreeMap::is_empty", flatten)]
28    pub extra: ExtraFields,
29}