Fix Get Agent Secret Using Key (flyteorg#2644)

Signed-off-by: Future-Outlier <[email protected]> Signed-off-by: mao3267 <[email protected]>
mao3267 · Aug 2, 2024 · 0b48963 · 0b48963
1 parent 8ff53b9
commit 0b48963
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: