"""
Stability AI Image Edit Module

Factory function for getting the appropriate config class.
"""

from litellm.llms.base_llm.image_edit.transformation import (
    BaseImageEditConfig,
)

from .transformations import StabilityImageEditConfig

__all__ = [
    "StabilityImageEditConfig",
    "get_stability_image_edit_config",
]


def get_stability_image_edit_config(model: str) -> BaseImageEditConfig:
    """
    Get the appropriate Stability AI config for the given model.

    Currently all models use the same config class, but this factory
    allows for model-specific configs in the future.

    Args:
        model: The model name (e.g., "stability/inpaint", "stability/outpaint")

    Returns:
        BaseImageEditConfig instance for Stability AI
    """
    # For now, all models use the same config
    # In the future, we could have model-specific configs:
    # - StabilityInpaintConfig for Inpaint models
    # - StabilityOutpaintConfig for Outpaint models
    # - etc.
    return StabilityImageEditConfig()