o ưiXE@sddlmZmZmZmZmZmZddlZddlZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZddlmZddlZdd lmZmZmZmZdd lmZdd lm Z erodd l!m"Z#d dl$m%Z&e#Z'e&Z%neZ'eZ%dee(e(ffddZ)Gddde Z*dS)) TYPE_CHECKINGAnyDictOptionalTupleUnionN) RequestFiles) VideoCreateOptionalRequestParams VideoObject)GenericLiteLLMParams)get_secret_str)encode_video_id_with_providerextract_original_video_id)ImageEditRequestUtils)"GeminiLongRunningOperationResponseGeminiVideoGenerationInstanceGeminiVideoGenerationParametersGeminiVideoGenerationRequest)%DEFAULT_GOOGLE_VIDEO_DURATION_SECONDS)BaseVideoConfig)Logging)BaseLLMExceptionreturncCs@t|}t|dr|d|}t|d}||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)ZbytesBase64EncodedZmimeType)rZget_image_content_typehasattrrreadbase64 b64encodedecode)Z image_fileZ mime_typeZ image_bytesZbase64_encodedr `/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/gemini/videos/transformation.py_convert_image_to_gemini_formats  r"cseZdZdZfddZdedefddZdeded e de ee ffd d Z d ede efd dZ  dAdedede ede edef ddZdede ededefddZdededede dededee eeffddZ  dAdedejdede ede e def d d!Zd"ededededeee ff d#d$Z dBdejdede edefd%d&Z dBd"edededed'e edeee ff d(d)Zdejdedefd*d+Z dBd"ededededed,e e ee fdeee ffd-d.Z dBdejdede edefd/d0Z!    dCdededed1e ed2e e"d3e ed4e e ee fdeee ffd5d6Z# dBdejdede ede eeffd7d8Z$d"ededededeee ff d9d:Z%dejdedefd;d<Z&d=ed>e"de'eej(fde)fd?d@Z*Z+S)DGeminiVideoConfiga Configuration class for Gemini (Veo) video generation. Veo uses a long-running operation model: 1. POST to :predictLongRunning returns operation name 2. Poll operation until done=true 3. Extract video URI from response 4. Download video using file API cstdSN)super__init__)self __class__r r!r&AszGeminiVideoConfig.__init__modelrcCsgdS)z Get the list of supported OpenAI parameters for Veo video generation. Veo supports minimal parameters compared to OpenAI. )r*promptinput_referencesecondssizer )r'r*r r r!get_supported_openai_paramsDsz-GeminiVideoConfig.get_supported_openai_paramsvideo_create_optional_params drop_paramsc Csi}||}dd|D}d|vr|d|d<d|vr/|d}|dur/||}|r/||d<d|vrX|d} zt| trAt| n| } | durK| |d <Wn ttfyWYnw|D]\} } | |vrl| |vrl| || <q\|S) u Map OpenAI-style parameters to Veo format. Mappings: - prompt → prompt - input_reference → image - size → aspectRatio (e.g., "1280x720" → "16:9") - seconds → durationSeconds (defaults to 4 seconds if not provided) All other params are passed through as-is to support Gemini-specific parameters. cSsh|]}|dvr|qS)>r+r*r ).0paramr r r! fs z6GeminiVideoConfig.map_openai_params..r,imager.NZ aspectRatior-durationSeconds)r/_convert_size_to_aspect_ratio isinstancestrint ValueError TypeErroritems) r'r0r*r1Z mapped_paramsZsupported_openai_paramsZopenai_params_to_mapr.Z aspect_ratior-durationkeyvaluer r r!map_openai_paramsQs8   z#GeminiVideoConfig.map_openai_paramsr.cCs"|sdSddddd}||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) Nz16:9z9:16)Z1280x720Z 1920x1080Z720x1280Z 1080x1920)get)r'r.Zaspect_ratio_mapr r r!r7s z/GeminiVideoConfig._convert_size_to_aspect_ratioNheadersapi_keylitellm_paramscCsN|r |jr |p |j}|ptjptdptd}|std||dd|S)z Validate environment and add Gemini API key to headers. Gemini uses x-goog-api-key header for authentication. ZGOOGLE_API_KEYZGEMINI_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)rDlitellmr r;update)r'rCr*rDrEr r r!validate_environments$ z&GeminiVideoConfig.validate_environmentapi_basecCsP|dur tdp d}|r|dkr|dS|dd}|dd|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 NZGEMINI_API_BASEz)https://generativelanguage.googleapis.com/zgemini/z/v1beta/models/z:predictLongRunning)r rstripreplace)r'r*rIrEZ model_nameurlr r r!get_complete_urls    z"GeminiVideoConfig.get_complete_urlr+$video_create_optional_request_paramsc Cslt|d}|}d|vr|ddurt|d} | |d<tdi|} t|g| d} | jdd} | g|fS)a Transform the video creation request for Veo API. Veo expects: { "instances": [ { "prompt": "A cat playing with a ball of yarn" } ], "parameters": { "aspectRatio": "16:9", "durationSeconds": 8, "resolution": "720p" } } )r+r5N)Z instances parametersT)Z exclude_noner )rcopyr"rrZ model_dump) r'r*r+rIrPrErCinstanceZ params_copyZ image_datarQZrequest_body_obj request_datar r r!transform_video_create_requests    z0GeminiVideoConfig.transform_video_create_request raw_response logging_objcustom_llm_providerrTc Cs|}z td i|}Wnty }ztd|d}~ww|j} | s-td||r6t| ||} n| } t| dd|d} i} |ri|di} | dpPt}|durizt || d <Wn tt fyhYnw| | _ | S) 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 for cost calculation z$Failed to parse operation response: Nz#No operation name in Veo response: video processing)idobjectstatusr*rQr6Zduration_secondsr ) jsonr Exceptionr;namer r rBrfloatr<usage)r'r*rVrWrXrT response_dataoperation_responseeoperation_namevideo_id video_objZ usage_datarQr>r r r!transform_video_create_responses> z1GeminiVideoConfig.transform_video_create_responsergcCs(t|}|dd|}i}||fS)z Transform the video status retrieve request for Veo API. Veo polls operations at: GET https://generativelanguage.googleapis.com/v1beta/{operation_name} rK/v1beta/)rrL)r'rgrIrErCrfrNparamsr r r!'transform_video_status_retrieve_request@s z9GeminiVideoConfig.transform_video_status_retrieve_requestc CsR|}tdi|}|j}|j}|rt||d}n|}t|d|s#dndd} | 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..." } } ] } } } NrYrZ completed)r[r\r]r )r^rr`doner r ) r'rVrWrXrcrdrfZis_donergrhr r r!(transform_video_status_retrieve_responseSs  z:GeminiVideoConfig.transform_video_status_retrieve_responsevariantcCst|}|dd|}tj}|j||d} | | } tdi| } | js.t d| j s5t d| j j j } | dj j} i}| |fS) z Transform the video content request for Veo API. For Veo, we need to: 1. Get operation status to extract video URI 2. Return download URL for the video rKrj)rNrCzaVideo generation is not complete yet. Please check status with video_status() before downloading.z'No response data in completed operationrNr )rrLrFZmodule_level_clientrBraise_for_statusr^rrnr;responseZgenerateVideoResponseZgeneratedSamplesrYuri)r'rgrIrErCrprfZ status_urlclientZstatus_responsercrdZgenerated_samplesZ download_urlrkr r r!transform_video_content_requests"  z1GeminiVideoConfig.transform_video_content_requestcCs|jS)zn Transform the Veo video content download response. Returns the video bytes directly. )contentr'rVrWr r r! transform_video_content_responses z2GeminiVideoConfig.transform_video_content_response extra_bodycCtd)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'rgr+rIrErCryr r r!transform_video_remix_requests z/GeminiVideoConfig.transform_video_remix_requestcCrz)zVideo remix is not supported.z+Video remix is not supported by Google Veo.r{r'rVrWrXr r r!transform_video_remix_responsez0GeminiVideoConfig.transform_video_remix_responseafterlimitorder extra_querycCrz)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'rIrErCrrrrr r r!transform_video_list_requests z.GeminiVideoConfig.transform_video_list_requestcCrz)zVideo list is not supported.z*Video list is not supported by Google Veo.r{r~r r r!transform_video_list_responserz/GeminiVideoConfig.transform_video_list_responsecCrz)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'rgrIrErCr r r!transform_video_delete_requests z0GeminiVideoConfig.transform_video_delete_requestcCrz)zVideo delete is not supported.z,Video delete is not supported by Google Veo.r{rwr r r!transform_video_delete_responsesz1GeminiVideoConfig.transform_video_delete_response error_message status_codecCsddlm}||||dS)N) GeminiError)rmessagerC)Z common_utilsr)r'rrrCrr r r!get_error_classs z!GeminiVideoConfig.get_error_class)NNr$)NNNN),__name__ __module__ __qualname____doc__r&r9listr/r boolrrrArr7dictr rHrOrrrUhttpxResponseLiteLLMLoggingObjr rirlrorubytesrxr}rr:rrrrrZHeadersrr __classcell__r r r(r!r#6s    8 "   2 =   9  )            r#)+typingrrrrrrrrZ httpx._typesrZlitellm.types.videos.mainr r Zlitellm.types.routerr Zlitellm.secret_managers.mainr Zlitellm.types.videos.utilsr rZlitellm.images.utilsrrFZlitellm.types.llms.geminirrrrZlitellm.constantsrZ+litellm.llms.base_llm.videos.transformationrZ*litellm.litellm_core_utils.litellm_loggingrZ_LiteLLMLoggingObjZbase_llm.chat.transformationrZ_BaseLLMExceptionrr9r"r#r r r r!s,