nedo-vision-worker 1.0.0__tar.gz → 1.1.0__tar.gz

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.0.0
+Version: 1.1.0
 Summary: Nedo Vision Worker Service Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/__init__.py

@@ -6,5 +6,5 @@ A library for running worker agents in the Nedo Vision platform.
 
 from .worker_service import WorkerService
 
-__version__ = "1.0.0"
+__version__ = "1.1.0"
 __all__ = ["WorkerService"] 

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/cli.py

@@ -115,7 +115,7 @@ Examples:
     parser.add_argument(
         "--version",
         action="version",
-        version="nedo-vision-worker 1.0.0"
+        version="nedo-vision-worker 1.1.0"
     )
     
     parser.add_argument(

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/worker/CoreActionWorker.py

@@ -72,12 +72,23 @@ class CoreActionWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for core action messages...")
+                logger.info("📡 [APP] Starting core action message listener...")
                 self.listener.start_listening(
                     exchange_name="nedo.worker.core.action", 
                     queue_name=f"nedo.worker.core.{self.worker_id}"
                 )
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Core action listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Core action listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Core Action Worker loop.", exc_info=True)
 

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/worker/PipelineActionWorker.py

@@ -73,12 +73,23 @@ class PipelineActionWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for Pipeline action messages...")
+                logger.info("📡 [APP] Starting pipeline action message listener...")
                 self.listener.start_listening(
                     exchange_name="nedo.worker.pipeline.action", 
                     queue_name=f"nedo.worker.pipeline.{self.worker_id}"
                 )
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Pipeline action listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Pipeline action listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Pipeline Action Worker loop.", exc_info=True)
 

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/worker/PipelineImageWorker.py

@@ -78,9 +78,20 @@ class PipelineImageWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for image request messages...")
+                logger.info("📡 [APP] Starting image request message listener...")
                 self.listener.start_listening(exchange_name="nedo.pipeline.image.request", queue_name=f"nedo.pipeline.request.{self.worker_id}")
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Image request listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Image request listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Pipeline Image Worker loop.", exc_info=True)
 

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/worker/RabbitMQListener.py

@@ -91,6 +91,10 @@ class RabbitMQListener:
         if self.listener_thread and self.listener_thread.is_alive():
             logger.warning("⚠️ [APP] RabbitMQ listener is already running.")
             return
+            
+        # Clean up any dead thread reference
+        if self.listener_thread and not self.listener_thread.is_alive():
+            self.listener_thread = None
 
         def run():
             while not self.stop_event.is_set():
@@ -137,6 +141,13 @@ class RabbitMQListener:
             logger.error(f"🚨 [APP] Error during RabbitMQ shutdown: {e}")
 
         self._cleanup_connection()
+        
+        # Clean up the listener thread reference
+        if self.listener_thread:
+            if self.listener_thread.is_alive():
+                self.listener_thread.join(timeout=5)  # Wait up to 5 seconds for thread to finish
+            self.listener_thread = None
+            
         logger.info("🔌 [APP] RabbitMQ listener stopped.")
 
     def _cleanup_connection(self):

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker/worker/VideoStreamWorker.py

@@ -80,9 +80,20 @@ class VideoStreamWorker:
         """Main loop to manage RabbitMQ listener."""
         try:
             while not self.stop_event.is_set():
-                logger.info("📡 [APP] Waiting for video stream messages...")
+                logger.info("📡 [APP] Starting video stream message listener...")
                 self.listener.start_listening(exchange_name="nedo.worker.stream.preview", queue_name=f"nedo.worker.preview.{self.worker_id}")
-                safe_join_thread(self.listener.listener_thread)
+                
+                # Wait for the listener thread to finish (connection lost or stop requested)
+                while not self.stop_event.is_set() and self.listener.listener_thread and self.listener.listener_thread.is_alive():
+                    self.listener.listener_thread.join(timeout=5)  # Check every 5 seconds
+                
+                if not self.stop_event.is_set():
+                    logger.warning("⚠️ [APP] Video stream listener disconnected. Attempting to reconnect in 10 seconds...")
+                    self.stop_event.wait(10)  # Wait 10 seconds before reconnecting
+                else:
+                    logger.info("📡 [APP] Video stream listener stopped.")
+                    break
+                    
         except Exception as e:
             logger.error("🚨 [APP] Unexpected error in Stream Worker loop.", exc_info=True)
 

{nedo_vision_worker-1.0.0 → nedo_vision_worker-1.1.0}/nedo_vision_worker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.0.0
+Version: 1.1.0
 Summary: Nedo Vision Worker Service Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>