From ffe2c7833d420c480c38fcaa0da2e05ee7f5deba Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Fri, 15 May 2026 09:08:14 -0400
Subject: [PATCH] cleanup

---
 .../pipeline_v1/distill_image_request.rs      |  1 +
 .../image/pipeline_v2/run_pipeline_v2.rs      | 39 +++----------------
 ...kup_media_files_as_cdn_url_list_and_map.rs |  4 ++
 3 files changed, 10 insertions(+), 34 deletions(-)

diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v1/distill_image_request.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v1/distill_image_request.rs
index ef690ba807..b0aec91434 100644
--- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v1/distill_image_request.rs
+++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/pipeline_v1/distill_image_request.rs
@@ -113,6 +113,7 @@ pub(crate) fn distill_image_request<M: DistillImageMediaInput>(
   if let Some(urls) = image_input_urls {
     initial.image_inputs = Some(ImageListRef::Urls(urls));
   }
+
   initial.provider = Provider::Fal;
   let fal_request = initial;
 
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 c30fb0312e..74ccb57707 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
@@ -46,7 +46,7 @@ pub async fn run_pipeline_v2(
   let hydrated_builder = apply_hydrated_media_inputs(
     router_builder,
     resolved_media,
-  )?;
+  );
 
   let draft_or_request = build_execution_request(&hydrated_builder)?;
   let cost = estimate_cost_in_credits(&hydrated_builder)?;
@@ -87,43 +87,14 @@ fn build_execution_request(
 fn apply_hydrated_media_inputs(
   router_builder: &GenerateImageRequestBuilder,
   resolved_media: &MediaFilesAsCdnUrlListAndMap,
-) -> Result<GenerateImageRequestBuilder, AdvancedCommonWebError> {
+) -> GenerateImageRequestBuilder {
   let mut hydrated_builder = router_builder.clone();
-  let image_input_urls = build_image_input_urls(
-    hydrated_builder.image_inputs.as_ref(),
-    resolved_media,
-  )?;
 
-  if let Some(urls) = image_input_urls {
-    hydrated_builder.image_inputs = Some(ImageListRef::Urls(urls));
+  if matches!(hydrated_builder.image_inputs.as_ref(), Some(ImageListRef::MediaFileTokens(tokens)) if !tokens.is_empty()) {
+    hydrated_builder.image_inputs = Some(ImageListRef::Urls(resolved_media.ordered_url_list.clone()));
   }
 
-  Ok(hydrated_builder)
-}
-
-fn build_image_input_urls(
-  image_inputs: Option<&ImageListRef>,
-  resolved_media: &MediaFilesAsCdnUrlListAndMap,
-) -> Result<Option<Vec<String>>, AdvancedCommonWebError> {
-  let tokens = match image_inputs {
-    Some(ImageListRef::MediaFileTokens(tokens)) if !tokens.is_empty() => tokens,
-    _ => return Ok(None),
-  };
-
-  let mut urls: Vec<String> = Vec::with_capacity(tokens.len());
-  for token in tokens {
-    match resolved_media.token_to_url_map.get(token) {
-      Some(url) => urls.push(url.clone()),
-      None => {
-        return Err(AdvancedCommonWebError::BadInputWithSimpleMessage(format!(
-          "Image media token not found in hydration map: {:?}",
-          token
-        )));
-      },
-    }
-  }
-
-  Ok(Some(urls))
+  hydrated_builder
 }
 
 fn estimate_cost_in_credits(
diff --git a/crates/service/web/storyteller_web/src/util/lookup/lookup_media_files_as_cdn_url_list_and_map.rs b/crates/service/web/storyteller_web/src/util/lookup/lookup_media_files_as_cdn_url_list_and_map.rs
index be83ec5459..af67066bf1 100644
--- a/crates/service/web/storyteller_web/src/util/lookup/lookup_media_files_as_cdn_url_list_and_map.rs
+++ b/crates/service/web/storyteller_web/src/util/lookup/lookup_media_files_as_cdn_url_list_and_map.rs
@@ -15,7 +15,11 @@ use crate::http_server::common_responses::media::media_links_builder::MediaLinks
 use crate::http_server::endpoints::media_files::helpers::get_media_domain::get_media_domain;
 
 pub struct MediaFilesAsCdnUrlListAndMap {
+  /// This is an in-order list of all the media files as their asset URLs.
+  /// This is in the order they were requested.
   pub ordered_url_list: Vec<String>,
+
+  /// This is an unordered map of media file token to asset URL.
   pub token_to_url_map: HashMap<MediaFileToken, String>,
 }