use composite action

Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>

Author: sirutBuasai

.github/actions/container-pull/action.yaml

@@ -0,0 +1,37 @@
+name: Container Pull
+description: Login to ECR, get image URI from current workflow artifacts, then pull the image onto host runner.
+
+inputs:
+  aws_region:
+    required: true
+    default: ${{ secrets.AWS_REGION }}
+  aws_account_id:
+    required: true
+    default: ${{ secrets.AWS_ACCOUNT_ID }}
+  artifact_name:
+    description: Docker image URI that is built and saved as artifact within workflows
+    required: true
+    type: string
+
+runs:
+  using: composite
+  steps:
+
+    - name: ECR login
+      run: |
+        aws ecr get-login-password --region ${{ inputs.aws_region }} | docker login --username AWS --password-stdin ${{ inputs.aws_account_id }}.dkr.ecr.${{ inputs.aws_region }}.amazonaws.com
+
+    - name: Download image URI artifact
+      uses: actions/download-artifact@v4
+      with:
+        name: ${{ inputs.artifact_name }}
+
+    - name: Resolve image URI for test
+      run: |
+        IMAGE_URI=$(cat image_uri.txt)
+        echo "Resolved image URI: $IMAGE_URI"
+        echo "IMAGE_URI=$IMAGE_URI" >> $GITHUB_ENV
+
+    - name: Pull image
+      run: |
+        docker pull $IMAGE_URI

.github/actions/pr-permission-gate/action.yml

@@ -1,11 +1,13 @@
 name: PR Permission Gate
 description: Fails the workflow if the PR sender lacks the required repository permission
+
 inputs:
   required-level:
     description: Minimum permission level required (read|triage|write|maintain|admin)
     default: write
+
 runs:
-  using: "composite"
+  using: composite
   steps:
     - name: Check PR sender permission
       uses: actions/github-script@v7

.github/workflows/pr-sglang.yml

@@ -87,24 +87,29 @@ jobs:
       - name: Checkout DLC source
         uses: actions/checkout@v5
 
-      - name: ECR login
-        run: |
-          aws ecr get-login-password --region ${{ secrets.AWS_REGION }} | docker login --username AWS --password-stdin ${{ secrets.AWS_ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com
-
-      - name: Download image URI artifact
-        uses: actions/download-artifact@v4
+      - name: Container pull
+        uses: ./.github/actions/container-pull
         with:
-          name: sglang-sagemaker-image-uri
-
-      - name: Resolve image URI for test
-        run: |
-          IMAGE_URI=$(cat image_uri.txt)
-          echo "Resolved image URI: $IMAGE_URI"
-          echo "IMAGE_URI=$IMAGE_URI" >> $GITHUB_ENV
-
-      - name: Pull image
-        run: |
-          docker pull $IMAGE_URI
+          artifact_name: sglang-sagemaker-image-uri
+
+      # - name: ECR login
+      #   run: |
+      #     aws ecr get-login-password --region ${{ secrets.AWS_REGION }} | docker login --username AWS --password-stdin ${{ secrets.AWS_ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com
+      #
+      # - name: Download image URI artifact
+      #   uses: actions/download-artifact@v4
+      #   with:
+      #     name: sglang-sagemaker-image-uri
+      #
+      # - name: Resolve image URI for test
+      #   run: |
+      #     IMAGE_URI=$(cat image_uri.txt)
+      #     echo "Resolved image URI: $IMAGE_URI"
+      #     echo "IMAGE_URI=$IMAGE_URI" >> $GITHUB_ENV
+      #
+      # - name: Pull image
+      #   run: |
+      #     docker pull $IMAGE_URI
 
       - name: Checkout SGLang Tests
         uses: actions/checkout@v5
@@ -138,6 +143,8 @@ jobs:
             -e HF_TOKEN=${{ secrets.HUGGING_FACE_HUB_TOKEN }} \
             ${IMAGE_URI})
           echo "CONTAINER_ID=$CONTAINER_ID" >> $GITHUB_ENV
+          echo "Printing container startup logs ..."
+          docker logs ${CONTAINER_ID}
 
       - name: Setup for SGLang Test
         run: |