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

ray_embedding/node_health.py

@@ -78,15 +78,15 @@ class NodeHealthTracker:
             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):
-                # self.logger.info(f"Checking actor: {actor}")
-                if (actor.state in ["DEPENDENCIES_UNREADY", 'PENDING_CREATION', 'ALIVE', 'RESTARTING'] and
-                        actor.node_id == target_node_id and
-                            any(actor.name.startswith(prefix) for prefix in prefixes)):
-                    self.logger.info(f"There is at least one replica of tracked_deployments={self.tracked_model_deployments} "
-                        f"running on node {node_ip}")
-                    return True
-
+            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
             self.logger.info(f"No replicas of tracked deployments={self.tracked_model_deployments} running on node: {node_ip}.")
             return False
         except Exception as e:

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

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

{ray_embedding-0.13.8.dist-info → ray_embedding-0.13.9.dist-info}/RECORD

@@ -3,8 +3,8 @@ 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=2sd-5CBjOYeB3ScdAlc7Dxq2n61uYU1nDqhX2lBgz7w,5244
-ray_embedding-0.13.8.dist-info/METADATA,sha256=L9Nij86q5EwWuZVIHBacsrVtoP6P48Y4B3bPndXTnjw,1074
-ray_embedding-0.13.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ray_embedding-0.13.8.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
-ray_embedding-0.13.8.dist-info/RECORD,,
+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,,