From 755c024be0effda4bfa72bd70b9fe0d0fdf351f4 Mon Sep 17 00:00:00 2001
From: Nyralei <93216976+Nyralei@users.noreply.github.com>
Date: Sat, 17 Aug 2024 03:32:09 +0300
Subject: [PATCH] Add the ability to set strategy for Deployment

---
 charts/local-ai/templates/deployment.yaml | 4 ++++
 charts/local-ai/values.yaml               | 7 +++++++
 2 files changed, 11 insertions(+)

diff --git a/charts/local-ai/templates/deployment.yaml b/charts/local-ai/templates/deployment.yaml
index 4756ab1..b4aae64 100644
--- a/charts/local-ai/templates/deployment.yaml
+++ b/charts/local-ai/templates/deployment.yaml
@@ -18,6 +18,10 @@ spec:
       app.kubernetes.io/name: {{ include "local-ai.name" . }}
       app.kubernetes.io/instance: {{ .Release.Name }}
   replicas: {{ .Values.replicaCount }}
+  {{- with .Values.deployment.strategy }}
+  strategy:
+    {{- toYaml . | nindent 4 }}
+  {{- end }}
   template:
     metadata:
       name: {{ template "local-ai.fullname" . }}
diff --git a/charts/local-ai/values.yaml b/charts/local-ai/values.yaml
index 3c35f10..d47bec0 100644
--- a/charts/local-ai/values.yaml
+++ b/charts/local-ai/values.yaml
@@ -16,6 +16,13 @@ deployment:
   #       name: some-secret
   #       key: hf-token
 
+  # # Set strategy for Deployment. The default values are below
+  # strategy:
+  #   type: RollingUpdate
+  #   rollingUpdate:
+  #     maxSurge: 25%
+  #     maxUnavailable: 25%
+
   modelsPath: "/models"
   download_model:
     # To use cloud provided (eg AWS) image, provide it like: 1234356789.dkr.ecr.us-REGION-X.amazonaws.com/busybox