vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix totp parser for totp secret stored in credential item (#3146)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-08-08 13:38:07 -07:00
committed by GitHub
parent 035722c750
commit 4da3efdffa
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/sdk/workflow/context_manager.py
View File

@@ -259,7 +259,7 @@ class WorkflowRunContext:
totp_secret_id = f"{random_secret_id}_totp" totp_secret_id = f"{random_secret_id}_totp"
self.secrets[totp_secret_id] = BitwardenConstants.TOTP self.secrets[totp_secret_id] = BitwardenConstants.TOTP
totp_secret_value = self.totp_secret_value_key(totp_secret_id) totp_secret_value = self.totp_secret_value_key(totp_secret_id)
self.secrets[totp_secret_value] = credential_item.totp self.secrets[totp_secret_value] = parse_totp_secret(credential_item.totp)
self.values[parameter.key]["totp"] = totp_secret_id self.values[parameter.key]["totp"] = totp_secret_id
except Exception as e: except Exception as e:
LOG.error(f"Failed to get credential from database: {credential_id}. Error: {e}") LOG.error(f"Failed to get credential from database: {credential_id}. Error: {e}")
@@ -305,7 +305,7 @@ class WorkflowRunContext:
totp_secret_id = f"{random_secret_id}_totp" totp_secret_id = f"{random_secret_id}_totp"
self.secrets[totp_secret_id] = BitwardenConstants.TOTP self.secrets[totp_secret_id] = BitwardenConstants.TOTP
totp_secret_value = self.totp_secret_value_key(totp_secret_id) totp_secret_value = self.totp_secret_value_key(totp_secret_id)
self.secrets[totp_secret_value] = credential_item.totp self.secrets[totp_secret_value] = parse_totp_secret(credential_item.totp)
self.values[parameter.key]["totp"] = totp_secret_id self.values[parameter.key]["totp"] = totp_secret_id
async def register_aws_secret_parameter_value( async def register_aws_secret_parameter_value(