model.yml

name: AWS Model Builder

on:
    push:
      branches:
        - main
    pull_request:
        types:
            - opened
            - synchronize
            - reopened
            - ready_for_review

jobs:
    build:
        runs-on: ubuntu-latest
        if: github.event.pull_request.draft == false
        steps:
            - uses: actions/checkout@v3
              with:
                  ref: ${{github.event.pull_request.head.sha || github.sha}}

            - name: Docker Build Task
              run: docker build -t segment-build . -f Dockerfile-build

            - name: Configure AWS Credentials
              uses: aws-actions/configure-aws-credentials@v1
              with:
                  aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
                  aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
                  aws-region: ${{ secrets.AWS_REGION }}

            - name: Run Container and Copy .mar Files to S3
              run: |
                  GITSHA=${{github.event.pull_request.head.sha || github.sha}}
                  docker run \
                      -e AWS_SECRET_ACCESS_KEY=${{secrets.AWS_SECRET_ACCESS_KEY}} \
                      -e AWS_ACCESS_KEY_ID=${{secrets.AWS_ACCESS_KEY_ID}} \
                      -e GITSHA=$GITSHA \
                      segment-build \
                      bash -c "\
                          aws configure set aws_access_key_id \$AWS_ACCESS_KEY_ID && \
                          aws configure set aws_secret_access_key \$AWS_SECRET_ACCESS_KEY && \
                          aws configure set region us-east-1 && \
                          aws s3 cp /home/model-store/sam_vit_h_decode.mar s3://segment-anything-services-prod-685980093638-us-east-1/\${GITSHA}/ && \
                          aws s3 cp /home/model-store/sam_vit_h_encode.mar s3://segment-anything-services-prod-685980093638-us-east-1/\${GITSHA}/"