Implement BitwardenSensitiveInformationParameter (#589)

skyvern/forge/sdk/db/models.py

@@ -22,6 +22,7 @@ from skyvern.forge.sdk.db.id import (
     generate_artifact_id,
     generate_aws_secret_parameter_id,
     generate_bitwarden_login_credential_parameter_id,
+    generate_bitwarden_sensitive_information_parameter_id,
     generate_org_id,
     generate_organization_auth_token_id,
     generate_output_parameter_id,
@@ -293,6 +294,35 @@ class BitwardenLoginCredentialParameterModel(Base):
     deleted_at = Column(DateTime, nullable=True)
 
 
+class BitwardenSensitiveInformationParameterModel(Base):
+    __tablename__ = "bitwarden_sensitive_information_parameters"
+
+    bitwarden_sensitive_information_parameter_id = Column(
+        String,
+        primary_key=True,
+        index=True,
+        default=generate_bitwarden_sensitive_information_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)
+    bitwarden_collection_id = Column(String, nullable=False)
+    bitwarden_identity_key = Column(String, nullable=False)
+    # This is a list of fields to extract from the Bitwarden Identity.
+    bitwarden_identity_fields = Column(JSON, 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"