uPjW<UdZddlZddlmZddlmZddlmZmZm Z ddl Z ddl Z ddl m Z ddlmZmZmZmZddlmZmZmZdd lmZer dd lmZdd lmZeeedZe ee e!fe"d <eezeze#zee e$fzZ%d efdZ&de d edzfdZ'de de dee e fd e fdZ(de)dee efd ee e ffdZ*de jVd e fdZ,de jVde-de d e%fdZ.dedee efd e/fdZ0Gd d!Z1e1Z2e e"d"<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) Coroutine)Path) TYPE_CHECKINGAnyFinal)encode_url_path_segment)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 p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/custom_httpx/container_handler.py_load_endpoints_configr/sRh..55<<|KN^^K 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'6sA*,F;' F } ,0F-0WO api_base path_template path_paramsc<|jd}|jD](\}}t||}|jd|d|}*t j |}|j jd|}|j|}t|S)zBuild the full URL by substituting path parameters. The api_base from get_complete_url already includes /containers and may include query parameters. We need to parse the URL, append the path, then preserve the query parameters. z /containers) field_name{}/)path) removeprefixitemsrreplacehttpxURLr1rstrip copy_withstr) r)r*r+paramvalue encoded_value parsed_basenew_path final_urls r _build_urlr@?s"..}=M$))+M u/%H %--5'nmL M 8,K%))0056}oFH#,,(,;I y>r(query_param_nameskwargsci}|D]6}|j|}|t|ts t|n|||<8|S)z#Build query parameters from kwargs.)get isinstancer9)rArBparams param_namer;s r_build_query_paramsrH_sN F'U  :&  3=eS3IUuF: U Mr(responsec |j}t|trEt|j dtr&|dj d}t|t r|S|jS#t$r|jcYSwxYw)Nerrormessage)r ValueErrortextrEdictrDr9)rIbodyrLs r_error_message_from_responserQlsxmmo$*TXXg->"Eg**95 gs #N == }}sA33B  B returns_binaryresponse_type_namec ddlm}tjj |j r1||j t |t|j|r |jS|j}d|vrQ||j |jdijdt|t|jtj|}|r|di|S|S)Nr)BaseLLMException) status_coderLheadersrKrL))litellm.llms.base_llm.chat.transformationrUr5codesis_errorrVrQrOrWcontentrrDr9r)rIrRrSrU response_json response_types r_transform_responser_zs K {{H001 ,,0:))*  #==?M- ,,!%%gr266y#mBTU))*  *--.@AM-}-- r(filerWcddlm}||}|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_datafilenamer`r\r( content_typezapplication/octet-streamz content-typeNz Content-Type)8litellm.litellm_core_utils.prompt_templates.common_utilsrbrDcopypop) r`rWrb extractedrcr\rdfiles headers_copys r_prepare_multipart_file_uploadrks).ImmJ/96H]]9-4G#--7U;ULXw =>E",,.L^T*^T* , r(ceZdZdZ ddedddeddd eeefdzd eeefdzd ee jzd e d e e zdzdeeeeefzfdZ ddedddeddd eeefdzd eeefdzd ee jzd e e zdzdefdZ ddedddeddd eeefdzd eeefdzd ee jzd e e zdzdefdZy)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 rnrorprrrsrtrvrX) _async_handle _sync_handle) selfr rnrorprrrsrtrurvrBs rhandlezGenericContainerHandler.handles8 %4%% +*C-'+'   !t   '&?)#'#    r(c  Zt|} | 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}|xsd} |dk(r| j || |}np|dk(r| j|| |}nV|dk(rC|r,d| vr(t| d| \}} | j!|| ||}n#| j!|| |}ntd|t#||| dScc}w#t$$r}|d}~wwxYw)zSynchronous request handler.Unknown endpoint: N ssl_verify)rFapi_keyrWrr)r)ror+r1 query_paramsr)rWrFinputradditional_argsmethodrRF is_multipartGETurlrWrFDELETEPOSTr`rrWrFriUnsupported HTTP method: r^rIrRrS)r'rMrEr r rDvalidate_environmentupdateget_complete_urlrOr@rHpre_callupperdeleterkpostr_ Exceptionrzr rnrorprrrsrtrvrBendpoint_config http_clientrWr)pr+rrrrRreffective_paramsrIries rryz$GenericContainerHandler._sync_handles"6m!D1-AB B >FK!@+L.BTBTUacgBh3ijK K,@@!'R"&&y$7A   NN= )3DD#'' D9/E  =LF4D!E0$1188$n&8&8t&LMK !K,@@!'R"&&y$7A   NN= )3DD#'' D9/E  =Ldf vh G sC"I0$I>BI0I1I2!II:III+I,#II0IIII I-&I((I--I0)NNXFN)NNrN)__name__ __module__ __qualname____doc__r9rrOrfloatr5Timeoutboolr r rr{ryrxrXr(rrmrms04-1),8<3 3 $93 - 3 ) 3 CH~, 3 #s(^d*3 &3 3 ..53  yc3' '3 v04-1),8<WW$9W- W ) W CH~, W#s(^d*W&W..5W W~04-1),8<ZZ$9Z- Z ) Z CH~, Z#s(^d*Z&Z..5Z Zr(rmgeneric_container_handler)3rrcollections.abcrpathlibrtypingrrrr5r$litellm.litellm_core_utils.url_utilsr&litellm.llms.custom_httpx.http_handlerr r r r litellm.types.containers.mainr rrlitellm.types.routerr*litellm.litellm_core_utils.litellm_loggingrrq/litellm.llms.base_llm.containers.transformationrrrOr9type__annotations__bytesobjectContainerEndpointResponserr'r@listrHResponserQrr_tuplerkrmrrXr(rrs %,, H  6WS ";.#>*d39o& 336QQTYY\`adflal\mm t c3h @   cN  #s(^  5>> c nn >  #s(^ 8pph$;#<5