uPj d Z ddlmZ ddlmZ ddlZddlmZmZ ddl Z ddl
m Z m Z m
Z
ddlmZ dd lmZ dd
lmZ d d lmZ G d
d Zy)z!
OpenAI Image Variations Handler
)Callable)FinalN) AsyncOpenAIOpenAI) FileTypes
ImageResponse LlmProviders)ProviderConfigManager )BaseImageVariationConfig)LiteLLMLoggingObj ) OpenAIErrorc e Zd Zdedz defdZdedz dedefdZdeded edz dedz d
ed ed edz d
e dz de
de de dedede
dede f dZ dde deded edz de
d
e dz dede dedededz d edz d edz de f dZy) OpenAIImageVariationsHandlerclientNinit_client_paramsc ( |
t di |}|S |}|S N )r selfr r
openai_clients q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/openai/image_variations/handler.pyget_sync_clientz,OpenAIImageVariationsHandler.get_sync_client s0
> " $ M
#M returnc ( |
t di |}|S |}|S r )r r s r get_async_clientz-OpenAIImageVariationsHandler.get_async_client" s. > ' $ M
#M r api_keyapi_base organizationdataheadersmodeltimeout max_retries logging_objmodel_responseoptional_paramslitellm_paramsimageprovider_configc K ||t j || |d}| j || } |j j j
di | d { }|j
}|j }|
j ||||d |j |t di |t j dt j dd |
||| |
d |
S 7 # t $ ra}t |d d }t |d
d }t |dt! | }t |dd }||r
t |d
d }t# ||| d }~ww xY ww)Nr base_url http_clientr& r' r" r r r$ r! r original_responseadditional_args GEThttps://litellm.aimethodurl status_coderequest
r% r) raw_responser( request_datar, r* r+ encodingr r> r$ textresponser> messager$ r )litellmclient_sessionr imageswith_raw_responsecreate_variationparse
model_dump post_call"transform_response_image_variationr httpxResponseRequest Exceptiongetattrstrr )r r r! r" r r# r$ r% r& r' r( r) r* r+ r, r- r rA rF
response_jsoner>
error_headers
error_texterror_responses r async_image_variationsz3OpenAIImageVariationsHandler.async_image_variations+ s $1 b" $ & 5 5" * ,
) **&M_*`F(X
(G(G(X(X(`[_(`"`L * 0 0 2H#+#6#6#8M
! ! "/ & (!
"
#EE , =} = "^^ # !MM rD r$ rE rF rG r ) r
#get_provider_image_variation_configr OPENAI
ValueErrorpop!transform_request_image_variationgetpre_callr] rI rJ r rK rL rM rN rO rP rQ r rR rS rT rU rV rW r ) r r) r r! r% r, r& r^ r( r* r+ r_ logger_fnr r" r$ r- r' r# json_datar rA rF rX rY r> rZ r[ r\ s r image_variationsz-OpenAIImageVariationsHandler.image_variationsp s $a b%:%^%^kr%,,&O
& #GH[G\\]!^__!0!4!4]A!FK)KK /
2 L D $xx /I #\]a\b!cdd & (+/!
!
!!,622 %" #Mr#1 # + # +!-!$3 $3#1 3 & # $ & 5 5" * ,
) ))L^)_F"R&--"A"A"R"R"_U^"_L * 0 0 2H#+#6#6#8M
! ! "/ & (!
"
#EE , =} = "^^ # !MM
r s; % & F F / P > &Ob Obr