From 6ef3434be48e0ed68fdd1f1ebfe77faaa2917d5f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 8 Apr 2025 21:40:51 -0700 Subject: [PATCH] fix debug_sso_callback --- litellm/proxy/management_endpoints/ui_sso.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/litellm/proxy/management_endpoints/ui_sso.py b/litellm/proxy/management_endpoints/ui_sso.py index e1a9c2e6e5..fea931fc35 100644 --- a/litellm/proxy/management_endpoints/ui_sso.py +++ b/litellm/proxy/management_endpoints/ui_sso.py @@ -311,7 +311,7 @@ def get_disabled_non_admin_personal_key_creation(): @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""" from litellm.proxy.management_endpoints.key_management_endpoints import ( generate_key_helper_fn, @@ -603,9 +603,9 @@ async def insert_sso_user( if user_defined_values.get("max_budget") is None: user_defined_values["max_budget"] = litellm.max_internal_user_budget if user_defined_values.get("budget_duration") is None: - user_defined_values[ - "budget_duration" - ] = litellm.internal_user_budget_duration + user_defined_values["budget_duration"] = ( + litellm.internal_user_budget_duration + ) if user_defined_values["user_role"] is None: user_defined_values["user_role"] = LitellmUserRoles.INTERNAL_USER_VIEW_ONLY @@ -798,9 +798,9 @@ class SSOAuthenticationHandler: if state: redirect_params["state"] = state elif "okta" in generic_authorization_endpoint: - redirect_params[ - "state" - ] = uuid.uuid4().hex # set state param for okta - required + redirect_params["state"] = ( + uuid.uuid4().hex + ) # set state param for okta - required return await generic_sso.get_login_redirect(**redirect_params) # type: ignore @staticmethod @@ -1077,8 +1077,8 @@ async def debug_sso_callback(request: Request): try: # Try to convert to string or another JSON serializable format filtered_result[key] = str(value) - except: - filtered_result[key] = "Complex value (not displayable)" + except Exception as e: + filtered_result[key] = f"Complex value (not displayable): {str(e)}" # Replace the placeholder in the template with the actual data html_content = jwt_display_template.replace(