nedo-vision-worker 1.2.1__tar.gz → 1.2.3__tar.gz

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.2.1
+Version: 1.2.3
 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>
@@ -165,6 +165,8 @@ pip install -e .
 pip install -e .[dev]
 ```
 
+See [INSTALL.md](INSTALL.md) for detailed installation instructions.
+
 ## 🔍 System Diagnostics
 
 Before running the worker service, use the built-in diagnostic tool to verify your system:

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/README.md

@@ -112,6 +112,8 @@ pip install -e .
 pip install -e .[dev]
 ```
 
+See [INSTALL.md](INSTALL.md) for detailed installation instructions.
+
 ## 🔍 System Diagnostics
 
 Before running the worker service, use the built-in diagnostic tool to verify your system:

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/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.2.1"
+__version__ = "1.2.3"
 __all__ = ["WorkerService"] 

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/nedo_vision_worker/doctor.py

@@ -87,8 +87,8 @@ class WorkerServiceDoctor:
     def check_python_environment(self) -> None:
         """Comprehensive Python environment validation."""
         version = sys.version_info
-        min_version = (3, 8)
-        recommended_version = (3, 9)
+        min_version = (3, 10)
+        recommended_version = (3, 10)
         
         details = [
             f"Python {version.major}.{version.minor}.{version.micro}",
@@ -891,6 +891,36 @@ class WorkerServiceDoctor:
             performance_impact=performance_impact
         ))
     
+    def check_psutil_installation(self) -> None:
+        """Check if psutil is installed for system monitoring."""
+        details = []
+        recommendations = []
+        try:
+            import psutil
+            version = getattr(psutil, '__version__', 'N/A')
+            details.append(f"psutil version: {version}")
+            status = HealthStatus.GOOD
+            message = "psutil is installed"
+            is_blocking = False
+            performance_impact = "None"
+        except ImportError:
+            status = HealthStatus.WARNING
+            message = "psutil not installed"
+            details.append("System resource monitoring will be disabled.")
+            recommendations.append("Install psutil for system monitoring: pip install psutil")
+            is_blocking = False
+            performance_impact = "Low"
+
+        self._add_result(HealthCheck(
+            name="System Monitoring (psutil)",
+            status=status,
+            message=message,
+            details=details,
+            recommendations=recommendations if recommendations else None,
+            is_blocking=is_blocking,
+            performance_impact=performance_impact
+        ))
+    
     def run_comprehensive_health_check(self) -> List[HealthCheck]:
         """Execute all health checks with progress indication."""
         print("🏥 Nedo Vision Worker Service - Comprehensive Health Check")

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/nedo_vision_worker/repositories/WorkerSourceRepository.py

@@ -17,6 +17,15 @@ class WorkerSourceRepository:
         except Exception as e:
             logger.error(f"🚨 [APP] Database error while fetching worker sources: {e}", exc_info=True)
             return []
+        
+    
+    def get_worker_sources_by_worker_id(self, worker_id: str):
+        """Retrieve all worker sources from the database."""
+        try:
+            return self.session.query(WorkerSourceEntity).filter_by(worker_id=worker_id).all()
+        except Exception as e:
+            logger.error(f"🚨 [APP] Database error while fetching worker sources: {e}", exc_info=True)
+            return []
     
     def bulk_update_worker_sources(self, updated_records):
         """Batch update worker sources in the database."""
@@ -43,4 +52,4 @@ class WorkerSourceRepository:
                 return None
         except Exception as e:
             logger.error(f"🚨 [APP] Database error while fetching worker source by ID {worker_source_id}: {e}", exc_info=True)
-            return None
+            return None

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/nedo_vision_worker/services/AIModelClient.py

@@ -160,8 +160,12 @@ class AIModelClient(GrpcClientBase):
 
     def _handle_existing_model(self, server_model, local_model, updated_models: List):
         if not self._model_file_exists(local_model.file):
-            logging.warning(f"⚠️ Model file missing for {local_model.name}. Re-downloading...")
-            self._schedule_download(server_model)
+            # Only schedule download if not already downloading
+            if not self._is_downloading(server_model.id):
+                logging.warning(f"⚠️ Model file missing for {local_model.name}. Re-downloading...")
+                self._schedule_download(server_model)
+            else:
+                logging.debug(f"⏳ Model {local_model.name} is already being downloaded, skipping...")
             return
 
         needs_update, changes = self._check_model_changes(server_model, local_model)
@@ -246,6 +250,12 @@ class AIModelClient(GrpcClientBase):
         local_model.set_main_class(server_model.main_class)
 
     def _schedule_download(self, model):
+        # Check if already downloading - don't cancel and restart
+        if self._is_downloading(model.id):
+            logging.debug(f"⏳ Model {model.name} download already in progress, skipping reschedule")
+            return
+            
+        # Only cancel if there's a stale download entry
         self._cancel_download(model.id)
         
         download_info = DownloadInfo(

{nedo_vision_worker-1.2.1 → nedo_vision_worker-1.2.3}/nedo_vision_worker/services/GrpcClientBase.py

@@ -68,7 +68,7 @@ class GrpcClientBase:
                 self.connected = False
                 error_msg = str(e)
                 
-                logger.error(f"⚠️ Connection failed ({attempts}/{self.max_retries}): {error_msg}")
+                logger.error(f"⚠️ Connection failed ({attempts}/{self.max_retries}): {error_msg}", exc_info=True)
 
                 if attempts < self.max_retries:
                     sleep_time = retry_interval * (2 ** (attempts - 1))