diff --git a/crates/lib/database_queries/src/queries/tts/tts_results/insert_tts_result.rs b/crates/lib/database_queries/src/queries/tts/tts_results/insert_tts_result.rs index 6bc42f95de..8839e9bd6d 100644 --- a/crates/lib/database_queries/src/queries/tts/tts_results/insert_tts_result.rs +++ b/crates/lib/database_queries/src/queries/tts/tts_results/insert_tts_result.rs @@ -24,6 +24,7 @@ pub async fn insert_tts_result>( duration_millis: u64, is_on_prem: bool, worker_hostname: &str, + is_debug_worker: bool, ) -> AnyhowResult<(u64, String)> { let inference_result_token = Tokens::new_tts_result()?; @@ -125,7 +126,9 @@ SET duration_millis = ?, is_generated_on_prem = ?, - generated_by_worker = ? + generated_by_worker = ?, + + is_debug_request = ? "#, inference_result_token, job.model_token.clone(), @@ -147,6 +150,7 @@ SET duration_millis, is_on_prem, worker_hostname, + is_debug_worker, ) .execute(&mut transaction) .await; diff --git a/crates/service/tts_inference_job/src/job_steps/process_single_job.rs b/crates/service/tts_inference_job/src/job_steps/process_single_job.rs index 0c19a29b52..de9954f3f1 100644 --- a/crates/service/tts_inference_job/src/job_steps/process_single_job.rs +++ b/crates/service/tts_inference_job/src/job_steps/process_single_job.rs @@ -409,7 +409,8 @@ pub async fn process_single_job( file_metadata.file_size_bytes, file_metadata.duration_millis.unwrap_or(0), inferencer.worker_details.is_on_prem, - &inferencer.worker_details.worker_hostname) + &inferencer.worker_details.worker_hostname, + inferencer.worker_details.is_debug_worker) .await .map_err(|e| ProcessSingleJobError::Other(e))?;