o ưi@sBddlZddlmZmZmZmZmZddlmZGdddZ dS)N)ListDictAnyOptionalUnion)UnauthorizedErrorc @sdeZdZd dedeefddZdeeeffddZdd ede e eee fe j ffd d ZdS)ModelGroupsManagementClientNbase_urlapi_keycCs|d|_||_dS)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 rX/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/client/model_groups.py__init__s  z$ModelGroupsManagementClient.__init__returncCsi}|jr d|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)rheadersrrr _get_headerssz(ModelGroupsManagementClient._get_headersFreturn_requestc Cs|jd}tjd||d}|r|St}z||}||dWStj j yC}z |j j dkr>t |d}~ww)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$errrinfos  z ModelGroupsManagementClient.info)N)F)__name__ __module__ __qualname__strrrrrboolrrrrrr*rrrrr s . r ) rtypingrrrrrr"rr rrrrs