tPjlHdZddlZddlmZmZddlZddlmZGddZy)z*Teams management client for LiteLLM proxy.N)AnyFinal)UnauthorizedErrorceZdZdZddededzdefdZdeeeffdZ dd edzd edzde eee ffd Z dd edzd edzd edzd edzdedededzdedeee ffdZ de jeee ffdZ y)TeamsManagementClientz+Client for managing teams in LiteLLM proxy.Nbase_urlapi_keytimeoutcL|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. 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/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)*<= ==)0'@GO $ruser_idorganization_idc|jd}i}|r||d<|r||d<tj||j||j}|j dk(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/listrrrparamsr *Authentication failed. Check your API key. rrequestsgetrr status_coderraise_for_statusjson)rrrurlrresponses rlistzTeamsManagementClient.list's&'z2  'F9  (7F$ %",,sD4E4E4GPV`d`m`mn   3 &#$PQ Q!!#}}rteam_id team_aliaspage page_sizesort_by sort_orderc H|jd} |||d} |r|| d<|r|| d<|r|| d<|r|| d<|r|| d<tj| |j| |j} | j d 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.r0rrr+r,r/rr r!r") rrrr+r,r-r.r/r0r(rr)s rlist_v2zTeamsManagementClient.list_v2Is>'}5"$/   'F9  (7F$ %  'F9  #-F<  'F9 ",,sD4E4E4GPV`d`m`mn   3 &#$PQ Q!!#}}rc|jd}tj||j|j}|j dk(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!r")rr(r)s r get_availablez#TeamsManagementClient.get_availablesi'7",,sD4E4E4GQUQ^Q^_   3 &#$PQ Q!!#}}r)N)NN)NNNNr Nasc)__name__ __module__ __qualname____doc__strintrdictrr*rr2builtinsr4rrrr s'5    sTz  3   d38n #&* t t  d38n  H#&*"!%"7t7t7t 7 $J 7  77t77 c3h7rx}}T#s(^<rr) r;r?typingrrr# exceptionsrrr@rrrCs 0)JJr