diff --git a/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_failed_job.rs b/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_failed_job.rs index 49cb890d67..8e0815474a 100644 --- a/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_failed_job.rs +++ b/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_failed_job.rs @@ -27,7 +27,7 @@ pub async fn process_failed_job( let frontend_failure_category = if platform_rules_violation { Some(FrontendFailureCategory::ModelRulesViolation) } else { - None + Some(FrontendFailureCategory::GenerationFailed) }; warn!( diff --git a/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_successful_job.rs b/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_successful_job.rs index 6da125d426..c29d6099f1 100644 --- a/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_successful_job.rs +++ b/crates/service/job/grok_api_job/src/jobs/video_polling_job/process_job/process_successful_job.rs @@ -7,6 +7,7 @@ use enums::by_table::media_files::media_file_class::MediaFileClass; use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory; use enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory; use enums::by_table::media_files::media_file_type::MediaFileType; +use enums::common::generation_provider::GenerationProvider; use errors::AnyhowResult; use hashing::sha256::sha256_hash_bytes::sha256_hash_bytes; use mysql_queries::queries::generic_inference::api_providers::grok_api::list_pending_grok_api_jobs::PendingGrokApiJob; @@ -87,18 +88,19 @@ pub async fn process_successful_job( // --- Step 3: Create media file record. --- let media_file_result = MediaFileInsertBuilder::new() - .maybe_creator_user(job.maybe_creator_user_token.as_ref()) - .maybe_creator_anonymous_visitor(job.maybe_creator_anonymous_visitor_token.as_ref()) + .checksum_sha2(&checksum) .creator_ip_address(&job.creator_ip_address) .creator_set_visibility(job.creator_set_visibility) + .file_size_bytes(video_bytes.len() as u64) + .maybe_creator_anonymous_visitor(job.maybe_creator_anonymous_visitor_token.as_ref()) + .maybe_creator_user(job.maybe_creator_user_token.as_ref()) + .maybe_generation_provider(Some(GenerationProvider::Artcraft)) + .maybe_prompt_token(job.maybe_prompt_token.as_ref()) .media_file_class(MediaFileClass::Video) - .media_file_type(MediaFileType::Mp4) .media_file_origin_category(MediaFileOriginCategory::Inference) .media_file_origin_product_category(MediaFileOriginProductCategory::VideoGeneration) + .media_file_type(MediaFileType::Mp4) .mime_type("video/mp4") - .file_size_bytes(video_bytes.len() as u64) - .checksum_sha2(&checksum) - .maybe_prompt_token(job.maybe_prompt_token.as_ref()) .public_bucket_directory_hash(&bucket_path) .insert_pool(&deps.mysql_pool) .await;