Add option to specify manually compiled plan repo

Author: vraiti

.github/workflows/build-triton.yml

@@ -1,8 +1,20 @@
+name: "Build vLLM Server"
+on:
+  workflow_dispatch:
+    inputs:
+      models:
+        type: string
+        required: true
+
 name: Build Triton Server
 on:
   workflow_dispatch:
+    inputs:
+      plan-repository:
+        type: string
+        required: false
+        description: "URL of plan repository (see README.md)"
   workflow_call:
-
 jobs:
   build-engine:
     uses: ./.github/workflows/run-jumpstarter-workflow.yml
@@ -11,9 +23,20 @@ jobs:
       jumpstarter-selector: ${{ vars.JUMPSTARTER_SELECTOR }}
       image-url: ${{ vars.DEST_REGISTRY_HOST }}/${{ vars.DEST_REGISTRY_REPO }}:base-raw
       workflow-cmd: |
-        bash triton/build/jmp-build-engine.sh \
-          ${{ vars.ARTIFACT_REGISTRY_HOST }}/${{ vars.ARTIFACT_REGISTRY_REPO }}:onnx-repository \
-          ${{ vars.ARTIFACT_REGISTRY_HOST }}/${{ vars.ARTIFACT_REGISTRY_REPO }}:plan-repository
+        SRC="${{ inputs.plan-repository }}"
+        DST="${{ vars.ARTIFACT_REGISTRY_HOST }}/${{ vars.ARTIFACT_REGISTRY_REPO }}:plan-repository"
+
+        if [ -n ${{ inputs.plan-repository }} ];
+          # this could probably be done more elegantly with ORAS
+          podman artifact pull $SRC
+          podman artifact extract $SRC plan-repository.tar
+          podman artifact add $DST plan-repository.tar
+          podman artifact push $DST  
+        else 
+          bash triton/build/jmp-build-engine.sh \
+            ${{ vars.ARTIFACT_REGISTRY_HOST }}/${{ vars.ARTIFACT_REGISTRY_REPO }}:onnx-repository \
+            ${{ vars.ARTIFACT_REGISTRY_HOST }}/${{ vars.ARTIFACT_REGISTRY_REPO }}:plan-repository
+        fi
 
   build-triton-base:
     uses: ./.github/workflows/build-rhel-bootc-image.yml

.github/workflows/run-jumpstarter-workflow.yml

@@ -96,7 +96,14 @@ jobs:
         run: |
           source jumpstarter/.venv/bin/activate
 
-          jmp shell --lease $LEASE -- ${{ inputs.workflow-cmd }}
+          FILE=$(mktmp)
+          cat <<EOF > $FILE
+          ${{ inputs.workflow-cmd }}
+          EOF
+
+          cat $FILE | jmp shell --lease $LEASE
+          rm $FILE
+
       - name: Release lease
         if: always()
         run: |