scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

fix(azure_ai): narrow MAI edit size param type for mypy

Browse source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Sameer Kankute 2026-06-08 11:59:46 +05:30
parent c1316a4d82
commit 7c8ed6d100
No known key found for this signature in database
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/azure_ai/image_edit/mai_transformation.py
View file

@ -1,4 +1,4 @@
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, cast
import httpx
from httpx._types import RequestFiles
@ -43,8 +43,9 @@ class AzureFoundryMAIImageEditConfig(OpenAIImageEditConfig):
if key in supported_params:
if key == "size" and value:
self._validate_size_param(value)
optional_params[key] = value
size_param = cast(str, value)
self._validate_size_param(size_param)
optional_params[key] = size_param
else:
optional_params[key] = value
elif not drop_params: