diff --git a/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/handle_artcraft_seedance_2p0.rs b/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/handle_artcraft_seedance_2p0.rs deleted file mode 100644 index fb948c8436..0000000000 --- a/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/handle_artcraft_seedance_2p0.rs +++ /dev/null @@ -1,81 +0,0 @@ -use crate::core::commands::enqueue::generate_error::GenerateError; -use crate::core::commands::generate::generate_video::request::TauriGenerateVideoRequest; -use crate::core::commands::enqueue::task_enqueue_success::TaskEnqueueSuccess; -use crate::core::events::generation_events::common::GenerationModel; -use crate::core::state::app_env_configs::app_env_configs::AppEnvConfigs; -use artcraft_client::credentials::storyteller_credential_set::StorytellerCredentialSet; -use artcraft_router::api::audio_list_ref::AudioListRef; -use artcraft_router::api::character_list_ref::CharacterListRef; -use artcraft_router::api::common_video_model::CommonVideoModel; -use artcraft_router::api::image_list_ref::ImageListRef; -use artcraft_router::api::image_ref::ImageRef; -use artcraft_router::api::provider::Provider; -use artcraft_router::api::video_list_ref::VideoListRef; -use artcraft_router::client::request_mismatch_mitigation_strategy::RequestMismatchMitigationStrategy; -use artcraft_router::client::router_artcraft_client::RouterArtcraftClient; -use artcraft_router::client::router_client::RouterClient; -use artcraft_router::generate::generate_video::generate_video_request_builder::GenerateVideoRequestBuilder; -use enums::common::generation_provider::GenerationProvider; -use enums::tauri::tasks::task_type::TaskType; -use log::{error, info}; - -pub(super) async fn handle_artcraft_seedance_2p0( - request: &TauriGenerateVideoRequest, - app_env_configs: &AppEnvConfigs, - creds: &StorytellerCredentialSet, -) -> Result { - let client = RouterClient::Artcraft(RouterArtcraftClient::new( - app_env_configs.storyteller_host.clone(), - creds.clone(), - )); - - let start_frame = request.image_media_token.clone().map(ImageRef::MediaFileToken); - let end_frame = request.end_frame_image_media_token.clone().map(ImageRef::MediaFileToken); - let reference_images = request.reference_image_media_tokens.clone().map(ImageListRef::MediaFileTokens); - - let router_request = GenerateVideoRequestBuilder { - model: CommonTauriVideoModel::Seedance2p0, - provider: Provider::Artcraft, - prompt: request.prompt.clone(), - negative_prompt: None, - start_frame, - end_frame, - reference_images, - reference_videos: request.reference_video_media_tokens.clone().map(VideoListRef::MediaFileTokens), - reference_audio: request.reference_audio_media_tokens.clone().map(AudioListRef::MediaFileTokens), - reference_character_tokens: request.reference_character_tokens.clone().map(CharacterListRef::CharacterTokens), - resolution: None, - aspect_ratio: request.aspect_ratio, - duration_seconds: request.duration_seconds, - video_batch_count: request.video_batch_count, - generate_audio: None, - request_mismatch_mitigation_strategy: RequestMismatchMitigationStrategy::PayMoreUpgrade, - idempotency_token: None, - }; - - let plan = router_request.build()?; - - info!("Video Generation Plan: {:?}", plan); - - let response = match plan.generate_video(&client).await { - Ok(resp) => { - info!("Successfully enqueued."); - resp - } - Err(err) => { - error!("Failed to enqueue: {:?}", err); - return Err(GenerateError::from(err)); - } - }; - - let job_id = response.get_artcraft_payload() - .map(|p| p.inference_job_token.to_string()) - .ok_or(GenerateError::ResponseHadNoJobTokens)?; - - Ok(TaskEnqueueSuccess { - task_type: TaskType::VideoGeneration, - model: Some(GenerationModel::Seedance2p0), - provider: GenerationProvider::Artcraft, - provider_job_id: Some(job_id), - }) -} diff --git a/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/mod.rs b/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/mod.rs index 8e87cf4f2e..af73548dee 100644 --- a/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/mod.rs +++ b/crates/desktop/artcraft/src/core/commands/generate/generate_video/artcraft/mod.rs @@ -6,7 +6,6 @@ mod handle_artcraft_kling_2p5_turbo_pro; mod handle_artcraft_kling_2p6_pro; mod handle_artcraft_video_via_router; mod handle_artcraft_seedance_1_lite; -mod handle_artcraft_seedance_2p0; mod handle_artcraft_sora2; mod handle_artcraft_sora2_pro; mod handle_artcraft_veo2;