jC UddlZddlZddlmZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlZddlmZddlmZmZddlmZddlmZmZddlmZdd lmZmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z&dd l'm(Z(ddl)m*Z*ddl+m,Z,e*Z-e*e.d<e d@de/dee/dee dee/dee/dee/dee e/efdee e/efdee e/efdefdZ0e dAde/dee/dee/dee/dee/dee/dee/dee/ded deeeeffd!Z1e d@d"d#de/dee/dee/dee/dee/dee/dee/dee/dee/ded"defd$Z1e d@de/dee/dee dee/dee/dee/dee e/efdee e/efdee e/efde eeeeefffd%Z1e dBd&e/dee/d'ee2d(ee/dee e/efdee e/efdee e/efde e3eeee3fffd)Z4e dCd&e/dee/dee/d'ee2d(ee/dee e/efdee e/efdee e/efde3fd*Z5e dDd&e/de/dee e/efdee e/efdee e/efdef d+Z6e dDd&e/de/dee/dee/dee/d,ed deeeeffd-Z7e dDd"d.d&e/de/dee/dee/dee/d,ed"defd/Z7e dDd&e/de/dee e/efdee e/efdee e/efde eeeeefff d0Z7e dAd1ee/d2ee8d3ee/dee/dee e/efdee e/efdee e/efde efd4Z9e dEd1ee/d2ee8d3ee/dee/dee/dee/d5ed deeee effd6Z:e dEd"d7d1ee/d2ee8d3ee/dee/dee/dee/d5ed"de efd8Z:e dEd1ee/d2ee8d3ee/dee e/efdee e/efdee e/efde e eeeee efffd9Z:e dDd&e/dee e/efdee e/efdee e/efdef d:Z;e dDd&e/dee/dee/dee/d;ed deeeeff d<ZZs rHr6r6ss F)rIc  yrMrN) rrrr r!r"r+rJr)rKr,rIr>s rHr6r6s"rOc  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_paramsrr"optional_paramslitellm_paramsr,r) rrrX(video_generation_optional_request_paramsr,r\ logging_objr#r%r+ _is_asyncrr-rN)r1popr5 isinstancestrjsonloadsrrr rrget_provider_video_configr4 LlmProviders ValueErrorupdater-get_requested_video_generation_optional_param$get_optional_params_video_generationupdate_environment_variablesdictr create_videovalue call_typervideo_generation_handlerDEFAULT_REQUEST_TIMEOUTr<r=)rrrr r!r"r+r,r#r$r%r>r?rSrTr_rUrFr\rArXrYvideo_generation_request_paramsrGs rHr6r6s2JU 17(>(D(D%%==-M5T 3)+'!66::h'    $$77 3 (    sAr?rSrTr_decodedr\video_provider_configvideo_content_request_paramsrGs rH video_contentr|sZJ? 17!?.!4 9 %::*? 3)+66'::h'    $$ 3 (    sDD&& E/E  Ec Kt} tj} d|d<|d}ttf||||||||d|} t j } t| j| } | jd| d{}tj|r |d{}|S|}|S7)7 #t$r}tjd||| |d}~wwxYww)a Asynchronously download video content. Parameters: - `video_id` (str): The identifier of the video whose content to download - `api_key` (Optional[str]): The API key to use for authentication - `api_base` (Optional[str]): The base URL for the API - `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(Nru)rsrJr)r+r,r#r$r%rvr-) r1r2r3rr|r7r8r9r:r;r<r4r=)rsrJr)r+r,r#r$r%r>r?r@rBrCrDrErFrGs rHavideo_contentr~s >J' %%'#|  &"*     3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA5B1B-B1"B/#B1(C)B1,C-B1/B11 C:CCCc 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()rsrr+r,r#r$r%Nrvr-) r1r2r3r video_remixr7r8r9r:r;r<r4r=)rsrr+r,r#r$r%r>r?r@rBrCrDrErFrGs rH avideo_remixrs:J! %%'#|    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA0B,>B(?B,B*B,#C$B,'C(B,*B,, C5CCCrc yrMrN rsrr+rJr)rKr,rr>s rHrrrO)rc yrMrNrs rHrrrrOc rt} |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 rSrTNr(FTrUr,rurVz!video remix is not supported for )rsrrvr"rZr) rsrvideo_remix_provider_configr,r\r^r#r%r+r_rr-rN)r1r`r5rarbrcrdrrrrrer4rfrgrhrkrlrrrnrorvideo_remix_handlerrqr<r=)rsrr+r,r#r$r%r>r?rSrTr_rUrFryr\rvideo_remix_request_paramsrGs rHrr0s*JM 17r?r@rArBrCrDrErFrGs rH avideo_listrs7BJ) %%'#|  &+2+C+C:>>*d#C, (A"Aq    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY DB C.C/C CCDCDCC D%C??DDrc yrMrN rrrr+rJr)rKr,rr>s rHrrrO)rc yrMrNrs rHrrrrOc \t} |jd} |jdd} |jdddu} |jdd} | >t| trt j | } | Dcgc] }tdi|c}S|d}tdi|}tjdtj| }|td || j||||d }| jd |jd t!|d| i||t"j$j&| _t+j,||||||| |||xst.| |jd Scc}w#t0$r}tj2d ||| |d}~wwxYw)rRrSrTNr(FTrUrurVz video list is not supported for )rrrrvr"rZr) rrrvideo_list_provider_configr,r\r^r#r$r+r_rr-rN)r1r5r`rarbrcrdrrrrer4rfrgrhrkrlrrrnrorvideo_list_handlerrqr<r=)rrrr+r,r#r$r%r>r?rSrTr_rUitemr\rvideo_list_request_paramsrGs rHrr s,JL 17FC(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()rsr+r,r#r$r%Nrvr-) r1r2r3r video_statusr7r8r9r:r;r<r4r=)rsr+r,r#r$r%r>r?r@rBrCrDrErFrGs rH avideo_statusrss8J! %%'#|    3'#!    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA/B+=B'>B+B)B+"C#B+&C'B+)B++ C4CCCrc yrMrNrsr+rJr)rKr,rr>s rHrrrO)rc yrMrNrs rHrrrrOc nt} |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}%") ``` rSrTNr(FTrUr,rurVz"video status is not supported for rsrvr"rZr) rsvideo_status_provider_configr,r\r^r#r%r+r_rr-rN)r1r5r`rarbrcrdrrrrrer4rfrgrhrkrlrvideo_retrievernrorvideo_status_handlerrqr<r=)rsr+r,r#r$r%r>r?rSrTr_rUrFryr\rvideo_status_request_paramsrGs rHrrsTJK 17rsw QQQ N=f.745SGID(:';$; +/! /3,0+/P P C=P i(P c] P 3- P 3- P DcN+P $sCx.)P c3h(P P P l  %) !"!%  C=c] 3-  3- c]sm#t}sC$% $  %)! !"!%).  C=c]c]  3-  3- c]sm#u~ * +/! /3,0+/n n C=n i(n c] n 3- n 3- n DcN+n $sCx.)n c3h(n   c3 #$&n n b##)-/3,0+/l l sml e_l "# l DcN+l $sCx.)l c3h(l    c3o l l `""#)-/3,0+/F F c]F smF e_ F "# F DcN+F $sCx.)F c3h(F  F F R /3,0+/> > > DcN+> $sCx.)> c3h(> > > H  !"!%    c]  sm  # $- sC$%     !"!% $)    c]  sm  # %.    " /3,0+/b b b DcN+b $sCx.)b c3h(b   c3 #$&b b L! /3,0+/J C=J C=J  C=J c] J DcN+J $sCx.)J c3h(J  +J J `  !"!%  C=  C=  C=  c]  sm #  sCk**+      !"!% #(  C=  C=  C=  c]  sm #   +   $ /3,0+/b C=b C=b  C=b DcN+b $sCx.)b c3h(b   c3[))*,b b L /3,0+/= = DcN+ = $sCx.)= c3h(= = = F  !"!%  c] sm  # 4= sC$%     !"!% %*  c] sm  # 5>      /3,0+/u u DcN+ u $sCx.)u c3h(u   c3 #$&u u rO