From 173680e8e4c22611b3ab408337103c0757a5e44a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E3=83=84?= <1109954+Xeroxxx@users.noreply.github.com>
Date: Sat, 22 Mar 2025 16:25:21 +0100
Subject: [PATCH] feat: Added runtimeClassName to open-webui deployment.

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

diff --git a/charts/open-webui/Chart.yaml b/charts/open-webui/Chart.yaml
index 2d0f020..7fd38c8 100644
--- a/charts/open-webui/Chart.yaml
+++ b/charts/open-webui/Chart.yaml
@@ -1,6 +1,6 @@
 apiVersion: v2
 name: open-webui
-version: 5.25.0
+version: 5.26.0
 appVersion: 0.5.20
 home: https://www.openwebui.com/
 icon: >-
diff --git a/charts/open-webui/README.md b/charts/open-webui/README.md
index 6429888..1d0176a 100644
--- a/charts/open-webui/README.md
+++ b/charts/open-webui/README.md
@@ -1,6 +1,6 @@
 # open-webui
 
-![Version: 5.24.0](https://img.shields.io/badge/Version-5.24.0-informational?style=flat-square) ![AppVersion: 0.5.20](https://img.shields.io/badge/AppVersion-0.5.20-informational?style=flat-square)
+![Version: 5.26.0](https://img.shields.io/badge/Version-5.26.0-informational?style=flat-square) ![AppVersion: 0.5.20](https://img.shields.io/badge/AppVersion-0.5.20-informational?style=flat-square)
 
 Open WebUI: A User-Friendly Web Interface for Chat Interactions 👋
 
@@ -98,6 +98,7 @@ helm upgrade --install open-webui open-webui/open-webui
 | redis-cluster.replica.replicaCount | int | `3` | Number of Redis replica instances |
 | replicaCount | int | `1` |  |
 | resources | object | `{}` |  |
+| runtimeClassName | string | `""` | Allows changing the Runtime Class. For ex. to "nvidia" if nvidia container runtime is installed but not default. |
 | service | object | `{"annotations":{},"containerPort":8080,"labels":{},"loadBalancerClass":"","nodePort":"","port":80,"type":"ClusterIP"}` | Service values to expose Open WebUI pods to cluster |
 | serviceAccount.annotations | object | `{}` |  |
 | serviceAccount.automountServiceAccountToken | bool | `false` |  |
diff --git a/charts/open-webui/templates/workload-manager.yaml b/charts/open-webui/templates/workload-manager.yaml
index fe6290b..7bff4ab 100644
--- a/charts/open-webui/templates/workload-manager.yaml
+++ b/charts/open-webui/templates/workload-manager.yaml
@@ -74,6 +74,9 @@ spec:
       {{- end }}
       enableServiceLinks: false
       automountServiceAccountToken: {{ .Values.serviceAccount.automountServiceAccountToken }}
+      {{- if .Values.runtimeClassName }}
+      runtimeClassName: {{ .Values.runtimeClassName | quote }}
+      {{- end }}
       {{- if .Values.serviceAccount.enable }}
       serviceAccountName: {{ .Values.serviceAccount.name | default (include "open-webui.name" .) }}
       {{- end }}
diff --git a/charts/open-webui/values.yaml b/charts/open-webui/values.yaml
index 02007df..183dd01 100644
--- a/charts/open-webui/values.yaml
+++ b/charts/open-webui/values.yaml
@@ -266,6 +266,10 @@ extraEnvVars:
   # - name: OLLAMA_DEBUG
   #   value: "1"
 
+# -- Configure runtime class
+# ref: <https://kubernetes.io/docs/concepts/containers/runtime-class/>
+runtimeClassName: ""
+
 # -- Configure container volume mounts
 # ref: <https://kubernetes.io/docs/tasks/configure-pod-container/configure-volume-storage/>
 volumeMounts: