[Bugfix][core] replace heartbeat with pid check (#9818)

Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
2024-10-30 10:34:07 -06:00
parent 9ff4511e43
commit 3b3f1e7436
5 changed files with 62 additions and 62 deletions
--- a/tests/mq_llm_engine/utils.py
+++ b/tests/mq_llm_engine/utils.py
@@ -68,7 +68,7 @@ class RemoteMQLLMEngine:

    async def make_client(self) -> MQLLMEngineClient:
        engine_config = self.engine_args.create_engine_config()
-        client = MQLLMEngineClient(self.ipc_path, engine_config)
+        client = MQLLMEngineClient(self.ipc_path, engine_config, self.proc.pid)
        while True:
            try:
                await client.setup()