From 1a1ba58a2c34e1f79ba0b9e1ce3669d791462640 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Wed, 23 Oct 2024 22:47:11 -0700
Subject: [PATCH] enforce collection id for login parameters

---
 skyvern/forge/sdk/workflow/exceptions.py | 8 ++++++++
 skyvern/forge/sdk/workflow/service.py    | 7 +++++++
 2 files changed, 15 insertions(+)

diff --git a/skyvern/forge/sdk/workflow/exceptions.py b/skyvern/forge/sdk/workflow/exceptions.py
index f9e64defa..96e09813d 100644
--- a/skyvern/forge/sdk/workflow/exceptions.py
+++ b/skyvern/forge/sdk/workflow/exceptions.py
@@ -99,3 +99,11 @@ def __init__(self, file_url: str, file_type: str, error: str) -> None:
             f"File URL {file_url} is not a valid {file_type} file. Error: {error}",
             status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
         )
+
+
+class WorkflowParameterMissingRequiredValue(BaseWorkflowHTTPException):
+    def __init__(self, workflow_parameter_type: str, workflow_parameter_key: str, required_value: str) -> None:
+        super().__init__(
+            f"Missing required value for workflow parameter. Workflow parameter type: {workflow_parameter_type}. workflow_parameter_key: {workflow_parameter_key}. Required value: {required_value}",
+            status_code=status.HTTP_400_BAD_REQUEST,
+        )
diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
index 6404cd375..4d9b4016d 100644
--- a/skyvern/forge/sdk/workflow/service.py
+++ b/skyvern/forge/sdk/workflow/service.py
@@ -19,6 +19,7 @@
     InvalidWorkflowDefinition,
     WorkflowDefinitionHasDuplicateParameterKeys,
     WorkflowDefinitionHasReservedParameterKeys,
+    WorkflowParameterMissingRequiredValue,
 )
 from skyvern.forge.sdk.workflow.models.block import (
     BlockStatus,
@@ -957,6 +958,12 @@ async def create_workflow_from_request(
                         description=parameter.description,
                     )
                 elif parameter.parameter_type == ParameterType.BITWARDEN_LOGIN_CREDENTIAL:
+                    if not parameter.bitwarden_collection_id:
+                        raise WorkflowParameterMissingRequiredValue(
+                            workflow_parameter_type=ParameterType.BITWARDEN_LOGIN_CREDENTIAL,
+                            workflow_parameter_key=parameter.key,
+                            required_value="bitwarden_collection_id",
+                        )
                     parameters[parameter.key] = await self.create_bitwarden_login_credential_parameter(
                         workflow_id=workflow.workflow_id,
                         bitwarden_client_id_aws_secret_key=parameter.bitwarden_client_id_aws_secret_key,