diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs index f745c37835..b2a91d3e7d 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs @@ -59,7 +59,7 @@ pub async fn omni_gen_image_generate_handler( // ==================== TRANSFORM REQUEST + PLAN ==================== // - let generate_request = transform_request(&request)?; + let mut generate_request = transform_request(&request)?; // ==================== COST ==================== // @@ -67,7 +67,7 @@ pub async fn omni_gen_image_generate_handler( // ==================== PLAN ==================== // - let plan = request_to_plan(&generate_request)?; + let plan = request_to_plan(&mut generate_request)?; // ==================== SESSION ==================== // diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/request_to_plan.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/request_to_plan.rs index 91822d90ab..8a76e18271 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/request_to_plan.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/request_to_plan.rs @@ -1,3 +1,4 @@ +use artcraft_router::api::provider::Provider; use artcraft_router::generate::generate_image::generate_image_request::GenerateImageRequest; use artcraft_router::generate::generate_image::image_generation_plan::ImageGenerationPlan; use log::warn; @@ -5,9 +6,12 @@ use log::warn; use crate::http_server::common_responses::advanced_common_web_error::AdvancedCommonWebError; /// Build an image generation plan from a transformed request. +/// Overrides the provider to Fal for execution (costing uses Artcraft separately). pub fn request_to_plan<'a>( - request: &'a GenerateImageRequest<'a>, + request: &'a mut GenerateImageRequest<'a>, ) -> Result, AdvancedCommonWebError> { + request.provider = Provider::Fal; + request.build() .map_err(|e| { warn!("Failed to build image generation plan: {}", e); diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs index 713663a26e..ab1340f305 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs @@ -59,7 +59,7 @@ pub async fn omni_gen_video_generate_handler( // ==================== TRANSFORM REQUEST + PLAN ==================== // - let generate_request = transform_request(&request)?; + let mut generate_request = transform_request(&request)?; // ==================== COST ==================== // @@ -67,7 +67,7 @@ pub async fn omni_gen_video_generate_handler( // ==================== PLAN ==================== // - let plan = request_to_plan(&generate_request)?; + let plan = request_to_plan(&mut generate_request)?; // ==================== SESSION ==================== // diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/request_to_plan.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/request_to_plan.rs index 26bdac5aef..e7f992d454 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/request_to_plan.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/request_to_plan.rs @@ -1,3 +1,4 @@ +use artcraft_router::api::provider::Provider; use artcraft_router::generate::generate_video::generate_video_request::GenerateVideoRequest; use artcraft_router::generate::generate_video::video_generation_plan::VideoGenerationPlan; use log::warn; @@ -5,9 +6,12 @@ use log::warn; use crate::http_server::common_responses::advanced_common_web_error::AdvancedCommonWebError; /// Build a video generation plan from a transformed request. +/// Overrides the provider to Fal for execution (costing uses Artcraft separately). pub fn request_to_plan<'a>( - request: &'a GenerateVideoRequest<'a>, + request: &'a mut GenerateVideoRequest<'a>, ) -> Result, AdvancedCommonWebError> { + request.provider = Provider::Fal; + request.build() .map_err(|e| { warn!("Failed to build video generation plan: {}", e);