From fab19a9666b1a33e48d095d352d5694deb596c19 Mon Sep 17 00:00:00 2001
From: Blake Driggers <bdriggers@torchtechnologies.com>
Date: Thu, 12 Jun 2025 10:46:36 -0500
Subject: [PATCH] feat(open-webui): add optional command args overrides for
 open-webui copyAppData init container

---
 charts/open-webui/README.md                       | 2 ++
 charts/open-webui/templates/workload-manager.yaml | 7 ++++++-
 charts/open-webui/values.yaml                     | 5 +++++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/charts/open-webui/README.md b/charts/open-webui/README.md
index d54634d..1415123 100644
--- a/charts/open-webui/README.md
+++ b/charts/open-webui/README.md
@@ -172,6 +172,8 @@ helm upgrade --install open-webui open-webui/open-webui
 | command | list | `[]` | Open WebUI container command (overrides default entrypoint) |
 | commonEnvVars | list | `[]` | Env vars added to the Open WebUI deployment, common across environments. Most up-to-date environment variables can be found here: https://docs.openwebui.com/getting-started/env-configuration/ (caution: environment variables defined in both `extraEnvVars` and `commonEnvVars` will result in a conflict. Avoid duplicates) |
 | containerSecurityContext | object | `{}` | Configure container security context ref: <https://kubernetes.io/docs/tasks/configure-pod-container/security-context/#set-the-security-context-for-a-containe> |
+| copyAppData.args | list | `[]` | Open WebUI copy-app-data init container arguments (overrides default) |
+| copyAppData.command | list | `[]` | Open WebUI copy-app-data init container command (overrides default) |
 | copyAppData.resources | object | `{}` |  |
 | databaseUrl | string | `""` | Configure database URL, needed to work with Postgres (example: `postgresql://<user>:<password>@<service>:<port>/<database>`), leave empty to use the default sqlite database |
 | enableOpenaiApi | bool | `true` | Enables the use of OpenAI APIs |
diff --git a/charts/open-webui/templates/workload-manager.yaml b/charts/open-webui/templates/workload-manager.yaml
index fdffd70..ccabbe8 100644
--- a/charts/open-webui/templates/workload-manager.yaml
+++ b/charts/open-webui/templates/workload-manager.yaml
@@ -52,7 +52,12 @@ spec:
         image: {{ .repository }}:{{ .tag | default $.Chart.AppVersion }}
         imagePullPolicy: {{ .pullPolicy }}
         {{- end }}
-        command: ['sh', '-c', 'cp -R -n /app/backend/data/* /tmp/app-data/']
+        command:
+          {{- toYaml (.Values.copyAppData.command | default (list "sh" "-c" "cp -R -n /app/backend/data/* /tmp/app-data/")) | nindent 10 }}
+        {{- with .Values.copyAppData.args }}
+        args:
+          {{- toYaml . | nindent 10 }}
+        {{- end }}
         {{- with .Values.containerSecurityContext }}
         securityContext:
           {{- toYaml . | nindent 10 }}
diff --git a/charts/open-webui/values.yaml b/charts/open-webui/values.yaml
index 460550b..7e173eb 100644
--- a/charts/open-webui/values.yaml
+++ b/charts/open-webui/values.yaml
@@ -181,6 +181,11 @@ startupProbe: {}
 resources: {}
 
 copyAppData:
+  # -- Open WebUI copy-app-data init container command (overrides default)
+  command: []
+  # -- Open WebUI copy-app-data init container arguments (overrides default)
+  args: []
+
   resources: {}
 
 managedCertificate: