from litellm.llms.base_llm.image_edit.transformation import BaseImageEditConfig

from .transformation import GeminiImageEditConfig
from .cost_calculator import cost_calculator

__all__ = ["GeminiImageEditConfig", "get_gemini_image_edit_config", "cost_calculator"]


def get_gemini_image_edit_config(model: str) -> BaseImageEditConfig:
    return GeminiImageEditConfig()