From 98547df063d2d5ea15a3870d472253a035347c33 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 24 Oct 2023 05:12:42 -0400 Subject: [PATCH] propagate avt from jobs to media_files --- .../job/list_available_generic_inference_jobs.rs | 5 +++++ .../media_files/insert_media_file_from_face_animation.rs | 2 ++ .../media_files/insert_media_file_from_voice_conversion.rs | 2 ++ 3 files changed, 9 insertions(+) diff --git a/crates/lib/mysql_queries/src/queries/generic_inference/job/list_available_generic_inference_jobs.rs b/crates/lib/mysql_queries/src/queries/generic_inference/job/list_available_generic_inference_jobs.rs index f895de8265..82f54884ce 100644 --- a/crates/lib/mysql_queries/src/queries/generic_inference/job/list_available_generic_inference_jobs.rs +++ b/crates/lib/mysql_queries/src/queries/generic_inference/job/list_available_generic_inference_jobs.rs @@ -38,6 +38,7 @@ pub struct AvailableInferenceJob { // User information to propagate downstream pub maybe_creator_user_token: Option, + pub maybe_creator_anonymous_visitor_token: Option, pub creator_ip_address: String, pub creator_set_visibility: Visibility, @@ -111,6 +112,7 @@ pub async fn list_available_generic_inference_jobs( uuid_idempotency_token: record.uuid_idempotency_token, creator_ip_address: record.creator_ip_address, maybe_creator_user_token: record.maybe_creator_user_token, + maybe_creator_anonymous_visitor_token: record.maybe_creator_anonymous_visitor_token, creator_set_visibility: Visibility::from_str(&record.creator_set_visibility) .map_err(|e| anyhow!("error: {:?}", e))?, // TODO/FIXME: This is a gross fix. inference_category: InferenceCategory::from_str(&record.inference_category) @@ -181,6 +183,7 @@ SELECT maybe_raw_inference_text, maybe_creator_user_token, + maybe_creator_anonymous_visitor_token, creator_ip_address, creator_set_visibility, @@ -274,6 +277,7 @@ SELECT maybe_raw_inference_text, maybe_creator_user_token, + maybe_creator_anonymous_visitor_token, creator_ip_address, creator_set_visibility, @@ -364,6 +368,7 @@ struct AvailableInferenceJobRawInternal { // User information to propagate downstream pub maybe_creator_user_token: Option, + pub maybe_creator_anonymous_visitor_token: Option, pub creator_ip_address: String, //pub creator_set_visibility: Visibility, pub creator_set_visibility: String, diff --git a/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs b/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs index e77adbf8b9..60e3b24371 100644 --- a/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs +++ b/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs @@ -94,6 +94,7 @@ SET maybe_public_bucket_extension = ?, maybe_creator_user_token = ?, + maybe_creator_anonymous_visitor_token = ?, creator_ip_address = ?, creator_set_visibility = ?, @@ -123,6 +124,7 @@ SET args.maybe_public_bucket_extension, args.job.maybe_creator_user_token, + args.job.maybe_creator_anonymous_visitor_token, args.job.creator_ip_address, args.job.creator_set_visibility.to_str(), diff --git a/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_voice_conversion.rs b/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_voice_conversion.rs index db79e2b734..968ab09732 100644 --- a/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_voice_conversion.rs +++ b/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_voice_conversion.rs @@ -108,6 +108,7 @@ SET maybe_public_bucket_extension = ?, maybe_creator_user_token = ?, + maybe_creator_anonymous_visitor_token = ?, creator_ip_address = ?, creator_set_visibility = ?, @@ -139,6 +140,7 @@ SET args.maybe_public_bucket_extension, args.job.maybe_creator_user_token, + args.job.maybe_creator_anonymous_visitor_token, args.job.creator_ip_address, args.job.creator_set_visibility.to_str(),