From ff6642e883bafb65681f7961658af4cef6d6d184 Mon Sep 17 00:00:00 2001 From: vincbeck Date: Mon, 2 Feb 2026 14:36:34 -0500 Subject: [PATCH] Fix `host` in AWS auth manager --- .../providers/amazon/aws/auth_manager/routes/login.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/providers/amazon/src/airflow/providers/amazon/aws/auth_manager/routes/login.py b/providers/amazon/src/airflow/providers/amazon/aws/auth_manager/routes/login.py index 2c65d5ad236a1..6c70786bd03bb 100644 --- a/providers/amazon/src/airflow/providers/amazon/aws/auth_manager/routes/login.py +++ b/providers/amazon/src/airflow/providers/amazon/aws/auth_manager/routes/login.py @@ -19,6 +19,7 @@ import logging from typing import Any +from urllib.parse import urlparse import anyio from fastapi import HTTPException, Request, status @@ -133,7 +134,9 @@ def _init_saml_auth(request: Request) -> OneLogin_Saml2_Auth: def _prepare_request(request: Request) -> dict: - host = request.headers.get("host", request.client.host if request.client else "localhost") + parsed = urlparse(conf.get("api", "base_url", fallback="http://localhost")) + host = parsed.hostname + data: dict[str, Any] = { "https": "on" if request.url.scheme == "https" else "off", "http_host": host,