@@ -48,23 +48,24 @@ function enable_health_check {
4848}
4949
5050VLLM_INSTALL_PATH=" /usr/local/lib/python3.12/dist-packages/vllm" 
51+ VLLM_V1_ENGINE_PATH=" $VLLM_INSTALL_PATH /v1/engine" 
5152
5253function  mock_vllm_async_llm_engine  {
5354    #  backup original file
54-     mv $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py.backup
55-     cp $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py.backup $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
55+     mv $VLLM_V1_ENGINE_PATH  /async_llm .py $VLLM_V1_ENGINE_PATH  /async_llm .py.backup
56+     cp $VLLM_V1_ENGINE_PATH  /async_llm .py.backup $VLLM_V1_ENGINE_PATH  /async_llm .py
5657    #  overwrite the original check_health method
57-     echo  -e " "   >>  $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
58-     echo  -e "     async def check_health(self, check_count=[0]):"   >>  $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
59-     echo  -e "         check_count[0] += 1"   >>  $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
60-     echo  -e "         if check_count[0] > 1:"   >>  $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
61-     echo  -e "             raise RuntimeError(\" Simulated vLLM check_health() failure\" )"   >>  $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
58+     echo  -e " "   >>  $VLLM_V1_ENGINE_PATH  /async_llm .py
59+     echo  -e "     async def check_health(self, check_count=[0]):"   >>  $VLLM_V1_ENGINE_PATH  /async_llm .py
60+     echo  -e "         check_count[0] += 1"   >>  $VLLM_V1_ENGINE_PATH  /async_llm .py
61+     echo  -e "         if check_count[0] > 1:"   >>  $VLLM_V1_ENGINE_PATH  /async_llm .py
62+     echo  -e "             raise RuntimeError(\" Simulated vLLM check_health() failure\" )"   >>  $VLLM_V1_ENGINE_PATH  /async_llm .py
6263}
6364
6465function  unmock_vllm_async_llm_engine  {
6566    #  restore from backup
66-     rm -f $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
67-     mv $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py.backup $VLLM_INSTALL_PATH  /engine/multiprocessing/client .py
67+     rm -f $VLLM_V1_ENGINE_PATH  /async_llm .py
68+     mv $VLLM_V1_ENGINE_PATH  /async_llm .py.backup $VLLM_V1_ENGINE_PATH  /async_llm .py
6869}
6970
7071function  test_check_health  {
0 commit comments