hHddlZddlmZmZmZmZmZddlmZGddZ y)N)ListDictAnyOptionalUnion)UnauthorizedErrorc zeZdZd dedeefdZdeeeffdZd dede e eee fe jffdZy) ModelGroupsManagementClientNbase_urlapi_keyc>|jd|_||_y)a" Initialize the ModelGroupsManagementClient. 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 s ]/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/client/model_groups.py__init__z$ModelGroupsManagementClient.__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(ModelGroupsManagementClient._get_headerss* ==)0'@GO $rreturn_requestc|jd}tjd||j}|r|Stj} |j |j }|j|jdS#tjj$r*}|jjdk(r t|d}~wwxYw)aG Get detailed information about all model groups from the server. Args: return_request (bool): If True, returns the prepared request object instead of executing it Returns: Union[List[Dict[str, Any]], requests.Request]: Either a list of model group 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/model_group/infoGET)rdataiN)rrequestsRequestrSessionsendprepareraise_for_statusjson exceptions HTTPErrorresponse status_coder )rrurlrequestsessionr)es rinfoz ModelGroupsManagementClient.infos 12""5#t7H7H7JK N""$ ||GOO$56H  % % '==?6* *"",, zz%%,'**  sABC.%CC)N)F)__name__ __module__ __qualname__strrrrrboolrrrr r!r/rrr r s^    x}   d38n 4E$tCH~:NPXP`P`:`4arr ) r typingrrrrrr'r r r5rrr7s33)66r