11import base64
2- import struct
3- from typing import Dict , Optional , Protocol , Sequence , Tuple
2+ from typing import Optional , Protocol , Sequence , Tuple
43from workos .types .vault import VaultObject , ObjectVersion
54from workos .types .vault .key import DataKey , DataKeyPair , KeyContext , DecodedKeys
65from workos .types .list_resource import (
@@ -215,7 +214,8 @@ def read_object(
215214 object_id : str ,
216215 ) -> VaultObject :
217216 if not object_id :
218- raise ValueError ("Incomplete arguments: 'object_id' is a required argument" )
217+ raise ValueError (
218+ "Incomplete arguments: 'object_id' is a required argument" )
219219
220220 response = self ._http_client .request (
221221 RequestHelper .build_parameterized_url (
@@ -304,7 +304,8 @@ def update_object(
304304 version_check : Optional [str ] = None ,
305305 ) -> VaultObject :
306306 if not object_id :
307- raise ValueError ("Incomplete arguments: 'object_id' is a required argument" )
307+ raise ValueError (
308+ "Incomplete arguments: 'object_id' is a required argument" )
308309
309310 request_data = {
310311 "value" : value ,
@@ -329,7 +330,8 @@ def delete_object(
329330 object_id : str ,
330331 ) -> None :
331332 if not object_id :
332- raise ValueError ("Incomplete arguments: 'object_id' is a required argument" )
333+ raise ValueError (
334+ "Incomplete arguments: 'object_id' is a required argument" )
333335
334336 self ._http_client .request (
335337 RequestHelper .build_parameterized_url (
@@ -385,7 +387,8 @@ def encrypt(
385387 key = self ._base64_to_bytes (key_pair .data_key .key )
386388 key_blob = self ._base64_to_bytes (key_pair .encrypted_keys )
387389 prefix_len_buffer = self ._encode_u32 (len (key_blob ))
388- aad_buffer = associated_data .encode ("utf-8" ) if associated_data else None
390+ aad_buffer = associated_data .encode (
391+ "utf-8" ) if associated_data else None
389392 iv = self ._crypto_provider .random_bytes (12 )
390393
391394 result = self ._crypto_provider .encrypt (
@@ -409,7 +412,8 @@ def decrypt(
409412 data_key = self .decrypt_data_key (keys = decoded .keys )
410413
411414 key = self ._base64_to_bytes (data_key .key )
412- aad_buffer = associated_data .encode ("utf-8" ) if associated_data else None
415+ aad_buffer = associated_data .encode (
416+ "utf-8" ) if associated_data else None
413417
414418 decrypted_bytes = self ._crypto_provider .decrypt (
415419 ciphertext = decoded .ciphertext ,
@@ -491,7 +495,8 @@ def _decode_u32(self, buf: bytes) -> Tuple[int, int]:
491495
492496 for i , b in enumerate (buf ):
493497 if i > 4 :
494- raise ValueError ("LEB128 integer overflow (was more than 4 bytes)" )
498+ raise ValueError (
499+ "LEB128 integer overflow (was more than 4 bytes)" )
495500
496501 res |= (b & 0x7F ) << (7 * bit )
497502
0 commit comments