fix

Signed-off-by: Junpu Fan <junpu@amazon.com>

Author: junpuf

.github/workflows/pr-sglang.yml

@@ -193,7 +193,7 @@ jobs:
 
       - name: Run tests
         run: |
-          docker exec ${CONTAINER_ID} bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "sglang"
+          docker exec ${CONTAINER_ID} bash -c "export TEST_MODE=1 && bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}""
 
 
   sglang-local-benchmark-test:

.github/workflows/pr-vllm.yml

@@ -195,7 +195,7 @@ jobs:
 
       - name: Run tests
         run: |
-          docker exec ${CONTAINER_ID} bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}"
+          docker exec ${CONTAINER_ID} bash -c "export TEST_MODE=1 && bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}""
 
   vllm-ec2-regression-test:
     needs: [build-vllm-ec2-image, set-ec2-test-environment]
@@ -514,7 +514,7 @@ jobs:
 
       - name: Run tests
         run: |
-          docker exec ${CONTAINER_ID} bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}"
+          docker exec ${CONTAINER_ID} bash -c "export TEST_MODE=1 && bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}""
 
   vllm-rayserve-regression-test:
     needs: [build-vllm-rayserve-image, set-rayserve-test-environment]
@@ -833,7 +833,7 @@ jobs:
 
       - name: Run tests
         run: |
-          docker exec ${CONTAINER_ID} bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}"
+          docker exec ${CONTAINER_ID} bash -c "export TEST_MODE=1 && bash /usr/local/bin/bash_telemetry.sh && cat /tmp/test_request.txt | grep "${{ env.FRAMEWORK }}""
 
   vllm-sagemaker-regression-test:
     needs: [build-vllm-sagemaker-image, set-sagemaker-test-environment]

docker/sglang/Dockerfile

@@ -49,7 +49,9 @@ RUN chmod +x /usr/local/bin/deep_learning_container.py \
     /tmp/bash_telemetry.sh.template >/usr/local/bin/bash_telemetry.sh \
   && chmod +x /usr/local/bin/bash_telemetry.sh \
   && chmod +x /usr/local/bin/start_cuda_compat.sh \
-  && rm /tmp/bash_telemetry.sh.template
+  && rm /tmp/bash_telemetry.sh.template \
+  && uv pip install --system botocore \
+  && uv cache clean
 
 # Install EFA and remove vulnerable nvjpeg
 # =========================================