From ffaf3ab4547be799089906088f80bac9cd3c275a Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 19 Sep 2024 15:20:15 -0700 Subject: [PATCH] make totp_identifier parametrized in workflow (#847) --- skyvern/forge/sdk/workflow/models/block.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/skyvern/forge/sdk/workflow/models/block.py b/skyvern/forge/sdk/workflow/models/block.py index 91382d76..648917e6 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -240,6 +240,20 @@ class TaskBlock(Block): ) self.url = task_url_parameter_value + if ( + self.totp_identifier + and workflow_run_context.has_parameter(self.totp_identifier) + and workflow_run_context.has_value(self.totp_identifier) + ): + totp_identifier_parameter_value = workflow_run_context.get_value(self.totp_identifier) + if totp_identifier_parameter_value: + LOG.info( + "TOTP identifier is parameterized, using parameter value", + totp_identifier_parameter_value=totp_identifier_parameter_value, + totp_identifier_parameter_key=self.totp_identifier, + ) + self.totp_identifier = totp_identifier_parameter_value + if self.download_suffix and workflow_run_context.has_parameter(self.download_suffix): download_suffix_parameter_value = workflow_run_context.get_value(self.download_suffix) if download_suffix_parameter_value: