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

fix debug_sso_callback

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-08 21:40:51 -07:00
parent 9d8ae741b5
commit 6ef3434be4
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -311,7 +311,7 @@ def get_disabled_non_admin_personal_key_creation():
@router.get("/sso/callback", tags=["experimental"], include_in_schema=False) @router.get("/sso/callback", tags=["experimental"], include_in_schema=False)
async def auth_callback(request: Request): async def auth_callback(request: Request): # noqa: PLR0915
"""Verify login""" """Verify login"""
from litellm.proxy.management_endpoints.key_management_endpoints import ( from litellm.proxy.management_endpoints.key_management_endpoints import (
generate_key_helper_fn, generate_key_helper_fn,
@ -603,9 +603,9 @@ async def insert_sso_user(
if user_defined_values.get("max_budget") is None: if user_defined_values.get("max_budget") is None:
user_defined_values["max_budget"] = litellm.max_internal_user_budget user_defined_values["max_budget"] = litellm.max_internal_user_budget
if user_defined_values.get("budget_duration") is None: if user_defined_values.get("budget_duration") is None:
user_defined_values[ user_defined_values["budget_duration"] = (
"budget_duration" litellm.internal_user_budget_duration
] = litellm.internal_user_budget_duration )
if user_defined_values["user_role"] is None: if user_defined_values["user_role"] is None:
user_defined_values["user_role"] = LitellmUserRoles.INTERNAL_USER_VIEW_ONLY user_defined_values["user_role"] = LitellmUserRoles.INTERNAL_USER_VIEW_ONLY
@ -798,9 +798,9 @@ class SSOAuthenticationHandler:
if state: if state:
redirect_params["state"] = state redirect_params["state"] = state
elif "okta" in generic_authorization_endpoint: elif "okta" in generic_authorization_endpoint:
redirect_params[ redirect_params["state"] = (
"state" uuid.uuid4().hex
] = uuid.uuid4().hex # set state param for okta - required ) # set state param for okta - required
return await generic_sso.get_login_redirect(**redirect_params) # type: ignore return await generic_sso.get_login_redirect(**redirect_params) # type: ignore
@staticmethod @staticmethod
@ -1077,8 +1077,8 @@ async def debug_sso_callback(request: Request):
try: try:
# Try to convert to string or another JSON serializable format # Try to convert to string or another JSON serializable format
filtered_result[key] = str(value) filtered_result[key] = str(value)
except: except Exception as e:
filtered_result[key] = "Complex value (not displayable)" filtered_result[key] = f"Complex value (not displayable): {str(e)}"
# Replace the placeholder in the template with the actual data # Replace the placeholder in the template with the actual data
html_content = jwt_display_template.replace( html_content = jwt_display_template.replace(