Save debug worker data to tts result record

This commit is contained in:
Brandon Thomas committed 2022-06-25 00:50:54 -04:00
1 parent 43824972e5
commit ade3162e53
2 files changed
+7 -2

No files matched your search

@@ -24,6 +24,7 @@ pub async fn insert_tts_result<P: AsRef<Path>>(
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;
@@ -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))?;