From 934c4ff6004cc22eb6f23ea8d4b061f3a14ca488 Mon Sep 17 00:00:00 2001
From: Sara Angel-Murphy <sara.angelmurphy@bluevoyant.com>
Date: Mon, 10 Mar 2025 16:12:56 -0400
Subject: [PATCH] feat: add support for extra init containers

---
 charts/open-webui/Chart.yaml                      |  2 +-
 charts/open-webui/README.md                       |  1 +
 charts/open-webui/templates/workload-manager.yaml |  3 +++
 charts/open-webui/values.yaml                     | 10 ++++++++++
 4 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/charts/open-webui/Chart.yaml b/charts/open-webui/Chart.yaml
index e3feba1..ca9dbdc 100644
--- a/charts/open-webui/Chart.yaml
+++ b/charts/open-webui/Chart.yaml
@@ -1,6 +1,6 @@
 apiVersion: v2
 name: open-webui
-version: 5.21.0
+version: 5.22.0
 appVersion: 0.5.16
 home: https://www.openwebui.com/
 icon: >-
diff --git a/charts/open-webui/README.md b/charts/open-webui/README.md
index 2d3e5d0..3162e76 100644
--- a/charts/open-webui/README.md
+++ b/charts/open-webui/README.md
@@ -50,6 +50,7 @@ helm upgrade --install open-webui open-webui/open-webui
 | copyAppData.resources | object | `{}` |  |
 | extraEnvVars | list | `[{"name":"OPENAI_API_KEY","value":"0p3n-w3bu!"}]` | Env vars added to the Open WebUI deployment. Most up-to-date environment variables can be found here: https://docs.openwebui.com/getting-started/env-configuration/ |
 | extraEnvVars[0] | object | `{"name":"OPENAI_API_KEY","value":"0p3n-w3bu!"}` | Default API key value for Pipelines. Should be updated in a production deployment, or be changed to the required API key if not using Pipelines |
+| extraInitContainers | list | `[]` | Additional init containers to add to the deployment/statefulset ref: <https://kubernetes.io/docs/concepts/workloads/pods/init-containers/> |
 | extraResources | list | `[]` | Extra resources to deploy with Open WebUI |
 | hostAliases | list | `[]` | HostAliases to be added to hosts-file of each container |
 | image | object | `{"pullPolicy":"IfNotPresent","repository":"ghcr.io/open-webui/open-webui","tag":""}` | Open WebUI image tags can be found here: https://github.com/open-webui/open-webui |
diff --git a/charts/open-webui/templates/workload-manager.yaml b/charts/open-webui/templates/workload-manager.yaml
index 781ca89..e5ea8f3 100644
--- a/charts/open-webui/templates/workload-manager.yaml
+++ b/charts/open-webui/templates/workload-manager.yaml
@@ -69,6 +69,9 @@ spec:
         {{- with .Values.volumeMounts.initContainer }}
         {{- toYaml . | nindent 8 }}
         {{- end }}
+      {{- with .Values.extraInitContainers }}
+      {{- toYaml . | nindent 6 }}
+      {{- end }}
       enableServiceLinks: false
       automountServiceAccountToken: {{ .Values.serviceAccount.automountServiceAccountToken }}
       {{- if .Values.serviceAccount.enable }}
diff --git a/charts/open-webui/values.yaml b/charts/open-webui/values.yaml
index 3632f4f..d294955 100644
--- a/charts/open-webui/values.yaml
+++ b/charts/open-webui/values.yaml
@@ -303,6 +303,16 @@ volumeMounts:
   # - name: ""
   #   mountPath: ""
 
+# -- Additional init containers to add to the deployment/statefulset
+# ref: <https://kubernetes.io/docs/concepts/workloads/pods/init-containers/>
+extraInitContainers: []
+# - name: custom-init
+#   image: busybox:latest
+#   command: ['sh', '-c', 'echo "Custom init container running"']
+#   volumeMounts:
+#   - name: data
+#     mountPath: /data
+
 # -- Configure pod volumes
 # ref: <https://kubernetes.io/docs/tasks/configure-pod-container/configure-volume-storage/>
 volumes: []