Add NIM flag logic

config/base/params.env

@@ -4,4 +4,4 @@ caikit-standalone-image=quay.io/opendatahub/caikit-nlp:fast
 tgis-image=quay.io/opendatahub/text-generation-inference:fast
 ovms-image=quay.io/opendatahub/openvino_model_server:2024.3-release
 vllm-image=quay.io/opendatahub/vllm:fast
-
+nim-state=removed

config/manager/manager.yaml

@@ -71,6 +71,12 @@ spec:
                   name: service-mesh-refs
                   key: MESH_NAMESPACE
                   optional: true
+            - name: NIM_STATE
+              valueFrom:
+                configMapKeyRef:
+                  name: odh-model-controller-parameters
+                  key: nim-state
+                  optional: true
           livenessProbe:
             httpGet:
               path: /healthz

main.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"flag"
 	"os"
+	"slices"
 	"strconv"
 
 	corev1 "k8s.io/api/core/v1"
@@ -242,13 +243,17 @@ func main() {
 	}
 
 	ctx := ctrl.SetupSignalHandler()
-	if err = (&controllers.NimAccountReconciler{
-		Client:  mgr.GetClient(),
-		Log:     ctrl.Log.WithName("controllers").WithName("NimAccountReconciler"),
-		KClient: kclient,
-	}).SetupWithManager(mgr, ctx); err != nil {
-		setupLog.Error(err, "unable to create controller NIM Account controller")
-		os.Exit(1)
+	nimState := os.Getenv("NIM_STATE")
+
+	if !slices.Contains([]string{"removed", ""}, nimState) {
+		if err = (&controllers.NimAccountReconciler{
+			Client:  mgr.GetClient(),
+			Log:     ctrl.Log.WithName("controllers").WithName("NimAccountReconciler"),
+			KClient: kclient,
+		}).SetupWithManager(mgr, ctx); err != nil {
+			setupLog.Error(err, "unable to create controller NIM Account controller")
+			os.Exit(1)
+		}
 	}
 
 	if err = builder.WebhookManagedBy(mgr).