AHjLdZddlmZmZmZmZmZddlZddlm Z GddZ y)z*Teams management client for LiteLLM proxy.)AnyDictListOptionalUnionN)UnauthorizedErrorceZdZdZddedeefdZdeeeffdZ ddeed eede eee ffd Z ddeed eed eed eed e de deededeee ffdZ de eee ffdZy)TeamsManagementClientz+Client for managing teams in LiteLLM proxy.Nbase_urlapi_keyc>|jd|_||_y)a Initialize the TeamsManagementClient. Args: base_url (str): The base URL of the LiteLLM proxy server (e.g., "http://localhost:4000") 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 R/root/tools/cai/cai_env/lib/python3.12/site-packages/litellm/proxy/client/teams.py__init__zTeamsManagementClient.__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"TeamsManagementClient._get_headerss0"#56 ==)0'@GO $ruser_idorganization_idc|jd}i}|r||d<|r||d<tj||j|}|jdk(r t d|j |jS)a List teams that the user belongs to. Args: user_id (Optional[str]): Only return teams which this user belongs to organization_id (Optional[str]): Only return teams which belong to this organization Returns: List[Dict[str, Any]]: List of team objects Raises: requests.exceptions.HTTPError: If the request fails UnauthorizedError: If authentication fails z /team/listrrrparams*Authentication failed. Check your API key.rrequestsgetr status_coder raise_for_statusjson)rrrurlr responses rlistzTeamsManagementClient.list$s& +  'F9  (7F$ %<<T->->-@P   3 &#$PQ Q!!#}}rteam_id team_aliaspage page_sizesort_by sort_orderc 2|jd} |||d} |r|| d<|r|| d<|r|| d<|r|| d<|r|| d<tj| |j| } | jd k(r t d | j | jS) a Get a paginated list of teams with filtering and sorting options. Args: user_id (Optional[str]): Only return teams which this user belongs to organization_id (Optional[str]): Only return teams which belong to this organization team_id (Optional[str]): Filter teams by exact team_id match team_alias (Optional[str]): Filter teams by partial team_alias match page (int): Page number for pagination page_size (int): Number of teams per page sort_by (Optional[str]): Column to sort by (e.g. 'team_id', 'team_alias', 'created_at') sort_order (str): Sort order ('asc' or 'desc') Returns: Dict[str, Any]: Paginated response containing teams and pagination info Raises: requests.exceptions.HTTPError: If the request fails UnauthorizedError: If authentication fails z /v2/team/list)r.r/r1rrr,r-r0rr!r"r#) rrrr,r-r.r/r0r1r)r r*s rlist_v2zTeamsManagementClient.list_v2Fs> ."$.   'F9  (7F$ %  'F9  #-F<  'F9 <<T->->-@P   3 &#$PQ Q!!#}}rc|jd}tj||j}|jdk(r t d|j |jS)a  Get list of available teams that the user can join. Returns: List[Dict[str, Any]]: List of available team objects Raises: requests.exceptions.HTTPError: If the request fails UnauthorizedError: If authentication fails z/team/available)rr!r"r#)rr)r*s r get_availablez#TeamsManagementClient.get_availables` 0<<T->->-@A   3 &#$PQ Q!!#}}r)N)NN)NNNNr Nasc)__name__ __module__ __qualname____doc__strrrrrrrr+intr3r5rrr r s5    x}   d38n "&)- # "#  d38n  H"&)-!%$(!%7#7"#7# 7 SM 7  77#77 c3h7rtDcN3rr ) r;typingrrrrrr$ exceptionsr r r>rrrAs 033)HHr