From 06644108786db68ca8b1eaf1d320c01663c4ac16 Mon Sep 17 00:00:00 2001
From: Vineeth Voruganti <13438633+VVoruganti@users.noreply.github.com>
Date: Thu, 31 Jul 2025 16:50:19 -0400
Subject: [PATCH] fix (queue): Add explicit rollback on integrity error (#174)

---
 src/deriver/queue_manager.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/deriver/queue_manager.py b/src/deriver/queue_manager.py
index 1e144d61..9cf83e7c 100644
--- a/src/deriver/queue_manager.py
+++ b/src/deriver/queue_manager.py
@@ -266,7 +266,8 @@ class QueueManager:
                                         )
                                         self.add_task(task)
                                 except IntegrityError:
-                                    # Note: rollback is handled by tracked_db dependency
+                                    # Rollback the failed transaction to clear the error state
+                                    await db.rollback()
                                     logger.debug(
                                         f"Failed to claim work unit {work_unit}, already owned"
                                     )