uPj.<ddlmZddlZddlmZmZmZddlZddlm Z ddl Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZmZmZmZdd lmZd dlm Z d dl!m"Z"erddl#m$Z$neZ$Gdde Z%e&ee&eze&eze&ezZ'Gdde Z(y)) annotationsN) TYPE_CHECKINGAnyFinal) BaseModel)verbose_logger)Logging)AmazonNovaCanvasConfig)AmazonStabilityConfig)AmazonStability3Config) AmazonTitanImageGenerationConfig)AsyncHTTPHandler HTTPHandler_get_httpx_clientget_async_httpx_client) ImageResponse) BaseAWSLLM) BedrockError)AWSPreparedRequestc:eZdZUdZded<ded<ded<ded <y ) BedrockImagePreparedRequestzV Internal/Helper class for preparing the request for bedrock image generation str endpoint_urlrpreppedbytesbodydictdataN)__name__ __module__ __qualname____doc____annotations__x/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/bedrock/image_generation/image_handler.pyrr)s  K Jr&rceZdZdZed dZ d d dZ d ddZddZ ddZ ddZ dd Z y)BedrockImageGenerationz* Bedrock Image Generation handler ctj|rtStj|rtSt j |rtSt jSN)r _is_titan_modelr _is_nova_modelr _is_stability_3_modellitellmr )clsmodels r'get_config_classz'BedrockImageGeneration.get_config_classAsJ + ; ;E B3 3 # 2 25 9) ) # 9 9% @) )00 0r&Nc |j|||| ||| } |dur1|j| |||||| t| tr| SdS| t| ts t } | j | j| jj| j} | j|j'||||| | j(}|S#tj$r<}|jj}t!||jj"d}~wtj$$rt!ddwxYw) N)r1optional_paramsapi_base extra_headers logging_objpromptapi_keyT)prepared_requesttimeoutr1r7r8model_responseclienturlheadersr status_codemessageTimeout error occurred.)r<r1r7r8responser)_prepare_requestasync_image_generation isinstancerrrpostrrr@rraise_for_statushttpxHTTPStatusErrorrFrBrtextTimeoutException+_transform_response_dict_to_openai_responser)selfr1r8r<r4r7r;aimg_generationr5r6r=r9r:rFerr error_codes r'image_generationz'BedrockImageGeneration.image_generationLs#'"7"7+'##8#  d "..!1'-"("4FL\9]/ dh/  >FK!@&(F S$kk$11(0088%***H  % % 'II)#!&& J $$ R # 8 8J:s||?P?PQ Q%% S38QR R Ss+A CE-7D$$$Ec6K|xs'ttjjd|i} |j |j |j j|jd{} | j|j#|||| |j$|}|S77#tj$r<} | jj} t| | jjd} ~ wtj $rtddwxYww) z Asynchronous handler for bedrock image generation Awaits the response from the bedrock image generation endpoint r;) llm_providerparamsr>NrArDrE)r1r7r8rFrr<)rr/ LlmProvidersBEDROCKrJrrr@rrKrLrMrFrBrrNrOrPr) rQr:r;r1r7r8r<r= async_clientrFrSrTs r'rHz-BedrockImageGeneration.async_image_generations% (> --55w') S$0$5$5$11(0088%**%6%H  % % 'II#!&&) J + $$ R # 8 8J:s||?P?PQ Q%% S38QR R Ss:,D?B(.B&/B(#D&B((D;7C22$DDcni}|jdd}|jdd}|||d<|||d<|S)z` Extract guardrail parameters from optional_params and convert them to headers. guardrailIdentifierNguardrailVersionz"x-amz-bedrock-guardrail-identifierzx-amz-bedrock-guardrail-version)pop)rQr4r@guardrail_identifierguardrail_versions r'%_extract_headers_from_optional_paramszrs" ,, +P .%'6) )* !"# !"# !"QZQr&