[CI] feat: add early exit to wait_for_server when process dies (#18602)

examples/runtime/hidden_states/hidden_states_server.py

@@ -25,7 +25,7 @@ def main():
     server_process, port = launch_server_cmd(
         "python -m sglang.launch_server --model-path Alibaba-NLP/gte-Qwen2-1.5B-instruct --enable-return-hidden-states --host 0.0.0.0"
     )
-    wait_for_server(f"http://localhost:{port}")
+    wait_for_server(f"http://localhost:{port}", process=server_process)
 
     prompts = [
         "Hello, my name is",

examples/runtime/token_in_token_out/token_in_token_out_llm_server.py

@@ -25,7 +25,7 @@ def main():
     server_process, port = launch_server_cmd(
         f"python -m sglang.launch_server --model-path {MODEL_PATH} --skip-tokenizer-init --host 0.0.0.0"
     )
-    wait_for_server(f"http://localhost:{port}")
+    wait_for_server(f"http://localhost:{port}", process=server_process)
 
     # Sample prompts.
     prompts = [

examples/runtime/token_in_token_out/token_in_token_out_vlm_server.py

@@ -47,7 +47,7 @@ def main():
     server_process, port = launch_server_cmd(
         f"python -m sglang.launch_server --model-path {MODEL_PATH} --skip-tokenizer-init --host 0.0.0.0"
     )
-    wait_for_server(f"http://localhost:{port}")
+    wait_for_server(f"http://localhost:{port}", process=server_process)
 
     input_ids, image_data = get_input_ids()