From c96fc2fef54f06320c0c1519b2f68f8b5d0fa503 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Fri, 15 May 2026 09:20:05 -0400 Subject: [PATCH] cleanup pipeline v2 --- .../image/pipeline_v2/run_pipeline_v2.rs | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v2/run_pipeline_v2.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v2/run_pipeline_v2.rs index 4becb13063..bd724dd1c5 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v2/run_pipeline_v2.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v2/run_pipeline_v2.rs @@ -109,8 +109,23 @@ fn estimate_cost_in_credits( let mut cost_builder = router_builder.clone(); cost_builder.provider = Provider::Artcraft; + // Attempt pipeline_v2 cost + match cost_builder.clone().build2() { + Err(err) => { + warn!("Failed to build image cost plan for v2 pipeline: {}", err); + }, + Ok(request) => { + let cost_plan = request.estimate_cost().map_err(|e| { + warn!("Failed to build image cost plan for v2 pipeline: {}", e); + AdvancedCommonWebError::from_error(e) + })?; + return Ok(cost_plan.cost_in_credits.unwrap_or(0)); + }, + } + + // Fall back to pipeline_v1 cost let cost_plan = cost_builder.build().map_err(|e| { - warn!("Failed to build image cost plan for v2 pipeline: {}", e); + warn!("Failed to build image cost plan for v1 pipeline: {}", e); AdvancedCommonWebError::from_error(e) })?;