tPjw<ddlmZmZddlZddlmZGddZy))AnyFinalN)UnauthorizedErrorc FeZdZddededzdefdZdeeeffdZ ddedeee fe jzfd Z dd ed eee fd eee fdedeee fe jzf d Z dd ededeee fe jzfdZ dd ededeee fe jzfdZy)CredentialsManagementClientNbase_urlapi_keytimeoutcL|jd|_||_||_y)af 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. timeout (int): Request timeout in seconds (default: 30) /N)rstrip _base_url_api_key_timeout)selfr r r s e/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/client/credentials.py__init__z$CredentialsManagementClient.__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(CredentialsManagementClient._get_headerss0)*<= ==)0'@GO $rreturn_requestc|jd}tjd||j}|r|Stj} |j |j |j}|j|jS#tjj$r*}|jjdk(r t|d}~wwxYw)a List all credentials. Args: 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 /credentialsGETrr NrrequestsRequestrSessionsendpreparerraise_for_statusjson exceptions HTTPErrorresponse status_coder)rrurlrequestsessionr,es rlistz CredentialsManagementClient.list"s$'|4!))%d>O>O>QR N!))+ %ll7??+%C##C(c|jd|}tjd||j}|r|Stj} |j |j |j}|j|jS#tjj$r*}|jjdk(r t|d}~wwxYw)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/DELETErr r!Nr"rr3rr.r/r0r,r1s rdeletez"CredentialsManagementClient.deletets('}_4EF!))(CARARATU N!))+ %ll7??+O>O>QR N!))+ %ll7??+)N)F)__name__ __module__ __qualname__strintrdictrboolrr#r$r2r9r=r@rrrrs5    sTz  3   d38n  %!! c3h(** *!P % --c3h- S> -  - c3h(** * -d %### c3h(** * #P %### c3h(** * #rr)typingrrr#r*rrrIrrrKs)ttr