mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-04 01:18:34 +00:00
update to call fal
This commit is contained in:
1 parent
6d12c06f1d
commit
2f2a20e607
4 files changed
+14
-6
No files matched your search
+2
-2
@@ -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 ==================== //
|
||||
|
||||
|
||||
+5
-1
@@ -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);
|
||||
|
||||
+2
-2
@@ -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 ==================== //
|
||||
|
||||
|
||||
+5
-1
@@ -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);
|
||||
|
||||
Reference in new issue
Block a user