diff --git a/src/vl_apify_shared/data_store.py b/src/vl_apify_shared/data_store.py index 37032c3..bf96dac 100644 --- a/src/vl_apify_shared/data_store.py +++ b/src/vl_apify_shared/data_store.py @@ -70,6 +70,11 @@ class DataStore: raise TypeError(f"Unsupported data type for CSV upload: {type(data)}") await self.__s3_upload(key, body, "text/csv") + async def s3_upload_pdf( + self, key: str, data: bytes, metadata: dict[str, object] = {} + ): + await self.__s3_upload(key, data, "application/pdf", metadata=metadata) + async def __s3_upload( self, key: str, body: bytes, content_type: str, metadata: dict[str, object] = {} ):