scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fix ensure_form_action_url_matches_request_scheme

Browse source
This commit is contained in:
Ishaan Jaff 2025-08-02 08:58:31 -07:00
parent 44900e781a
commit d0d53384f6
1 changed files with 30 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -75,6 +75,31 @@ else:
router = APIRouter()
def ensure_form_action_url_matches_request_scheme(
form_action_url: str,
request: Request,
) -> str:
"""
Ensure the form action URL uses the same scheme as the request URL.
Args:
form_action: The form action URL to normalize
request_base_url: The base URL from the request
Returns:
The form action URL with scheme matching the request URL scheme
"""
from urllib.parse import urlparse
request_url_scheme = urlparse(str(request.base_url)).scheme
form_action_scheme = urlparse(form_action_url).scheme
if form_action_scheme != request_url_scheme:
form_action_url = form_action_url.replace(form_action_scheme, request_url_scheme)
return form_action_url
@router.get("/sso/key/generate", tags=["experimental"], include_in_schema=False)
async def serve_login_page(
request: Request,
@ -169,6 +194,11 @@ async def serve_login_page(
# Get the base URL for form action using proper URL construction
form_action = get_custom_url(request_base_url=str(request.base_url), route="login")
# ensure the form action url uses the same scheme as the request
form_action = ensure_form_action_url_matches_request_scheme(
form_action_url=form_action,
request=request
)
unified_login_html = f"""
<!DOCTYPE html>