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,10 +82,13 @@ spec:
           emptyDir:
             medium: Memory
             sizeLimit: 1Gi
-      {{- with $.Values.api.nodeSelector }}
       nodeSelector:
+        {{- if .Values.api.nodeGroupSelector }}
+        azimuth.stackhpc.com/node-group: {{ .Values.api.nodeGroupSelector | quote }}
+        {{- end }}
+        {{- with .Values.api.nodeSelector }}
         {{- toYaml . | nindent 8 }}
-      {{- end }}
+        {{- end }}
       {{- with $.Values.api.affinity }}
       affinity:
         {{- toYaml . | nindent 8 }}

charts/azimuth-llm/values.schema.json

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

charts/azimuth-llm/values.yaml

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