scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test: test_openai_image_edit_litellm_sdk

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-23 17:34:03 -07:00
parent 8c0054a77e
commit ba4148100f
2 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/model_prices_and_context_window_backup.json
View file

@ -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,

1
tests/image_gen_tests/test_image_edits.py
View file

@ -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