PyPI - altcodepro-polydb-python - Versions diffs - 2.3.28__tar.gz → 2.3.29__tar.gz - Mend

@@ -75,6 +75,23 @@ class TestReceiveReturnsReceiptHandle:
         assert "receipt_handle" in received
         assert received["receipt_handle"]  # truthy -- WorkerPool checks `if receipt_handle`
+    def test_receive_surfaces_dequeue_count(self):
+        """WorkerPool's poison-message guard thresholds on this to
+        dead-letter a message the queue keeps redelivering."""
+        adapter = _make_adapter()
+        client = _mock_queue_client(adapter)
+        raw_msg = MagicMock()
+        raw_msg.id = "msg-123"
+        raw_msg.pop_receipt = "pop-abc"
+        raw_msg.content = json.dumps({"hello": "world"})
+        raw_msg.dequeue_count = 7
+        client.receive_messages.return_value = [raw_msg]
+        [received] = adapter.receive(queue_name="default", max_messages=1)
+        assert received["dequeue_count"] == 7
     def test_receipt_handle_round_trips_through_decode(self):
         message_id, pop_receipt = AzureQueueAdapter._decode_receipt(
             AzureQueueAdapter._encode_receipt("msg-123", "pop-abc")
@@ -82,6 +99,33 @@ class TestReceiveReturnsReceiptHandle:
         assert message_id == "msg-123"
         assert pop_receipt == "pop-abc"
+    def test_receive_defaults_to_a_long_visibility_timeout(self):
+        """The Azure SDK's own default (30s) is shorter than a real task
+        can take (an LLM call alone can run 30-120s+), so a still-in-flight
+        message became visible again and got picked up by a second worker
+        mid-processing -- two concurrent executions of the same task. This
+        guards against that default silently regressing back to 30s (or
+        being dropped entirely, which is the SDK's fallback)."""
+        adapter = _make_adapter()
+        client = _mock_queue_client(adapter)
+        client.receive_messages.return_value = []
+        adapter.receive(queue_name="default", max_messages=1)
+        _, kwargs = client.receive_messages.call_args
+        assert kwargs["visibility_timeout"] == AzureQueueAdapter.DEFAULT_VISIBILITY_TIMEOUT
+        assert kwargs["visibility_timeout"] >= 300
+    def test_receive_honours_an_explicit_visibility_timeout(self):
+        adapter = _make_adapter()
+        client = _mock_queue_client(adapter)
+        client.receive_messages.return_value = []
+        adapter.receive(queue_name="default", max_messages=1, visibility_timeout=45)
+        _, kwargs = client.receive_messages.call_args
+        assert kwargs["visibility_timeout"] == 45
 class TestAckAcceptsSingleReceiptHandle:
     def test_ack_with_generic_single_arg_deletes_using_decoded_parts(self):

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.3.28
+Version: 2.3.29
 Summary: Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety
 Author: AltCodePro
 Project-URL: Homepage, https://github.com/altcodepro/polydb-python

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "altcodepro-polydb-python"
-version = "2.3.28"
+version = "2.3.29"
 description = "Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: altcodepro-polydb-python
-Version: 2.3.28
+Version: 2.3.29
 Summary: Production-ready multi-cloud database abstraction layer with connection pooling, retry logic, and thread safety
 Author: AltCodePro
 Project-URL: Homepage, https://github.com/altcodepro/polydb-python

@@ -110,15 +110,38 @@ class AzureQueueAdapter(QueueAdapter):
         except Exception as e:
             raise QueueError(f"Azure Queue send failed: {e}")
+    # How long a delivered message stays invisible to other consumers before
+    # Azure hands it out again. The SDK's own default is 30s — far shorter
+    # than a real task (an LLM generation call alone can run 30-120s+), so
+    # a still-in-flight message was becoming visible again and getting
+    # picked up by a second worker mid-processing: two concurrent
+    # executions of the same task, one of which stomps/duplicates the
+    # other's result. 300s covers realistic worst-case processing; ack()
+    # deletes the message immediately on completion regardless; a genuinely
+    # crashed worker still recovers the message after this timeout.
+    DEFAULT_VISIBILITY_TIMEOUT = 300
     @retry(max_attempts=3, delay=1.0, exceptions=(QueueError,))
-    def receive(self, queue_name: str = "default", max_messages: int = 1) -> List[Dict[str, Any]]:
+    def receive(
+        self,
+        queue_name: str = "default",
+        max_messages: int = 1,
+        visibility_timeout: Optional[int] = None,
+    ) -> List[Dict[str, Any]]:
         """Receive messages"""
         try:
             queue_name = self._normalize_queue_name(queue_name)
             queue_client = self._get_queue(queue_name)
-            messages = queue_client.receive_messages(max_messages=max_messages)
+            messages = queue_client.receive_messages(
+                max_messages=max_messages,
+                visibility_timeout=(
+                    visibility_timeout
+                    if visibility_timeout is not None
+                    else self.DEFAULT_VISIBILITY_TIMEOUT
+                ),
+            )
             results = []
@@ -135,6 +158,12 @@ class AzureQueueAdapter(QueueAdapter):
                         # Azure's visibility timeout elapses — replaying the
                         # same task forever even after it already ran.
                         "receipt_handle": self._encode_receipt(msg.id, msg.pop_receipt),
+                        # How many times Azure has handed this same message
+                        # out (1 on first delivery). Lets a consumer (e.g.
+                        # WorkerPool) detect a poison message that keeps
+                        # failing to ack/process and dead-letter it instead
+                        # of retrying forever.
+                        "dequeue_count": msg.dequeue_count,
                         "body": payload,
                     }
                 )

altcodepro-polydb-python 2.3.28__tar.gz → 2.3.29__tar.gz

altcodepro-polydb-python 2.3.28tar.gz → 2.3.29tar.gz