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

ray_embedding/node_health.py

@@ -76,17 +76,17 @@ 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 {node_ip}, ID {target_node_id}")
+            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):
+            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'] and
-                        actor.node_id == target_node_id and
-                            actor.name.startswith(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
-
+                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.7.dist-info → ray_embedding-0.13.9.dist-info}/METADATA

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

{ray_embedding-0.13.7.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=D8FdDRCnd86hdZxFeLB-C6gbOHBDtm-vcJQuYl75J-M,5213
-ray_embedding-0.13.7.dist-info/METADATA,sha256=hFarAec6loLv0cWpyZyl8xdzb3f5frxg44QYEODNyds,1074
-ray_embedding-0.13.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ray_embedding-0.13.7.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
-ray_embedding-0.13.7.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,,