update to ack

worker.py

@@ -83,7 +83,6 @@ class RabbitMQWorker:
         logger.info("[Worker %s] Received message: %s, headers: %s", thread_id, body, headers)
 
         try:
-            ch.basic_ack(delivery_tag=method.delivery_tag)
             contexts = []
 
             body_dict = json.loads(body)
@@ -116,6 +115,7 @@ class RabbitMQWorker:
                 # Publish results
                 if self.publish_message(body_dict, headers):
                     logger.info("[Worker %s] Successfully published results to ml_server.", thread_id)
+                    ch.basic_ack(delivery_tag=method.delivery_tag)
                 else:
                     ch.basic_nack(delivery_tag=method.delivery_tag, requeue=True)
                     logger.error("[Worker %s] Failed to publish results.", thread_id)
@@ -123,7 +123,7 @@ class RabbitMQWorker:
                 logger.info("[Worker %s] Contexts: %s", thread_id, contexts)
                 
             else:
-                ch.basic_ack(delivery_tag=method.delivery_tag)
+                ch.basic_ack(delivery_tag=method.delivery_tag, requeue=False)
                 logger.warning("[Worker %s] Unknown pattern type in headers: %s", thread_id, pattern)
                 
         except Exception as e: