chore: initial public snapshot for github upload

llm-gateway-competitors/litellm-wheel-src/litellm/llms/runwayml/cost_calculator.py

@@ -0,0 +1,30 @@
+from typing import Any
+
+import litellm
+from litellm.types.utils import ImageResponse
+
+
+def cost_calculator(
+    model: str,
+    image_response: Any,
+) -> float:
+    """
+    RunwayML image generation cost calculator.
+
+    RunwayML charges per image generated, not per pixel.
+    Pricing is stored in model_prices_and_context_window.json with output_cost_per_image.
+    """
+    _model_info = litellm.get_model_info(
+        model=model,
+        custom_llm_provider=litellm.LlmProviders.RUNWAYML.value,
+    )
+    output_cost_per_image: float = _model_info.get("output_cost_per_image") or 0.0
+    num_images: int = 0
+    if isinstance(image_response, ImageResponse):
+        if image_response.data:
+            num_images = len(image_response.data)
+        return output_cost_per_image * num_images
+    else:
+        raise ValueError(
+            f"image_response must be of type ImageResponse, got type={type(image_response)}"
+        )