o ưi@sFdZddlmZmZmZmZmZddlZddlm Z GdddZ dS)z*Teams management client for LiteLLM proxy.)AnyDictListOptionalUnionN)UnauthorizedErrorc@seZdZdZddedeefddZdeeeffdd Z  dd eed eede eee ffd d Z        dd eed eedeedeede de deededeee ffddZ de eee ffddZdS)TeamsManagementClientz+Client for managing teams in LiteLLM proxy.Nbase_urlapi_keycCs|d|_||_dS)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 rQ/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/client/teams.py__init__ s  zTeamsManagementClient.__init__returncCs"ddi}|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)rheadersrrr _get_headerssz"TeamsManagementClient._get_headersuser_idorganization_idcCs^|jd}i}|r||d<|r||d<tj|||d}|jdkr'td||S)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_coderraise_for_statusjson)rrrurlrresponserrrlist$s  zTeamsManagementClient.listr ascteam_id team_aliaspage page_sizesort_by sort_orderc Cs|jd} |||d} |r|| d<|r|| d<|r|| d<|r$|| d<|r*|| d<tj| || d} | jd kr=td | | S) 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,r.rrr)r*r-rrrr) rrrr)r*r+r,r-r.r$rr%rrrlist_v2Fs(  zTeamsManagementClient.list_v2cCs@|jd}tj||d}|jdkrtd||S)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)rrrr)rr$r%rrr get_availables  z#TeamsManagementClient.get_available)N)NN)NNNNrr'Nr()__name__ __module__ __qualname____doc__strrrrrrrr&intr/r0rrrrr sR  $  9r ) r4typingrrrrrr exceptionsrr rrrrs