mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 03:34:10 +00:00
Litellm dev 01 08 2025 p1 (#7640)
* feat(ui_sso.py): support reading team ids from sso token * feat(ui_sso.py): working upsert sso user teams membership in litellm - if team exists Adds user to relevant teams, if user is part of teams and team exists on litellm * fix(ui_sso.py): safely handle add team member task * build(ui/): support setting team id when creating team on UI * build(ui/): teams.tsx allow setting team id on ui * build(circle_ci/requirements.txt): add fastapi-sso to ci/cd testing * fix: fix linting errors
This commit is contained in:
parent
6d8cfeaf14
commit
b77832a793
10 changed files with 269 additions and 120 deletions
|
@ -1327,3 +1327,35 @@ async def test_get_user_info_for_proxy_admin(mock_team_data, mock_key_data):
|
|||
# Verify the result structure
|
||||
assert isinstance(result, UserInfoResponse)
|
||||
assert len(result.keys) == 2
|
||||
|
||||
|
||||
def test_custom_openid_response():
|
||||
from litellm.proxy.management_endpoints.ui_sso import generic_response_convertor
|
||||
from litellm.proxy.management_endpoints.ui_sso import JWTHandler
|
||||
from litellm.proxy._types import LiteLLM_JWTAuth
|
||||
from litellm.caching import DualCache
|
||||
|
||||
jwt_handler = JWTHandler()
|
||||
jwt_handler.update_environment(
|
||||
prisma_client={},
|
||||
user_api_key_cache=DualCache(),
|
||||
litellm_jwtauth=LiteLLM_JWTAuth(
|
||||
team_ids_jwt_field="department",
|
||||
),
|
||||
)
|
||||
response = {
|
||||
"sub": "3f196e06-7484-451e-be5a-ea6c6bb86c5b",
|
||||
"email_verified": True,
|
||||
"name": "Krish Dholakia",
|
||||
"preferred_username": "krrishd",
|
||||
"given_name": "Krish",
|
||||
"department": ["/test-group"],
|
||||
"family_name": "Dholakia",
|
||||
"email": "krrishdholakia@gmail.com",
|
||||
}
|
||||
|
||||
resp = generic_response_convertor(
|
||||
response=response,
|
||||
jwt_handler=jwt_handler,
|
||||
)
|
||||
assert resp.team_ids == ["/test-group"]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue