h&HddlZddlmZmZmZmZmZddlmZGddZ y)N)DictAnyOptionalUnionList)UnauthorizedErrorcBeZdZddedeefdZdeeeffdZ ddeedeed eed eed eed eed eedee dee de de eee fe jffdZ ddeeedeeeefdeedeed eed eed eedeedeeee fde de eee fe jffdZ d deeedeeede de eee fe jffdZd!dede de eee fe jffdZy)"KeysManagementClientNbase_urlapi_keyc>|jd|_||_y)a Initialize the KeysManagementClient. Args: base_url (str): The base URL of the LiteLLM proxy server (e.g., "http://localhost:8000") api_key (Optional[str]): API key for authentication. If provided, it will be sent as a Bearer token. /N)rstrip _base_url_api_key)selfr r s U/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/client/keys.py__init__zKeysManagementClient.__init__s"- returncJddi}|jrd|j|d<|S)z Get the headers for API requests, including authorization if api_key is set. Returns: Dict[str, str]: Headers to use for API requests z Content-Typezapplication/jsonzBearer Authorization)r)rheaderss r _get_headersz!KeysManagementClient._get_headerss0"#56 ==)0'@GO $rpagesizeuser_idteam_idorganization_idkey_hash key_aliasreturn_full_objectinclude_team_keysreturn_requestc |jd} i} ||| d<||| d<||| d<||| d<||| d<||| d<||| d <|t|j| d <| t| j| d <tjd | |j | } | r| Stj } |j| j}|j|jS#tjj$r*}|jjdk(r t|d}~wwxYw)a List all API keys with optional filtering and pagination. Args: page (Optional[int]): Page number for pagination size (Optional[int]): Number of items per page user_id (Optional[str]): Filter keys by user ID team_id (Optional[str]): Filter keys by team ID organization_id (Optional[str]): Filter keys by organization ID key_hash (Optional[str]): Filter by specific key hash key_alias (Optional[str]): Filter by key alias return_full_object (Optional[bool]): Whether to return the full key object include_team_keys (Optional[bool]): Whether to include team keys in the response return_request (bool): If True, returns the prepared request object instead of executing it Returns: Union[Dict[str, Any], requests.Request]: Either the response from the server or a prepared request object if return_request is True. The response contains a list of API keys with their configurations. Raises: UnauthorizedError: If the request fails with a 401 status code requests.exceptions.RequestException: If the request fails with any other error z /key/listNrrrrr r!r"r#r$GET)rparams)rstrlowerrequestsRequestrSessionsendprepareraise_for_statusjson exceptions HTTPErrorresponse status_coder )rrrrrr r!r"r#r$r%urlr(requestsessionr5es rlistzKeysManagementClient.listspJ *!#  !F6N  !F6N   'F9    'F9   &(7F$ %  !)F:   "+F;   )+./A+B+H+H+JF' (  (*-.?*@*F*F*HF& '""5#t7H7H7JSYZ N""$ ||GOO$56H  % % '==? """,, zz%%,'**  s?>C>>E%EEmodelsaliasesspendduration budget_idconfigc 4|jd} i} ||| d<||| d<||| d<||| d<||| d<||| d<||| d <||| d <| | | d <tjd | |j| } | r| Stj} |j | j }|j|jS#tjj$r*}|jjdk(r t|d}~wwxYw)a Generate an API key based on the provided data. Docs: https://docs.litellm.ai/docs/proxy/virtual_keys Args: models (Optional[List[str]]): List of allowed models for this key aliases (Optional[Dict[str, str]]): Model alias mappings spend (Optional[float]): Maximum spend limit for this key duration (Optional[str]): Duration for which the key is valid (e.g. "24h", "7d") key_alias (Optional[str]): Alias/name for the key for easier identification team_id (Optional[str]): Team ID to associate the key with user_id (Optional[str]): User ID to associate the key with budget_id (Optional[str]): Budget ID to associate the key with config (Optional[Dict[str, Any]]): Additional configuration parameters return_request (bool): If True, returns the prepared request object instead of executing it Returns: Union[Dict[str, Any], requests.Request]: Either the response from the server or a prepared request object if return_request is True Raises: UnauthorizedError: If the request fails with a 401 status code requests.exceptions.RequestException: If the request fails with any other error z /key/generateNr<r=r>r?r"rrr@rAPOSTrr2r)rr,r-rr.r/r0r1r2r3r4r5r6r )rr<r=r>r?r"rrr@rAr%r7datar8r9r5r:s rgeneratezKeysManagementClient.generateisIL .!  #DN  %DO  !DM  'D   )D   %DO  %DO  )D   #DN""638I8I8KRVW N""$ ||GOO$56H  % % '==? """,, zz%%,'**  s>CD-%DDkeys key_aliasesc|jd}||d}tjd||j|}|r|Stj} |j |j }|j|jS#tjj$r*} | jjdk(r t| d} ~ wwxYw)a Delete existing keys Args: keys (List[str]): List of API keys to delete key_aliases (List[str]): List of key aliases to delete return_request (bool): If True, returns the prepared request object instead of executing it Returns: Union[Dict[str, Any], requests.Request]: Either the response from the server or a prepared request object if return_request is True Raises: UnauthorizedError: If the request fails with a 401 status code requests.exceptions.RequestException: If the request fails with any other error z /key/delete)rHrIrCrDr)NrE) rrHrIr%r7rFr8r9r5r:s rdeletezKeysManagementClient.deletes, ,&  ""638I8I8KRVW N""$ ||GOO$56H  % % '==? """,, zz%%,'**  s>BC1%CCkeyc|jd|}tjd||j}|r|Stj} |j |j }|j|jS#tjj$r*}|jjdk(r t|d}~wwxYw)a? Get information about API keys. Args: key (str): The key hash to get information about return_request (bool): If True, returns the prepared request object instead of executing it Returns: Union[Dict[str, Any], requests.Request]: Either the response from the server or a prepared request object if return_request is True Raises: UnauthorizedError: If the request fails with a 401 status code requests.exceptions.RequestException: If the request fails with any other error z/keys/info?key=r')rr)NrE)rrLr%r7r8r9r5r:s rinfozKeysManagementClient.infos u5""5#t7H7H7JK N""$ ||GOO$56H  % % '==? """,, zz%%,'**  s>BC-%CC)N) NNNNNNNNNF)NNF)F)__name__ __module__ __qualname__r*rrrrintboolrrr,r-r;rfloatrGrKrNrrr r s    x}   d38n #"!%!%)-"&#'-1,0$IsmIsmI# I # I "# I3-IC=I%TNI$D>II tCH~x/// 0IZ'+,0!%"&#'!%!%#'+/$Ic#I$sCx.)I I 3- I C= I#I#IC=Ic3h(II tCH~x/// 0IZ%)+/$ *tCy!*d3i(* * tCH~x/// 0 *XTeDcNT\TdTdDd>err ) r,typingrrrrrr3r r rUrrrWs33)wwr