chore: initial public snapshot for github upload

llm-gateway-competitors/litellm-wheel-src/litellm/llms/litellm_proxy/image_generation/transformation.py

@@ -0,0 +1,41 @@
+from typing import Optional
+
+from litellm.llms.openai.image_generation.gpt_transformation import (
+    GPTImageGenerationConfig,
+)
+from litellm.secret_managers.main import get_secret_str
+
+
+class LiteLLMProxyImageGenerationConfig(GPTImageGenerationConfig):
+    """Configuration for image generation requests routed through LiteLLM Proxy."""
+
+    def validate_environment(
+        self,
+        headers: dict,
+        model: str,
+        messages,
+        optional_params: dict,
+        litellm_params: dict,
+        api_key: Optional[str] = None,
+        api_base: Optional[str] = None,
+    ) -> dict:
+        api_key = api_key or get_secret_str("LITELLM_PROXY_API_KEY")
+        headers.update({"Authorization": f"Bearer {api_key}"})
+        return headers
+
+    def get_complete_url(
+        self,
+        api_base: Optional[str],
+        api_key: Optional[str],
+        model: str,
+        optional_params: dict,
+        litellm_params: dict,
+        stream: Optional[bool] = None,
+    ) -> str:
+        api_base = api_base or get_secret_str("LITELLM_PROXY_API_BASE")
+        if api_base is None:
+            raise ValueError(
+                "api_base not set for LiteLLM Proxy route. Set in env via `LITELLM_PROXY_API_BASE`"
+            )
+        api_base = api_base.rstrip("/")
+        return f"{api_base}/images/generations"