Add nodeGroupSelector option to LLM UI

Author: Alex-Welsh

charts/azimuth-chat/azimuth-ui.schema.yaml

@@ -21,6 +21,7 @@ controls:
 sortOrder:
   - /azimuth-llm/huggingface/model
   - /azimuth-llm/huggingface/token
+  - /azimuth-llm/api/nodeGroupSelector
   - /azimuth-llm/ui/appSettings/model_instruction
   - /azimuth-llm/ui/appSettings/page_title
   - /azimuth-llm/api/image/version

charts/azimuth-llm/azimuth-ui.schema.yaml

@@ -21,6 +21,7 @@ controls:
 sortOrder:
   - /huggingface/model
   - /huggingface/token
+  - /api/nodeGroupSelector
   - /ui/appSettings/model_instruction
   - /ui/appSettings/page_title
   - /api/image/version

charts/azimuth-llm/templates/api/deployment.yml

@@ -82,9 +82,14 @@ spec:
           emptyDir:
             medium: Memory
             sizeLimit: 1Gi
-      {{- with $.Values.api.nodeSelector }}
+      {{- with .Values.api }}
+      {{- if .nodeGroupSelector }}
       nodeSelector:
-        {{- toYaml . | nindent 8 }}
+        capi.stackhpc.com/node-group: {{ .nodeGroupSelector | quote }}
+      {{- else if .nodeSelector }}
+      nodeSelector:
+        {{- toYaml .nodeSelector | nindent 8 }}
+      {{- end }}
       {{- end }}
       {{- with $.Values.api.affinity }}
       affinity:

charts/azimuth-llm/values.schema.json

@@ -40,6 +40,11 @@
                             "default": "v0.11.0"
                         }
                     }
+                },
+                "nodeGroupSelector": {
+                    "type": "string",
+                    "title": "Node Group",
+                    "description": "The node group to deploy the API backend to. Must match a node group label in the cluster."
                 }
             }
         },

charts/azimuth-llm/values.yaml

@@ -104,6 +104,8 @@ api:
   # - name: FOO
   #   value: "bar"
   extraEnv: []
+  # Pod node group selector label value
+  nodeGroupSelector: ""
   # Pod node selector labels
   nodeSelector: {}
   # Pod tolerations