build-and-push-image.yaml

name: Build the LLM Finetuning Docker image

on:
  push:
    tags:
      - v*
    branches:
      - update-workflow

permissions:
  id-token: write
  contents: read

concurrency:
  group: build-llm-finetune-${{ github.ref }}
  cancel-in-progress: true

env:
  GITHUB_SHA: ${{ github.sha }}
  TAG_PREFIX: "v"

jobs:
  set_version_tags:
    name: Set version tags
    runs-on: ubuntu-latest
    outputs:
      version_tag: ${{ steps.set_version_tags.outputs.version_tag }}
      jupyter_version_tag: ${{ steps.set_version_tags.outputs.jupyter_version_tag }}
      alt_version_tag: ${{ steps.set_version_tags.outputs.alt_version_tag }}
    steps:
      - name: Set version tags
        id: set_version_tags
        run: |
          echo "Ref name is ${{ github.ref_name }}"
          echo "TAG_PREFIX is ${{ env.TAG_PREFIX }}"

          REF_NAME=${{ github.ref_name }}
          TAG_PREFIX=${{ env.TAG_PREFIX }}
          VERSION_TAG=$(echo $REF_NAME | awk -F$TAG_PREFIX '{print $2}')
          echo "Setting VERSION_TAG to $VERSION_TAG"
          echo "version_tag=$VERSION_TAG" >> $GITHUB_OUTPUT
          
          JUPYTER_VERSION_TAG=$(echo $REF_NAME | cut -d $TAG_PREFIX -f 2-)-jupyter
          echo "Setting JUPYTER VERSION_TAG equal to JUPYTER_VERSION_TAG"
          echo "jupyter_version_tag=JUPYTER_VERSION_TAG" >> $GITHUB_OUTPUT 

          ALT_VERSION_TAG=$(echo $VERSION_TAG | awk -F- '{print $1}')-${GITHUB_SHA::7}
          echo "Setting ALT_VERSION_TAG equal to $ALT_VERSION_TAG"
          echo "alt_version_tag=$ALT_VERSION_TAG" >> $GITHUB_OUTPUT

  build_finetune_image:
    - name: Build and Push llm-finetune image
      needs: set_version_tags
      uses: truefoundry/workflows/.github/workflows/build.yml@v0.1.1
      with:
        image_tag: ${{ github.sha }}
        extra_image_tag: |
          ${{ needs.set_version_tags.outputs.version_tag }}
          ${{ needs.set_version_tags.outputs.alt_version_tag }}
        platforms: linux/amd64
        free_disk_space: true
        image_artifact_name: ${{ github.event.repository.name }}
        artifactory_registry_url: ${{ vars.TRUEFOUNDRY_ARTIFACTORY_REGISTRY_URL }}
        artifactory_repository_url: ${{ vars.TRUEFOUNDRY_ARTIFACTORY_PUBLIC_REPOSITORY }}
      secrets:
        artifactory_username: ${{ secrets.TRUEFOUNDRY_ARTIFACTORY_PUBLIC_USERNAME }}
        artifactory_password: ${{ secrets.TRUEFOUNDRY_ARTIFACTORY_PUBLIC_PASSWORD }}

  build_jupyter_image:
    - name: Build and Push jupyter image
      uses: truefoundry/workflows/.github/workflows/build.yml@v0.1.1
      with:
        image_tag: ${{ needs.set_version_tags.outputs.jupyter_version_tag }}
        platforms: linux/amd64
        free_disk_space: true
        image_artifact_name: ${{ github.event.repository.name }}
        artifactory_registry_url: ${{ vars.TRUEFOUNDRY_ARTIFACTORY_REGISTRY_URL }}
        artifactory_repository_url: ${{ vars.TRUEFOUNDRY_ARTIFACTORY_PUBLIC_REPOSITORY }}
      secrets:
        artifactory_username: ${{ secrets.TRUEFOUNDRY_ARTIFACTORY_PUBLIC_USERNAME }}
        artifactory_password: ${{ secrets.TRUEFOUNDRY_ARTIFACTORY_PUBLIC_PASSWORD }}