diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/stats/get_unified_queue_stats.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/stats/get_unified_queue_stats.rs index e38e037566..1af0ae3d2b 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/stats/get_unified_queue_stats.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/stats/get_unified_queue_stats.rs @@ -45,6 +45,7 @@ pub struct ByQueueStats { pub pending_face_animation_jobs: u64, pub pending_rvc_jobs: u64, pub pending_svc_jobs: u64, + pub pending_tacotron2_jobs: u64, } #[derive(Debug)] @@ -95,6 +96,7 @@ pub async fn get_unified_queue_stats_handler( pending_face_animation_jobs: 10_000, pending_rvc_jobs: 10_000, pending_svc_jobs: 10_000, + pending_tacotron2_jobs: 10_000, } }, legacy_tts: LegacyQueueDetails { @@ -194,6 +196,7 @@ pub async fn get_unified_queue_stats_handler( pending_face_animation_jobs: cacheable_stats_result.queues.sad_talker, pending_rvc_jobs: cacheable_stats_result.queues.rvc_v2, pending_svc_jobs: cacheable_stats_result.queues.so_vits_svc, + pending_tacotron2_jobs: cacheable_stats_result.queues.tacotron2, } }, legacy_tts: LegacyQueueDetails { diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/stats/result_transformer.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/stats/result_transformer.rs index 29cc6d30e9..cf68587ad3 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/stats/result_transformer.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/stats/result_transformer.rs @@ -21,11 +21,12 @@ impl Default for CacheableQueueStats { #[derive(Serialize, Deserialize, Clone, Default)] pub struct Queues { - pub legacy_tts: u64, // Tacotron2 - pub total_generic: u64, // Everything except Tacotron2 summed together + pub legacy_tts: u64, // Tacotron2 (as tts_inference_jobs) + pub total_generic: u64, // Everything except tts_inference_jobs' Tacotron2 summed together pub rvc_v2: u64, pub sad_talker: u64, pub so_vits_svc: u64, + pub tacotron2: u64, // Tacotron2 (as generic_inference_jobs; NB: included in `total_generic`) } pub fn database_result_to_cacheable(database_records: Vec) -> CacheableQueueStats { @@ -43,6 +44,7 @@ pub fn database_result_to_cacheable(database_records: Vec) -> Cac let rvc_v2 = unwrap(queue_lengths.get("rvc_v2")); let sad_talker = unwrap(queue_lengths.get("sad_talker")); let so_vits_svc = unwrap(queue_lengths.get("so_vits_svc")); + let tacotron2 = unwrap(queue_lengths.get("tacotron2")); let total_generic = rvc_v2 + so_vits_svc; @@ -54,6 +56,7 @@ pub fn database_result_to_cacheable(database_records: Vec) -> Cac rvc_v2, sad_talker, so_vits_svc, + tacotron2, }, } }