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: