fix pricing data handler

crates/schema/public/enums/src/common/generation/common_image_model.rs

@@ -5,6 +5,22 @@ use utoipa::ToSchema;
 #[derive(Copy, Clone, Debug, Serialize, Deserialize, ToSchema)]
 #[serde(rename_all = "snake_case")]
 pub enum CommonImageModel {
+  #[serde(rename = "flux_1_dev")]
+  Flux1Dev,
+  #[serde(rename = "flux_1_schnell")]
+  Flux1Schnell,
+  #[serde(rename = "flux_pro_1p1")]
+  FluxPro11,
+  #[serde(rename = "flux_pro_1p1_ultra")]
+  FluxPro11Ultra,
+  #[serde(rename = "gpt_image_1p5")]
+  GptImage1p5,
+  #[serde(rename = "nano_banana")]
+  NanaBanana,
   #[serde(rename = "nano_banana_pro")]
   NanaBananaPro,
+  #[serde(rename = "seedream_4")]
+  Seedream4,
+  #[serde(rename = "seedream_4p5")]
+  Seedream4p5,
 }

crates/service/web/storyteller_web/src/http_server/endpoints/generate/cost_estimate/image/estimate_image_cost_handler.rs

@@ -121,7 +121,15 @@ fn map_provider(
 
 fn map_image_model(model: CommonImageModel) -> Result<RouterImageModel, HandlerError> {
   let router_model = match model {
+    CommonImageModel::Flux1Dev => RouterImageModel::Flux1Dev,
+    CommonImageModel::Flux1Schnell => RouterImageModel::Flux1Schnell,
+    CommonImageModel::FluxPro11 => RouterImageModel::FluxPro11,
+    CommonImageModel::FluxPro11Ultra => RouterImageModel::FluxPro11Ultra,
+    CommonImageModel::GptImage1p5 => RouterImageModel::GptImage1p5,
+    CommonImageModel::NanaBanana => RouterImageModel::NanaBanana,
     CommonImageModel::NanaBananaPro => RouterImageModel::NanaBananaPro,
+    CommonImageModel::Seedream4 => RouterImageModel::Seedream4,
+    CommonImageModel::Seedream4p5 => RouterImageModel::Seedream4p5,
   };
   Ok(router_model)
 }