From 8155be9ff202e070a4b38dd08aa7c04fa9b5216f Mon Sep 17 00:00:00 2001 From: Kerem Yilmaz Date: Wed, 26 Jun 2024 08:55:48 -0700 Subject: [PATCH] Parameterize bitwarden_collection_id (#520) --- skyvern/forge/sdk/workflow/context_manager.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/workflow/context_manager.py b/skyvern/forge/sdk/workflow/context_manager.py index 03f12e8b..e5ba6ce4 100644 --- a/skyvern/forge/sdk/workflow/context_manager.py +++ b/skyvern/forge/sdk/workflow/context_manager.py @@ -151,13 +151,22 @@ class WorkflowRunContext: LOG.error(f"URL parameter {parameter.url_parameter_key} not found or has no value") raise ValueError("URL parameter for Bitwarden login credentials not found or has no value") + collection_id = None + if parameter.bitwarden_collection_id: + if self.has_parameter(parameter.bitwarden_collection_id) and self.has_value( + parameter.bitwarden_collection_id + ): + collection_id = self.values[parameter.bitwarden_collection_id] + else: + collection_id = parameter.bitwarden_collection_id + try: secret_credentials = BitwardenService.get_secret_value_from_url( client_id, client_secret, master_password, url, - collection_id=parameter.bitwarden_collection_id, + collection_id=collection_id, ) if secret_credentials: self.secrets[BitwardenConstants.URL] = url