o ưiɨ@s ddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZddlmZddlmZdd lmZmZmZmZm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*m+Z+gdZ,e+       dMde-de ee-efde e e-de dde ee-efde ee-efde ee-efde!fddZ.e       dNde-de ee-efde e e-de e-de e-de e-de dde ddeeee!ffdd Z/e       dNd!d"de-de ee-efde e e-de e-de e-de e-de dde d!de!fd#d Z/e+          dOde-de ee-efde e e-de e-de e-de e-de dde ee-efde ee-efde ee-efde e!eeee!fffd$d Z/e+        dPd%e e-d&e e0d'e e-de dde ee-efde ee-efde ee-efde fd(d)Z1e        dQd%e e-d&e e0d'e e-de e-de e-de e-de dd)e ddeeee ffd*d+Z2e        dQd!d,d%e e-d&e e0d'e e-de e-de e-de e-de dd)e d!de fd-d+Z2e+           dRd%e e-d&e e0d'e e-de e-de e-de e-de dde ee-efde ee-efde ee-efde e eeee fffd.d+Z2e+     dSd/e-de dde ee-efde ee-efde ee-efde!f d0d1Z3e     dTd/e-de e-de e-de e-de dd1e ddeeee!ffd2d3Z4e     dTd!d4d/e-de e-de e-de e-de dd1e d!de!fd5d3Z4e+        dUd/e-de e-de e-de e-de dde ee-efde ee-efde ee-efde e!eeee!fffd6d3Z4e+     dSd/e-de dde ee-efde ee-efde ee-efde"f d7d8Z5e     dTd/e-de e-de e-de e-de dd8e ddeeee"ffd9d:Z6e     dTd!d;d/e-de e-de e-de e-de dd8e d!de"fdd?Z7e        dQd/e-d%e e-d&e e0d'e e-de e-de e-de e-de dd?e ddeeeeffd@dAZ8e        dQd!dBd/e-d%e e-d&e e0d'e e-de e-de e-de e-de dd?e d!defdCdAZ8e+           dRd/e-d%e e-d&e e0d'e e-de e-de e-de e-de dde ee-efde ee-efde ee-efde eeeeefffdDdAZ8e+     dSd/e-dEe$de dde ee-efde ee-efde ee-efdefdFdGZ9e     dTd/e-dEe$de e-de e-de e-de ddGe ddeeeeffdHdIZ:e     dTd!dJd/e-dEe$de e-de e-de e-de ddGe d!defdKdIZ:e+        dUd/e-dEe$de e-de e-de e-de dde ee-efde ee-efde ee-efde eeeeefffdLdIZ:dS)VN)partial)Any CoroutineDictListLiteralOptionalUnionoverload)request_timeout)ContainerRequestUtils)Logging)BaseContainerConfig)base_llm_http_handler)$ContainerCreateOptionalRequestParamsContainerFileListResponseContainerFileObject"ContainerListOptionalRequestParamsContainerListResponseContainerObjectDeleteContainerResult) FileTypes)GenericLiteLLMParams) CallTypes)ProviderConfigManagerclient) acreate_containeradelete_containeralist_container_filesalist_containersaretrieve_containeraupload_container_filecreate_containerdelete_containerlist_container_fileslist_containersretrieve_containerupload_container_fileXopenainame expires_afterfile_idscustom_llm_provider extra_headers extra_query extra_bodyreturnc t} z>t} d|d<ttf||||||||d|} t} t| j| } | d| IdH}t |r>|IdH}|WS|}|WSt yX}z t j d||| |dd}~ww)a4Asynchronously calls the `create_container` function with the given arguments and keyword arguments. Parameters: - `name` (str): Name of the container to create - `expires_after` (Optional[Dict[str, Any]]): Container expiration time settings - `file_ids` (Optional[List[str]]): IDs of files to copy to the container - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Optional[Literal["openai"]]): 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` (ContainerObject): The created container object T async_call)r*r+r,timeoutr-r.r/r0Nmodelr-Zoriginal_exceptionZcompletion_kwargs extra_kwargs) localsasyncioget_event_looprr" contextvars copy_contextrunrun_in_executor iscoroutine Exceptionlitellmexception_type)r*r+r,r4r-r.r/r0kwargs local_varsloopfuncctxfunc_with_context init_responseresponseerMN/home/app/Keep/.python/lib/python3.10/site-packages/litellm/containers/main.pyr+sJ    rapi_keyapi_base api_versionTc KdSNrM r*r+r,r4rOrPrQr-rrDrMrMrNr"sr"F)rc KrRrSrMrTrMrMrNr"rUc  Ks:t} z| d} | d}| dddu}| d}|dur3t|tr)t|}tdi|}|WStd|||d| }t j t |d }|durRt d || | t| }tj||d }| jd t|d|i||d tjj| _tj||||| ||pt|dWSty}z t jd ||| | dd}~ww)aCreate a container using the OpenAI Container API. Currently supports OpenAI Example: ```python import litellm response = litellm.create_container( name="My Container", custom_llm_provider="openai", ) print(response) ``` litellm_logging_objlitellm_call_idr3FT mock_responseNrOrPrQproviderz+container operations are not supported for )container_provider_config container_create_optional_paramsr5r7Zoptional_paramslitellm_paramsr-)r*container_create_request_paramsr\r_ logging_objr.r4 _is_asyncr6rM)r9popget isinstancestrjsonloadsrrrget_provider_container_configrB LlmProviders ValueErrorupdater Z-get_requested_container_create_optional_paramZ$get_optional_params_container_createupdate_environment_variablesdictrr"value call_typerZcontainer_create_handlerDEFAULT_REQUEST_TIMEOUTrArC)r*r+r,r4rOrPrQr-r.r/r0rDrErVrWrbrXrKr_r\r]r`rLrMrMrNr"s|#        afterlimitorderc r2)aAsynchronously list containers. Parameters: - `after` (Optional[str]): A cursor for pagination - `limit` (Optional[int]): Number of items to return (1-100, default 20) - `order` (Optional[str]): Sort order ('asc' or 'desc', default 'desc') - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Literal["openai"]): 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` (ContainerListResponse): The list of containers Tr3)rrrsrtr4r-r.r/r0Nr5r6) r9r:r;rr%r<r=r>r?r@rArBrC)rrrsrtr4r-r.r/r0rDrErFrGrHrIrJrKrLrMrMrNr sJ    rc KrRrSrM rrrsrtr4rOrPrQr-rrDrMrMrNr%SrUr%)rc KrRrSrMrurMrMrNr%drUc  Ks&t} zy| d} | d}| dddu}| d}|dur3t|tr)t|}tdi|}|WStd|||d| }t j t |d }|durRt d |t| }| jd t|d|i||d tjj| _tj||| ||||| |pxt|d WSty}z t jd ||| | dd}~ww)zSList containers using the OpenAI Container API. Currently supports OpenAI rVrWr3FTrXNrYrZ2Container provider config not found for provider: r5r^) r\r_rarrrsrtr.r/r4rbr6rM)r9rcrdrerfrgrhrrrrirBrjrkr Z+get_requested_container_list_optional_paramrmrnrr%rorprZcontainer_list_handlerrqrArC)rrrsrtr4rOrPrQr-r.r/r0rDrErVrWrbrXrKr_r\Zcontainer_list_optional_paramsrLrMrMrNr%wst        container_idc t}zr?r@rArBrCrwr4r-r.r/r0rDrErFrGrHrIrJrKrLrMrMrNr F   r cKrRrSrMrwr4rOrPrQr-r rDrMrMrNr& r&)r cKrRrSrMr|rMrMrNr&*r}c  Kt} zn| d} | d} | dddu} | d}|dur3t|tr)t|}tdi|}|WStd|||d| }t j t |d }|durRt d || jd id| i|d tjj| _tj|||| |||pmt| d WSty}z t jd ||| | dd}~ww)zXRetrieve a container using the OpenAI Container API. Currently supports OpenAI rVrWr3FTrXNrYrZrvr5r^rwr\r_rar.r/r4rbr6rM)r9rcrdrerfrgrhrrrrirBrjrkrmrr&rorprZcontainer_retrieve_handlerrqrArCrwr4rOrPrQr-r.r/r0rDrErVrWrbrXrKr_r\rLrMrMrNr&;h       c rx)aLAsynchronously delete a container. Parameters: - `container_id` (str): The ID of the container to delete - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Literal["openai"]): 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` (DeleteContainerResult): The deletion result Tr3ryNr5r6) r9r:r;rr#r<r=r>r?r@rArBrCrzrMrMrNrr{rcKrRrSrMrwr4rOrPrQr-rrDrMrMrNr#r}r#)rcKrRrSrMrrMrMrNr#r}c  Kr~)zVDelete a container using the OpenAI Container API. Currently supports OpenAI rVrWr3FTrXNrYrZrvr5r^rr6rM)r9rcrdrerfrgrhrrrrirBrjrkrmrr#rorprZcontainer_delete_handlerrqrArCrrMrMrNr#rc  st} z?t} d| d<ttf|||||||||d | } t} t| j| }| d|IdH}t |r?|IdH}|WS|}|WSt yY}z t j d||| | dd}~ww)a%Asynchronously list files in a container. Parameters: - `container_id` (str): The ID of the container - `after` (Optional[str]): A cursor for pagination - `limit` (Optional[int]): Number of items to return (1-100, default 20) - `order` (Optional[str]): Sort order ('asc' or 'desc', default 'desc') - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Literal["openai"]): 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` (ContainerFileListResponse): The list of container files Tr3) rwrrrsrtr4r-r.r/r0Nr5r6) r9r:r;rr$r<r=r>r?r@rArBrC)rwrrrsrtr4r-r.r/r0rDrErFrGrHrIrJrKrLrMrMrNrOsL    rc  KrRrSrM rwrrrsrtr4rOrPrQr-rrDrMrMrNr$r$)rc  KrRrSrMrrMrMrNr$rc  Ks t} zv| d}| d}| dddu}| d}|dur3t|tr)t|}tdi|}|WStd|||d| }t j t |d }|durRt d ||jd ||||d d|i|d tjj|_tj|||||||| | |put|d WSty}z t jd ||| | dd}~ww)z]List files in a container using the OpenAI Container API. Currently supports OpenAI rVrWr3FTrXNrYrZrvr5)rwrrrsrtr^) rwr\r_rarrrsrtr.r/r4rbr6rM)r9rcrdrerfrgrhrrrrirBrjrkrmrr$rorprZcontainer_file_list_handlerrqrArC)rwrrrsrtr4rOrPrQr-r.r/r0rDrErVrWrbrXrKr_r\rLrMrMrNr$sn       filec 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 upload a file to a container. This endpoint allows uploading files directly to a container session, supporting various file types like CSV, Excel, Python scripts, etc. Parameters: - `container_id` (str): The ID of the container to upload the file to - `file` (FileTypes): The file to upload. Can be: - A tuple of (filename, content, content_type) - A tuple of (filename, content) - A file-like object with read() method - Bytes - A string path to a file - `timeout` (int): Request timeout in seconds - `custom_llm_provider` (Literal["openai"]): 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` (ContainerFileObject): The uploaded file object Example: ```python import litellm # Upload a CSV file response = await litellm.aupload_container_file( container_id="container_abc123", file=("data.csv", open("data.csv", "rb").read(), "text/csv"), custom_llm_provider="openai", ) print(response) ``` Tr3)rwrr4r-r.r/r0Nr5r6) r9r:r;rr'r<r=r>r?r@rArBrC)rwrr4r-r.r/r0rDrErFrGrHrIrJrKrLrMrMrNr!sH/    r!c KrRrSrM rwrr4rOrPrQr-r!rDrMrMrNr'p r')r!c KrRrSrMrrMrMrNr'rc  Ks$ddlm} t} zr| d} | d}| dddu}| d}|d ur9t|tr/t|}t di|}|WSt d|||d | }t j t |d }|d urXtd || jd d|id|i|dtjj| _| jd||| |||put|||d WSty}z t jd ||| | dd }~ww)aUpload a file to a container using the OpenAI Container API. This endpoint allows uploading files directly to a container session, supporting various file types like CSV, Excel, Python scripts, JSON, etc. This is useful when /chat/completions or /responses sends files to the container but the input file type is limited to PDF. This endpoint lets you work with other file types. Currently supports OpenAI Example: ```python import litellm # Upload a CSV file response = litellm.upload_container_file( container_id="container_abc123", file=("data.csv", open("data.csv", "rb").read(), "text/csv"), custom_llm_provider="openai", ) print(response) # Upload a Python script response = litellm.upload_container_file( container_id="container_abc123", file=("script.py", b"print('hello world')", "text/x-python"), custom_llm_provider="openai", ) print(response) ``` r)generic_container_handlerrVrWr3FTrXNrYrZrvr5rwr^r') Z endpoint_namer\r_rar.r/r4rbrwrr6rM)Z+litellm.llms.custom_httpx.container_handlerrr9rcrdrerfrgrhrrrrirBrjrkrmrr'rorphandlerqrArC)rwrr4rOrPrQr-r.r/r0rDrrErVrWrbrXrKr_r\rLrMrMrNr'sn 0       )NNr(r)NNN)NNr(NNNr)) NNr(NNNr)NNN)NNNr(r)NNN)NNNr(NNNr)) NNNr(NNNr)NNN)r(r)NNN)r(NNNr))r(NNNr)NNN);r:r<rg functoolsrtypingrrrrrrr r rBZlitellm.constantsr rqZlitellm.containers.utilsr Z*litellm.litellm_core_utils.litellm_loggingr ZLiteLLMLoggingObjZ/litellm.llms.base_llm.containers.transformationrZ litellm.mainrZlitellm.types.containers.mainrrrrrrrZlitellm.types.llms.openairZlitellm.types.routerrZlitellm.types.utilsrZ litellm.utilsrr__all__rfrr"intrr%r r&rr#rr$r!r'rMrMrMrNsV (     $       G             u   E          c ?        Y ?        Y  F             ] U