From b6f3f55579f06eedac16ffdf40f3187af1e29803 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Fri, 17 Apr 2026 00:10:23 -0400 Subject: [PATCH] cleanup panicky code --- ...lan_generate_image_artcraft_flux_2_lora_angles.rs | 12 ++++++------ ..._generate_image_artcraft_qwen_edit_2511_angles.rs | 12 ++++++------ 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs index 599645574f..5328bedb60 100644 --- a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs +++ b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_flux_2_lora_angles.rs @@ -54,13 +54,13 @@ fn resolve_single_image_input( })) } Some(ImageListRef::MediaFileTokens(tokens)) => { - if tokens.len() != 1 { - return Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption { - field: "image_inputs", - value: format!("Angle models require exactly one input image, got {}", tokens.len()), - })); + if let Some(token) = tokens.first() { + return Ok(token.clone()); } - Ok(tokens.into_iter().next().unwrap()) + Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption { + field: "image_inputs", + value: format!("Angle models require exactly one input image, got {}", tokens.len()), + })) } Some(ImageListRef::Urls(_)) => { Err(ArtcraftRouterError::Client(ClientError::ArtcraftOnlySupportsMediaTokens)) diff --git a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs index d439652f3c..b20cc80e62 100644 --- a/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs +++ b/crates/api_clients/artcraft/artcraft_router/src/generate/generate_image/plan/artcraft/plan_generate_image_artcraft_qwen_edit_2511_angles.rs @@ -56,13 +56,13 @@ fn resolve_single_image_input( })) } Some(ImageListRef::MediaFileTokens(tokens)) => { - if tokens.len() != 1 { - return Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption { - field: "image_inputs", - value: format!("Angle models require exactly one input image, got {}", tokens.len()), - })); + if let Some(token) = tokens.first() { + return Ok(token.clone()); } - Ok(tokens.into_iter().next().unwrap()) + Err(ArtcraftRouterError::Client(ClientError::ModelDoesNotSupportOption { + field: "image_inputs", + value: format!("Angle models require exactly one input image, got {}", tokens.len()), + })) } Some(ImageListRef::Urls(_)) => { Err(ArtcraftRouterError::Client(ClientError::ArtcraftOnlySupportsMediaTokens))