ray-embedding 0.13.9__py3-none-any.whl → 0.13.10__py3-none-any.whl

ray_embedding/node_health.py

@@ -76,19 +76,22 @@ class NodeHealthTracker:
             target_node_id = next(node["NodeID"] for node in ray.nodes() if node["Alive"] and node["NodeManagerAddress"] == node_ip)
             assert target_node_id, f"No node found with IP {node_ip}"
             prefixes = tuple(f"SERVE_REPLICA::{self.app_name}#{d}" for d in self.tracked_model_deployments)
-            self.logger.info(f"Checking actors with prefixes: {prefixes} in node IP {node_ip}, ID {target_node_id}")
 
-            for actor in list_actors(detail=False, filters=[("node_id", "=", target_node_id)]):
-                self.logger.info(f"Checking actor: {actor}")
-                if actor.state in ["DEPENDENCIES_UNREADY", 'PENDING_CREATION', 'ALIVE', 'RESTARTING']:
-                    for prefix in prefixes:
-                        if actor.name.startswith(prefix):
-                            self.logger.info(f"Found a replica {actor.name} of "
-                                             f"tracked_deployments={self.tracked_model_deployments} "
-                                             f"running in node IP {node_ip}, node ID {target_node_id}.")
-                            return True
+            actors_list = list_actors(detail=True, filters=[("node_id", "=", target_node_id), ("state", "!=", "DEAD")])
+            if actors_list:
+                self.logger.info(f"Checking non-dead actors with prefixes: {prefixes} in node IP {node_ip}, ID {target_node_id}")
+                for actor in actors_list:
+                    self.logger.info(f"Checking actor: {actor}")
+                    if actor.state in ["DEPENDENCIES_UNREADY", 'PENDING_CREATION', 'ALIVE', 'RESTARTING']:
+                        for prefix in prefixes:
+                            if actor.name.startswith(prefix):
+                                self.logger.info(f"Found a replica {actor.name} of "
+                                                 f"tracked_deployments={self.tracked_model_deployments} "
+                                                 f"running in node IP {node_ip}, node ID {target_node_id}.")
+                                return True
+
             self.logger.info(f"No replicas of tracked deployments={self.tracked_model_deployments} running on node: {node_ip}.")
-            return False
         except Exception as e:
             self.logger.error(f"An error occurred while checking replicas on node {node_ip}: {e}")
-            return False
+
+        return False

{ray_embedding-0.13.9.dist-info → ray_embedding-0.13.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ray-embedding
-Version: 0.13.9
+Version: 0.13.10
 Summary: Deploy SentenceTransformers embedding models to a ray cluster
 Author: Crispin Almodovar
 Author-email: 

ray_embedding-0.13.10.dist-info/RECORD

@@ -0,0 +1,10 @@
+ray_embedding/__init__.py,sha256=YS5LAZfRIwwVvE3C9g7hsauvjgIkqKtHyxkwMFFfAGY,46
+ray_embedding/deploy.py,sha256=2R7bQ7aPc9G8H9KVoemxum6-9YxmlXQogWbhFhuslko,3762
+ray_embedding/dto.py,sha256=lk_LuVQPq3MLIMTMddqHviYXILY6V5dvbzDJuD_D_qc,1573
+ray_embedding/embedding_model.py,sha256=P2xyXCznxXmdQBK6zodOJEMvxGVRMA8Ra3O5Qi7RCh0,6013
+ray_embedding/model_router.py,sha256=fmaeXzaAJeCemzL9nUoXfdCrU-ZaCe_29fx5ayDCTC0,6845
+ray_embedding/node_health.py,sha256=uySBHjqdHM4VQfuAu6xTVD_B4o8LJJYDIxe218coszE,5462
+ray_embedding-0.13.10.dist-info/METADATA,sha256=ZmdnpvTf3_hQZ660rmmNtR0oPoXR3eVdWUQH2dBTBhs,1075
+ray_embedding-0.13.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ray_embedding-0.13.10.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
+ray_embedding-0.13.10.dist-info/RECORD,,

ray_embedding-0.13.9.dist-info/RECORD

@@ -1,10 +0,0 @@
-ray_embedding/__init__.py,sha256=YS5LAZfRIwwVvE3C9g7hsauvjgIkqKtHyxkwMFFfAGY,46
-ray_embedding/deploy.py,sha256=2R7bQ7aPc9G8H9KVoemxum6-9YxmlXQogWbhFhuslko,3762
-ray_embedding/dto.py,sha256=lk_LuVQPq3MLIMTMddqHviYXILY6V5dvbzDJuD_D_qc,1573
-ray_embedding/embedding_model.py,sha256=P2xyXCznxXmdQBK6zodOJEMvxGVRMA8Ra3O5Qi7RCh0,6013
-ray_embedding/model_router.py,sha256=fmaeXzaAJeCemzL9nUoXfdCrU-ZaCe_29fx5ayDCTC0,6845
-ray_embedding/node_health.py,sha256=bKRoFHS6cVRQBOYTcv0dRA61VDeiJjmIPT8tA0hbRIU,5350
-ray_embedding-0.13.9.dist-info/METADATA,sha256=O1ObZ9JwO7eI-6Vke5hwxmyLy4VvRWIb0IgiTD6GZzQ,1074
-ray_embedding-0.13.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ray_embedding-0.13.9.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
-ray_embedding-0.13.9.dist-info/RECORD,,