helm: support adding labels to models (#338)

substratusai · Dec 4, 2024 · a12fecf · a12fecf
1 parent 5c73336
commit a12fecf
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/charts/models/templates/models.yaml b/charts/models/templates/models.yaml
@@ -5,6 +5,10 @@ apiVersion: kubeai.org/v1
 kind: Model
 metadata:
   name: {{ $name }}
+  {{- with $model.labels }}
+  labels:
+  {{- toYaml . | nindent 4 }}
+  {{- end }}
 spec:
   features: {{ $model.features }}
   {{- with $model.owner }}

diff --git a/charts/models/values.yaml b/charts/models/values.yaml
@@ -4,6 +4,9 @@ all:
 
 catalog:
   llama-3.2-11b-vision-instruct-l4:
+    # You can optionally add metadata.labels.
+    # labels:
+    #   tenancy: public
     enabled: false
     features: [TextGeneration]
     url: hf://neuralmagic/Llama-3.2-11B-Vision-Instruct-FP8-dynamic

diff --git a/docs/how-to/architect-for-multitenancy.md b/docs/how-to/architect-for-multitenancy.md
@@ -22,6 +22,15 @@ spec:
 # ...
 ```
 
+Example Model using Helm chart:
+```yaml
+catalog:
+  llama-3.2:
+    labels:
+      tenancy: public
+    # ...
+```
+
 Example HTTP requests:
 
 ```bash