o ưi8@s~ddlmZmZddlmZmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZmZddlmZGdd d ZdS) )BufferedReaderBytesIO)AnyDictListOptionalcastget_type_hintsN)get_image_type)BaseImageEditConfig)FILE_MIME_TYPESFileType)ImageEditOptionalRequestParamsc@sxeZdZe  ddedededeedee ede f dd Z ed e ee fdefd d Z ed e defddZdS)ImageEditRequestUtilsNmodelimage_edit_provider_configimage_edit_optional_params drop_paramsadditional_drop_paramsreturnc s||tjdup |du}t|}|r|D]}||dqfdd|D}|rH|r8|D]}||dq.ntj|d|dd|d|jtt |||d } | S) a" Get optional parameters for the image edit API. Args: model: The model name image_edit_provider_config: The provider configuration for image edit API image_edit_optional_params: The optional parameters for the image edit API drop_params: If True, silently drop unsupported parameters instead of raising additional_drop_params: List of additional parameter names to drop Returns: A dictionary of supported parameters for the image edit API TNcsg|]}|vr|qSr).0paramZsupported_paramsrK/home/app/Keep/.python/lib/python3.10/site-packages/litellm/images/utils.py *s zHImageEditRequestUtils.get_optional_params_image_edit..z5The following parameters are not supported for model z: z, )rmessage)rrr) Zget_supported_openai_paramslitellmrdictpopZUnsupportedParamsErrorjoinZmap_openai_paramsrr) rrrrrZ should_dropZfiltered_optional_paramsrZunsupported_paramsZ mapped_paramsrrrget_optional_params_image_edit s4  z4ImageEditRequestUtils.get_optional_params_image_editparamscs,ttfdd|D}tt|S)a Filter parameters to only include those defined in ImageEditOptionalRequestParams. Args: params: Dictionary of parameters to filter Returns: ImageEditOptionalRequestParams instance with only the valid parameters cs&i|]\}}|vr|dur||qS)Nr)rkvZ valid_keysrr Rs zQImageEditRequestUtils.get_requested_image_edit_optional_param..)r rkeysitemsr)r"Zfiltered_paramsrr%r'get_requested_image_edit_optional_paramDs  z=ImageEditRequestUtils.get_requested_image_edit_optional_param image_datacCsTzt|tr|}|d|d}||nSt|tr3|}|d|d}||n:t|tr?|dd}n.t|drgt|ddd}t|drW|d|d}t|drf||nt t j WSt |}|dur{t t j WSt j t j t jt jt jd }||}|durt t j WSt |WStyt t j YSw) a Detect the content type of image data using existing LiteLLM utils. Args: image_data: Can be BytesIO, bytes, BufferedReader, or other file-like objects Returns: The MIME type string (e.g., "image/png", "image/jpeg") rdNreadtellcSsdS)Nrrrrrrysz>ImageEditRequestUtils.get_image_content_type..seek)ZpngZjpegZgifZwebpZheic) isinstancerr-r/r,rbyteshasattrgetattrr r ZPNGr ZJPEGZGIFZWEBPZHEICget Exception)r*Z current_pos bytes_dataZimage_type_strZ type_mapping file_typerrrget_image_content_typeWsP                    z,ImageEditRequestUtils.get_image_content_type)NN)__name__ __module__ __qualname__ staticmethodstrr rrboolrrr!rr)r8rrrrr s2  7  r)iorrtypingrrrrrr rZ(litellm.litellm_core_utils.token_counterr Z/litellm.llms.base_llm.image_edit.transformationr Zlitellm.types.filesr r Zlitellm.types.images.mainrrrrrrs