diff --git a/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs b/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs index 80d4fed182..e88d08ca43 100644 --- a/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs +++ b/crates/lib/mysql_queries/src/payloads/prompt_args/prompt_inner_payload.rs @@ -1,4 +1,5 @@ use chrono::Duration; + use enums::no_table::style_transfer::style_transfer_name::StyleTransferName; use errors::AnyhowResult; use tokens::tokens::media_files::MediaFileToken; @@ -72,6 +73,12 @@ pub struct PromptInnerPayload { #[serde(alias = "global_ipa_token")] #[serde(skip_serializing_if = "Option::is_none")] pub global_ipa_token: Option, + + /// Prompt travel prompt + #[serde(rename = "tp")] // NB: DO NOT CHANGE: IT WILL BREAK MYSQL RECORDS. Renamed to consume fewer bytes. + #[serde(alias = "travel_prompt")] + #[serde(skip_serializing_if = "Option::is_none")] + pub travel_prompt: Option, } pub struct PromptInnerPayloadBuilder { @@ -87,6 +94,7 @@ pub struct PromptInnerPayloadBuilder { pub face_detailer_workflow: Option, pub upscaler_workflow: Option, pub global_ipa_token: Option, + pub travel_prompt: Option, } impl PromptInnerPayloadBuilder { @@ -104,6 +112,7 @@ impl PromptInnerPayloadBuilder { face_detailer_workflow: None, upscaler_workflow: None, global_ipa_token: None, + travel_prompt: None, } } @@ -120,6 +129,7 @@ impl PromptInnerPayloadBuilder { && self.face_detailer_workflow.is_none() && self.upscaler_workflow.is_none() && self.global_ipa_token.is_none() + && self.travel_prompt.is_none() { return None; } @@ -140,6 +150,7 @@ impl PromptInnerPayloadBuilder { face_detailer_workflow: self.face_detailer_workflow, upscaler_workflow: self.upscaler_workflow, global_ipa_token: self.global_ipa_token, + travel_prompt: self.travel_prompt, }) } @@ -210,6 +221,10 @@ impl PromptInnerPayloadBuilder { pub fn set_global_ipa_token(&mut self, token: Option) { self.global_ipa_token = token; } + + pub fn set_travel_prompt(&mut self, travel_prompt: Option) { + self.travel_prompt = travel_prompt; + } } impl PromptInnerPayload{ diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs index 8a7a3f71df..7f3b0dc532 100644 --- a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs +++ b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/steps/validate_and_save_results.rs @@ -238,6 +238,7 @@ pub async fn validate_and_save_results(args: SaveResultsArgs<'_>) -> Result) -> Result