From 52bf3e0ce1aca048c060e9f3d1d72ad9fdc08946 Mon Sep 17 00:00:00 2001 From: Martynas Asipauskas Date: Fri, 30 Aug 2024 13:02:13 +0100 Subject: [PATCH] Fix Bearer auth to work when authenticating against K8S from Airflow Operator --- third_party/airflow/armada/log_manager.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/third_party/airflow/armada/log_manager.py b/third_party/airflow/armada/log_manager.py index 0a94ddeabbb..b7c763eb085 100644 --- a/third_party/airflow/armada/log_manager.py +++ b/third_party/airflow/armada/log_manager.py @@ -47,15 +47,12 @@ def _k8s_client(self, k8s_context) -> client.CoreV1Api: k8s_client = client.CoreV1Api( api_client=client.ApiClient(configuration=configuration) ) - k8s_client.api_client.configuration.api_key_prefix["authorization"] = ( - "Bearer" - ) KubernetesPodLogManager.CLIENTS[k8s_context] = k8s_client return KubernetesPodLogManager.CLIENTS[k8s_context] def _with_bearer_auth(self, client): client.api_client.configuration.api_key["authorization"] = ( - self._token_retriever.get_token() + f"Bearer {self._token_retriever.get_token()}" ) def fetch_container_logs(