Revert "remove azure logins"

This reverts commit a58f273.

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

.github/actions/build-container/action.yml

@@ -15,9 +15,22 @@
 name: Build container
 
 inputs:
+  azure-client-id:
+    description: "Azure Client ID"
+    required: true
+  azure-tenant-id:
+    description: "Azure Tenant ID"
+    required: true
+  azure-subscription-id:
+    description: "Azure Subscription ID"
+    required: true
   dockerfile-path:
     description: "Path to dockerfile to build"
     required: true
+  has-azure-credentials:
+    description: "Has Azure credentials"
+    required: false
+    default: "false"
   PAT:
     description: "GitHub Personal Access Token"
     required: true
@@ -46,6 +59,26 @@ runs:
       if: startsWith(github.ref, 'refs/heads/pull-request/')
       uses: nv-gha-runners/get-pr-info@main
 
+    - name: Install Azure CLI
+      shell: bash
+      run: |
+        echo "::group::Install Azure CLI"
+        # Create systemd override for proper dependencies
+        curl -sL https://aka.ms/InstallAzureCLIDeb | sudo bash
+        echo "::endgroup::"
+
+    - name: Azure Login
+      uses: azure/login@v2
+      with:
+        client-id: ${{ inputs.azure-client-id }}
+        tenant-id: ${{ inputs.azure-tenant-id }}
+        subscription-id: ${{ inputs.azure-subscription-id }}
+
+    - name: Azure ACR Login
+      shell: bash
+      run: |
+        az acr login --name nemoci
+
     - name: Install GH CLI
       shell: bash
       run: |

.github/actions/test-template/action.yml

@@ -34,6 +34,19 @@ inputs:
     description: "Run tests on CPU only"
     required: false
     default: "false"
+  azure-client-id:
+    description: "Azure Client ID"
+    required: true
+  azure-tenant-id:
+    description: "Azure Tenant ID"
+    required: true
+  azure-subscription-id:
+    description: "Azure Subscription ID"
+    required: true
+  has-azure-credentials:
+    description: "Has Azure credentials"
+    required: false
+    default: "false"
   PAT:
     description: "GitHub Personal Access Token"
     required: true
@@ -55,6 +68,27 @@ runs:
       with:
         path: NeMo-Automodel
 
+    - name: Install Azure CLI
+      if: ${{ inputs.has-azure-credentials == 'true' }}
+      shell: bash
+      run: |
+        echo "::group::Install Azure CLI"
+        curl -sL https://aka.ms/InstallAzureCLIDeb | sudo bash
+        echo "::endgroup::"
+
+    - name: Azure Login
+      if: ${{ inputs.has-azure-credentials == 'true' }}
+      uses: azure/login@v2
+      with:
+        client-id: ${{ inputs.azure-client-id }}
+        tenant-id: ${{ inputs.azure-tenant-id }}
+        subscription-id: ${{ inputs.azure-subscription-id }}
+
+    - name: Azure ACR Login
+      shell: bash
+      run: |
+        az acr login --name nemoci
+
     - name: Install uuidgen
       shell: bash -x -e -u -o pipefail {0}
       if: ${{ contains(inputs.runner, 'aws') }}

.github/workflows/cicd-main.yml

@@ -152,7 +152,11 @@ jobs:
       - name: main
         uses: ./.github/actions/build-container
         with:
+          azure-client-id: ${{ secrets.AZURE_CLIENT_ID }}
+          azure-tenant-id: ${{ secrets.AZURE_TENANT_ID }}
+          azure-subscription-id: ${{ secrets.AZURE_SUBSCRIPTION_ID }}
           dockerfile-path: "./docker/Dockerfile"
+          has-azure-credentials: "true"
           PAT: ${{ secrets.PAT }}
           repo-name: "Automodel"
           registry: ${{ vars.ECR }}
@@ -189,6 +193,10 @@ jobs:
           timeout: ${{ matrix.timeout || 10 }}
           is-unit-test: "true"
           cpu-only: ${{ matrix.cpu-only || false }}
+          has-azure-credentials: "true"
+          azure-client-id: ${{ secrets.AZURE_CLIENT_ID }}
+          azure-tenant-id: ${{ secrets.AZURE_TENANT_ID }}
+          azure-subscription-id: ${{ secrets.AZURE_SUBSCRIPTION_ID }}
           PAT: ${{ secrets.PAT }}
           test-data-path: ${{ needs.pre-flight.outputs.test_data_path }}
           container-image: ${{ vars.ECR }}/automodel:${{ github.sha }}
@@ -259,6 +267,10 @@ jobs:
           test-folder: ${{ matrix.test-folder }}
           timeout: ${{ matrix.timeout }}
           is_unit_test: "false"
+          has-azure-credentials: "true"
+          azure-client-id: ${{ secrets.AZURE_CLIENT_ID }}
+          azure-tenant-id: ${{ secrets.AZURE_TENANT_ID }}
+          azure-subscription-id: ${{ secrets.AZURE_SUBSCRIPTION_ID }}
           PAT: ${{ secrets.PAT }}
           test-data-path: ${{ matrix.test-data-path }}
           container-image: ${{ vars.ECR }}/automodel:${{ github.sha }}