test: test_openai_image_edit_litellm_sdk

litellm/model_prices_and_context_window_backup.json

@@ -3815,6 +3815,19 @@
         "supports_assistant_prefill": true,
         "supports_tool_choice": true
     },
+    "mistral/devstral-small-2505": {
+        "max_tokens": 128000,
+        "max_input_tokens": 128000,
+        "max_output_tokens": 128000,
+        "input_cost_per_token": 0.0000001,
+        "output_cost_per_token": 0.0000003,
+        "litellm_provider": "mistral",
+        "mode": "chat",
+        "source": "https://mistral.ai/news/devstral",
+        "supports_function_calling": true,
+        "supports_assistant_prefill": true,
+        "supports_tool_choice": true
+    },
     "mistral/mistral-embed": {
         "max_tokens": 8192,
         "max_input_tokens": 8192,

tests/image_gen_tests/test_image_edits.py

@@ -22,6 +22,7 @@ TEST_IMAGES = [
 ]
 
 @pytest.mark.parametrize("sync_mode", [True])
+@pytest.mark.flaky(retries=3, delay=2)
 @pytest.mark.asyncio
 async def test_openai_image_edit_litellm_sdk(sync_mode):
     from litellm import image_edit, aimage_edit