AHj! UddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"dd l#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-eZ.ee/d<e+ dLde0de e0de e"de e0de e0de e0de ee0efde ee0efde ee0efde&fdZ1e dLde0de e0de e"de e0de e0de e0de2de e0de ee0efde ee0efde ee0efde d d!edeeee&ffd"Z3e dLd#d$de0de e0de e"de e0de e0de e0de2de e0de ee0efde ee0efde ee0efde d#d!ede&fd%Z3e+ dLde0de e0de e"de e0de e0de e0de ee0efde ee0efde ee0efde e&eeee&fffd&Z3e+ dMd'e0de e4de e0d(e e0de ee0efde ee0efde ee0efde e5eeee5fffd)Z6e+ dMd'e0de e4de e0d(e e0de ee0efde ee0efde ee0efde5fd*Z7e+ dNd'e0de0de ee0efde ee0efde ee0efde&f d+Z8e dNd'e0de0de2de e0de ee0efde ee0efde ee0efd,e d d!edeeee&ffd-Z9e dNd#d.d'e0de0de2de e0de ee0efde ee0efde ee0efd,e d#d!ede&fd/Z9e+ dNd'e0de0de ee0efde ee0efde ee0efde e&eeee&fff d0Z9e+ dOd1e e0d2e e2d3e e0d4e e0de ee0efde ee0efde ee0efde e&fd5Z:e dPd1e e0d2e e2d3e e0de2de e0de ee0efde ee0efde ee0efd6e d d!edeeee e&ffd7Z;e dPd#d8d1e e0d2e e2d3e e0de2de e0de ee0efde ee0efde ee0efd6e d#d!ede e&fd9Z;e+ dPd1e e0d2e e2d3e e0de ee0efde ee0efde ee0efde e e&eeee e&fffd:Z;e+ dNd'e0de ee0efde ee0efde ee0efde&f d;Zd'e0de2de e0de ee0efde ee0efde ee0efde+ dNdAe0dBede ee0efde ee0efde ee0efde e$eeee$fff dDZ?e+ dNdEe0de ee0efde ee0efde ee0efde$f dFZ@e+ dNdEe0de ee0efde ee0efde ee0efde e$eeee$fff dGZAe+ dNd'e0de0de ee0efde ee0efde ee0efde&f dHZBe+ dNd'e0de0de ee0efde ee0efde ee0efde e&eeee&fff dIZCe+ dNd'e0de0de0de ee0efde ee0efde ee0efde&fdJZDe+ dNd'e0de0de0de ee0efde ee0efde ee0efde e&eeee&fffdKZEy)QN)partial)Any CoroutineDictListLiteralOptionalUnionoverload)DEFAULT_VIDEO_ENDPOINT_MODEL)request_timeout)get_llm_provider)Logging)BaseVideoConfig)BaseLLMHTTPHandler)base_llm_http_handler)GenericLiteLLMParams) CallTypes FileTypes)CharacterObject VideoCreateOptionalRequestParams VideoObject)decode_video_id_with_provider)ProviderConfigManagerclient)VideoGenerationRequestUtilsllm_http_handlerpromptmodelinput_referencesecondssizeuser extra_headers extra_query extra_bodyreturnc &Kt} tj} d| d<|4tj|xst | j dd\}}}}ttf|||||||||| | d | }tj}t|j|}| jd|d{}tj|r |d{}|S|}|S7)7 #t$r}tj|||| | d}~wwxYww)a) Asynchronously calls the `video_generation` function with the given arguments and keyword arguments. Parameters: - `prompt` (str): Text prompt that describes the video to generate - `model` (Optional[str]): The video generation model to use - `input_reference` (Optional[FileTypes]): Optional image reference that guides generation - `seconds` (Optional[str]): Clip duration in seconds - `size` (Optional[str]): Output resolution formatted as width x height - `user` (Optional[str]): A unique identifier representing your end-user - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Optional[str]): The LLM provider to use - `extra_headers` (Optional[Dict[str, Any]]): Additional headers - `extra_query` (Optional[Dict[str, Any]]): Additional query parameters - `extra_body` (Optional[Dict[str, Any]]): Additional body parameters - `kwargs` (dict): Additional keyword arguments Returns: - `response` (VideoResponse): The response returned by the `video_generation` function. T async_callNapi_baserr*) rrr r!r"r#timeoutcustom_llm_providerr$r%r&rr-original_exceptioncompletion_kwargs extra_kwargs)localsasyncioget_event_looplitellmrr getrvideo_generation contextvars copy_contextrunrun_in_executor iscoroutine Exceptionexception_type)rrr r!r"r#r,r-r$r%r&kwargs local_varsloop_funcctxfunc_with_context init_responseresponsees K/root/tools/cai/cai_env/lib/python3.12/site-packages/litellm/videos/main.pyavideo_generationrJsDJJ, %%'#|  &+2+C+C;;# D9, (A"Aq   + 3'#!   &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY DB*C&8C"9C&C$C&DC&!D"C&$C&& D/D  DDr,r-rJTr?c  yN rrr r!r"r#r,r-r$r%r&rJr?s rIr7r7w"F)rJc  yrLrMrNs rIr7r7rOrPc  t} | jd} | jdd}| jdddu}| jdd}|2t|trt j |}tdi|}|Stdi| }t|xst|\}}}}tj|tj| }|td || j!| t#j$| }t#j&||| }| j)| ||t+|d|i|| t,j.j0| _t5j6||||||| || |xst8|| jd  S#t:$r'}tj<|xst||| | d}~wwxYw)\ Maps the https://api.openai.com/v1/videos endpoint. Currently supports OpenAI litellm_logging_objlitellm_call_idNr)FT mock_response)rr-rproviderz&video generation is not supported for )r video_generation_provider_config video_generation_optional_paramsr?rr#optional_paramslitellm_paramsr-r) rrrY(video_generation_optional_request_paramsr-r] logging_objr$r&r, _is_asyncrr.rM)r2popr6 isinstancestrjsonloadsrrrr rget_provider_video_configr5 LlmProviders ValueErrorupdater-get_requested_video_generation_optional_param$get_optional_params_video_generationupdate_from_kwargsdictr create_videovalue call_typervideo_generation_handlerDEFAULT_REQUEST_TIMEOUTr=r>)rrr r!r"r#r,r-r$r%r&r?r@rTrUr`rVrGr]rBrYrZvideo_generation_request_paramsrHs rIr7r7s,JZ 17  ) , 389L8MN  &! ( U U  ) ( L L1Q1Q  ( .. !@A!?1!4 / )2(>(>(D(D%%==-M5T 3)+'!66::h'    $$77 3 (    sA)rtr,r-rur$r%r&r?r@rTrUr`decodedr]video_provider_configvideo_content_request_paramsrHs rI video_contentr~sTJC 17   ! (>?R>ST  &! . $ ..F# !=>!?.!4 / %::*? 3)+66'::h'    $$ 3 (    sDD(( E1E  Ec Kt} tj} d|d<| t|} | j dxsd}t t f|||||||d|} tj} t | j| } | jd| d{}tj|r |d{}|S|}|S7)7 #t$r}tjd||||d}~wwxYww) ai Asynchronously download video content. Parameters: - `video_id` (str): The identifier of the video whose content to download - `timeout` (Optional[float]): The timeout for the request in seconds - `custom_llm_provider` (Optional[str]): The LLM provider to use - `extra_headers` (Optional[Dict[str, Any]]): Additional headers - `extra_query` (Optional[Dict[str, Any]]): Additional query parameters - `extra_body` (Optional[Dict[str, Any]]): Additional body parameters - `kwargs` (dict): Additional keyword arguments Returns: - `bytes`: The raw video content as bytes Tr)Nr-rw)rtr,r-rur$r%r&rxr.)r2r3r4rr6rr~r8r9r:r;r<r=r5r>)rtr,r-rur$r%r&r?r@rAr{rCrDrErFrGrHs rIavideo_contentrs#8J& %%'#|  &3H=G")++.C"D"P     3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY C9BC C !C?C CC9C C9 C C C6C11C66C9c Kt} tj} d|d<ttf|||||||d|} t j } t| j| } | jd| d{} tj| r | d{}|S| }|S7)7 #t$r}tjd||||d}~wwxYww)a Asynchronously calls the `video_remix` function with the given arguments and keyword arguments. Parameters: - `video_id` (str): The identifier of the completed video to remix - `prompt` (str): Updated text prompt that directs the remix generation - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Optional[str]): The LLM provider to use - `extra_headers` (Optional[Dict[str, Any]]): Additional headers - `extra_query` (Optional[Dict[str, Any]]): Additional query parameters - `extra_body` (Optional[Dict[str, Any]]): Additional body parameters - `kwargs` (dict): Additional keyword arguments Returns: - `response` (VideoObject): The response returned by the `video_remix` function. Tr)rtrr,r-r$r%r&Nrxr.) r2r3r4r video_remixr8r9r:r;r<r=r5r>rtrr,r-r$r%r&r?r@rArCrDrErFrGrHs rI avideo_remixrs:J! %%'#|    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    Y CA0B,>B(?B,B*B,#C$B,'C(B,*B,, C5CCCrc yrLrM rtrr,r-r$r%r&rr?s rIrrrP)rc yrLrMrs rIrr#rrPc tt} |jd} |jdd} |jdddu} |jdd} | 2t| trt j | } tdi| } | S| t|}|jdxsd }tdi|}tjdtj| }|td ||j|||d }| j!|d |jdt#|d| i||t$j&j(| _t-j.|||||| |||xst0| |jd S#t2$r}tj4d ||||d}~wwxYw)zm Maps the https://api.openai.com/v1/videos/{video_id}/remix endpoint. Currently supports OpenAI rTrUNr)FTrVr-rwrWz!video remix is not supported for rtrrxr#r[r) rtrvideo_remix_provider_configr-r]r_r$r&r,r`rr.rM)r2rar6rbrcrdrerrrrrfr5rgrhrirlrmrrrorprvideo_remix_handlerrrr=r>)rtrr,r-r$r%r&r?r@rTrUr`rVrGr{r]rvideo_remix_request_paramsrHs rIrr5s$JN 17  $ ' .@AT@UVW W&!!, " ..F# !;<!?,!4 / )2(=(=(C(C%%88(C 3)+'!66::h'    $$ 3 (    sA)rrrrr,r-r$r%r&r?r@rArBrCrDrErFrGrHs rI avideo_listrs7BJ) %%'#|  &+2+C+C:>>*d#C, (A"Aq    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY DB C.C/C CCDCDCC D%C??DDrc yrLrM rrrr,r-r$r%r&rr?s rIrrrP)rc yrLrMrs rIrrrrPc ^t} |jd} |jdd} |jdddu} |jdd} | >t| trt j | } | Dcgc] }tdi|c}S|d}tdi|}tjdtj| }|td || j||||d }| j|d |jd t!|d| i||t"j$j&| _t+j,||||||| |||xst.| |jd Scc}w#t0$r}tj2d ||| |d}~wwxYw)rSrTrUNr)FTrVrwrWz video list is not supported for )rrrrxr#r[r) rrrvideo_list_provider_configr-r]r_r$r%r,r`rr.rM)r2r6rarbrcrdrerrrrfr5rgrhrirlrmrrrorprvideo_list_handlerrrr=r>)rrrr,r-r$r%r&r?r@rTrUr`rVitemr]rvideo_list_request_paramsrHs rIrrs&JM 17  # & -?@S?TUV V&!+ ! ..F# !:;!?+!4 / )2(<(<(B(B%%77'A 3)+'#66::h'  [Cx  $$ 3 (    s1A4FE?FC)F?F F, F''F,c Kt} tj}d|d<ttf||||||d|} t j } t| j| } |jd| d{} tj| r | d{} | S| } | S7)7 #t$r}tjd||||d}~wwxYww)a Asynchronously retrieve video status from OpenAI's video API. Parameters: - `video_id` (str): The identifier of the video whose status to retrieve - `model` (Optional[str]): The model to use. If not provided, will be auto-detected - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Optional[str]): The LLM provider to use - `extra_headers` (Optional[Dict[str, Any]]): Additional headers - `extra_query` (Optional[Dict[str, Any]]): Additional query parameters - `extra_body` (Optional[Dict[str, Any]]): Additional body parameters - `kwargs` (dict): Additional keyword arguments Returns: - `response` (VideoObject): The response returned by the `video_status` function. Tr))rtr,r-r$r%r&Nrxr.) r2r3r4r video_statusr8r9r:r;r<r=r5r>)rtr,r-r$r%r&r?r@rArCrDrErFrGrHs rI avideo_statusrts8J  %%'#|    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    Y CA/B+=B'>B+B)B+"C#B+&C'B+)B++ C4CCCrc yrLrMrtr,r-r$r%r&rr?s rIrrrP)rc yrLrMrs rIrrrrPc pt} |jd}|jdd} |jdddu} |jdd} | 2t| trt j | } tdi| } | S| t|} | jdxsd }tdi|}tjdtj| }|td ||j|d |i}|j!|d |jdt#|d| i||t$j&j(|_t-j.||||||||xst0| |jd S#t2$r}tj4d ||||d}~wwxYw)a Retrieve video status from OpenAI's video API. Args: video_id (str): The identifier of the video whose status to retrieve. timeout (int): The timeout for the request in seconds. custom_llm_provider (Optional[str]): The LLM provider to use. If not provided, will be auto-detected. extra_headers (Optional[Dict[str, Any]]): Additional headers to include in the request. extra_query (Optional[Dict[str, Any]]): Additional query parameters. extra_body (Optional[Dict[str, Any]]): Additional body parameters. Returns: VideoObject: The video status information. Example: ```python import litellm # Get video status video_status = litellm.video_status( video_id="video_123" ) print(f"Video status: {video_status.status}") print(f"Progress: {video_status.progress}%") ``` rTrUNr)FTrVr-rwrWz"video status is not supported for rtrxr#r[r) rtvideo_status_provider_configr-r]r_r$r&r,r`rr.rM)r2r6rarbrcrdrerrrrrfr5rgrhrirlrmrvideo_retrieverorprvideo_status_handlerrrr=r>)rtr,r-r$r%r&r?r@rTrUr`rVrGr{r]rvideo_status_request_paramsrHs rIrrsNJL 17  % ( /ABUAVWX X&! - # ..F# !<=!?-!4 / )2(@(@(F(F%%99)E 3)+'!66::h'    $$ 3 (    sA)rrr,r-r$r%r&r?r@rArCrDrErFrGrHs rIavideo_create_characterrNsJ$ %%'#|  &"*  "   3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA4B0B,B0!B."B0'C(B0+C,B0.B00 C9CCCc .t} |jd} |jdd} |jdddu} |jdd} | 0t| trt j | } tdi| S|d}tdi|} tjdtj| }|td ||j|d |i}| jd |jd t!|d| i||t"j$j&| _t+j,||||| | ||xst.| |jd S#t0$r}tj2d ||||d}~wwxYw)z` Create a character from an uploaded video file. Maps to POST /v1/videos/characters rTrUNr)FTrVrwrWz,video create character is not supported for rrxr#rr#r\r]r-r) rrr|r-r]r_r$r,r`rr.rM)r2rar6rbrcrdrerrrrfr5rgrhriupdate_environment_variablesrmrrrorprvideo_create_character_handlerrrr=r>)rrr,r-r$r%r&r?r@rTrUr`rVr]provider_configrequest_paramsrHs rIrrsJ= 17    ">?R>ST  &! &~88F# 0-Q.Q 3 9 )2(H(H(N(N%$CC"1 3)+'66::h'    $$ 3 (    sA:E,C$E,, F5FF character_idc Kt} tj}d|d<ttf||||||d|} t j } t| j| } |jd| d{} tj| r | d{} | S| } | S7)7 #t$r}tjd||||d}~wwxYww)zi Asynchronously retrieve a character by ID. Maps to GET /v1/videos/characters/{character_id} Tr))rr,r-r$r%r&Nrxr.) r2r3r4rvideo_get_characterr8r9r:r;r<r=r5r>)rr,r-r$r%r&r?r@rArCrDrErFrGrHs rIavideo_get_characterrsJ  %%'#|   % 3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    rc ,t} |jd}|jdd} |jdddu} |jdd} | 0t| trt j | } tdi| S|d}tdi|} tjdtj| } | td ||j|d |i}|jd |jd t!|d| i||t"j$j&|_t+j,|| || |||xst.| |jd S#t0$r}tj2d ||||d}~wwxYw)zZ Retrieve a character by ID. Maps to GET /v1/videos/characters/{character_id} rTrUNr)FTrVrwrWz)video get character is not supported for rrxr#rr) rr|r-r]r_r$r,r`rr.rM)r2rar6rbrcrdrerrrrfr5rgrhrirrmrrrorprvideo_get_character_handlerrrr=r>)rr,r-r$r%r&r?r@rTrUr`rVr]rrrHs rIrrsJ< 17    ";rs rI avideo_editrUsJ! %%'#|    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    rc nt} |jd} |jdd} |jdddu} |jdd} | 0t| trt j | } tdi| S| t|} | jdxsd }tdi|}tjdtj| }|td ||j|||d }| j!d |jdt#|d| i||t$j&j(| _t-j.|||||| |||xst0| |jd S#t2$r}tj4d ||||d}~wwxYw)zD Create a video edit job. Maps to POST /v1/videos/edits rTrUNr)FTrVr-rwrWz video edit is not supported for rrxr#rr) rrtr|r-r]r_r$r&r,r`rr.rM)r2rar6rbrcrdrerrrrrfr5rgrhrirrmrrrorprvideo_edit_handlerrrr=r>)rtrr,r-r$r%r&r?r@rTrUr`rVr{r]rrrHs rIrrsJ= 17    "?@S?TUV V&!,4G88F# 0-Q.Q 3 9 )2(<(<(B(B%$77"1 3)+'!66::h'    $$ 3 (    sA:F DF F4F//F4c Kt} tj} d|d<ttf||||||||d|} t j } t| j| } | jd| d{}tj|r |d{}|S|}|S7)7 #t$r}tjd||| |d}~wwxYww)zY Asynchronously create a video extension. Maps to POST /v1/videos/extensions Tr))rtrr!r,r-r$r%r&Nrxr.) r2r3r4rvideo_extensionr8r9r:r;r<r=r5r>)rtrr!r,r-r$r%r&r?r@rArCrDrErFrGrHs rIavideo_extensionrs J" %%'#|    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA1B-?B)B-B+B-$C%B-(C)B-+B-- C6CCCc rt} |jd} |jdd} |jdddu} |jdd} | 0t| trt j | } tdi| S| t|}|jdxsd }tdi|}tjdtj| }|td || j||||d }| j!d |jdt#|d| i||t$j&j(| _t-j.||||||| |||xst0| |jd S#t2$r}tj4d ||| |d}~wwxYw)zJ Create a video extension. Maps to POST /v1/videos/extensions rTrUNr)FTrVr-rwrWz%video extension is not supported for )rtrr!rxr#rr) rrtr!r|r-r]r_r$r&r,r`rr.rM)r2rar6rbrcrdrerrrrrfr5rgrhrirrmrrrorprvideo_extension_handlerrrr=r>)rtrr!r,r-r$r%r&r?r@rTrUr`rVr{r]rrrHs rIrrs JD 17    "78K7LM  &!   88F# 0-Q.Q 3 9 )2(A(A(G(G%$<<"1 3)+'!66::h'    $$ 3 (    sA:FDF F6F11F6) NNNNNXNNNN)NNNNNN)rNNNN) NNNNrNNNN)NNNrNNNN)Fr3r8rd functoolsrtypingrrrrrr r r r5litellm.constantsr r rr1litellm.litellm_core_utils.get_llm_provider_logicr*litellm.litellm_core_utils.litellm_loggingrLiteLLMLoggingObj+litellm.llms.base_llm.videos.transformationr*litellm.llms.custom_httpx.llm_http_handlerr litellm.mainrlitellm.types.routerrlitellm.types.utilsrrlitellm.types.videos.mainrrrlitellm.types.videos.utilsr litellm.utilsrrlitellm.videos.utilsrr__annotations__rcrJintr7floatbytesr~rrrrrrrrrrrrrrrrMrPrIrs QQQ:HNSGI.54 E7<(:';$; +/! /3,0+/Q Q C=Q i(Q c] Q 3- Q 3- Q DcN+Q $sCx.)Q c3h(Q Q Q n  +/!)-.2,0+/  C=i(c]  3-  3- "#DcN+$sCx.)c3h(t}sC$% &  +/!)-.2,0+/).  C=i(c]  3-  3- "#DcN+$sCx.)c3h(u~ * +/! /3,0+/p p C=p i(p c] p 3- p 3- p DcN+p $sCx.)p c3h(p  ; #sK"78 9:p p f $)-!/3,0+/m m e_m "#m c] m DcN+m $sCx.)m c3h(m  5)CeO, -.m m b $)-!/3,0+/B B e_B "#B c] B DcN+B $sCx.)B c3h(B  B B L /3,0+/> > > DcN+> $sCx.)> c3h(> > > H )-.2,0+/     "#  DcN+  $sCx.) c3h( $-  sC$%    )-.2,0+/ $)     "#  DcN+  $sCx.) c3h( %.     " /3,0+/` ` ` DcN+` $sCx.)` c3h(`  ; #sK"78 9:` ` H! /3,0+/J C=J C=J  C=J c] J DcN+J $sCx.)J c3h(J  +J J ` )-.2,0+/  C=  C=  C=   "#  DcN+ $sCx.) c3h(   sCk**+     )-.2,0+/ #(  C=  C=  C=   "#  DcN+ $sCx.) c3h(    +   $ /3,0+/` C=` C=`  C=` DcN+` $sCx.)` c3h(`  4 iS${2C(CD EF` ` H /3,0+/< < DcN+ < $sCx.)< c3h(< < < D )-.2,0+/    "# DcN+  $sCx.)  c3h( 4=  sC$%    )-.2,0+/ %*    "# DcN+  $sCx.)  c3h( 5>       /3,0+/s s DcN+ s $sCx.)s c3h(s  ; #sK"78 9:s s l .2,0+/3 3 3 DcN+ 3 $sCx.) 3 c3h(3 3 3 l .2,0+/L L L DcN+ L $sCx.) L c3h(L  ?Ic3&?@ @AL L ^ .2,0+/ . . DcN+ . $sCx.) . c3h( . . . b .2,0+/ J J DcN+ J $sCx.) J c3h( J  ?Ic3&?@ @AJ J Z .2,0+/0 0 0 DcN+ 0 $sCx.) 0 c3h(0 0 0 f .2,0+/L L L DcN+ L $sCx.) L c3h(L  ; #sK"78 89L L ^ .2,0+/2 2 2 2 DcN+ 2 $sCx.)2 c3h(2 2 2 j .2,0+/T T T T DcN+ T $sCx.)T c3h(T  ; #sK"78 89T T rP