From b350cd306ade2fad8d51754f39706500b9e05dba Mon Sep 17 00:00:00 2001
From: Emerson Gomes <emerson.gomes@thalesgroup.com>
Date: Fri, 23 May 2025 00:45:50 -0500
Subject: [PATCH] Add Azure Mistral Medium 25.05 (#11063)

* Add Azure Mistral Medium 25.05

* fix provider
---
 model_prices_and_context_window.json | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index eca0bbcfc89..a37a2de9cf5 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -3117,6 +3117,18 @@
         "supports_function_calling": true,
         "source": "https://azuremarketplace.microsoft.com/en/marketplace/apps/000-000.mistral-nemo-12b-2407?tab=PlansAndPrice"
     },
+    "azure_ai/mistral-medium-2505": {
+        "max_tokens": 8191,
+        "max_input_tokens": 131072,
+        "max_output_tokens": 8191,
+        "input_cost_per_token": 0.0000004,
+        "output_cost_per_token": 0.000002,
+        "litellm_provider": "azure_ai",
+        "mode": "chat",
+        "supports_function_calling": true,
+        "supports_assistant_prefill": true,
+        "supports_tool_choice": true
+    },
     "azure_ai/mistral-large": {
         "max_tokens": 8191,
         "max_input_tokens": 32000,