From 9fe6442172972ceade871c721e6f78c50c1da5f9 Mon Sep 17 00:00:00 2001
From: "berriai-litellm-provider-info-sync[bot]"
 <328147090+berriai-litellm-provider-info-sync[bot]@users.noreply.github.com>
Date: Sat, 3 Oct 2026 11:40:44 -0700
Subject: [PATCH] fix(azure): add azure_ai/flux.2-pro input token limit from
 the models sold directly page (#44405)

Price-Sync: litellm-providers

Co-authored-by: berriai-litellm-provider-info-sync[bot] <328147090+berriai-litellm-provider-info-sync[bot]@users.noreply.github.com>
---
 litellm/model_prices_and_context_window_backup.json | 4 +++-
 model_prices_and_context_window.json                | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json
index fd8074f4744..3f7f5194514 100644
--- a/litellm/model_prices_and_context_window_backup.json
+++ b/litellm/model_prices_and_context_window_backup.json
@@ -11542,9 +11542,11 @@
     },
     "azure_ai/flux.2-pro": {
         "litellm_provider": "azure_ai",
+        "max_input_tokens": 32000,
+        "max_tokens": 32000,
         "mode": "image_generation",
         "output_cost_per_image": 0.04,
-        "source": "https://ai.azure.com/explore/models/flux.2-pro/version/1/registry/azureml-blackforestlabs",
+        "source": "https://learn.microsoft.com/en-us/azure/foundry/foundry-models/concepts/models-sold-directly-by-azure",
         "supported_endpoints": [
             "/v1/images/generations"
         ]
diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index fd8074f4744..3f7f5194514 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -11542,9 +11542,11 @@
     },
     "azure_ai/flux.2-pro": {
         "litellm_provider": "azure_ai",
+        "max_input_tokens": 32000,
+        "max_tokens": 32000,
         "mode": "image_generation",
         "output_cost_per_image": 0.04,
-        "source": "https://ai.azure.com/explore/models/flux.2-pro/version/1/registry/azureml-blackforestlabs",
+        "source": "https://learn.microsoft.com/en-us/azure/foundry/foundry-models/concepts/models-sold-directly-by-azure",
         "supported_endpoints": [
             "/v1/images/generations"
         ]