diff --git a/crates/_types/enums/src/by_table/generic_inference_jobs/frontend_failure_category.rs b/crates/_types/enums/src/by_table/generic_inference_jobs/frontend_failure_category.rs index f7d7b6652a..44d3494651 100644 --- a/crates/_types/enums/src/by_table/generic_inference_jobs/frontend_failure_category.rs +++ b/crates/_types/enums/src/by_table/generic_inference_jobs/frontend_failure_category.rs @@ -4,6 +4,7 @@ use std::collections::BTreeSet; use strum::EnumCount; #[cfg(test)] use strum::EnumIter; +use utoipa::ToSchema; /// Used in the `generic_inference_jobs` table in `VARCHAR(32)` field `frontend_failure_category`. /// @@ -20,7 +21,7 @@ use strum::EnumIter; /// /// YOU CAN ADD NEW VALUES, BUT DO NOT CHANGE EXISTING VALUES WITHOUT A MIGRATION STRATEGY. #[cfg_attr(test, derive(EnumIter, EnumCount))] -#[derive(Clone, Copy, Eq, PartialEq, Hash, Ord, PartialOrd, Deserialize, Serialize)] +#[derive(Clone, Copy, Eq, PartialEq, Hash, Ord, PartialOrd, Deserialize, Serialize, ToSchema)] pub enum FrontendFailureCategory { /// When a face is not detected in the image used for animation. /// For SadTalker (and possibly Wav2Lip) diff --git a/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs b/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs index 2c6e9dd3b1..e402126bed 100644 --- a/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs +++ b/crates/_types/enums/src/by_table/generic_inference_jobs/inference_category.rs @@ -4,6 +4,7 @@ use std::collections::BTreeSet; use strum::EnumCount; #[cfg(test)] use strum::EnumIter; +use utoipa::ToSchema; /// Used in the `generic_inference_jobs` table in `VARCHAR(32)` field `inference_category`. /// @@ -13,7 +14,7 @@ use strum::EnumIter; /// /// YOU CAN ADD NEW VALUES, BUT DO NOT CHANGE EXISTING VALUES WITHOUT A MIGRATION STRATEGY. #[cfg_attr(test, derive(EnumIter, EnumCount))] -#[derive(Clone, Copy, Eq, PartialEq, Hash, Ord, PartialOrd, Deserialize, Serialize, Default)] +#[derive(Clone, Copy, Eq, PartialEq, Hash, Ord, PartialOrd, Deserialize, Serialize, Default, ToSchema)] pub enum InferenceCategory { /// Eg. SadTalker and possibly Wav2Lip #[serde(rename = "lipsync_animation")] diff --git a/crates/service/web/storyteller_web/src/api_doc.rs b/crates/service/web/storyteller_web/src/api_doc.rs index 7e7786835a..7eac693d0c 100644 --- a/crates/service/web/storyteller_web/src/api_doc.rs +++ b/crates/service/web/storyteller_web/src/api_doc.rs @@ -1,5 +1,7 @@ use utoipa::OpenApi; +use enums::by_table::generic_inference_jobs::frontend_failure_category::FrontendFailureCategory; +use enums::by_table::generic_inference_jobs::inference_category::InferenceCategory; use enums::by_table::media_files::media_file_class::MediaFileClass; use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory; use enums::by_table::media_files::media_file_origin_model_type::MediaFileOriginModelType; @@ -148,6 +150,8 @@ use crate::http_server::web_utils::response_success_helpers::*; ZsVoiceDatasetToken, // Enums + FrontendFailureCategory, + InferenceCategory, JobStatusPlus, MediaFileClass, MediaFileOriginCategory,