g@dZddlmZddlmZmZddlmZdedefdZy) a Example Custom SSO Handler Use this if you want to run custom code after litellm has retrieved information from your IDP (Identity Provider). Flow: - User lands on Admin UI - LiteLLM redirects user to your SSO provider - Your SSO provider redirects user back to LiteLLM - LiteLLM has retrieved user information from your IDP - Your custom SSO handler is called and returns an object of type SSOUserDefinedValues - User signed in to UI )OpenID)LitellmUserRolesSSOUserDefinedValues) user_info userIDPInforeturnctK tdtd||jtd|t|jd{}td|t g|j|j t jjddS7M#t$r td wxYww) Nzinside custom sso handlerz userIDPInfo: z-No ID found for user. userIDPInfo.id is None )user_idz_user_info from litellm DB 1d)modelsr user_email user_role max_budgetbudget_durationzFailed custom auth) printid ValueErrorrremailr INTERNAL_USERvalue Exception)r _user_infos M/var/www/openai/venv/lib/python3.12/site-packages/litellm/proxy/custom_sso.pycustom_sso_handlerrs. )* k]+, >> !? }M  %[^^<<  +Z8#NN"((&44::   = .,--.s0B8A B BA B B8B B55B8N) __doc__fastapi_sso.sso.baserlitellm.proxy._typesrr:litellm.proxy.management_endpoints.internal_user_endpointsrrrr"s) (GP.&.5I.r!