Docker Build and Publish #21
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Docker Build and Publish | |
| on: | |
| workflow_dispatch: | |
| inputs: | |
| version: | |
| description: 'TestPyPI or PyPI version to build (e.g., 0.0.63.dev20250114)' | |
| required: true | |
| type: string | |
| jobs: | |
| build-and-push: | |
| runs-on: ubuntu-latest | |
| permissions: | |
| contents: read | |
| packages: write | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v3 | |
| - name: Log in to the Container registry | |
| uses: docker/login-action@v3 | |
| with: | |
| username: ${{ secrets.DOCKERHUB_USERNAME }} | |
| password: ${{ secrets.DOCKERHUB_TOKEN }} | |
| - name: Set version | |
| id: version | |
| run: | | |
| if [ "${{ github.event_name }}" = "push" ]; then | |
| echo "VERSION=0.0.63.dev20250114" >> $GITHUB_OUTPUT | |
| else | |
| echo "VERSION=${{ inputs.version }}" >> $GITHUB_OUTPUT | |
| fi | |
| - name: Check package version availability | |
| run: | | |
| # Function to check if version exists in a repository | |
| check_version() { | |
| local repo=$1 | |
| local status_code=$(curl -s -o /dev/null -w "%{http_code}" "https://$repo.org/project/llama-stack/${{ steps.version.outputs.version }}") | |
| return $([ "$status_code" -eq 200 ]) | |
| } | |
| # Check TestPyPI first, then PyPI | |
| if check_version "test.pypi"; then | |
| echo "Version ${{ steps.version.outputs.version }} found in TestPyPI" | |
| echo "PYPI_SOURCE=testpypi" >> $GITHUB_ENV | |
| elif check_version "pypi"; then | |
| echo "Version ${{ steps.version.outputs.version }} found in PyPI" | |
| echo "PYPI_SOURCE=pypi" >> $GITHUB_ENV | |
| else | |
| echo "Error: Version ${{ steps.version.outputs.version }} not found in either TestPyPI or PyPI" | |
| exit 1 | |
| fi | |
| - name: Install llama-stack | |
| run: | | |
| if [ "${{ github.event_name }}" = "push" ]; then | |
| pip install -e . | |
| else | |
| if [ "$PYPI_SOURCE" = "testpypi" ]; then | |
| pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple llama-stack==${{ steps.version.outputs.version }} | |
| else | |
| pip install llama-stack==${{ steps.version.outputs.version }} | |
| fi | |
| fi | |
| - name: Build docker image | |
| run: | | |
| TEMPLATES=("ollama" "bedrock" "remote-vllm" "fireworks" "together" "tgi" "meta-reference-gpu") | |
| for template in "${TEMPLATES[@]}"; do | |
| if [ "$PYPI_SOURCE" = "testpypi" ]; then | |
| TEST_PYPI_VERSION=${{ steps.version.outputs.version }} llama stack build --template $template --image-type docker | |
| else | |
| PYPI_VERSION=${{ steps.version.outputs.version }} llama stack build --template $template --image-type docker | |
| fi | |
| done | |
| - name: List docker images | |
| run: | | |
| docker images | |
| - name: Push to dockerhub | |
| run: | | |
| TEMPLATES=("ollama" "bedrock" "remote-vllm" "fireworks" "together" "tgi" "meta-reference-gpu") | |
| for template in "${TEMPLATES[@]}"; do | |
| if [ "$PYPI_SOURCE" = "testpypi" ]; then | |
| docker tag distribution-$template:test-${{ steps.version.outputs.version }} llamastack/distribution-$template:test-${{ steps.version.outputs.version }} | |
| docker push llamastack/distribution-$template:test-${{ steps.version.outputs.version }} | |
| else | |
| docker tag distribution-$template:${{ steps.version.outputs.version }} llamastack/distribution-$template:${{ steps.version.outputs.version }} | |
| docker push llamastack/distribution-$template:${{ steps.version.outputs.version }} | |
| fi | |
| done |