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
@@ -59,4 +73,4 @@ spec:
       - name: data
         persistentVolumeClaim:
           claimName: {{ include "open-webui.name" . }}
-      {{- end }}
+      {{- end }}

charts/open-webui/values.yaml

@@ -9,7 +9,7 @@ ollama:
   #     enabled: true
   #     type: 'nvidia'
   #     number: 1
-  #   models: 
+  #   models:
   #     - llama3
   # runtimeClassName: nvidia
   # persistentVolume:
@@ -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: {}
@@ -58,4 +66,9 @@ service:
   containerPort: 8080
   nodePort: ""
   labels: {}
-  loadBalancerClass: "" 
+  loadBalancerClass: ""
+
+# -- Additional environments variables on the output Deployment definition.
+extraEnvVars: []
+#  - name: OLLAMA_DEBUG
+#    value: "1"