o ưiΘ@svUddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"dd l#m$Z$m%Z%ddl&m'Z'ddl(m)Z)m*Z*ddl+m,Z,eZ-ee.d<e*          dIde/de e/de e"de e/de e/de e/de ee/efde ee/efde ee/efde%fddZ0e         dJde/de e/de e/de e/de e/de e/d e e/d!e e/de d"deeee%ffd#d$Z1e          dId%d&de/de e/de e/de e/de e/de e/de e/d e e/d!e e/de d%de%fd'd$Z1e*          dIde/de e/de e"de e/de e/de e/de ee/efde ee/efde ee/efde e%eeee%fffd(d$Z1e*      dKd)e/d*e e2d+e e/d,e e/de ee/efde ee/efde ee/efde e3eeee3fffd-d.Z4e*      dKd)e/d*e e2d+e e/d,e e/de ee/efde ee/efde ee/efde3fd/d0Z5e*     dLd)e/de/de ee/efde ee/efde ee/efde%f d1d2Z6e     dLd)e/de/de e/d e e/d!e e/d2e d"deeee%ffd3d4Z7e     dLd%d5d)e/de/de e/d e e/d!e e/d2e d%de%fd6d4Z7e*     dLd)e/de/de ee/efde ee/efde ee/efde e%eeee%fff d7d4Z7e*         dJd8e e/d9e e8d:e e/de e/de ee/efde ee/efde ee/efde e%fd;d<Z9e        dMd8e e/d9e e8d:e e/de e/d e e/d!e e/dZ:e        dMd%d?d8e e/d9e e8d:e e/de e/d e e/d!e e/dZ:e*        dMd8e e/d9e e8d:e e/de ee/efde ee/efde ee/efde e e%eeee e%fffdAd>Z:e*     dLd)e/de ee/efde ee/efde ee/efde%f dBdCZ;e     dLd)e/de e/d e e/d!e e/dCe d"deeee%ff dDdEZrRrSZvideo_generation_request_paramsrDrErErFr3s        video_idr*r+variantc Kst}zo|d} |dd} |dddu} |dur&t|} | dp%d}tdi|} tjdt|d }|durBt d || |d |i}| j d |d t |d| i||dt j|||| | |pht||d| |d WSty}z tjd ||||dd}~ww)aN Download video content from OpenAI's video API. Args: 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. If not provided, will be auto-detected. variant (Optional[str]): Which downloadable asset to return. Defaults to the MP4 video. 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: bytes: The raw video content as bytes. Example: ```python import litellm video_bytes = litellm.video_content( video_id="video_123" ) with open("video.mp4", "wb") as f: f.write(video_bytes) ``` rMrNNr'FTr+openairPz,video support download is not supported for rfr"rTr) rfZvideo_content_provider_configr+rUrVr*r#rrWrgr,rE)r.r2rXrrrr]r1r^r_r`rarbrZvideo_content_handlerrer9r:)rfr*r+rgr#r$r%r;r<rMrNrWdecodedrUZvideo_provider_configZvideo_content_request_paramsrDrErErF video_contentsf-     rkc st}zLt} d|d<|durt|} | dpd}ttf|||||||d|} t} t| j | } | d| IdH}t |rL|IdH}|WS|}|WSt yf}z t jd||||dd}~ww) 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+rh)rfr*r+rgr#r$r%rir,)r.r/r0rr2rrkr4r5r6r7r8r9r1r:)rfr*r+rgr#r$r%r;r<r=rjr?r@rArBrCrDrErErFavideo_contentsN    rlc st}z=t} d|d<ttf|||||||d|} t} t| j| } | d| IdH} t | r=| IdH}|WS| }|WSt yW}z t j d||||dd}~ww)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')rfrr*r+r#r$r%Nrir,) r.r/r0r video_remixr4r5r6r7r8r9r1r:)rfrr*r+r#r$r%r;r<r=r?r@rArBrCrDrErErF avideo_remixsH    rnc KrJrKrE rfrr*rHr(rIr+rnr;rErErFrm  rm)rnc KrJrKrErorErErFrmrpcKs\t}z|d} |dd} |dddu} |dd} | dur5t| tr+t| } tdi| } | WS|durDt|}|dpCd }t di|}t j dt |d }|dur`td |||||d }| jd |dt|d| i||dtjj| _tj|||||| |||pt| |dd WSty}z t jd ||||dd}~ww)zm Maps the https://api.openai.com/v1/videos/{video_id}/remix endpoint. Currently supports OpenAI rMrNNr'FTrOr+rhrPz!video remix is not supported for )rfrrir"rTr) rfrvideo_remix_provider_configr+rUrVr#r%r*rWrr,rE)r.rXr2rYrZr[r\rrrrr]r1r^r_r`rarbrrmrcrdrZvideo_remix_handlerrer9r:)rfrr*r+r#r$r%r;r<rMrNrWrOrCrjrUrqZvideo_remix_request_paramsrDrErErFrm.sx       afterlimitorderc  st} zRt} d| d<|dur tjd| ddd\} }} } ttf|||||||||d | } t }t|j | }| d|IdH}t |rR|IdH}|WS|}|WSt yl}z tjd||| | dd}~ww) a Asynchronously calls the `video_list` function with the given arguments and keyword arguments. Parameters: - `after` (Optional[str]): Identifier for the last item from the previous pagination request - `limit` (Optional[int]): Number of items to retrieve - `order` (Optional[str]): Sort order of results by timestamp. Use asc for ascending order or desc for descending order - `api_key` (Optional[str]): The API key to use for authentication - `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` (Dict[str, Any]): The response returned by the `video_list` function. Tr'Nrir(r)) rrrsrtrHr*r+r#r$r%r,)r.r/r0r1r r2r video_listr4r5r6r7r8r9r:)rrrsrtrHr*r+r#r$r%r;r<r=r>r?r@rArBrCrDrErErF avideo_listsT!     rvc KrJrKrE rrrsrtr*rHr(rIr+rvr;rErErFruru)rvc KrJrKrErwrErErFrurxcKsJt} z|d} |dd} |dddu} |dd} | dur3t| tr+t| } dd | DWS|dur9d }tdi|}tj dt |d }|durUt d || ||||d }| jd|dt|d| i||dtjj| _tj||||||| |||pt| |dd WSty}z t jd||| |dd}~ww)rLrMrNNr'FTrOcSsg|] }tdi|qS)rE)r).0itemrErErF +szvideo_list..rhrPz video list is not supported for )rrrsrtrir"rTr) rrrsrtvideo_list_provider_configr+rUrVr#r$r*rWrr,rE)r.r2rXrYrZr[r\rrr]r1r^r_r`rarbrrurcrdrZvideo_list_handlerrer9r:)rrrsrtr*r+r#r$r%r;r<rMrNrWrOrUr|Zvideo_list_request_paramsrDrErErFru sx       c st}zs  (              V              q    q  E  D         f    P         f C