From 2a091e50225dadda24bb530724abed28739eda0a Mon Sep 17 00:00:00 2001
From: Justin Law <justin.law@defenseunicorns.com>
Date: Wed, 2 Oct 2024 14:17:33 -0400
Subject: [PATCH] setup tasks for k3d cluster and test user, pt.2

---
 tasks/create.yaml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tasks/create.yaml b/tasks/create.yaml
index 4ab551bee..7690e7e8d 100644
--- a/tasks/create.yaml
+++ b/tasks/create.yaml
@@ -50,6 +50,10 @@ tasks:
       dockerfile:
         description: "Path to the Dockerfile"
         required: true
+      buildContext:
+        description: "Path to the Docker build context"
+        default: "."
+        required: false
       options:
         description: "Extra Docker CLI options"
         default: "-q"
@@ -79,7 +83,7 @@ tasks:
               -t ${{ .inputs.imageRepository }}:${{ .inputs.version }} \
               -f ${{ .inputs.dockerfile }} \
               --build-arg LOCAL_VERSION=${{ .inputs.version }} \
-              ${{ .inputs.buildOptions }} .
+              ${{ .inputs.buildOptions }} ${{ .inputs.buildContext }}
           else
             echo "Non-upstream flavor detected, skipping image build"
           fi
@@ -168,6 +172,7 @@ tasks:
           architecture: "amd64"
           version: ${VERSION}
           imageRepository: ghcr.io/defenseunicorns/leapfrogai/k3d-gpu
+          buildContext: "packages/k3d-gpu"
 
   - name: sdk-image
     description: "Create the LeapfrogAI SDK Docker image"