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

feat(proxy_server.py): allow restricting allowed email domains for the UI

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-06 12:53:03 -07:00
parent 2ec7cb4153
commit 5ffadd631d
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/proxy_server.py
View file

@ -8557,6 +8557,19 @@ async def auth_callback(request: Request):
user_email = getattr(result, "email", None) user_email = getattr(result, "email", None)
user_id = getattr(result, "id", None) user_id = getattr(result, "id", None)
if user_email is not None and os.getenv("ALLOWED_EMAIL_DOMAINS") is not None:
email_domain = user_email.split("@")[1]
allowed_domains = os.getenv("ALLOWED_EMAIL_DOMAINS").split(",") # type: ignore
if email_domain not in allowed_domains:
raise HTTPException(
status_code=401,
detail={
"message": "The email domain={}, is not an allowed email domain={}. Contact your admin to change this.".format(
email_domain, allowed_domains
)
},
)
# generic client id # generic client id
if generic_client_id is not None: if generic_client_id is not None:
user_id = getattr(result, "id", None) user_id = getattr(result, "id", None)