chore: Add Path parameter

NVIDIA · Aug 5, 2024 · 45943c6 · 45943c6
1 parent 518908b
commit 45943c6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/api/v1alpha1/nimcache_types.go b/api/v1alpha1/nimcache_types.go
@@ -76,6 +76,8 @@ type GITSource struct {
 	PullSecret string `json:"pullSecret,omitempty"`
 	// Model spec for caching
 	Model ModelSpec `json:"model,omitempty"`
+	// Path
+	Path string `json:"path"`
 }
 
 // ModelSpec is the spec required to cache selected models

diff --git a/internal/controller/nimcache_controller_test.go b/internal/controller/nimcache_controller_test.go
@@ -307,7 +307,11 @@ var _ = Describe("NIMCache Controller", func() {
 			}, time.Second*10).Should(Succeed())
 
 			nimCache.Spec.Source.NGC = nil
-			nimCache.Spec.Source.GIT = &appsv1alpha1.GITSource{ModelPuller: "nvcr.io/nim:test-git-puller", PullSecret: "my-secret"}
+			nimCache.Spec.Source.GIT = &appsv1alpha1.GITSource{
+				ModelPuller: "nvcr.io/nim:test-git-puller",
+				PullSecret:  "my-secret",
+				Path:        "https://github.com/modelx",
+			}
 
 			err = client.Delete(context.TODO(), job)
 			Expect(err).ToNot(HaveOccurred())