From 05305682f79ff14495b2beda67467ba13730160d Mon Sep 17 00:00:00 2001
From: John Adams <john.adams@sixgen.io>
Date: Wed, 6 Sep 2023 11:43:10 -0400
Subject: [PATCH] add imagepullpolicy

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

diff --git a/charts/local-ai/templates/deployment.yaml b/charts/local-ai/templates/deployment.yaml
index 3da08a4..121fdcc 100644
--- a/charts/local-ai/templates/deployment.yaml
+++ b/charts/local-ai/templates/deployment.yaml
@@ -31,6 +31,7 @@ spec:
         {{- if .Values.promptTemplates }}
         - name: prompt-templates
           image: {{ .Values.deployment.prompt_templates.image }}
+          imagePullPolicy: {{ .Values.image.pullPolicy }}
           command: ["/bin/sh", "-c"]
           args:
             - |
@@ -43,6 +44,7 @@ spec:
         {{- end }}
         - name: download-model
           image: {{ .Values.deployment.download_model.image }}
+          imagePullPolicy: {{ .Values.image.pullPolicy }}
           command: ["/bin/sh", "-c"]
           args:
             - |
@@ -88,6 +90,7 @@ spec:
       containers:
         - name: {{ template "local-ai.fullname" . }}
           image: {{ .Values.deployment.image }}
+          imagePullPolicy: {{ .Values.image.pullPolicy }}
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           env:
diff --git a/charts/local-ai/values.yaml b/charts/local-ai/values.yaml
index 88bfcc6..523c58b 100644
--- a/charts/local-ai/values.yaml
+++ b/charts/local-ai/values.yaml
@@ -12,6 +12,7 @@ deployment:
   prompt_templates:
     # To use cloud provided (eg AWS) image, provide it like: 1234356789.dkr.ecr.us-REGION-X.amazonaws.com/busybox
     image: busybox
+  pullPolicy: IfNotPresent
 
 resources:
   {}