o ưi8@s`UdZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl Z ddl Z ddlmZmZmZmZddlmZmZmZddlmZerQddlmZdd lmZeeedZeee fe d <d efd d Z!ded e efddZ"dededeeefd efddZ#de$deeefd eeeffddZ%dedeeefd e&fddZ'GdddZ(e(Z)dS) 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_TYPESreturncCsLttjjjdd}t| }t|WdS1swYdS)z0Load the endpoints configuration from JSON file. containerszendpoints.jsonN)r__file__parentopenjsonload)Z config_pathfrb/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/custom_httpx/container_handler.py_load_endpoints_config)s $r endpoint_namecCs8t}|dD]}|d|ks|d|kr|SqdS)z+Get config for a specific endpoint by name.Z endpointsnameZ async_nameN)r)r configZendpointrrr_get_endpoint_config0s  r#api_base path_template path_paramscCsV|dr |tdd}|d|}|D]\}}|d|d|}q|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&urlparamvaluerrr _build_url9s r2query_param_nameskwargscCs>i}|D]}||}|durt|tst|n|||<q|S)z#Build query parameters from kwargs.N)get isinstancestr)r3r4params param_namer1rrr_build_query_paramsNs r:fileheadersc Cstddlm}||}|dpd}|dpd}|dpd}d|||fi}|}|d d |d d ||fS) z Prepare file and headers for multipart upload. Returns: Tuple of (files_dict, headers_without_content_type) r)extract_file_datafilenamer;content content_typezapplication/octet-streamz content-typeNz Content-Type)Z8litellm.litellm_core_utils.prompt_templates.common_utilsr=r5copypop) r;r<r=Z extractedr>r?rAfilesZ headers_copyrrr_prepare_multipart_file_upload[s   rEc@sZeZdZdZ     ddeddded d d eeeefd eeeefd e e e j fde dee eefde eeeeefffddZ    ddeddded d d eeeefd eeeefd e e e j fdee eefdefddZ    ddeddded d d eeeefd eeeefd e e e j fdee eefdefddZdS)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. NXFr container_provider_configrlitellm_params logging_objLiteLLMLoggingObj extra_headers extra_querytimeout _is_asyncclientrc KsH|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 rHrIrJrLrMrNrPNr) _async_handle _sync_handle) selfr rHrIrJrLrMrNrOrPr4rrrhandles2   zGenericContainerHandler.handlec  s>t|} | s td||dust|ts"td|ddid} n|} |j|p)i|ddd} |r8| ||j|ddt |d} fd d | d gD}t | | d |}t | d g}|rj|||j dd|| |dd| d }| dd}| dd}z|dkr| j|| |d}n:|dkr| j|| |d}n-|dkr|rdvrtd| \}} | j|| ||d}n| j|| |d}ntd||r|jWS|}d|vrddlm}|didt|}||j|t |jd t| d!}|r |d"i|WS|WSty}z|d}~ww)#zSynchronous request handler.Unknown endpoint: N ssl_verify)r8api_keyr<rWr$r$rIci|] }||dqSr5.0pr4rr z8GenericContainerHandler._sync_handle..r&path query_paramsr\r$r<r8inputrWZadditional_argsmethodreturns_binaryF is_multipartGETr/r<r8DELETEPOSTr;r/r<r8rDUnsupported HTTP method: errorrBaseLLMExceptionmessage status_coderur< response_typer)r# ValueErrorr6r r r5validate_environmentupdateget_complete_urldictr2r:pre_callupperdeleterEpostr?r)litellm.llms.base_llm.chat.transformationrtr7rwr<r ExceptionrSr rHrIrJrLrMrNrPr4Zendpoint_config http_clientr<r$r&r/rerirjrkresponserDZ response_jsonrt error_msgrxerrarrRs         z$GenericContainerHandler._sync_handlec  s`t|} | std||dust|ts&ttjjd|ddid} n|} |j |p-i|ddd} |r<| ||j |ddt |d} fd d | d gD}t | | d |}t| d g}|rn| ||jdd|| |dd| d}| dd}| dd}z|dkr| j|| |dIdH}nC|dkr| j|| |dIdH}n3|dkr|rdvrtd| \}} | j|| ||dIdH}n| j|| |dIdH}ntd||r|jWS|}d|vr ddlm}|didt|}||j|t |jd t| d!}|r|d"i|WS|WSty/}z|d}~ww)#zAsynchronous request handler.rUNrV)Z llm_providerr8rWrXr$rYcrZr[r]r^rarrrbArcz9GenericContainerHandler._async_handle..r&rdrer\rfrgrirjFrkrlrmrnror;rprqrrrrsrurvrxr)r#ryr6r r litellmZ LlmProvidersZOPENAIr5rzr{r|r}r2r:r~rrrErr?rrrtr7rwr<rrrrrarrQs          z%GenericContainerHandler._async_handle)NNrGFN)NNrGN)__name__ __module__ __qualname____doc__r7rrrrr floathttpxTimeoutboolr r rrTrRrQrrrrrFws     ;   k   rF)*rrpathlibrtypingrrrrrrr rrZ&litellm.llms.custom_httpx.http_handlerr r r r Zlitellm.types.containers.mainrrrZlitellm.types.routerrZ*litellm.litellm_core_utils.litellm_loggingrrKZ/litellm.llms.base_llm.containers.transformationrrr7__annotations__rr#r2listr:tuplerErFZgeneric_container_handlerrrrrsZ $