diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
index b2a91d3e7d..7944cc0dcc 100644
--- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
+++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/image/omni_gen_image_generate_handler.rs
@@ -10,6 +10,7 @@ use artcraft_api_defs::omni_gen::generate_response::omni_gen_image_generate_resp
 use enums::by_table::prompt_context_items::prompt_context_semantic_type::PromptContextSemanticType;
 use enums::by_table::prompts::prompt_type::PromptType;
 use enums::common::generation::common_generation_mode::CommonGenerationMode;
+use enums::common::generation::common_model_type::CommonModelType;
 use enums::common::generation_provider::GenerationProvider;
 use enums::common::visibility::Visibility;
 use http_server_common::request::get_request_ip::get_request_ip;
@@ -61,6 +62,11 @@ pub async fn omni_gen_image_generate_handler(
 
   let mut generate_request = transform_request(&request)?;
 
+  let maybe_prompt_model_type: Option<CommonModelType> =
+    serde_json::to_string(&generate_request.model)
+      .ok()
+      .and_then(|json| serde_json::from_str(&json).ok());
+
   // ==================== COST ==================== //
 
   let cost_estimate = request_to_costs(&generate_request)?;
@@ -174,7 +180,7 @@ pub async fn omni_gen_image_generate_handler(
     maybe_apriori_prompt_token: None,
     prompt_type: PromptType::ArtcraftApp,
     maybe_creator_user_token: Some(user_token),
-    maybe_model_type: None,
+    maybe_model_type: maybe_prompt_model_type,
     maybe_generation_provider: Some(GenerationProvider::Artcraft),
     maybe_positive_prompt: request.prompt.as_deref(),
     maybe_negative_prompt: None,
diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs
index ab1340f305..6bf8f6abee 100644
--- a/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs
+++ b/crates/service/web/storyteller_web/src/http_server/endpoints/omni_gen/generate/video/omni_gen_video_generate_handler.rs
@@ -10,6 +10,7 @@ use artcraft_api_defs::omni_gen::generate_response::omni_gen_video_generate_resp
 use enums::by_table::prompt_context_items::prompt_context_semantic_type::PromptContextSemanticType;
 use enums::by_table::prompts::prompt_type::PromptType;
 use enums::common::generation::common_generation_mode::CommonGenerationMode;
+use enums::common::generation::common_model_type::CommonModelType;
 use enums::common::generation_provider::GenerationProvider;
 use enums::common::visibility::Visibility;
 use http_server_common::request::get_request_ip::get_request_ip;
@@ -61,6 +62,11 @@ pub async fn omni_gen_video_generate_handler(
 
   let mut generate_request = transform_request(&request)?;
 
+  let maybe_prompt_model_type: Option<CommonModelType> =
+    serde_json::to_string(&generate_request.model)
+      .ok()
+      .and_then(|json| serde_json::from_str(&json).ok());
+
   // ==================== COST ==================== //
 
   let cost_estimate = request_to_costs(&generate_request)?;
@@ -176,7 +182,7 @@ pub async fn omni_gen_video_generate_handler(
     maybe_apriori_prompt_token: None,
     prompt_type: PromptType::ArtcraftApp,
     maybe_creator_user_token: Some(user_token),
-    maybe_model_type: None,
+    maybe_model_type: maybe_prompt_model_type,
     maybe_generation_provider: Some(GenerationProvider::Artcraft),
     maybe_positive_prompt: request.prompt.as_deref(),
     maybe_negative_prompt: request.negative_prompt.as_deref(),