From 46258e508bf0852acff45ce6ce628003924abc90 Mon Sep 17 00:00:00 2001
From: Anuradha Karuppiah <anuradhak@nvidia.com>
Date: Fri, 20 Sep 2024 21:50:41 +0000
Subject: [PATCH] Update github workflow to build/upload the morpheus_dfp conda
 package

Signed-off-by: Anuradha Karuppiah <anuradhak@nvidia.com>
---
 .github/workflows/ci_pipe.yml | 44 ++++++++++++++++++++++++++++++++---
 .github/workflows/pr.yaml     |  8 +++----
 2 files changed, 45 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/ci_pipe.yml b/.github/workflows/ci_pipe.yml
index 3482fd2b8d..ccfc29cabf 100644
--- a/.github/workflows/ci_pipe.yml
+++ b/.github/workflows/ci_pipe.yml
@@ -21,8 +21,8 @@ on:
       run_check:
         required: true
         type: boolean
-      conda_core_run_build:
-        description: 'Runs the stage to build the morpheus-core conda package'
+      conda_lib_build:
+        description: 'Runs the stage to build the conda packages for all morpheus libraries - core, dfp etc.'
         required: true
         type: boolean
       conda_upload_label:
@@ -206,7 +206,7 @@ jobs:
 
   package-core:
     name: Package Core
-    if: ${{ inputs.conda_core_run_build }}
+    if: ${{ inputs.conda_lib_build }}
     needs: [documentation, test]
     runs-on: linux-amd64-cpu16
     timeout-minutes: 60
@@ -242,6 +242,44 @@ jobs:
           CONDA_PKG_LABEL: "${{ inputs.conda_upload_label }}"
         run: ./morpheus/ci/scripts/github/conda_core.sh $SCRIPT_ARGS
 
+  package-dfp:
+    name: Package DFP library
+    if: ${{ inputs.conda_lib_build }}
+    needs: [package-core]
+    runs-on: linux-amd64-cpu16
+    timeout-minutes: 60
+    container:
+      credentials:
+        username: '$oauthtoken'
+        password: ${{ secrets.NGC_API_KEY }}
+      image: ${{ inputs.container }}
+    strategy:
+      fail-fast: true
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v3
+        with:
+          lfs: false
+          path: 'morpheus'
+          fetch-depth: 0
+          submodules: 'recursive'
+
+      - name: Get AWS credentials using OIDC
+        uses: aws-actions/configure-aws-credentials@v1-node16
+        with:
+          role-to-assume: ${{ vars.AWS_ROLE_ARN }}
+          aws-region: ${{ vars.AWS_REGION }}
+          role-duration-seconds: 43200 # 12h
+
+      - name: Build morpheus-dfp conda package
+        shell: bash
+        env:
+          CONDA_TOKEN: "${{ secrets.CONDA_TOKEN }}"
+          SCRIPT_ARGS: "${{ inputs.conda_upload_label != '' && 'upload' || '' }}"
+          CONDA_PKG_LABEL: "${{ inputs.conda_upload_label }}"
+        run: ./morpheus/ci/scripts/github/conda_dfp.sh $SCRIPT_ARGS
+
   package:
     name: Package All
     if: ${{ inputs.conda_run_build }}
diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
index 097d3c9e69..203c9185ea 100644
--- a/.github/workflows/pr.yaml
+++ b/.github/workflows/pr.yaml
@@ -68,7 +68,7 @@ jobs:
       is_main_branch: ${{ github.ref_name == 'main' }}
       is_dev_branch: ${{ startsWith(github.ref_name, 'branch-') }}
       has_conda_build_label: ${{ steps.get-pr-info.outcome == 'success' && contains(fromJSON(steps.get-pr-info.outputs.pr-info).labels.*.name, 'conda-build') || false }}
-      has_conda_core_build_label: ${{ steps.get-pr-info.outcome == 'success' && contains(fromJSON(steps.get-pr-info.outputs.pr-info).labels.*.name, 'conda-core-build') || false }}
+      has_conda_lib_build_label: ${{ steps.get-pr-info.outcome == 'success' && contains(fromJSON(steps.get-pr-info.outputs.pr-info).labels.*.name, 'conda-lib-build') || false }}
       has_skip_ci_label: ${{ steps.get-pr-info.outcome == 'success' && contains(fromJSON(steps.get-pr-info.outputs.pr-info).labels.*.name, 'skip-ci') || false }}
       pr_info: ${{ steps.get-pr-info.outcome == 'success' && steps.get-pr-info.outputs.pr-info || '' }}
 
@@ -90,9 +90,9 @@ jobs:
     with:
       # Run checks for any PR branch
       run_check: ${{ fromJSON(needs.prepare.outputs.is_pr) }}
-      # Build morpheus-core conda package. This is done for main/dev branches and
-      # for PRs with the conda-core-build label
-      conda_core_run_build: ${{ !fromJSON(needs.prepare.outputs.is_pr) || fromJSON(needs.prepare.outputs.has_conda_core_build_label) }}
+      # Build conda packages for all the morpheus libraries - core, dfp, llm. This is done for
+      # main/dev branches and for PRs with the conda-lib-build label
+      conda_lib_build: ${{ !fromJSON(needs.prepare.outputs.is_pr) || fromJSON(needs.prepare.outputs.has_conda_lib_build_label) }}
       # Upload morpheus-core conda package only for non PR branches. Use 'main' for main branch and 'dev' for all other branches
       conda_upload_label: ${{ !fromJSON(needs.prepare.outputs.is_pr) && (fromJSON(needs.prepare.outputs.is_main_branch) && 'main' || 'dev') || '' }}
       # Run morpheus conda package, with all components. This is done for main/dev