mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-03 00:47:53 +00:00
Save debug worker data to tts result record
This commit is contained in:
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))?;
|
||||
|
||||
|
||||
Reference in new issue
Block a user