o ưi#/@sBddlmZmZmZmZmZddlZddlmZGdddZ dS))AnyDictListOptionalUnionN)UnauthorizedErrorc@s8eZdZd'dedeefddZdeeeffddZ          d(d eed eed eed eedeedeedeedee dee de de eee fe j ffddZ          d(deeedeeeefdeedeedeed eed eedeedeeee fde de eee fe j ffddZ   d)deeedeeede de eee fe j ffd d!Z       d*d"edeeedeeeefdeedeedeed eed eede eee fe j ffd#d$Zd+d"ede de eee fe j ffd%d&ZdS),KeysManagementClientNbase_urlapi_keycCs|d|_||_dS)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 rP/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/client/keys.py__init__ s  zKeysManagementClient.__init__returncCs"ddi}|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)rheadersrrr _get_headerssz!KeysManagementClient._get_headersFpagesizeuser_idteam_idorganization_idkey_hash key_aliasreturn_full_objectinclude_team_keysreturn_requestc  Cs*|jd} i} |dur|| d<|dur|| d<|dur || d<|dur(|| d<|dur0|| d<|dur8|| d<|dur@|| d <|durLt|| d <| durXt| | d <tjd | || d } | rg| St}z|| }| | WStj j y}z |j jdkrt|d}~ww)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/listNrrrrrrrrr GET)rparams)rstrlowerrequestsRequestrSessionsendprepareraise_for_statusjson exceptions HTTPErrorresponse status_coder)rrrrrrrrrr r!urlr#requestsessionr0errrlist sD %  zKeysManagementClient.listmodelsaliasesspendduration budget_idconfigc  Cs|jd} i} |dur|| d<|dur|| d<|dur || d<|dur(|| d<|dur0|| d<|dur8|| d<|dur@|| d <|durH|| d <| durP| | d <tjd | || d } | r_| St}z|| }||WStj j y}z |j j dkrt |d}~ww)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/generateNr7r8r9r:rrrr;r<POSTrr-r$rr'r(rr)r*r+r,r-r.r/r0r1r)rr7r8r9r:rrrr;r<r!r2datar3r4r0r5rrrgenerateksD &  zKeysManagementClient.generatekeys key_aliasesc Cs|jd}||d}tjd|||d}|r|St}z||}||WStj j yG} z | j j dkrBt | d} ~ ww)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)rBrCr=r>r$Nr?) rrBrCr!r2r@r3r4r0r5rrrdeletes$   zKeysManagementClient.deletekeyc Cs|jd} d|i} |dur|| d<|dur|| d<|dur"|| d<|dur*|| d<|dur2|| d<|dur:|| d <|durB|| d <tjd | || d } t} d} z| | }|j} || WSt ytt d | w)a Update an existing API key's parameters. Args: models: Optional[List[str]] = None, aliases: Optional[Dict[str, str]] = None, spend: Optional[float] = None, duration: Optional[str] = None, key_alias: Optional[str] = None, team_id: Optional[str] = None, user_id: Optional[str] = None, 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/updaterENrrrr7r9r:r8r=r>zError updating key: ) rr'r(rr)r*r+textr,r- Exception)rrEr7r8r9r:rrrr2r@r3r4Z response_textr0rrrupdates6   zKeysManagementClient.updatec Cs|jd|}tjd||d}|r|St}z||}||WStj j yC}z |j j dkr>t |d}~ww)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/key/info?key=r")rr$Nr?)rrEr!r2r3r4r0r5rrrinfos  zKeysManagementClient.info)N) NNNNNNNNNF)NNF)NNNNNNN)F)__name__ __module__ __qualname__r%rrrrintboolrrr'r(r6rfloatrArDrHrIrrrrr s     M     M   /   .=r ) typingrrrrrr'r.rr rrrrs