From 603ee8cde56c4ee5321136f8363a22faad41bbd5 Mon Sep 17 00:00:00 2001
From: stepanbaghdasaryan <styop.b@hotmail.com>
Date: Tue, 7 May 2024 22:10:29 +0200
Subject: [PATCH] feat: add feature to configure ollama service labels via
 values

This commit adds the ability to configure labels for the ollama service in the values.yaml file. Users can now specify custom labels for the ollama service, providing greater flexibility in managing its configuration.
---
 charts/open-webui/templates/ollama-service.yaml | 3 +++
 charts/open-webui/values.yaml                   | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/charts/open-webui/templates/ollama-service.yaml b/charts/open-webui/templates/ollama-service.yaml
index 32c93ca..d2848a3 100644
--- a/charts/open-webui/templates/ollama-service.yaml
+++ b/charts/open-webui/templates/ollama-service.yaml
@@ -5,6 +5,9 @@ metadata:
   name: {{ include "ollama.name" . }}
   labels:
     {{- include "ollama.labels" . | nindent 4 }}
+    {{- with .Values.ollama.service.labels }}
+    {{- toYaml . | nindent 4 }}
+    {{- end }}
   {{- with .Values.ollama.service.annotations }}
   annotations:
     {{- toYaml . | nindent 4 }}
diff --git a/charts/open-webui/values.yaml b/charts/open-webui/values.yaml
index 4437973..81a4197 100644
--- a/charts/open-webui/values.yaml
+++ b/charts/open-webui/values.yaml
@@ -29,6 +29,7 @@ ollama:
   service:
     type: ClusterIP
     annotations: {}
+    labels: {}
     port: 80
     containerPort: 11434
   gpu:
@@ -72,4 +73,4 @@ webui:
     containerPort: 8080
     nodePort: ""
     labels: {}
-    loadBalancerClass: "" 
+    loadBalancerClass: ""