Route Comfy workflows through video API

2026-06-03 11:39:14 +08:00
parent f8741b544e
commit 5ce5ea1687
3 changed files with 417 additions and 7 deletions
@@ -347,12 +347,7 @@ async fn video_task_handler(
        .unwrap_or("default")
        .to_string();
    let route = resolve_route(&state, &model).await?;
-    let suffix = if route.base_url.contains("ark.cn-beijing.volces.com") {
-        format!("/contents/generations/tasks/{id}")
-    } else {
-        format!("/videos/{id}")
-    };
-    let result = executors::execute_proxy_get(route, &suffix).await?;
+    let result = executors::execute_video_task_status(state.clone(), route, &id).await?;
    record_billing(&state, key_id, &model, 0, 0, 0.0).await;
    Ok(Json(result))
 }