feat(mr): enable model registry inference reconcile (#326)

Signed-off-by: Alessio Pragliola <[email protected]>
opendatahub-io · Dec 13, 2024 · 6136ed3 · 6136ed3
1 parent c4ab582
commit 6136ed3
Show file tree

Hide file tree

Showing 9 changed files with 174 additions and 540 deletions.
diff --git a/config/overlays/odh/kustomization.yaml b/config/overlays/odh/kustomization.yaml
@@ -5,6 +5,6 @@ resources:
 
 patches:
 - path: odh_model_controller_manager_patch.yaml
-  
+
 configurations:
-  - params.yaml
+  - params.yaml
diff --git a/config/overlays/odh/odh_model_controller_manager_patch.yaml b/config/overlays/odh/odh_model_controller_manager_patch.yaml
@@ -8,5 +8,6 @@ spec:
       containers:
         - args:
             - --leader-elect
-          image: $(odh-model-controller) 
+            - --model-registry-inference-reconcile
+          image: $(odh-model-controller)
           name: manager
diff --git a/controllers/constants/constants.go b/controllers/constants/constants.go
@@ -52,9 +52,12 @@ var (
 const (
 	MLMDAddressEnv                       = "MLMD_ADDRESS"
 	ModelRegistryNamespaceLabel          = "modelregistry.opendatahub.io/namespace"
+	ModelRegistryNameLabel               = "modelregistry.opendatahub.io/name"
+	ModelRegistryUrlAnnotation           = "modelregistry.opendatahub.io/url"
 	ModelRegistryInferenceServiceIdLabel = "modelregistry.opendatahub.io/inference-service-id"
 	ModelRegistryModelVersionIdLabel     = "modelregistry.opendatahub.io/model-version-id"
 	ModelRegistryRegisteredModelIdLabel  = "modelregistry.opendatahub.io/registered-model-id"
+	ModelRegistryFinalizer               = "modelregistry.opendatahub.io/finalizer"
 )
 
 const (