Merge pull request #13 from abhishek-ch/missing_affinity

Added provision of affinity and env vars

charts/open-webui/templates/deployment.yaml

@@ -42,11 +42,25 @@ spec:
         env:
         - name: OLLAMA_BASE_URLS
           value: {{ include "ollamaBaseUrls" . | quote }}
+        {{- with .Values.extraEnvVars }}
+        {{- range . }}
+        - name: {{ .name }}
+          value: {{ .value | quote }}
+        {{- end }}
+        {{- end }}
         tty: true
       {{- with .Values.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.tolerations }}
+      tolerations:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      {{- with .Values.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       volumes:
       {{- if and .Values.persistence.enabled .Values.persistence.existingClaim }}
       - name: data

charts/open-webui/values.yaml

@@ -49,8 +49,16 @@ persistence:
   storageClass: ""
   selector: {}
   annotations: {}
+
+# -- Node labels for pod assignment.
 nodeSelector: {}
+
+# -- Tolerations for pod assignment
 tolerations: []
+
+# -- Affinity for pod assignment
+affinity: {}
+
 service:
   type: ClusterIP
   annotations: {}
@@ -59,3 +67,8 @@ service:
   nodePort: ""
   labels: {}
   loadBalancerClass: ""
+
+# -- Additional environments variables on the output Deployment definition.
+extraEnvVars: []
+#  - name: OLLAMA_DEBUG
+#    value: "1"