tPjO-HddlZddlmZmZddlZddlmZmZGddZy)N)AnyFinal) NotFoundErrorUnauthorizedErrorc eZdZddededzdefdZdeeeffdZddede eee fe jzfd Z dd ed eee fd eee fdzdedeee fe jzf d Z ddededeee fe jzfdZ ddedzd edzdedeee fe jzfdZddedejeee fe jzfdZ dded eee fd eee fdzdedeee fe jzf dZy)ModelsManagementClientNbase_urlapi_keytimeoutcL|jd|_||_||_y)aa Initialize the ModelsManagementClient. 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/models.py__init__zModelsManagementClient.__init__ s""-  returncFi}|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 zBearer Authorization)r)rheaderss r _get_headersz#ModelsManagementClient._get_headerss* ==)0'@GO $rreturn_requestc|jd}tjd||j}|r|Stj} |j |j |j}|j|jdS#tjj$r*}|jjdk(r t|d}~wwxYw)a Get the list of models supported by the server. Args: return_request (bool): If True, returns the prepared request object instead of executing it. Useful for inspection or modification before sending. Returns: Union[List[Dict[str, Any]], requests.Request]: Either a list of model information dictionaries 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/modelsGETrr dataNrrequestsRequestrSessionsendpreparerraise_for_statusjson exceptions HTTPErrorresponse status_coderrrurlrequestsessionr-es rlistzModelsManagementClient.list#s 'w/!))%d>O>O>QR N"))+ %ll7??+rr0r!r1r2r-r3s rdeletezModelsManagementClient.deleteus"'}5X&!))&#t?P?P?RY]^ N"))+ %ll7??+r6rresultmodelsmodelmsgs rrKzModelsManagementClient.gets0  !39MR\RhUV V  IITI:Ffh&6&67 77M &$''' EUYY|R8<O>O>QR N"))+ %ll7??+r7r8rr0r!r1r2r-r3s rupdatezModelsManagementClient.updates2'}5*  !+D !))&#t?P?P?RY]^ N"))+ %ll7??+r_s8^^r