diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/media_files/list_featured_media_files.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/media_files/list_featured_media_files.rs index 2ec77a257e..55dbc6dd95 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/media_files/list_featured_media_files.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/media_files/list_featured_media_files.rs @@ -9,6 +9,9 @@ use r2d2_redis::redis::Commands; use utoipa::ToSchema; use buckets::public::media_files::bucket_file_path::MediaFileBucketPath; +use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory; +use enums::by_table::media_files::media_file_origin_model_type::MediaFileOriginModelType; +use enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory; use enums::by_table::media_files::media_file_type::MediaFileType; use mysql_queries::queries::media_files::list::list_media_files_by_tokens::list_media_files_by_tokens; use tokens::tokens::media_files::MediaFileToken; @@ -33,11 +36,10 @@ pub struct MediaFile { /// URL to the media file pub public_bucket_path: String, - // TODO: Provenance data - hide some of the models we use, especially zero shot. - //pub origin_category: MediaFileOriginCategory, - //pub origin_product_category: MediaFileOriginProductCategory, - //pub maybe_origin_model_type: Option, - //pub maybe_origin_model_token: Option, + pub origin_category: MediaFileOriginCategory, + pub origin_product_category: MediaFileOriginProductCategory, + pub maybe_origin_model_type: Option, + pub maybe_origin_model_token: Option, pub maybe_creator: Option, @@ -136,10 +138,10 @@ pub async fn list_featured_media_files_handler( token: m.token, media_type: m.media_type, public_bucket_path, - //origin_category: m.origin_category, - //origin_product_category: m.origin_product_category, - //maybe_origin_model_type: m.maybe_origin_model_type, - //maybe_origin_model_token: m.maybe_origin_model_token, + origin_category: m.origin_category, + origin_product_category: m.origin_product_category, + maybe_origin_model_type: m.maybe_origin_model_type, + maybe_origin_model_token: m.maybe_origin_model_token, maybe_creator: UserDetailsLight::from_optional_db_fields_owned( m.maybe_creator_user_token, m.maybe_creator_username,