From 52a42e172874f40623a1037a64519c9238fe3f63 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Fri, 14 Nov 2025 12:01:49 +0530
Subject: [PATCH] Add all imagen variants in fal ai in model map (#16579)

---
 .../image_generation/imagen4_transformation.py   |  2 +-
 .../model_prices_and_context_window_backup.json  | 16 ++++++++++++++++
 model_prices_and_context_window.json             | 16 ++++++++++++++++
 3 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/litellm/llms/fal_ai/image_generation/imagen4_transformation.py b/litellm/llms/fal_ai/image_generation/imagen4_transformation.py
index f38ced65313..4e7708c9f40 100644
--- a/litellm/llms/fal_ai/image_generation/imagen4_transformation.py
+++ b/litellm/llms/fal_ai/image_generation/imagen4_transformation.py
@@ -23,7 +23,7 @@ class FalAIImagen4Config(FalAIBaseConfig):
     
     Model variants:
     - fal-ai/imagen4/preview (Standard): $0.05 per image
-    - fal-ai/imagen4/preview/fast (Fast): $0.04 per image
+    - fal-ai/imagen4/preview/fast (Fast): $0.02 per image
     - fal-ai/imagen4/preview/ultra (Ultra): $0.06 per image
     
     Documentation: https://fal.ai/models/fal-ai/imagen4/preview
diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index c16980211ea..7c86e570c31 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -8539,6 +8539,22 @@
             "/v1/images/generations"
         ]
     },
+    "fal_ai/fal-ai/imagen4/preview/fast": {
+        "litellm_provider": "fal_ai",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.02,
+        "supported_endpoints": [
+            "/v1/images/generations"
+        ]
+    },
+    "fal_ai/fal-ai/imagen4/preview/ultra": {
+        "litellm_provider": "fal_ai",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.06,
+        "supported_endpoints": [
+            "/v1/images/generations"
+        ]
+    },
     "fal_ai/fal-ai/recraft/v3/text-to-image": {
         "litellm_provider": "fal_ai",
         "mode": "image_generation",
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index c16980211ea..7c86e570c31 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -8539,6 +8539,22 @@
             "/v1/images/generations"
         ]
     },
+    "fal_ai/fal-ai/imagen4/preview/fast": {
+        "litellm_provider": "fal_ai",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.02,
+        "supported_endpoints": [
+            "/v1/images/generations"
+        ]
+    },
+    "fal_ai/fal-ai/imagen4/preview/ultra": {
+        "litellm_provider": "fal_ai",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.06,
+        "supported_endpoints": [
+            "/v1/images/generations"
+        ]
+    },
     "fal_ai/fal-ai/recraft/v3/text-to-image": {
         "litellm_provider": "fal_ai",
         "mode": "image_generation",