my-aws-helpers 4.6.0__tar.gz → 4.8.0__tar.gz

{my_aws_helpers-4.6.0 → my_aws_helpers-4.8.0}/PKG-INFO

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

{my_aws_helpers-4.6.0 → my_aws_helpers-4.8.0}/my_aws_helpers/bedrock.py

@@ -13,6 +13,13 @@ import concurrent.futures
 from dataclasses import dataclass
 
 
+class ImageType(str, Enum):
+    gif = "gif"
+    jpeg = "jpeg"
+    png = "png"
+    webp = "webp"
+
+
 class PromptType(str, Enum):
     transaction_headers = "transactions_headers_prompt_v2.txt"
     transactions = "transactions_prompt.txt"
@@ -239,30 +246,25 @@ class Bedrock:
             self.logger.exception(e)
             return []
 
+    def _get_image_block(self, image: bytes, image_content_type: ImageType) -> dict:
+        return {
+            "image": {
+                "format": image_content_type,
+                "source": {
+                    "bytes": image,
+                },
+            }
+        }
 
-    def image_analysis(self, images: List[bytes]) -> OCRResult:
-        instruction = (
-            "You are an assistant for real estate investors. Analyze the following property images "
-            "and return structured insights. Focus only on what can reasonably be inferred from the images. "
-            "Do not make up financial or legal data. Be objective and consistent.\n\n"
-            "Return ONLY valid JSON with this structure:\n"
-            "{\n"
-            '  "condition_score": "Integer from 1-10 (1 = very poor, 10 = excellent)",\n'
-            '  "materials": {\n'
-            '    "exterior_walls": "string", "roof": "string", "windows": "string",\n'
-            '    "flooring": "string", "kitchen_benchtops": "string", "bathroom_fittings": "string"\n'
-            "  },\n"
-            '  "features": [], "maintenance_risks": [], "energy_efficiency": [],\n'
-            '  "location_clues": [], "upgrade_potential": []\n'
-            "}"
-        )
-
+    def image_analysis(
+        self, images: List[bytes], prompt: str, image_content_type: ImageType
+    ) -> OCRResult:
 
-        system_prompt = [{"text": instruction}]
+        system_prompt = [{"text": prompt}]
         message = [
             {
                 "role": "user",
-                "content": [self._get_image_block(image=image) for image in images],
+                "content": [self._get_image_block(image=image, image_content_type=image_content_type) for image in images],
             }
         ]
         response = self.client.converse(
@@ -274,4 +276,4 @@ class Bedrock:
             response["output"]["message"]["content"][0]["text"]
         )
         result["token_usage"] = response["usage"]
-        return OCRResult.from_dict(data=result) 
+        return OCRResult.from_dict(data=result)

{my_aws_helpers-4.6.0 → my_aws_helpers-4.8.0}/my_aws_helpers/s3.py

@@ -41,6 +41,7 @@ class S3Serialiser:
             return {k: S3Serialiser.object_serialiser(v) for k, v in obj.items()}
         return S3Serialiser._serialise(obj=obj)
 
+
 class S3Location:
     bucket: str
     file_name: str
@@ -96,13 +97,14 @@ class S3:
 
     def list_objects_by_prefix(self, bucket_name: str, prefix: str) -> List[Dict]:
         try:
-            response = self.client.client.list_objects_v2(
-                Bucket=bucket_name,
-                Prefix=prefix
+            response = self.client.list_objects_v2(
+                Bucket=bucket_name, Prefix=prefix
             )
-            return response['Contents']
+            return response["Contents"]
         except Exception as e:
-            logger.exception(f"Failed to get objects from s3: {bucket_name}/{prefix} due to {e}")
+            logger.exception(
+                f"Failed to get objects from s3: {bucket_name}/{prefix} due to {e}"
+            )
             return []
 
     def get_object(self, bucket_name: str, file_name: str):
@@ -122,7 +124,7 @@ class S3:
                 "Key": file_name,
             },
             ExpiresIn=expires_in,
-        ) 
+        )
 
     def get_s3_location_from_bucket_file(
         bucket_name: str, file_name: str
@@ -258,7 +260,9 @@ class S3:
             )
         except Exception as e:
             logger.exception(f"Failed to save jpeg to s3 due to {e}")
-            return None        
+            return None
 
     def read_dict_from_s3(self, s3_location: S3Location) -> dict:
-        return json.loads(self.read_binary_from_s3(s3_location=s3_location).decode("utf-8"))
+        return json.loads(
+            self.read_binary_from_s3(s3_location=s3_location).decode("utf-8")
+        )

{my_aws_helpers-4.6.0 → my_aws_helpers-4.8.0}/my_aws_helpers.egg-info/PKG-INFO

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

{my_aws_helpers-4.6.0 → my_aws_helpers-4.8.0}/setup.py

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