phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

create_litellm_teams_from_service_principal_team_ids

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-09 17:49:08 -07:00
parent 958c284957
commit f85767e4af
1 changed files with 34 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

61
litellm/proxy/management_endpoints/ui_sso.py
View file

@ -1193,35 +1193,42 @@ class MicrosoftSSOHandler:
litellm_team_name: Optional[str] = service_principal_team.get( litellm_team_name: Optional[str] = service_principal_team.get(
"principalDisplayName" "principalDisplayName"
) )
if litellm_team_id: if not litellm_team_id:
try: verbose_proxy_logger.debug(
verbose_proxy_logger.debug( f"Skipping team creation for {litellm_team_name} because it has no principalId"
f"Creating Litellm Team: {litellm_team_id} - {litellm_team_name}" )
) continue
team_obj = await prisma_client.db.litellm_teamtable.find_first( try:
where={"team_id": litellm_team_id} verbose_proxy_logger.debug(
f"Creating Litellm Team: {litellm_team_id} - {litellm_team_name}"
)
team_obj = await prisma_client.db.litellm_teamtable.find_first(
where={"team_id": litellm_team_id}
)
verbose_proxy_logger.debug(f"Team object: {team_obj}")
# only create a new team if it doesn't exist
if team_obj:
verbose_proxy_logger.debug(
f"Team already exists: {litellm_team_id} - {litellm_team_name}"
) )
verbose_proxy_logger.debug(f"Team object: {team_obj}") continue
if team_obj: await new_team(
verbose_proxy_logger.debug( data=NewTeamRequest(
f"Team already exists: {litellm_team_id} - {litellm_team_name}" team_id=litellm_team_id,
) team_alias=litellm_team_name,
continue ),
await new_team( # params used for Audit Logging
data=NewTeamRequest( http_request=Request(scope={"type": "http", "method": "POST"}),
team_id=litellm_team_id, user_api_key_dict=UserAPIKeyAuth(
team_alias=litellm_team_name, token="",
), key_alias=f"litellm.{MicrosoftSSOHandler.__name__}",
# params used for Audit Logging ),
http_request=Request(scope={"type": "http", "method": "POST"}), )
user_api_key_dict=UserAPIKeyAuth( except Exception as e:
token="", verbose_proxy_logger.exception(f"Error creating Litellm Team: {e}")
key_alias=f"litellm.{MicrosoftSSOHandler.__name__}",
),
)
except Exception as e:
verbose_proxy_logger.exception(f"Error creating Litellm Team: {e}")
class GoogleSSOHandler: class GoogleSSOHandler: