refactor: consolidate media helpers

Author: nabinchha

packages/data-designer-config/src/data_designer/config/__init__.py

@@ -107,9 +107,8 @@
     )
     from data_designer.config.seed_source_dataframe import DataFrameSeedSource  # noqa: F401
     from data_designer.config.utils.code_lang import CodeLang  # noqa: F401
-    from data_designer.config.utils.image_helpers import ImageFormat  # noqa: F401
     from data_designer.config.utils.info import InfoType  # noqa: F401
-    from data_designer.config.utils.media_helpers import AudioFormat, VideoFormat  # noqa: F401
+    from data_designer.config.utils.media_helpers import AudioFormat, ImageFormat, VideoFormat  # noqa: F401
     from data_designer.config.utils.trace_type import TraceType  # noqa: F401
     from data_designer.config.validator_params import (  # noqa: F401
         CodeValidatorParams,
@@ -171,7 +170,7 @@
     "EmbeddingInferenceParams": (_MOD_MODELS, "EmbeddingInferenceParams"),
     "GenerationType": (_MOD_MODELS, "GenerationType"),
     "ImageContext": (_MOD_MODELS, "ImageContext"),
-    "ImageFormat": (f"{_MOD_UTILS}.image_helpers", "ImageFormat"),
+    "ImageFormat": (f"{_MOD_UTILS}.media_helpers", "ImageFormat"),
     "ImageInferenceParams": (_MOD_MODELS, "ImageInferenceParams"),
     "ManualDistribution": (_MOD_MODELS, "ManualDistribution"),
     "ManualDistributionParams": (_MOD_MODELS, "ManualDistributionParams"),

packages/data-designer-config/src/data_designer/config/models.py

@@ -21,25 +21,23 @@
     MIN_TEMPERATURE,
     MIN_TOP_P,
 )
-from data_designer.config.utils.image_helpers import (
-    ImageFormat,
-    decode_base64_image,
-    detect_image_format,
-    is_image_path,
-    is_image_url,
-    load_image_path_to_base64,
-)
 from data_designer.config.utils.io_helpers import smart_load_yaml
 from data_designer.config.utils.media_helpers import (
     AudioFormat,
+    ImageFormat,
     VideoFormat,
     audio_format_from_mime_type,
     audio_mime_type,
+    decode_base64_image,
+    detect_image_format,
     get_media_base64_context,
     get_media_url_context,
     is_audio_path,
+    is_image_path,
+    is_image_url,
     is_media_url,
     is_video_path,
+    load_image_path_to_base64,
     normalize_media_context_values,
     parse_base64_data_uri,
     video_format_from_mime_type,