i>9rUdZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl Z ddl Z ddlmZmZmZmZddlmZmZmZddlmZer ddlmZdd lmZeeedZeee fe d <d efd Z!d ed e efdZ"dededeeefd efdZ#de$deeefd eeeffdZ%dedeeefd e&fdZ'GddZ(e(Z)y)z Generic container file handler for LiteLLM. This module provides a single generic handler that can process any container file endpoint defined in endpoints.json, eliminating the need for individual handler methods. N)Path) TYPE_CHECKINGAny CoroutineDictOptionalTypeUnion)AsyncHTTPHandler HTTPHandler_get_httpx_clientget_async_httpx_client)ContainerFileListResponseContainerFileObjectDeleteContainerFileResponse)GenericLiteLLMParams)Logging)BaseContainerConfigRESPONSE_TYPESreturncttjjjdz dz }t|5}t j |cdddS#1swYyxYw)z0Load the endpoints configuration from JSON file. containerszendpoints.jsonN)r__file__parentopenjsonload) config_pathfs |/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/custom_httpx/container_handler.py_load_endpoints_configr!)sIx.''..55 DGWWK k ayy|   s AA' endpoint_namecTt}|dD]}|d|k(s |d|k(s|cSy)z+Get config for a specific endpoint by name. endpointsname async_nameN)r!)r"configendpoints r _get_endpoint_configr)0s< # %F;' F } ,0F-0WO( api_base path_template path_paramsc|jdr|tdd}|jd|}|jD]\}}|j d|d|}|S)zBuild the full URL by substituting path parameters. The api_base from get_complete_url already includes /containers, so we need to strip that prefix from the path_template. z /containersN/{}) startswithlenrstripitemsreplace)r+r,r-urlparamvalues r _build_urlr:9sp .%c-&8&:; __S ! "=/ 2C#))+ ukkBugR.%0, Jr*query_param_nameskwargsci}|D]6}|j|}|t|ts t|n|||<8|S)z#Build query parameters from kwargs.)get isinstancestr)r;r<params param_namer9s r _build_query_paramsrCNsG F'  :&  3=eS3IUuF: ( Mr*fileheaderscddlm}||}|jdxsd}|jdxsd}|jdxsd}d|||fi}|j}|j d d |j d d ||fS) z Prepare file and headers for multipart upload. Returns: Tuple of (files_dict, headers_without_content_type) r)extract_file_datafilenamerDcontentr* content_typezapplication/octet-streamz content-typeNz Content-Type)8litellm.litellm_core_utils.prompt_templates.common_utilsrGr>copypop) rDrErG extractedrHrIrJfiles headers_copys r _prepare_multipart_file_uploadrQ[s"$'I}}Z(2FHmmI&-#G==0N4NL h6 7E<<>L^T*^T* , r*ceZdZdZ ddedddeddd eeeefd eeeefd e e e jfd e d ee eefde eeeeefffdZ ddedddeddd eeeefd eeeefd e e e jfd ee eefdefdZ ddedddeddd eeeefd eeeefd e e e jfd ee eefdefdZy)GenericContainerHandlerz Generic handler for container file API endpoints. This single handler can process any endpoint defined in endpoints.json, eliminating the need for individual handler methods per endpoint. Nr"container_provider_configrlitellm_params logging_objLiteLLMLoggingObj extra_headers extra_querytimeout _is_asyncclientrc r|r|jd|||||||| d| S|jd|||||||| d| S)a Generic handler for any container file endpoint. Args: endpoint_name: Name of the endpoint (e.g., "list_container_files") container_provider_config: Provider-specific configuration litellm_params: LiteLLM parameters including api_key, api_base logging_obj: Logging object for request logging extra_headers: Additional HTTP headers extra_query: Additional query parameters timeout: Request timeout _is_async: Whether to make async request client: Optional HTTP client **kwargs: Path params and query params (e.g., container_id, file_id, after, limit) )r"rTrUrVrXrYrZr\) _async_handle _sync_handle) selfr"rTrUrVrXrYrZr[r\r<s r handlezGenericContainerHandler.handles8 %4%% +*C-'+'   !t   '&?)#'#    r*c  jt|} | std||t|tst d|j ddi} n|} |j |xsi|j dd} |r| j||j|j ddt|} | j d gDcic]}|| j |d }}t| | d |}t| j d g| }|r|j||jd d || |d | dj}| j dd}| j dd} |dk(r| j || |}np|dk(r| j|| |}nV|dk(rC|r,d| vr(t| d| \}} | j!|| ||}n#| j!|| |}ntd||r |j"S|j%}d|vrYddlm}|j dij dt+|}||j,|t|j.t0j | d}|r|d i|S|Scc}w#t2$r}|d}~wwxYw)!zSynchronous request handler.Unknown endpoint: N ssl_verify)rAapi_keyrErfr+r+rUr-path query_paramsr+rErAinputrfadditional_argsmethodreturns_binaryF is_multipartGETr7rErADELETEPOSTrDr7rErArOUnsupported HTTP method: errorrBaseLLMExceptionmessage status_coder|rE response_typer^)r) ValueErrorr?r r r>validate_environmentupdateget_complete_urldictr:rCpre_callupperdeleterQpostrIr)litellm.llms.base_llm.chat.transformationr{r@r~rEr Exceptionrar"rTrUrVrXrYrZr\r<endpoint_config http_clientrEr+pr-r7rkrprqrrresponserO response_jsonr{ error_msgres r r`z$GenericContainerHandler._sync_handles</}=1-AB B >FK!@+$n&8&8t&LMK!K,@@!'R"&&y$7A   NN= ),==#'' D9/> +:*=*=mR*P *PQAvzz!R *P  ?6#:KH+    3V      , "&  !*002(,,-=uE&**>5A 1 &??W\+8#&--W\.6!Ff$4%Cv&NE7 +//U 0 H +// 0 H!#>s=1 ' ( 4 4% !1!12+../OPM$5}55 U X G s, J4BJ" BJ"J"" J2+J--J2c Kt|} | std||t|ts8t t j jd|jddi} n|} |j|xsi|jdd} |r| j||j|jddt|} | jd gDcic]}|| j|d }}t| | d |}t| jd g| }|r|j||jd d || |d | dj!}| jdd}| jdd} |dk(r| j|| |d{}n|dk(r| j#|| |d{}nf|dk(rS|r4d| vr0t%| d| \}} | j'|| ||d{}n+| j'|| |d{}ntd||r |j(S|j+}d|vrYddlm}|jdijdt1|}||j2|t|j4t6j| d}|r|d i|S|Scc}w72777#t8$r}|d}~wwxYww)!zAsynchronous request handler.rdNre) llm_providerrArfrgr+rhr-rirjrkrlrmrprqFrrrsrtrurvrDrwrxryrrzr|r}rr^)r)rr?r rlitellm LlmProvidersOPENAIr>rrrrr:rCrrrrQrrIrrr{r@r~rErrrs r r_z%GenericContainerHandler._async_handle*sy/}=1-AB B >F4D!E0$1188$n&8&8t&LMK !K,@@!'R"&&y$7A   NN= ),==#'' D9/> +:*=*=mR*P *PQAvzz!R *P  ?6#:KH+    3V      , "&  !*002(,,-=uE&**>5A 1 !,W\"1"8#!,!3!3W\"4"6!Ff$4%Cv&NE7&1%5%5U&6& H&1%5%5&6& H!#>s=1 ' ( 4 4% !1!12+../OPM$5}55 U <  < G sC"K:$K>BK:K'+K,!K' K :K'K# K'%K%& K'K:BK'K:K'K:K' K'#K'%K'' K70K22K77K:)NNXFN)NNrN)__name__ __module__ __qualname____doc__r@rrrrr floathttpxTimeoutboolr r rrbr`r_r^r*r rSrSws(3704/2AE3 3 $93 - 3 ) 3  S#X/ 3 d38n-3 uemm+,3 3 {,<<=>3  sIc3m,, -3 v3704/2AEtt$9t- t ) t  S#X/ td38n-tuemm+,t{,<<=>t tx3704/2AEuu$9u- u ) u  S#X/ ud38n-uuemm+,u{,<<=>u ur*rS)*rrpathlibrtypingrrrrrr r rr&litellm.llms.custom_httpx.http_handlerr r r rlitellm.types.containers.mainrrrlitellm.types.routerr*litellm.litellm_core_utils.litellm_loggingrrW/litellm.llms.base_llm.containers.transformationrrr@__annotations__r!r)r:listrCtuplerQrSgeneric_container_handlerr^r*r rs& MMM   6WS ";.#>#S$Yc3h *   cN  #s(^   #s(^ 8hhX 45r*