data-designer-config 0.4.0rc1__py3-none-any.whl → 0.4.0rc2__py3-none-any.whl

data_designer/config/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.4.0rc1'
-__version_tuple__ = version_tuple = (0, 4, 0, 'rc1')
+__version__ = version = '0.4.0rc2'
+__version_tuple__ = version_tuple = (0, 4, 0, 'rc2')
 
 __commit_id__ = commit_id = None

data_designer/config/models.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import json
 import logging
 from abc import ABC, abstractmethod
 from enum import Enum
@@ -65,7 +66,7 @@ class ModalityContext(ABC, BaseModel):
     data_type: ModalityDataType
 
     @abstractmethod
-    def get_context(self, record: dict) -> dict[str, Any]: ...
+    def get_contexts(self, record: dict) -> list[dict[str, Any]]: ...
 
 
 class ImageContext(ModalityContext):
@@ -81,25 +82,53 @@ class ImageContext(ModalityContext):
     modality: Modality = Modality.IMAGE
     image_format: ImageFormat | None = None
 
-    def get_context(self, record: dict) -> dict[str, Any]:
-        """Get the context for the image modality.
+    def get_contexts(self, record: dict) -> list[dict[str, Any]]:
+        """Get the contexts for the image modality.
 
         Args:
-            record: The record containing the image data.
+            record: The record containing the image data. The data can be:
+                - A JSON serialized list of strings
+                - A list of strings
+                - A single string
 
         Returns:
-            The context for the image modality.
+            A list of image contexts.
         """
-        context = dict(type="image_url")
-        context_value = record[self.column_name]
-        if self.data_type == ModalityDataType.URL:
-            context["image_url"] = context_value
+        raw_value = record[self.column_name]
+
+        # Normalize to list of strings
+        if isinstance(raw_value, str):
+            # Try to parse as JSON first
+            try:
+                parsed_value = json.loads(raw_value)
+                if isinstance(parsed_value, list):
+                    context_values = parsed_value
+                else:
+                    context_values = [raw_value]
+            except (json.JSONDecodeError, TypeError):
+                context_values = [raw_value]
+        elif isinstance(raw_value, list):
+            context_values = raw_value
+        elif hasattr(raw_value, "__iter__") and not isinstance(raw_value, (str, bytes, dict)):
+            # Handle array-like objects (numpy arrays, pandas Series, etc.)
+            context_values = list(raw_value)
         else:
-            context["image_url"] = {
-                "url": f"data:image/{self.image_format.value};base64,{context_value}",
-                "format": self.image_format.value,
-            }
-        return context
+            context_values = [raw_value]
+
+        # Build context list
+        contexts = []
+        for context_value in context_values:
+            context = dict(type="image_url")
+            if self.data_type == ModalityDataType.URL:
+                context["image_url"] = context_value
+            else:
+                context["image_url"] = {
+                    "url": f"data:image/{self.image_format.value};base64,{context_value}",
+                    "format": self.image_format.value,
+                }
+            contexts.append(context)
+
+        return contexts
 
     @model_validator(mode="after")
     def _validate_image_format(self) -> Self:
@@ -399,12 +428,14 @@ class ModelConfig(ConfigBase):
         inference_parameters: Inference parameters for the model (temperature, top_p, max_tokens, etc.).
             The generation_type is determined by the type of inference_parameters.
         provider: Optional model provider name if using custom providers.
+        skip_health_check: Whether to skip the health check for this model. Defaults to False.
     """
 
     alias: str
     model: str
     inference_parameters: InferenceParamsT = Field(default_factory=ChatCompletionInferenceParams)
     provider: str | None = None
+    skip_health_check: bool = False
 
     @property
     def generation_type(self) -> GenerationType:

{data_designer_config-0.4.0rc1.dist-info → data_designer_config-0.4.0rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: data-designer-config
-Version: 0.4.0rc1
+Version: 0.4.0rc2
 Summary: Configuration layer for DataDesigner synthetic data generation
 License-Expression: Apache-2.0
 Classifier: Development Status :: 4 - Beta

{data_designer_config-0.4.0rc1.dist-info → data_designer_config-0.4.0rc2.dist-info}/RECORD

@@ -3,7 +3,7 @@ data_designer/lazy_heavy_imports.py,sha256=5X04vUj9TYbKgfDmY2qvhzRf5-IZWKOanIpi3
 data_designer/logging.py,sha256=_x-tDj34ClrgSnU57Dh0mZdDxrnA73vgs09KooUvbEA,5444
 data_designer/plugin_manager.py,sha256=C2ZkZiXlcMRiaxfrrho5Shz6DKdExVeBha7ch-d4CnU,2695
 data_designer/config/__init__.py,sha256=MWzRZhXA41sTpc0sL_xq2baA3kSlV37alT6g8RlP8dU,4919
-data_designer/config/_version.py,sha256=yib4WPM_pEWXdpIHBdFnf29aurTH5f4xrnwVlv7cijo,714
+data_designer/config/_version.py,sha256=FvItxCBzPigrdVpFPfL1gQeV1-km5r7nCNGUzrYebTU,714
 data_designer/config/base.py,sha256=IGj6sy_GnKzC94uu2rdxe12EqR_AmGJ6O3rl2MxOv6g,2449
 data_designer/config/column_configs.py,sha256=JBYIeqmIiwdZzA-NXTw4qghs2Y30RZPQ_-koqPVcQ9g,20384
 data_designer/config/column_types.py,sha256=xGXuu0EBy3Y5Jd74f2VM6x5jHq72GmK9leA6qOnAz8c,5423
@@ -14,7 +14,7 @@ data_designer/config/dataset_metadata.py,sha256=UTlEgnHWgjwPuc7bP95T7gaKmcr7pIhF
 data_designer/config/default_model_settings.py,sha256=c-llH2otfG0tMCMsxoz3ZcS1nFxIQQPfRedFXAydDbc,4868
 data_designer/config/errors.py,sha256=JhvUYecfLmP0gZjQzqA3OmfaSs9TRlC5E-ubnV_-3gs,560
 data_designer/config/interface.py,sha256=ikmpm_KwencTpM-yg0auo7XMgcmMSa67S75IqdpFLfk,1676
-data_designer/config/models.py,sha256=OekrXEVnI9WdHzEVk-8fO0NtxLZtjKVtCL03RY8qwYs,15457
+data_designer/config/models.py,sha256=_NctRk4brgBeb5q5V7r_hXE5OORlLh6SCVZP0eu2LGo,16721
 data_designer/config/preview_results.py,sha256=WnPlDcHElIHNfjV_P-nLu_Dpul8D3Eyb5qyi3E173Gs,1744
 data_designer/config/processors.py,sha256=lnyUZA1EhO9NWjjVFFioYxSgeYpoAaM1J7UzwOYkvms,6028
 data_designer/config/run_config.py,sha256=oJ163DpHXu9PzST5Hn9px-bIP9DYjIkCO7UGB93J7bI,2663
@@ -45,6 +45,6 @@ data_designer/plugins/__init__.py,sha256=qe1alcTEtnMSMdzknjb57vvjqKgFE5cEHXxBj8t
 data_designer/plugins/errors.py,sha256=d7FMed3ueQvZHwuhwyPLzF4E34bO1mdj3aBVEw6p34o,386
 data_designer/plugins/plugin.py,sha256=TVyyOaQBWAt0FQwUmtihTZ9MDJD85HwggrQ3L9CviPQ,5367
 data_designer/plugins/registry.py,sha256=Cnt33Q25o9bS2v2YDbV3QPM57VNrtIBKAb4ERQRE_dY,3053
-data_designer_config-0.4.0rc1.dist-info/METADATA,sha256=EvLUDu0U0oK7FuylbCr44meSR8H5Ty-bt-EdUIagmi4,2286
-data_designer_config-0.4.0rc1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-data_designer_config-0.4.0rc1.dist-info/RECORD,,
+data_designer_config-0.4.0rc2.dist-info/METADATA,sha256=sEtB9CR6HT9TCz7nptAitc_pH9N62tDBddnEFMw9MYM,2286
+data_designer_config-0.4.0rc2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+data_designer_config-0.4.0rc2.dist-info/RECORD,,