my-aws-helpers 5.0.1__tar.gz → 5.2.0__tar.gz

{my_aws_helpers-5.0.1 → my_aws_helpers-5.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my_aws_helpers
-Version: 5.0.1
+Version: 5.2.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-5.0.1 → my_aws_helpers-5.2.0}/my_aws_helpers/bedrock.py

@@ -5,12 +5,13 @@ import json
 import time
 import os
 import io
-import base64
+from copy import copy
 from typing import Optional, List, Dict
 from enum import Enum
 import pymupdf
 import concurrent.futures
 from dataclasses import dataclass
+from my_aws_helpers.s3 import S3Serialiser
 
 
 class ImageType(str, Enum):
@@ -56,6 +57,14 @@ class OCRResult:
             page_number=data.get("page_number", 0),
         )
 
+    @classmethod
+    def from_s3_representation(cls, obj: dict) -> OCRResult:
+        return cls(**obj)
+
+    def to_s3_representation(self) -> dict:
+        obj = copy(vars(self))
+        return S3Serialiser.object_serialiser(obj=obj)
+
 
 class Bedrock:
     def __init__(

{my_aws_helpers-5.0.1 → my_aws_helpers-5.2.0}/my_aws_helpers/s3.py

@@ -34,7 +34,7 @@ class S3Serialiser:
         if isinstance(obj, datetime) or isinstance(obj, date):
             return obj.isoformat()
         if isinstance(obj, S3Location):
-            return obj.location        
+            return obj.location
         return obj
 
     @staticmethod
@@ -103,7 +103,9 @@ class S3:
         try:
             response = self.client.list_objects_v2(Bucket=bucket_name, Prefix=prefix)
             contents = response["Contents"]
-            locations = [S3Location(bucket = c['Bucket'], file_name = c["Key"]) for c in contents]
+            locations = [
+                S3Location(bucket=bucket_name, file_name=c["Key"]) for c in contents
+            ]
             return locations
         except Exception as e:
             logger.exception(

{my_aws_helpers-5.0.1 → my_aws_helpers-5.2.0}/my_aws_helpers.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my-aws-helpers
-Version: 5.0.1
+Version: 5.2.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-5.0.1 → my_aws_helpers-5.2.0}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_namespace_packages, setup
 
 base_path = os.path.abspath(os.path.dirname(__file__))
 
-version = "5.0.1"
+version = "5.2.0"
 
 core = [
     "boto3==1.34.36",