tPj81PddlZddlmZmZddlZddlmZddlmZGddZ y)N)AnyFinal) redact_string)UnauthorizedErrorc eZdZd dededzdefdZdeeeffdZ d!dedzd edzd edzd edzd edzd edzdedzdedzdedzdedeee fe jzfdZ d!de jedzdeeefdzdedzdedzdedzd edzd edzdedzdeee fdzdedeee fe jzfdZ d"de jedzde jedzdedeee fe jzfdZ d#dede jedzdeeefdzdedzdedzdedzd edzd edzdeee fe jzfdZd$dededeee fe jzfdZy)%KeysManagementClientNbase_urlapi_keytimeoutcL|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. timeout (int): Request timeout in seconds (default: 30) /N)rstrip _base_url_api_key_timeout)selfr r r s ^/home/oem/CyberBrain26-unified/.venv/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)*<= ==)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}|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)rparamsr )rstrlowerrequestsRequestrSessionsendpreparerraise_for_statusjson exceptions HTTPErrorresponse status_coder)rrrrrr r!r"r#r$r%urlr(requestsessionr6es rlistzKeysManagementClient.list%sxJ'y1(*  !F6N  !F6N   'F9    'F9   &(7F$ %  !)F:   "+F;   )+./A+B+H+H+JF' (  (*-.?*@*F*F*HF& '!))%d>O>O>QZ`a N!))+ %ll7??+r?r@r"rrrArBPOSTrr3r)r*rr-r.rr/r0r1rr2r3r4r5r6r7r)rr=r>r?r@r"rrrArBr%r8datar9r:r6r;s rgeneratezKeysManagementClient.generatepsQL'}5&(  #DN  %DO  !DM  'D   )D   %DO  %DO  )D   #DN!))&#t?P?P?RY]^ N!))+ %ll7??+rDrEr)zError updating key: ) rr-r.rr/r0r1rtextr2r3 Exception)rrMr=r>r?r@r"rrr8rGr9r: response_textr6s rupdatezKeysManagementClient.updates$>'{3',cl  )D   %DO  %DO  #DN  !DM  'D   %DO!))&#t?P?P?RY]^!))+$(  D%ll7??+O>O>QR N!))+ a%ll7??+_arr ) r]typingrrr-+litellm.litellm_core_utils.secret_redactionrr4rr r_rrrbs"E)vavar