From 2229cfaef1aa35d0b9f1e871c16532c5cc872399 Mon Sep 17 00:00:00 2001
From: jyje <jyjeon@outlook.com>
Date: Sun, 29 Dec 2024 04:11:48 +0900
Subject: [PATCH] feat: add 3 container probes for workload-manager

- add livenessProbe, readinessProbe, startupProbe for main container
- add key rows in table of readme
- bump the chart version from 5.0.0 to 5.1.0
---
 charts/open-webui/Chart.yaml                      | 2 +-
 charts/open-webui/README.md                       | 3 +++
 charts/open-webui/templates/workload-manager.yaml | 9 +++++++++
 3 files changed, 13 insertions(+), 1 deletion(-)
 mode change 100755 => 100644 charts/open-webui/templates/workload-manager.yaml

diff --git a/charts/open-webui/Chart.yaml b/charts/open-webui/Chart.yaml
index 5c7a4f7..f868d32 100644
--- a/charts/open-webui/Chart.yaml
+++ b/charts/open-webui/Chart.yaml
@@ -1,6 +1,6 @@
 apiVersion: v2
 name: open-webui
-version: 5.0.0
+version: 5.1.0
 appVersion: 0.5.2
 home: https://www.openwebui.com/
 icon: >-
diff --git a/charts/open-webui/README.md b/charts/open-webui/README.md
index f808a1b..11565b6 100644
--- a/charts/open-webui/README.md
+++ b/charts/open-webui/README.md
@@ -49,6 +49,9 @@ helm upgrade --install open-webui open-webui/open-webui
 | 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 |
 | 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 |
 | imagePullSecrets | list | `[]` | Configure imagePullSecrets to use private registry ref: <https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry> |
+| livenessProbe | object | `{}` | Configure livenessProbe ref: <https://kubernetes.io/docs/tasks/configure-pod-container/configure-liveness-readiness-startup-probes> |
+| readinessProbe | object | `{}` | Configure readinessProbe ref: <https://kubernetes.io/docs/tasks/configure-pod-container/configure-liveness-readiness-startup-probes> |
+| startupProbe | object | `{}` | Configure startupProbe ref: <https://kubernetes.io/docs/tasks/configure-pod-container/configure-liveness-readiness-startup-probes> |
 | ingress.additionalHosts | list | `[]` |  |
 | ingress.annotations | object | `{}` | Use appropriate annotations for your Ingress controller, e.g., for NGINX: nginx.ingress.kubernetes.io/rewrite-target: / |
 | ingress.class | string | `""` |  |
diff --git a/charts/open-webui/templates/workload-manager.yaml b/charts/open-webui/templates/workload-manager.yaml
old mode 100755
new mode 100644
index 8ae3a77..8b5d14f
--- a/charts/open-webui/templates/workload-manager.yaml
+++ b/charts/open-webui/templates/workload-manager.yaml
@@ -87,6 +87,15 @@ spec:
         ports:
         - name: http
           containerPort: {{ .Values.service.containerPort }}
+        {{- with .Values.livenessProbe }}
+        livenessProbe: {{- toYaml . | nindent 10 }}
+        {{- end }}
+        {{- with .Values.readinessProbe }}
+        readinessProbe: {{- toYaml . | nindent 10 }}
+        {{- end }}
+        {{- with .Values.startupProbe }}
+        startupProbe: {{- toYaml . | nindent 10 }}
+        {{- end }}
         {{- with .Values.resources }}
         resources: {{- toYaml . | nindent 10 }}
         {{- end }}