From f132f2c811af30ae448f0e47540c6a566783d139 Mon Sep 17 00:00:00 2001
From: Chesars <cesarponce19544@gmail.com>
Date: Mon, 15 Dec 2025 17:59:03 -0300
Subject: [PATCH] chore: add Black Forest Labs models to model registry

Add BFL models to model_prices_and_context_window.json with pricing:
- flux-kontext-pro: $0.04/image
- flux-kontext-max: $0.08/image
- flux-pro-1.0-fill: $0.05/image
- flux-pro-1.0-expand: $0.05/image

Add black_forest_labs_models set to __init__.py for model discovery.
---
 litellm/__init__.py                  |  5 ++++
 model_prices_and_context_window.json | 36 ++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)

diff --git a/litellm/__init__.py b/litellm/__init__.py
index 84b8e47c462..1af1f68b5b6 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -575,6 +575,7 @@ v0_models: Set = set()
 morph_models: Set = set()
 lambda_ai_models: Set = set()
 hyperbolic_models: Set = set()
+black_forest_labs_models: Set = set()
 recraft_models: Set = set()
 cometapi_models: Set = set()
 oci_models: Set = set()
@@ -821,6 +822,8 @@ def add_known_models(model_cost_map: Optional[Dict] = None):
             lambda_ai_models.add(key)
         elif value.get("litellm_provider") == "hyperbolic":
             hyperbolic_models.add(key)
+        elif value.get("litellm_provider") == "black_forest_labs":
+            black_forest_labs_models.add(key)
         elif value.get("litellm_provider") == "recraft":
             recraft_models.add(key)
         elif value.get("litellm_provider") == "cometapi":
@@ -952,6 +955,7 @@ model_list = list(
     | v0_models
     | morph_models
     | lambda_ai_models
+    | black_forest_labs_models
     | recraft_models
     | cometapi_models
     | oci_models
@@ -1049,6 +1053,7 @@ models_by_provider: dict = {
     "morph": morph_models,
     "lambda_ai": lambda_ai_models,
     "hyperbolic": hyperbolic_models,
+    "black_forest_labs": black_forest_labs_models,
     "recraft": recraft_models,
     "cometapi": cometapi_models,
     "oci": oci_models,
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 412f99791a0..26a0f926e6b 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -7786,6 +7786,42 @@
         "supports_response_schema": true,
         "supports_tool_choice": true
     },
+    "black_forest_labs/flux-kontext-pro": {
+        "litellm_provider": "black_forest_labs",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.04,
+        "source": "https://bfl.ai/pricing",
+        "supported_endpoints": [
+            "/v1/images/edits"
+        ]
+    },
+    "black_forest_labs/flux-kontext-max": {
+        "litellm_provider": "black_forest_labs",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.08,
+        "source": "https://bfl.ai/pricing",
+        "supported_endpoints": [
+            "/v1/images/edits"
+        ]
+    },
+    "black_forest_labs/flux-pro-1.0-fill": {
+        "litellm_provider": "black_forest_labs",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.05,
+        "source": "https://bfl.ai/pricing",
+        "supported_endpoints": [
+            "/v1/images/edits"
+        ]
+    },
+    "black_forest_labs/flux-pro-1.0-expand": {
+        "litellm_provider": "black_forest_labs",
+        "mode": "image_generation",
+        "output_cost_per_image": 0.05,
+        "source": "https://bfl.ai/pricing",
+        "supported_endpoints": [
+            "/v1/images/edits"
+        ]
+    },
     "cerebras/llama-3.3-70b": {
         "input_cost_per_token": 8.5e-07,
         "litellm_provider": "cerebras",