o ưie9@sddlmZddlmZmZmZmZmZmZm Z m Z ddl Z ddl m Z ddlZddlmZddlmZddlmZddlmZdd lmZdd lmZmZdd lmZmZeridd lm Z!d dl"m#Z$e!Z%e$Z#neZ%eZ#GdddeZ&dS))BufferedReader) TYPE_CHECKINGAnyDictListOptionalTupleUnioncastN) RequestFiles)BaseVideoConfig)ImageEditRequestUtils)get_secret_str)CreateVideoRequest)GenericLiteLLMParams) VideoCreateOptionalRequestParams VideoObject)encode_video_id_with_providerextract_original_video_id)LoggingBaseLLMExceptioncseZdZdZfddZdedefddZdeded e de fd d Z dCd e dede ede ede f ddZdede ede defddZdededede ded e dee eeffddZ dCdedejdede ede e def ddZ dDdededed e d e edeee ff d!d"Z dDdedededed e d#e e eefdeee ffd$d%Zdejdedefd&d'Z dDdejdede edefd(d)Z dEdeded e d*e ed+e ed,e ed-e e eefdeee ffd.d/Z dDdejdede ede eeffd0d1Z!dededed e deee ff d2d3Z"dejdedefd4d5Z#dededed e deee ff d6d7Z$ dDdejdede edefd8d9Z%d:ed;ed e&e ej'fde(fde*eeefd?ed@edd fdAdBZ+Z,S)FOpenAIVideoConfigz: Configuration class for OpenAI video generation. cstdSN)super__init__)self __class__`/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/openai/videos/transformation.pyr$szOpenAIVideoConfig.__init__modelreturncCsgdS)zS Get the list of supported OpenAI parameters for video generation. )r"promptinput_referencesecondssizeuser extra_headersr )rr"r r r!get_supported_openai_params'sz-OpenAIVideoConfig.get_supported_openai_paramsvideo_create_optional_params drop_paramscCst|S)z:No mapping applied since inputs are in OpenAI spec already)dict)rr+r"r,r r r!map_openai_params5sz#OpenAIVideoConfig.map_openai_paramsNheadersapi_keylitellm_paramscCsD|r |jr |p |j}|ptjptjptd}|dd|i|S)NZOPENAI_API_KEY AuthorizationzBearer )r0litellmZ openai_keyrupdate)rr/r"r0r1r r r!validate_environment>s   z&OpenAIVideoConfig.validate_environmentapi_basecCs|durd}|ddS)zC Get the complete URL for OpenAI video generation. Nzhttps://api.openai.com/v1/z/videos)rstrip)rr"r6r1r r r!get_complete_urlVs z"OpenAIVideoConfig.get_complete_urlr$$video_create_optional_request_paramsc Csrdd|D}td||d|}tt|}|d} dd|D} g} | dur4|j| | dd| | |fS) zF Transform the video creation request for OpenAI API. cSi|] \}}|dvr||qS))r"r)r$r .0kvr r r! qs zDOpenAIVideoConfig.transform_video_create_request..)r"r$r%cSr;))r%r r<r r r!r@sN) files_listimage field_namer )itemsrr rget_add_image_to_files) rr"r$r6r:r1r/Zvideo_create_requestZ request_dictZ_input_referenceZdata_without_filesrAr r r!transform_video_create_requestds,    z0OpenAIVideoConfig.transform_video_create_request raw_response logging_objcustom_llm_provider request_datac Cs||}tdi|}|r|jrt|j|||_i}|r9t|dr9|jr9z t|j|d<Wn ttfy8Ynw||_ |S)z-Transform the OpenAI video creation response.r&duration_secondsNr jsonridrhasattrr&float ValueError TypeErrorusage) rr"rHrIrJrK response_data video_obj usage_datar r r!transform_video_create_responses  z1OpenAIVideoConfig.transform_video_create_responsevideo_idvariantc Cs@t|}|dd|d}|dur|d|}i}||fS)z Transform the video content request for OpenAI API. OpenAI API expects the following request: - GET /v1/videos/{video_id}/content - GET /v1/videos/{video_id}/content?variant=thumbnail r7z/contentNz ?variant=rr8) rrYr6r1r/rZoriginal_video_idurldatar r r!transform_video_content_requests z1OpenAIVideoConfig.transform_video_content_request extra_bodyc Cs<t|}|dd|d}d|i} |r| ||| fS)z Transform the video remix request for OpenAI API. OpenAI API expects the following request: - POST /v1/videos/{video_id}/remix r7z/remixr$)rr8r4) rrYr$r6r1r/r`r\r]r^r r r!transform_video_remix_requests  z/OpenAIVideoConfig.transform_video_remix_requestcCs|jS)z5Transform the OpenAI video content download response.)content)rrHrIr r r! transform_video_content_responsesz2OpenAIVideoConfig.transform_video_content_responsec Cs||}tdi|}|r|jrt|j|d|_i}|r9t|dr9|jr9z t|j|d<Wn ttfy8Ynw||_ |S)z< Transform the OpenAI video remix response. Nr&rLr rM)rrHrIrJrUrVrWr r r!transform_video_remix_responses  z0OpenAIVideoConfig.transform_video_remix_responseafterlimitorder extra_queryc CsV|}i} |durt|| d<|durt|| d<|dur || d<|r'| ||| fS)z Transform the video list request for OpenAI API. OpenAI API expects the following request: - GET /v1/videos Nrerfrg)rstrr4) rr6r1r/rerfrgrhr]paramsr r r!transform_video_list_requests   z.OpenAIVideoConfig.transform_video_list_requestc Cs|}|rqd|vrq|dgD]}t|tr(d|vr(t|d||d|d<q|dg}|drPd}|rFt|dtrF|dd}t|d|||d<|drqd}|rgt|dtrg|dd}t|d|||d<|S)Nr^rOr"Zfirst_idrZlast_id)rNrE isinstancer-r) rrHrIrJrUrVZ data_listZ first_modelZ last_modelr r r!transform_video_list_response+s<    z/OpenAIVideoConfig.transform_video_list_responsecC(t|}|dd|}i}||fS)z Transform the video delete request for OpenAI API. OpenAI API expects the following request: - DELETE /v1/videos/{video_id} r7r[rrYr6r1r/r\r]r^r r r!transform_video_delete_requestTs z0OpenAIVideoConfig.transform_video_delete_requestcCs|}tdi|}|S)z= Transform the OpenAI video delete response. Nr )rNr)rrHrIrUrVr r r!transform_video_delete_responseksz1OpenAIVideoConfig.transform_video_delete_responsecCro)z> Transform the OpenAI video retrieve request. r7r[rpr r r!'transform_video_status_retrieve_requestzs z9OpenAIVideoConfig.transform_video_status_retrieve_requestcCs4|}tdi|}|r|jrt|j|d|_|S)z? Transform the OpenAI video retrieve response. Nr )rNrrOr)rrHrIrJrUrVr r r!(transform_video_status_retrieve_responses   z:OpenAIVideoConfig.transform_video_status_retrieve_response error_message status_codecCsddlm}||||d)Nrr)rvmessager/)base_llm.chat.transformationr)rrurvr/rr r r!get_error_classs z!OpenAIVideoConfig.get_error_classrArBrCcCsFt|}t|tr|||j||ffdS||d||ffdS)zs( (