o ưiP@s>ddlZddlmZmZmZmZddlmZGdddZdS)N)DictAnyOptionalUnion)UnauthorizedErrorc@seZdZddedeefddZdeeeffddZ dd ede eee fe j ffd d Z dd edeee fdeee fd ede eee fe j ff ddZ dd ed ede eee fe j ffddZ dd ed ede eee fe j ffddZdS)CredentialsManagementClientNbase_urlapi_keycCs|d|_||_dS)a" Initialize the CredentialsManagementClient. 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 rW/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/client/credentials.py__init__s  z$CredentialsManagementClient.__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(CredentialsManagementClient._get_headersFreturn_requestc Cs|jd}tjd||d}|r|St}z||}||WStj j yA}z |j j dkrt |d}~ww)aO Delete a credential by name. Args: credential_name (str): Name of the credential 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 /credentials/DELETErrNrrr-rr(r)r*r&r+rrrdeleteq  z"CredentialsManagementClient.deletec Cr3)aN Get a credential by name. Args: credential_name (str): Name of the credential to retrieve 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/credentials/by_name/rrrNrr5rrrgetr7zCredentialsManagementClient.get)N)F)__name__ __module__ __qualname__strrrrrboolrrrrr,r2r6r8rrrrrsJ  (   2 (r) rtypingrrrrr$rrrrrrs