Fix Get Agent Secret Using Key (#2644)

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Aug 2, 2024 · bcfbb80 · bcfbb80
1 parent 3d96dd6
commit bcfbb80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: