From bcfbb80e00bdd3c6b14f1f193108d184389c7263 Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Fri, 2 Aug 2024 12:04:58 +0800 Subject: [PATCH] Fix Get Agent Secret Using Key (#2644) Signed-off-by: Future-Outlier --- flytekit/extend/backend/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flytekit/extend/backend/utils.py b/flytekit/extend/backend/utils.py index dcea3e6b34..4dcdf3174a 100644 --- a/flytekit/extend/backend/utils.py +++ b/flytekit/extend/backend/utils.py @@ -39,7 +39,7 @@ def is_terminal_phase(phase: TaskExecution.Phase) -> bool: def get_agent_secret(secret_key: str) -> str: - return flytekit.current_context().secrets.get(secret_key) + return flytekit.current_context().secrets.get(key=secret_key) def render_task_template(tt: TaskTemplate, file_prefix: str) -> TaskTemplate: