feature(helm): Add support for affinities

Signed-off-by: Dhouti <[email protected]>

Author: Dhouti

helm/kagent/templates/controller-deployment.yaml

@@ -39,6 +39,10 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.controller.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       containers:
         - name: controller
           args:

helm/kagent/templates/ui-deployment.yaml

@@ -34,6 +34,10 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.ui.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       containers:
         - name: ui
           securityContext:

helm/kagent/values.yaml

@@ -46,6 +46,9 @@ tolerations: []
 # -- Node labels to match for `Pod` [scheduling](https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/).
 nodeSelector: {}
 
+# Affinities for `Pod` https://kubernetes.io/docs/tasks/configure-pod-container/assign-pods-nodes-using-node-affinity/
+affinity: {}
+
 # ==============================================================================
 # DATABASE CONFIGURATION
 # ==============================================================================
@@ -85,6 +88,9 @@ controller:
 
   # -- Node labels to match for `Pod` [scheduling](https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/).
   nodeSelector: {}
+  
+  # Affintities for `Pod` https://kubernetes.io/docs/tasks/configure-pod-container/assign-pods-nodes-using-node-affinity/
+  affinity: {}
 
   image:
     registry: ""
@@ -136,6 +142,9 @@ ui:
   # -- Node labels to match for `Pod` [scheduling](https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/).
   nodeSelector: {}
 
+  # Affintities for `Pod` https://kubernetes.io/docs/tasks/configure-pod-container/assign-pods-nodes-using-node-affinity/
+  affinity: {}
+
 # ==============================================================================
 # LLM PROVIDERS CONFIGURATION
 # ==============================================================================
@@ -208,6 +217,15 @@ kagent-tools:
       memory: 1Gi
   tools:
     loglevel: "debug"
+  
+  # -- Node taints which will be tolerated for `Pod` [scheduling](https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/).
+  tolerations: []
+
+  # -- Node labels to match for `Pod` [scheduling](https://kubernetes.io/docs/concepts/scheduling-eviction/assign-pod-node/).
+  nodeSelector: {}
+  
+  # Affintities for `Pod` https://kubernetes.io/docs/tasks/configure-pod-container/assign-pods-nodes-using-node-affinity/
+  affinity: {}
 
 # ==============================================================================
 # AGENTS

helm/tools/grafana-mcp/templates/deployment.yaml

@@ -27,6 +27,10 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       containers:
         - name: grafana-mcp
           securityContext:

helm/tools/grafana-mcp/values.yaml

@@ -24,6 +24,8 @@ tolerations: []
 
 nodeSelector: {}
 
+affinity: {}
+
 service:
   type: ClusterIP
   port: 8000

helm/tools/querydoc/templates/deployment.yaml

@@ -29,6 +29,10 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       containers:
         - name: querydoc
           securityContext:

helm/tools/querydoc/values.yaml

@@ -20,6 +20,8 @@ tolerations: []
 
 nodeSelector: {}
 
+affinity: {}
+
 service:
   type: ClusterIP
   port: 8080