use secret image uri file

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

Author: sirutBuasai

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

@@ -8,13 +8,13 @@ inputs:
   aws_account_id:
     description: AWS Account ID for docker image registry
     required: true
-  image_uri:
+  image_uri_file:
+    description: File where image URI is stored
+    required: true
+  artifact_name:
     description: Docker image URI that is built and saved as artifact within workflows
     required: true
-  # artifact_name:
-  #   description: Docker image URI that is built and saved as artifact within workflows
-  #   required: true
-  #   type: string
+    type: string
 
 runs:
   using: composite
@@ -25,17 +25,17 @@ runs:
       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
-    #   shell: bash
-    #   run: |
-    #     IMAGE_URI=$(cat image_uri.txt)
-    #     echo "Resolved image URI: $IMAGE_URI"
-    #     echo "IMAGE_URI=$IMAGE_URI" >> $GITHUB_ENV
+    - name: Download image URI artifact
+      uses: actions/download-artifact@v4
+      with:
+        name: ${{ inputs.artifact_name }}
+
+    - name: Resolve image URI for test
+      shell: bash
+      run: |
+        IMAGE_URI=$(cat ${{ input.image_uri_file }})
+        echo "Resolved image URI: $IMAGE_URI"
+        echo "IMAGE_URI=$IMAGE_URI" >> $GITHUB_ENV
 
     - name: Pull image
       shell: bash

.github/workflows/pr-sglang.yml

@@ -41,8 +41,6 @@ jobs:
     runs-on:
       - codebuild-runner-${{ github.run_id }}-${{ github.run_attempt }}
         fleet:x86-build-runner
-    outputs:
-      my_output: ${{ steps.image_uri_build.outputs.image_uri }}
     steps:
       - uses: actions/checkout@v5
       - run: .github/scripts/runner_setup.sh
@@ -57,7 +55,6 @@ jobs:
           IMAGE_URI=${{ secrets.AWS_ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/ci:sglang-0.5.5-gpu-py312-cu129-ubuntu22.04-sagemaker-pr-${{ github.event.pull_request.number }}
           echo "Image URI to build: $IMAGE_URI"
           echo "IMAGE_URI=$IMAGE_URI" >> $GITHUB_ENV
-          echo "image_uri=$IMAGE_URI" >> $GITHUB_OUTPUT
 
       - name: Build image
         run: |
@@ -73,13 +70,13 @@ jobs:
         run: |
           docker push $IMAGE_URI
           docker rmi $IMAGE_URI
-          echo $IMAGE_URI > image_uri.txt
+          echo $IMAGE_URI > ${{ secrets.IMAGE_URI_FILE }}
 
       - name: Upload image URI artifact
         uses: actions/upload-artifact@v4
         with:
           name: sglang-sagemaker-image-uri
-          path: image_uri.txt
+          path: ${{ secrets.IMAGE_URI_FILE }}
 
   sglang-local-benchmark-test:
     needs: [build-sglang-image]
@@ -91,16 +88,13 @@ jobs:
       - name: Checkout DLC source
         uses: actions/checkout@v5
 
-      - name: echo container
-        run: |
-          echo ${{ needs.build-sglang-image.outputs.my_output }}
-
       - name: Container pull
         uses: ./.github/actions/container-pull
         with:
           aws_region: ${{ secrets.AWS_REGION }}
           aws_account_id: ${{ secrets.AWS_ACCOUNT_ID }}
-          image_uri: ${{ needs.build-sglang-image.outputs.my_output }}
+          image_uri_file: ${{ secrets.IMAGE_URI_FILE }}
+          artifact_name: sglang-sagemaker-image-uri
 
       - name: Setup for SGLang datasets
         run: |