diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 3dedc3a71..86d1d0fb9 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2798,7 +2798,7 @@ async def image_generation( ) -#### KEY MANAGEMENT #### +#### KEY MANAGEMENT ##### @router.post( @@ -3159,6 +3159,9 @@ async def info_key_fn( ) +#### SPEND MANAGEMENT ##### + + @router.get( "/spend/keys", tags=["budget & spend Tracking"], @@ -3946,6 +3949,60 @@ async def user_update(data: UpdateUserRequest): ) +#### TEAM MANAGEMENT #### + + +@router.post( + "/team/new", tags=["team management"], dependencies=[Depends(user_api_key_auth)] +) +async def new_team(): + """ + Create a new team + + Parameters: + - team_alias: Optional[str] - User defined team alias + - team_id: Optional[str] - The team id of the user. If none passed, we'll generate it. + - team_admins: list - A list of user IDs that will be owning the team + - metadata: Optional[dict] - Metadata for team, store information for team. Example metadata = {"team": "core-infra", "app": "app2", "email": "ishaan@berri.ai" } + + Returns: + - key: (str) The generated api key + - expires: (datetime) Datetime object for when key expires. + - team_id: (str) Unique team id - used for tracking spend across multiple keys for same team id. + """ + pass + + +@router.post( + "/team/update", tags=["team management"], dependencies=[Depends(user_api_key_auth)] +) +async def update_team(): + """ + update team and members + """ + pass + + +@router.post( + "/team/delete", tags=["team management"], dependencies=[Depends(user_api_key_auth)] +) +async def delete_team(): + """ + delete team and team keys + """ + pass + + +@router.post( + "/team/info", tags=["team management"], dependencies=[Depends(user_api_key_auth)] +) +async def info_team(): + """ + get info on team + related keys + """ + pass + + #### MODEL MANAGEMENT ####