uPjR*ddlZddlmZmZmZddlZddlmZddlZddl m Z ddl m Z ddl mZddlmZddlmZmZmZmZdd lmZdd lmZmZdd lmZmZerdd lmZ d dl!m"Z#e Z$e#Z"neZ$eZ"de%e&e&ffdZ'de%e&efde&dzfdZ(GddeZ)y)N) TYPE_CHECKINGAnyFinal) RequestFiles)%DEFAULT_GOOGLE_VIDEO_DURATION_SECONDS)ImageEditRequestUtils)BaseVideoConfig)get_secret_str)"GeminiLongRunningOperationResponseGeminiVideoGenerationInstanceGeminiVideoGenerationParametersGeminiVideoGenerationRequest)GenericLiteLLMParams) VideoCreateOptionalRequestParams VideoObject)encode_video_id_with_providerextract_original_video_id)Logging)BaseLLMExceptionreturnctj|}t|dr|jd|j }t j |jd}||dS)z Convert image file to Gemini format with base64 encoding and MIME type. Args: image_file: File-like object opened in binary mode (e.g., open("path", "rb")) Returns: Dict with bytesBase64Encoded and mimeType seekrzutf-8)bytesBase64EncodedmimeType)rget_image_content_typehasattrrreadbase64 b64encodedecode) image_file mime_type image_bytesbase64_encodeds n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gemini/videos/transformation.py_convert_image_to_gemini_formatr'%s\-CCJOIz6"#*K",,[9@@IN"0i HH parametersc|jd}||dk(ryt|jjS)zDNormalize Veo ``parameters.resolution`` for usage and cost tracking. resolutionN)getstrstriplower)r)ress r&'_usage_video_resolution_from_parametersr29s: -C {cRi s8>>  ! ! ##r(ceZdZUdZdddddZeeefed<fdZdede fd Z d e ded e deee ffd Zd ededzfdZd ededzfdZ d;dedededzdedzdef dZdededzdedefdZdededededededeeeeffdZ d;dedej0dededzdedzdef dZdededededeeeff d Z dd4Z, d;d5Z- dzGeminiVideoConfig.__init__Us r(modelrc gdS)z Get the list of supported OpenAI parameters for Veo video generation. Veo supports minimal parameters compared to OpenAI. )rApromptinput_referencesecondssize)r?rAs r&get_supported_openai_paramsz-GeminiVideoConfig.get_supported_openai_paramsXs IHr(video_create_optional_params drop_paramsci}|j|}|Dchc] }|dvs| }}d|vr|d|d<d|vrH|d}|A|j|} | r| |d<|jds|j|} | | |d<d|vr*|d} t | t r t | n| } | | |d<|jD]\} }| |vs | |vs||| <|Scc}w#ttf$rY?wxYw) u Map OpenAI-style parameters to Veo format. Mappings: - prompt → prompt - input_reference → image - size → aspectRatio (e.g., "1280x720" → "16:9") - size → resolution when inferable ("1280x720"/"720x1280" → "720p", "1920x1080"/"1080x1920" → "1080p"); skipped if ``resolution`` is already set - seconds → durationSeconds (defaults to 4 seconds if not provided) All other params are passed through as-is to support Gemini-specific parameters. >rArCrDimagerF aspectRatior+rEdurationSeconds) rH_convert_size_to_aspect_ratior-_convert_size_to_resolution isinstancer.int ValueError TypeErroritems)r?rIrArJ mapped_paramssupported_openai_paramsparamopenai_params_to_maprF aspect_ratioinferred_resolutionrEdurationkeyvalues r&map_openai_paramsz#GeminiVideoConfig.map_openai_params_s\&02 *.)I)I%)P:Q&vUZbuUuu&v&v  < <%ABS%TM' " 1 16v>D&*&H&H&N 3?M-0377 E151Q1QRV1W'*66I l3 4 49)DG 2 +JC..3m3K%* c" +E'w2 *  s CC $CC.-C.rFNc@|sy|jj|dS)z Convert OpenAI size format to Veo aspectRatio format. https://cloud.google.com/vertex-ai/generative-ai/docs/image/generate-videos Supported aspect ratios: 9:16 (portrait), 16:9 (landscape) Nr5)r:r-)r?rFs r&rOz/GeminiVideoConfig._convert_size_to_aspect_ratios#66::4HHr(c|r||jvry |jdd\}}tt|t|}|dk(ry|dk(ryy#tt f$rYywxYw)z Map OpenAI ``size`` (WxH) to Veo ``resolution`` for presets in ``_OPENAI_VIDEO_SIZE_TO_ASPECT_RATIO`` (720p / 1080p from the smaller edge). Unknown sizes return None so the API default applies (no forced resolution). Nxi720pi81080p)r:splitminrRrSrT)r?rFw_strh_strsmallers r&rPz-GeminiVideoConfig._convert_size_to_resolutionsvt4#J#JJ ::c1-LE5 USZ8G c> d? I&  s3AA%$A%headersapi_keylitellm_paramsc|r|jr|xs |j}|xs*tjxstdxs td}|s td|j |dd|S)z Validate environment and add Gemini API key to headers. Gemini uses x-goog-api-key header for authentication. GOOGLE_API_KEYGEMINI_API_KEYzGEMINI_API_KEY or GOOGLE_API_KEY is required for Veo video generation. Set it via environment variable or pass it as api_key parameter.zapplication/json)zx-goog-api-keyz Content-Type)rllitellmr rSupdate)r?rkrArlrms r&validate_environmentz&GeminiVideoConfig.validate_environmentsz n447!7!7GtW__t?O0PtTbcsTtS  ") 2  r(api_basec|tdxsd}|r|dk(r|jdS|jdd}|jdd|d}|S)z Get the complete URL for Veo video generation. For video creation: returns full URL with :predictLongRunning For status/delete: returns base URL only GEMINI_API_BASEz)https://generativelanguage.googleapis.comr,/zgemini/z/v1beta/models/z:predictLongRunning)r rstripreplace)r?rArtrm model_nameurls r&get_complete_urlz"GeminiVideoConfig.get_complete_urlse  %&78g(`j"k.99t9L R))r( raw_response logging_objcustom_llm_providerrc|j} td i|}|j} | std||rt | ||} n| } t | dd|} i} |rN|jdi} | jdxst}| t|| d <t| }||| d <| | _ | S#t$r}td|d}~wwxYw#ttf$rYKwxYw) a Transform the Veo video creation response. Veo returns: { "name": "operations/generate_1234567890", "metadata": {...}, "done": false, "error": {...} } We return this as a VideoObject with: - id: operation name (used for polling) - status: "processing" - usage: includes duration_seconds and optional video_resolution for cost calculation z$Failed to parse operation response: Nz#No operation name in Veo response: video processing)idobjectstatusrAr)rNduration_secondsvideo_resolutionrG) jsonr ExceptionrSnamerrr-rfloatrTr2usage)r?rArrrr response_dataoperation_responseeoperation_namevideo_id video_obj usage_datar)r\rs r&transform_video_create_responsez1GeminiVideoConfig.transform_video_create_response#s.0 ,002  I(J([]([ !3 7 7B=/RS S 4^EXZ_`H%H&  -/  , 0 0r BJ(nn->?hChH#5:8_J12'Nj&Y +1A -.$ C ICA3GH H I4#I.s) B5 C5 C>C  CC&%C&rcPt|}|jdd|}i}||fS)z Transform the video status retrieve request for Veo API. Veo polls operations at: GET https://generativelanguage.googleapis.com/v1beta/{operation_name} rw/v1beta/)rrx)r?rrtrmrkrr{paramss r&'transform_video_status_retrieve_requestz9GeminiVideoConfig.transform_video_status_retrieve_requestcs8!:( C ,-Xn5EF(*F{r(c|j}tdi|}|j}|j}|rt ||d}n|}t |d|sdnd} | S)a Transform the Veo operation status response. Veo returns: { "name": "operations/generate_1234567890", "done": false # or true when complete } When done=true: { "name": "operations/generate_1234567890", "done": true, "response": { "generateVideoResponse": { "generatedSamples": [ { "video": { "uri": "files/abc123..." } } ] } } } Nrr completed)rrrrG)rr rdonerr) r?rrrrrris_donerrs r&(transform_video_status_retrieve_responsez:GeminiVideoConfig.transform_video_status_retrieve_responsevsn@ ,002 $F$W$W 2 7 7+00 4^EXZ^_H%H&'.#>#T#T#e#e/288<< (*V##r(c|jS)zn Transform the Veo video content download response. Returns the video bytes directly. )contentr?rrs r& transform_video_content_responsez2GeminiVideoConfig.transform_video_content_responses###r( extra_bodyctd)z: Video remix is not supported by Veo API. z_Video remix is not supported by Google Veo. Please use video_generation() to create new videos.NotImplementedError)r?rrCrtrmrkrs r&transform_video_remix_requestz/GeminiVideoConfig.transform_video_remix_requests" m  r(ctd)zVideo remix is not supported.z+Video remix is not supported by Google Veo.rr?rrrs r&transform_video_remix_responsez0GeminiVideoConfig.transform_video_remix_responses""OPPr(afterlimitorder extra_queryctd)z9 Video list is not supported by Veo API. zoVideo list is not supported by Google Veo. Use the operations endpoint directly if you need to list operations.r)r?rtrmrkrrrrs r&transform_video_list_requestz.GeminiVideoConfig.transform_video_list_requests" S  r(ctd)zVideo list is not supported.z*Video list is not supported by Google Veo.rrs r&transform_video_list_responsez/GeminiVideoConfig.transform_video_list_responses""NOOr(ctd)z; Video delete is not supported by Veo API. z[Video delete is not supported by Google Veo. Videos are automatically cleaned up by Google.r)r?rrtrmrks r&transform_video_delete_requestz0GeminiVideoConfig.transform_video_delete_requests" i  r(ctd)zVideo delete is not supported.z,Video delete is not supported by Google Veo.rrs r&transform_video_delete_responsez1GeminiVideoConfig.transform_video_delete_responses ""PQQr(ctdNz2video create character is not supported for Geminir)r?rrrtrmrks r&(transform_video_create_character_requestz:GeminiVideoConfig.transform_video_create_character_request&!"VWWr(ctdrrrs r&)transform_video_create_character_responsez;GeminiVideoConfig.transform_video_create_character_response)rr(ctdNz/video get character is not supported for Geminir)r? character_idrtrmrks r&%transform_video_get_character_requestz7GeminiVideoConfig.transform_video_get_character_request,!"STTr(ctdrrrs r&&transform_video_get_character_responsez8GeminiVideoConfig.transform_video_get_character_response/rr(c tdNz&video edit is not supported for Geminir) r?rCrrtrmrk video_filerprefetched_source_datas r&transform_video_edit_requestz.GeminiVideoConfig.transform_video_edit_request2s""JKKr(ctdrr)r?rrrrs r&transform_video_edit_responsez/GeminiVideoConfig.transform_video_edit_response?s""JKKr(ctdNz+video extension is not supported for Geminir)r?rCrrErtrmrkrs r&!transform_video_extension_requestz3GeminiVideoConfig.transform_video_extension_requestHs""OPPr(ctdrrrs r&"transform_video_extension_responsez4GeminiVideoConfig.transform_video_extension_responseTs!"OPPr( error_message status_codec$ddlm}||||S)N) GeminiError)rmessagerk) common_utilsr)r?rrrkrs r&get_error_classz!GeminiVideoConfig.get_error_classWs.#!  r()NNr<)NNNN)NNN)4__name__ __module__ __qualname____doc__r:rr.__annotations__r>listrHrboolrr_rOrPrrsr|tuplerrhttpxResponseLiteLLMLoggingObjrrrrrbytesrrrrRrrrrrrrrrrrrHeadersrr __classcell__)r@s@r&r4r4Cs :&S#XIII9&F99 9 c3h 9v I# I#* Id 0#6: t  -t3   >*  ,1*1*1* 1* /3 1* - 1*1* t\3& '1*p+/$( >>nn>' > !4Z > Tk > >@-    sDy  .+/ 1nn1'1!4Z 1  1r# &$&$&$- &$  &$ t &$ sDy &$P $nn $' $  $$-1     -    cNT)  sDy  (+/ QnnQ'Q!4Z Q  Q!  -1  -   Tz  Tz  Tz #s(^d*  sDy  ,+/ PnnP'P!4Z P c3h P      -    sDy   RnnR'R  RXXUU# L"! L" QQ S s TTYTaTaMa fv r(r4)*rtypingrrrr httpx._typesrrqlitellm.constantsrlitellm.images.utilsr+litellm.llms.base_llm.videos.transformationr litellm.secret_managers.mainr litellm.types.llms.geminir r r rlitellm.types.routerrlitellm.types.videos.mainrrlitellm.types.videos.utilsrr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjbase_llm.chat.transformationr_BaseLLMExceptionrrr.r'r2r4rGr(r&rs ,, %C6G7 6S XU*(I4S>I($S#X$4Z$[ [ r(