Fix Bearer auth to work when authenticating against K8S from Airflow …

…Operator
armadaproject · Aug 30, 2024 · 52bf3e0 · 52bf3e0
1 parent 523771f
commit 52bf3e0
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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(