Refactoring and bug fixes

stackhpc · Nov 11, 2024 · 6e61451 · 6e61451
1 parent fecd477
commit 6e61451
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 3 deletions.
diff --git a/charts/flux-image-gen/Chart.yaml b/charts/flux-image-gen/Chart.yaml
@@ -21,4 +21,4 @@ version: 0.1.0
 # incremented each time you make changes to the application. Versions are not expected to
 # follow Semantic Versioning. They should reflect the version the application is using.
 # It is recommended to use it with quotes.
-appVersion: "02aeef3"
+appVersion: "fb1397e"
diff --git a/...flux-image-gen/templates/ui/configmap.yml → ...lux-image-gen/templates/ui/configmap.yaml b/...flux-image-gen/templates/ui/configmap.yml → ...lux-image-gen/templates/ui/configmap.yaml
@@ -11,4 +11,6 @@ data:
     {{- range $model := .Values.models }}
       - name: {{ . }}
         address: {{ printf "http://%s.%s.svc:%v" ( printf "%s-%s-api" (include "flux-image-gen.fullname" $) . ) $.Release.Namespace $.Values.api.service.port }}
-    {{- end -}}
+    {{- end }}
+    example_prompt: |
+      {{- .Values.examplePrompt | nindent 6 -}}
diff --git a/charts/flux-image-gen/templates/ui/deployment.yaml b/charts/flux-image-gen/templates/ui/deployment.yaml
@@ -14,6 +14,8 @@ spec:
     metadata:
       {{- with .Values.ui.podAnnotations }}
       annotations:
+        # Recreate pods if settings config map changes
+        checksum/config: {{ include (print $.Template.BasePath "/ui/configmap.yaml") . | sha256sum }}
         {{- toYaml . | nindent 8 }}
       {{- end }}
       labels:

diff --git a/charts/flux-image-gen/templates/ui/ingress.yaml b/charts/flux-image-gen/templates/ui/ingress.yaml
@@ -35,7 +35,7 @@ spec:
             {{- end }}
             backend:
               service:
-                name: {{ include "flux-image-gen.fullname" $ }}
+                name: {{ include "flux-image-gen.fullname" $ }}-ui
                 port:
                   number: {{ $.Values.ui.service.port }}
           {{- end }}

diff --git a/charts/flux-image-gen/values.yaml b/charts/flux-image-gen/values.yaml
@@ -6,6 +6,9 @@ models:
   - flux-schnell
   # - flux-dev
 
+examplePrompt: |
+  Yoda riding a tiny unicorn through space.
+
 # This sets the container image more information can be found here: https://kubernetes.io/docs/concepts/containers/images/
 image:
   repository: ghcr.io/stackhpc/flux-image-model-inference