1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
use chrono::{DateTime, Utc}; use serde::{Deserialize, Serialize}; use std::str::FromStr; use crate::error::{Error, Result}; use super::{ comment::{Comment, CommentFilter, Entity, LabelName, PredictedLabel}, dataset::{FullName as DatasetFullName, Id as DatasetId}, }; #[derive(Debug, Clone, Deserialize, Serialize)] pub struct Name(pub String); #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, Hash)] pub struct Id(pub String); #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, Hash)] pub struct SequenceId(pub String); #[derive(Debug, Clone, Deserialize, Serialize)] pub struct FullName { pub dataset: DatasetFullName, pub trigger: Name, } impl FromStr for FullName { type Err = Error; fn from_str(string: &str) -> Result<Self> { let mut splits = string.split('/'); match (splits.next(), splits.next(), splits.next(), splits.next()) { (Some(owner), Some(dataset), Some(trigger_name), None) => Ok(FullName { dataset: DatasetFullName(format!("{}/{}", owner, dataset)), trigger: Name(trigger_name.to_owned()), }), _ => Err(Error::BadTriggerName { identifier: string.into(), }), } } } #[derive(Debug, Clone, Deserialize, Serialize)] pub struct Trigger { pub id: Id, pub dataset_id: DatasetId, pub name: Name, pub title: String, pub description: String, pub created_at: DateTime<Utc>, pub updated_at: DateTime<Utc>, #[serde(default)] pub comment_filter: CommentFilter, #[serde(rename = "label_threshold_filter")] pub label_filter: Option<LabelFilter>, } #[derive(Debug, Clone, Deserialize, Serialize)] pub struct LabelFilter { pub label: LabelName, pub model_version: UserModelVersion, pub threshold: f64, } #[derive(Debug, Clone, Deserialize, Serialize)] pub struct UserModelVersion(pub u64); #[derive(Debug, Clone, Deserialize, Serialize)] pub struct Batch { pub results: Vec<TriggerResult>, pub filtered: u32, pub sequence_id: SequenceId, } #[derive(Debug, Clone, Serialize, Deserialize)] pub struct TriggerResult { pub comment: Comment, pub sequence_id: SequenceId, pub labels: Option<Vec<PredictedLabel>>, pub entities: Option<Vec<Entity>>, } #[derive(Debug, Clone, Deserialize)] pub(crate) struct GetResponse { pub triggers: Vec<Trigger>, } #[derive(Debug, Clone, Serialize)] pub(crate) struct FetchRequest { pub size: u32, } #[derive(Debug, Clone, Serialize)] pub(crate) struct AdvanceRequest { pub sequence_id: SequenceId, } #[derive(Debug, Clone, Serialize)] pub(crate) struct ResetRequest { pub to_comment_created_at: DateTime<Utc>, }