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

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

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

{nedo_vision_worker-1.2.2 → nedo_vision_worker-1.2.4}/nedo_vision_worker/services/AIModelClient.py

@@ -98,7 +98,7 @@ class AIModelClient(GrpcClientBase):
         if download_info.thread and download_info.thread.is_alive():
             download_info.thread.join(timeout=5)
         
-        self._update_model_status(model_id, "cancelled", "Download cancelled")
+        self._update_model_status(model_id, DownloadState.CANCELLED.value, "Download cancelled")
         logging.info(f"🛑 Cancelled download for {download_info.model_name}")
         return True
 
@@ -160,8 +160,11 @@ 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)
+            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.info(f"⏳ Model {local_model.name} is already being downloaded, skipping reschedule")
             return
 
         needs_update, changes = self._check_model_changes(server_model, local_model)
@@ -193,7 +196,6 @@ class AIModelClient(GrpcClientBase):
         self._schedule_download(server_model)
 
     def _check_model_changes(self, server_model, local_model) -> tuple[bool, List[str]]:
-        """Check if model needs update and return list of changes"""
         changes = []
         
         if server_model.version != local_model.version:
@@ -246,16 +248,22 @@ class AIModelClient(GrpcClientBase):
         local_model.set_main_class(server_model.main_class)
 
     def _schedule_download(self, model):
+        if self._is_downloading(model.id):
+            logging.info(f"⏳ Model {model.name} download already in progress, skipping reschedule")
+            return
+            
         self._cancel_download(model.id)
         
         download_info = DownloadInfo(
             model_id=model.id,
             model_name=model.name,
             version=model.version,
+            state=DownloadState.PENDING,
             start_time=time.time()
         )
         
-        self._update_model_status(model.id, "pending")
+        self._set_download_info(model.id, download_info)
+        self._update_model_status(model.id, DownloadState.PENDING.value)
         
         download_info.thread = threading.Thread(
             target=self._download_worker,
@@ -264,12 +272,11 @@ class AIModelClient(GrpcClientBase):
             name=f"ModelDownload-{model.id[:8]}"
         )
         download_info.thread.start()
-        self._set_download_info(model.id, download_info)
 
     def _download_worker(self, model, download_info: DownloadInfo):
         try:
             download_info.state = DownloadState.DOWNLOADING
-            self._update_model_status(model.id, "downloading")
+            self._update_model_status(model.id, DownloadState.DOWNLOADING.value)
             logging.info(f"📥 Downloading {model.name}...")
             
             success = self._download_model_file(model, download_info)
@@ -278,7 +285,7 @@ class AIModelClient(GrpcClientBase):
                 download_info.state = DownloadState.COMPLETED
                 download_info.end_time = time.time()
                 duration = download_info.end_time - download_info.start_time
-                self._update_model_status(model.id, "completed")
+                self._update_model_status(model.id, DownloadState.COMPLETED.value)
                 logging.info(f"✅ Downloaded {model.name} in {duration:.1f}s")
             else:
                 self._handle_download_failure(download_info, "Download failed")
@@ -291,7 +298,7 @@ class AIModelClient(GrpcClientBase):
     def _handle_download_failure(self, download_info: DownloadInfo, error: str):
         download_info.state = DownloadState.FAILED
         download_info.error_message = error
-        self._update_model_status(download_info.model_id, "failed", error)
+        self._update_model_status(download_info.model_id, DownloadState.FAILED.value, error)
         logging.error(f"❌ Failed to download {download_info.model_name}: {error}")
 
     def _save_changes(self, new_models: List, updated_models: List, models_to_delete: List):

{nedo_vision_worker-1.2.2 → nedo_vision_worker-1.2.4}/nedo_vision_worker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.2.2
+Version: 1.2.4
 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>