nedo-vision-worker 1.2.0__tar.gz → 1.2.2__tar.gz

{nedo_vision_worker-1.2.0 → nedo_vision_worker-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.2.0
+Version: 1.2.2
 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.0 → nedo_vision_worker-1.2.2}/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.0 → nedo_vision_worker-1.2.2}/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.0"
+__version__ = "1.2.2"
 __all__ = ["WorkerService"] 

{nedo_vision_worker-1.2.0 → nedo_vision_worker-1.2.2}/nedo_vision_worker/cli.py

@@ -49,7 +49,7 @@ Examples:
         parser.add_argument(
             "--version",
             action="version",
-            version="nedo-vision-worker 1.2.0"
+            version="nedo-vision-worker 1.2.1"
         )
         
         subparsers = parser.add_subparsers(

{nedo_vision_worker-1.2.0 → nedo_vision_worker-1.2.2}/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.0 → nedo_vision_worker-1.2.2}/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.0 → nedo_vision_worker-1.2.2}/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))

{nedo_vision_worker-1.2.0 → nedo_vision_worker-1.2.2}/nedo_vision_worker/services/VideoStreamClient.py

@@ -8,6 +8,7 @@ import fractions
 from urllib.parse import urlparse
 from .GrpcClientBase import GrpcClientBase
 from .SharedDirectDeviceClient import SharedDirectDeviceClient
+from ..util.FFmpegUtil import get_rtsp_ffmpeg_options, get_rtsp_probe_options
 from ..protos.VisionWorkerService_pb2_grpc import VideoStreamServiceStub
 from ..protos.VisionWorkerService_pb2 import VideoFrame
 
@@ -47,8 +48,10 @@ class VideoStreamClient(GrpcClientBase):
             ]
 
             if stream_type == "rtsp":
-                probe_cmd.insert(1, "-rtsp_transport")
-                probe_cmd.insert(2, "tcp")
+                probe_options = get_rtsp_probe_options()
+                # Insert options at the beginning (after ffprobe)
+                for i, option in enumerate(probe_options):
+                    probe_cmd.insert(1 + i, option)
 
             result = subprocess.run(probe_cmd, capture_output=True, text=True)
             probe_data = json.loads(result.stdout)
@@ -97,10 +100,8 @@ class VideoStreamClient(GrpcClientBase):
                 logging.error(f"Failed to create ffmpeg input for direct device: {e}")
                 return
         elif stream_type == "rtsp":
-            ffmpeg_input = (
-                ffmpeg
-                .input(url, rtsp_transport="tcp", fflags="nobuffer", timeout="5000000")
-            )
+            rtsp_options = get_rtsp_ffmpeg_options()
+            ffmpeg_input = ffmpeg.input(url, **rtsp_options)
         elif stream_type == "hls":
             ffmpeg_input = (
                 ffmpeg
@@ -148,8 +149,13 @@ class VideoStreamClient(GrpcClientBase):
             if stream_type == "direct":
                 self.shared_device_client.release_device_access(url)
             
-            stderr_output = process.stderr.read().decode()
-            logging.error(f"FFmpeg stderr: {stderr_output}")
+            try:
+                stderr_output = process.stderr.read().decode()
+                if stderr_output.strip():  # Only log if there's actual error content
+                    logging.error(f"FFmpeg stderr for {stream_type} stream: {stderr_output}")
+            except Exception as e:
+                logging.warning(f"Could not read FFmpeg stderr: {e}")
+            
             process.terminate()
             process.wait()