documente_shared 0.1.7__tar.gz → 0.1.9__tar.gz

{documente_shared-0.1.7 → documente_shared-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.7
+Version: 0.1.9
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech

{documente_shared-0.1.7 → documente_shared-0.1.9}/documente_shared/infrastructure/sqs_queue.py

@@ -7,6 +7,8 @@ from dataclasses import dataclass
 @dataclass
 class SQSQueue(object):
     queue_url: str
+    visibility_timeout: int =  60 * 10
+    waiting_timeout: int = 20
 
     def __post_init__(self):
         self._client = boto3.client('sqs')
@@ -35,3 +37,12 @@ class SQSQueue(object):
             QueueUrl=self.queue_url,
             ReceiptHandle=receipt_handle
         )
+
+    def fetch_messages(self, num_messages: int = 1) -> list[dict]:
+        response = self._client.receive_message(
+            QueueUrl=self.queue_url,
+            MaxNumberOfMessages=num_messages,
+            VisibilityTimeout=self.visibility_timeout,
+            WaitTimeSeconds=self.waiting_timeout,
+        )
+        return response.get('Messages', [])

{documente_shared-0.1.7 → documente_shared-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.7"
+version = "0.1.9"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"