my-aws-helpers 6.0.1__tar.gz → 6.0.2__tar.gz

{my_aws_helpers-6.0.1 → my_aws_helpers-6.0.2}/PKG-INFO

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

{my_aws_helpers-6.0.1 → my_aws_helpers-6.0.2}/my_aws_helpers/bedrock.py

@@ -65,12 +65,14 @@ class OCRResult(BaseS3Object):
 
     @classmethod
     def from_s3_representation(cls, obj: dict) -> OCRResult:
-        obj['token_usage'] = TokenUsage.from_dict(obj.get("token_usage", {})),
+        obj["token_usage"] = (TokenUsage.from_dict(obj.get("token_usage", {})),)
         return cls(**obj)
 
     def to_s3_representation(self) -> dict:
         obj = copy(vars(self))
-        obj['token_usage'] = S3Serialiser.object_serialiser(obj=vars(obj['token_usage']))
+        obj["token_usage"] = S3Serialiser.object_serialiser(
+            obj=vars(obj["token_usage"])
+        )
         return S3Serialiser.object_serialiser(obj=obj)
 
     def get_save_location(self, bucket_name: str) -> S3Location:
@@ -123,7 +125,7 @@ class OCRResultQueries(BaseS3Queries):
 class Bedrock:
     def __init__(
         self,
-        model_id: str = "apac.anthropic.claude-3-5-sonnet-20241022-v2:0",#anthropic.claude-sonnet-4-20250514-v1:0
+        model_id: str = "apac.anthropic.claude-3-5-sonnet-20241022-v2:0",  # anthropic.claude-sonnet-4-20250514-v1:0
         logger=None,
         sleep_time: float = 1.0,
     ):
@@ -354,7 +356,7 @@ class Bedrock:
 
         result = {}
         result["content"] = Bedrock.extract_json_from_markdown(
-            text = response["output"]["message"]["content"][0]["text"]
+            text=response["output"]["message"]["content"][0]["text"]
         )
         result["token_usage"] = response["usage"]
         return OCRResult.from_dict(data=result)

{my_aws_helpers-6.0.1 → my_aws_helpers-6.0.2}/my_aws_helpers/dynamo.py

@@ -113,7 +113,7 @@ class Dynamo:
         return self.table.get_item(Item=item)
 
     def delete_item(self, item: dict):
-        return self.table.delete_item(Item=item)
+        return self.table.delete_item(Key=item)
 
     def batch_put(self, items: List[dict]) -> None:
         with self.table.batch_writer() as batch:

{my_aws_helpers-6.0.1 → my_aws_helpers-6.0.2}/my_aws_helpers/s3.py

@@ -112,12 +112,11 @@ class S3:
                     response = self.client.list_objects_v2(
                         Bucket=bucket_name,
                         Prefix=prefix,
-                        ContinuationToken=continuation_token
+                        ContinuationToken=continuation_token,
                     )
                 else:
                     response = self.client.list_objects_v2(
-                        Bucket=bucket_name,
-                        Prefix=prefix
+                        Bucket=bucket_name, Prefix=prefix
                     )
 
                 # Append current batch

{my_aws_helpers-6.0.1 → my_aws_helpers-6.0.2}/my_aws_helpers.egg-info/PKG-INFO

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

{my_aws_helpers-6.0.1 → my_aws_helpers-6.0.2}/setup.py

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