daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-04 01:18:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Brandon Thomas committed 2026-04-23 14:51:29 -04:00
1 parent 31385ea2c7
commit 4d98c7e0d6
1 file changed
+8 -27
Download Patch File Download Diff File Expand all files Collapse all files
crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/pipeline_v1/run_pipeline_v1.rs
+8 -27
View File
@@ -72,38 +72,19 @@ pub async fn run_pipeline_v1(args: RunPipelineV1Args<'_>) -> Result<PipelineResu
cost_plan.estimate_costs()
};
// ── Resolve media tokens → URLs from handler map ──
// ── Build the execution request (clone builder, override provider + media URLs) ──
let empty_map = HashMap::new();
let url_map = media_url_map.as_ref().unwrap_or(&empty_map);
let start_frame = lookup_url(url_map, request.start_frame_image_media_token.as_ref()).map(ImageRef::Url);
let end_frame = lookup_url(url_map, request.end_frame_image_media_token.as_ref()).map(ImageRef::Url);
let reference_images = lookup_urls(url_map, request.reference_image_media_tokens.as_ref()).map(ImageListRef::Urls);
let reference_videos = lookup_urls(url_map, request.reference_video_media_tokens.as_ref()).map(VideoListRef::Urls);
let reference_audio = lookup_urls(url_map, request.reference_audio_media_tokens.as_ref()).map(AudioListRef::Urls);
let mut exec_request = router_builder.clone();
exec_request.provider = execution_provider;
// ── Build the execution request ──
let exec_request = GenerateVideoRequestBuilder {
model: router_builder.model,
provider: execution_provider,
prompt: router_builder.prompt.clone(),
negative_prompt: router_builder.negative_prompt.clone(),
start_frame,
end_frame,
reference_images,
reference_videos,
reference_audio,
reference_character_tokens: router_builder.reference_character_tokens.clone(),
resolution: router_builder.resolution,
aspect_ratio: router_builder.aspect_ratio,
duration_seconds: router_builder.duration_seconds,
video_batch_count: router_builder.video_batch_count,
generate_audio: router_builder.generate_audio,
request_mismatch_mitigation_strategy: router_builder.request_mismatch_mitigation_strategy,
idempotency_token: router_builder.idempotency_token.clone(),
};
exec_request.start_frame = lookup_url(url_map, request.start_frame_image_media_token.as_ref()).map(ImageRef::Url);
exec_request.end_frame = lookup_url(url_map, request.end_frame_image_media_token.as_ref()).map(ImageRef::Url);
exec_request.reference_images = lookup_urls(url_map, request.reference_image_media_tokens.as_ref()).map(ImageListRef::Urls);
exec_request.reference_videos = lookup_urls(url_map, request.reference_video_media_tokens.as_ref()).map(VideoListRef::Urls);
exec_request.reference_audio = lookup_urls(url_map, request.reference_audio_media_tokens.as_ref()).map(AudioListRef::Urls);
let plan = exec_request.build().map_err(|e| {
warn!("Failed to build video generation plan during distillation: {}", e);