update to call fal

This commit is contained in:
Brandon Thomas committed 2026-04-06 00:49:46 -04:00
1 parent 6d12c06f1d
commit 2f2a20e607
4 files changed
+14 -6

No files matched your search

@@ -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 ==================== //
@@ -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<ImageGenerationPlan<'a>, AdvancedCommonWebError> {
request.provider = Provider::Fal;
request.build()
.map_err(|e| {
warn!("Failed to build image generation plan: {}", e);
@@ -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 ==================== //
@@ -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<VideoGenerationPlan<'a>, AdvancedCommonWebError> {
request.provider = Provider::Fal;
request.build()
.map_err(|e| {
warn!("Failed to build video generation plan: {}", e);