Implement BitwardenLoginCredentialParameter (#151)

skyvern/forge/sdk/db/client.py

@@ -13,6 +13,7 @@ from skyvern.forge.sdk.db.exceptions import NotFoundError
 from skyvern.forge.sdk.db.models import (
     ArtifactModel,
     AWSSecretParameterModel,
+    BitwardenLoginCredentialParameterModel,
     OrganizationAuthTokenModel,
     OrganizationModel,
     OutputParameterModel,
@@ -28,6 +29,7 @@ from skyvern.forge.sdk.db.utils import (
     _custom_json_serializer,
     convert_to_artifact,
     convert_to_aws_secret_parameter,
+    convert_to_bitwarden_login_credential_parameter,
     convert_to_organization,
     convert_to_organization_auth_token,
     convert_to_output_parameter,
@@ -43,6 +45,7 @@ from skyvern.forge.sdk.models import Organization, OrganizationAuthToken, Step,
 from skyvern.forge.sdk.schemas.tasks import ProxyLocation, Task, TaskStatus
 from skyvern.forge.sdk.workflow.models.parameter import (
     AWSSecretParameter,
+    BitwardenLoginCredentialParameter,
     OutputParameter,
     WorkflowParameter,
     WorkflowParameterType,
@@ -844,6 +847,31 @@ class AgentDB:
             await session.refresh(aws_secret_parameter)
             return convert_to_aws_secret_parameter(aws_secret_parameter)
 
+    async def create_bitwarden_login_credential_parameter(
+        self,
+        workflow_id: str,
+        bitwarden_client_id_aws_secret_key: str,
+        bitwarden_client_secret_aws_secret_key: str,
+        bitwarden_master_password_aws_secret_key: str,
+        url_parameter_key: str,
+        key: str,
+        description: str | None = None,
+    ) -> BitwardenLoginCredentialParameter:
+        async with self.Session() as session:
+            bitwarden_login_credential_parameter = BitwardenLoginCredentialParameterModel(
+                workflow_id=workflow_id,
+                bitwarden_client_id_aws_secret_key=bitwarden_client_id_aws_secret_key,
+                bitwarden_client_secret_aws_secret_key=bitwarden_client_secret_aws_secret_key,
+                bitwarden_master_password_aws_secret_key=bitwarden_master_password_aws_secret_key,
+                url_parameter_key=url_parameter_key,
+                key=key,
+                description=description,
+            )
+            session.add(bitwarden_login_credential_parameter)
+            await session.commit()
+            await session.refresh(bitwarden_login_credential_parameter)
+            return convert_to_bitwarden_login_credential_parameter(bitwarden_login_credential_parameter)
+
     async def create_output_parameter(
         self,
         workflow_id: str,

skyvern/forge/sdk/db/id.py

@@ -38,6 +38,7 @@ WORKFLOW_RUN_PREFIX = "wr"
 WORKFLOW_PARAMETER_PREFIX = "wp"
 AWS_SECRET_PARAMETER_PREFIX = "asp"
 OUTPUT_PARAMETER_PREFIX = "op"
+BITWARDEN_LOGIN_CREDENTIAL_PARAMETER_PREFIX = "blc"
 
 
 def generate_workflow_id() -> str:
@@ -65,6 +66,11 @@ def generate_output_parameter_id() -> str:
     return f"{OUTPUT_PARAMETER_PREFIX}_{int_id}"
 
 
+def generate_bitwarden_login_credential_parameter_id() -> str:
+    int_id = generate_id()
+    return f"{BITWARDEN_LOGIN_CREDENTIAL_PARAMETER_PREFIX}_{int_id}"
+
+
 def generate_organization_auth_token_id() -> str:
     int_id = generate_id()
     return f"{ORGANIZATION_AUTH_TOKEN_PREFIX}_{int_id}"

skyvern/forge/sdk/db/models.py

@@ -8,6 +8,7 @@ from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType
 from skyvern.forge.sdk.db.id import (
     generate_artifact_id,
     generate_aws_secret_parameter_id,
+    generate_bitwarden_login_credential_parameter_id,
     generate_org_id,
     generate_organization_auth_token_id,
     generate_output_parameter_id,
@@ -177,6 +178,24 @@ class AWSSecretParameterModel(Base):
     deleted_at = Column(DateTime, nullable=True)
 
 
+class BitwardenLoginCredentialParameterModel(Base):
+    __tablename__ = "bitwarden_login_credential_parameters"
+
+    bitwarden_login_credential_parameter_id = Column(
+        String, primary_key=True, index=True, default=generate_bitwarden_login_credential_parameter_id
+    )
+    workflow_id = Column(String, ForeignKey("workflows.workflow_id"), index=True, nullable=False)
+    key = Column(String, nullable=False)
+    description = Column(String, nullable=True)
+    bitwarden_client_id_aws_secret_key = Column(String, nullable=False)
+    bitwarden_client_secret_aws_secret_key = Column(String, nullable=False)
+    bitwarden_master_password_aws_secret_key = Column(String, nullable=False)
+    url_parameter_key = Column(String, nullable=False)
+    created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
+    modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)
+    deleted_at = Column(DateTime, nullable=True)
+
+
 class WorkflowRunParameterModel(Base):
     __tablename__ = "workflow_run_parameters"
 

skyvern/forge/sdk/db/utils.py

@@ -9,6 +9,7 @@ from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType
 from skyvern.forge.sdk.db.models import (
     ArtifactModel,
     AWSSecretParameterModel,
+    BitwardenLoginCredentialParameterModel,
     OrganizationAuthTokenModel,
     OrganizationModel,
     OutputParameterModel,
@@ -24,6 +25,7 @@ from skyvern.forge.sdk.models import Organization, OrganizationAuthToken, Step,
 from skyvern.forge.sdk.schemas.tasks import ProxyLocation, Task, TaskStatus
 from skyvern.forge.sdk.workflow.models.parameter import (
     AWSSecretParameter,
+    BitwardenLoginCredentialParameter,
     OutputParameter,
     WorkflowParameter,
     WorkflowParameterType,
@@ -211,6 +213,30 @@ def convert_to_aws_secret_parameter(
     )
 
 
+def convert_to_bitwarden_login_credential_parameter(
+    bitwarden_login_credential_parameter_model: BitwardenLoginCredentialParameterModel, debug_enabled: bool = False
+) -> BitwardenLoginCredentialParameter:
+    if debug_enabled:
+        LOG.debug(
+            "Converting BitwardenLoginCredentialParameterModel to BitwardenLoginCredentialParameter",
+            bitwarden_login_credential_parameter_id=bitwarden_login_credential_parameter_model.bitwarden_login_credential_parameter_id,
+        )
+
+    return BitwardenLoginCredentialParameter(
+        bitwarden_login_credential_parameter_id=bitwarden_login_credential_parameter_model.bitwarden_login_credential_parameter_id,
+        workflow_id=bitwarden_login_credential_parameter_model.workflow_id,
+        key=bitwarden_login_credential_parameter_model.key,
+        description=bitwarden_login_credential_parameter_model.description,
+        bitwarden_client_id_aws_secret_key=bitwarden_login_credential_parameter_model.bitwarden_client_id_aws_secret_key,
+        bitwarden_client_secret_aws_secret_key=bitwarden_login_credential_parameter_model.bitwarden_client_secret_aws_secret_key,
+        bitwarden_master_password_aws_secret_key=bitwarden_login_credential_parameter_model.bitwarden_master_password_aws_secret_key,
+        url_parameter_key=bitwarden_login_credential_parameter_model.url_parameter_key,
+        created_at=bitwarden_login_credential_parameter_model.created_at,
+        modified_at=bitwarden_login_credential_parameter_model.modified_at,
+        deleted_at=bitwarden_login_credential_parameter_model.deleted_at,
+    )
+
+
 def convert_to_output_parameter(
     output_parameter_model: OutputParameterModel, debug_enabled: bool = False
 ) -> OutputParameter: