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

Introduce collectionid filter for bitwarden parameters (#454)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-06-10 22:06:58 -07:00
committed by GitHub
parent bb19a8ab8e
commit 0ede4fdfa0
9 changed files with 56 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/sdk/db/client.py
View File

@@ -1026,6 +1026,7 @@ class AgentDB:
url_parameter_key: str,
key: str,
description: str | None = None,
bitwarden_collection_id: str | None = None,
) -> BitwardenLoginCredentialParameter:
async with self.Session() as session:
bitwarden_login_credential_parameter = BitwardenLoginCredentialParameterModel(
@@ -1036,6 +1037,7 @@ class AgentDB:
url_parameter_key=url_parameter_key,
key=key,
description=description,
bitwarden_collection_id=bitwarden_collection_id,
)
session.add(bitwarden_login_credential_parameter)
await session.commit()

1
skyvern/forge/sdk/db/models.py
View File

@@ -278,6 +278,7 @@ class BitwardenLoginCredentialParameterModel(Base):
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=True, default=None)
url_parameter_key = Column(String, nullable=False)
created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
modified_at = Column(

2
skyvern/forge/sdk/db/utils.py
View File

@@ -241,6 +241,7 @@ def convert_to_bitwarden_login_credential_parameter(
LOG.debug(
"Converting BitwardenLoginCredentialParameterModel to BitwardenLoginCredentialParameter",
bitwarden_login_credential_parameter_id=bitwarden_login_credential_parameter_model.bitwarden_login_credential_parameter_id,
bitwarden_collection_id=bitwarden_login_credential_parameter_model.bitwarden_collection_id,
)
return BitwardenLoginCredentialParameter(
@@ -251,6 +252,7 @@ def convert_to_bitwarden_login_credential_parameter(
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,
bitwarden_collection_id=bitwarden_login_credential_parameter_model.bitwarden_collection_id,
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,