feat(leo): Add Leo definition

iterative · Aug 10, 2023 · e388207 · e388207
1 parent 3d59ad4
commit e388207
Show file tree

Hide file tree

Showing 5 changed files with 195 additions and 21 deletions.
diff --git a/charts/studio/templates/_helpers.tpl b/charts/studio/templates/_helpers.tpl
@@ -50,18 +50,18 @@ helm.sh/chart: {{ include "studio.chart" . }}
 app.kubernetes.io/version: {{ .Chart.AppVersion | quote }}
 {{- end }}
 app.kubernetes.io/managed-by: {{ .Release.Service }}
-
 {{- end }}
-{{- define "studio-worker.labels" -}}
+
+{{- define "studio-backend.labels" -}}
 helm.sh/chart: {{ include "studio.chart" . }}
-{{ include "studio-worker.selectorLabels" . }}
+{{ include "studio-backend.selectorLabels" . }}
 {{- if .Chart.AppVersion }}
 app.kubernetes.io/version: {{ .Chart.AppVersion | quote }}
 {{- end }}
 app.kubernetes.io/managed-by: {{ .Release.Service }}
 {{- end }}
 
-{{- define "studio-backend.labels" -}}
+{{- define "studio-leo.labels" -}}
 helm.sh/chart: {{ include "studio.chart" . }}
 {{ include "studio-backend.selectorLabels" . }}
 {{- if .Chart.AppVersion }}
@@ -79,6 +79,15 @@ app.kubernetes.io/version: {{ .Chart.AppVersion | quote }}
 app.kubernetes.io/managed-by: {{ .Release.Service }}
 {{- end }}
 
+{{- define "studio-worker.labels" -}}
+helm.sh/chart: {{ include "studio.chart" . }}
+{{ include "studio-worker.selectorLabels" . }}
+{{- if .Chart.AppVersion }}
+app.kubernetes.io/version: {{ .Chart.AppVersion | quote }}
+{{- end }}
+app.kubernetes.io/managed-by: {{ .Release.Service }}
+{{- end }}
+
 {{/*
 Selector labels
 */}}
@@ -88,11 +97,6 @@ app.kubernetes.io/name: {{ include "studio.name" . }}
 app.kubernetes.io/instance: {{ .Release.Name }}
 {{- end }}
 
-{{- define "studio-ui.selectorLabels" -}}
-app.kubernetes.io/name: studio-ui
-app.kubernetes.io/instance: {{ .Release.Name }}
-{{- end }}
-
 {{- define "studio-backend.selectorLabels" -}}
 app.kubernetes.io/name: studio-backend
 app.kubernetes.io/instance: {{ .Release.Name }}
@@ -103,6 +107,16 @@ app.kubernetes.io/name: studio-beat
 app.kubernetes.io/instance: {{ .Release.Name }}
 {{- end }}
 
+{{- define "studio-leo.selectorLabels" -}}
+app.kubernetes.io/name: studio-leo
+app.kubernetes.io/instance: {{ .Release.Name }}
+{{- end }}
+
+{{- define "studio-ui.selectorLabels" -}}
+app.kubernetes.io/name: studio-ui
+app.kubernetes.io/instance: {{ .Release.Name }}
+{{- end }}
+
 {{- define "studio-worker.selectorLabels" -}}
 app.kubernetes.io/name: studio-worker
 app.kubernetes.io/instance: {{ .Release.Name }}

diff --git a/charts/studio/templates/configmap-studio.yaml b/charts/studio/templates/configmap-studio.yaml
@@ -119,3 +119,9 @@ data:
   DQL_ENABLED: {{ $dvcx.enabled | default "False" | quote }}
   DVCX_ENABLED: {{ $dvcx.enabled | default "False" | quote }}
 
+  LEO_SERVICE_HOST: {{ .Release.Name }}-studio-leo
+  LEO_SERVICE_PORT: {{ .Values.studioLeo.service.port }}
+  LEO_HOST: "{{ .Release.Name }}-studio-leo:{{ .Values.studioLeo.service.port }}"
+  LEO_ENCODING: "utf-8"
+  LEO_API_PATH_PREFIX: "{{- if and .Values.global.basePath (not (eq .Values.global.basePath "/")) }}/{{ include "studio.basePath" . }}{{- end }}/leo"
+  LEO_ORIGIN: "http://{{.Release.Name}}-studio-leo"
diff --git a/charts/studio/templates/deployment-studio-leo.yaml b/charts/studio/templates/deployment-studio-leo.yaml
@@ -0,0 +1,94 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: {{ .Release.Name }}-studio-leo
+  labels:
+    {{- include "studio-leo.labels" . | nindent 4 }}
+spec:
+  {{- if not .Values.studioLeo.autoscaling.enabled }}
+  replicas: {{ .Values.studioLeo.replicaCount }}
+  {{- end }}
+  selector:
+    matchLabels:
+      {{- include "studio-leo.selectorLabels" . | nindent 6 }}
+  template:
+    metadata:
+      annotations:
+        {{- include "studio.checksum" . | indent 8 }}
+      {{- with .Values.studioLeo.podAnnotations }}
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      labels:
+        {{- include "studio-leo.selectorLabels" . | nindent 8 }}
+    spec:
+      {{- with .Values.imagePullSecrets }}
+      imagePullSecrets:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      serviceAccountName: {{ include "studio.serviceAccountName" . }}
+      securityContext:
+        {{- toYaml .Values.studioLeo.podSecurityContext | nindent 8 }}
+      containers:
+        - name: studio-leo
+          securityContext:
+            {{- toYaml .Values.studioLeo.securityContext | nindent 12 }}
+          image: "{{ .Values.studioLeo.image.repository }}:{{ .Values.studioLeo.image.tag | default .Chart.AppVersion }}"
+          imagePullPolicy: {{ .Values.studioBackend.image.pullPolicy }}
+          ports:
+            - name: http
+              containerPort: 80
+              protocol: TCP
+          startupProbe:
+            tcpSocket:
+              port: 80
+            initialDelaySeconds: 5
+            periodSeconds: 10
+          livenessProbe:
+            tcpSocket:
+              port: 80
+            initialDelaySeconds: 5
+            periodSeconds: 10
+          readinessProbe:
+            httpGet:
+              path: /leo/debug/version
+              port: 80
+            initialDelaySeconds: 5
+            periodSeconds: 25
+            successThreshold: 3
+            timeoutSeconds: 120
+          resources:
+            {{- toYaml .Values.studioLeo.resources | nindent 12 }}
+          envFrom:
+            - secretRef:
+                name: studio
+            {{- if .Values.global.envFromSecret }}
+            - secretRef:
+                name: {{ .Values.global.envFromSecret }}
+            {{- end }}
+            {{- if .Values.studioLeo.envFromSecret }}
+            - secretRef:
+                name: {{ .Values.studioLeo.envFromSecret }}
+            {{- end }}
+          volumeMounts:
+            {{- if .Values.global.customCaCert }}
+            - name: studio-ca-certificates
+              mountPath: /usr/local/share/ca-certificates
+            {{- end }}
+      volumes:
+        {{- if .Values.global.customCaCert }}
+        - name: studio-ca-certificates
+          configMap:
+            name: studio-ca-certificates
+        {{- end }}
+      {{- with .Values.studioLeo.nodeSelector }}
+      nodeSelector:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      {{- with .Values.studioLeo.affinity }}
+      affinity:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
+      {{- with .Values.studioLeo.tolerations }}
+      tolerations:
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
diff --git a/charts/studio/templates/service-leo.yaml b/charts/studio/templates/service-leo.yaml
@@ -0,0 +1,15 @@
+apiVersion: v1
+kind: Service
+metadata:
+  name: {{ .Release.Name}}-studio-leo
+  labels:
+    {{- include "studio-leo.labels" . | nindent 4 }}
+spec:
+  type: {{ .Values.studioLeo.service.type }}
+  ports:
+    - port: {{ .Values.studioLeo.service.port }}
+      targetPort: http
+      protocol: TCP
+      name: http
+  selector:
+    {{- include "studio-leo.selectorLabels" . | nindent 4 }}
diff --git a/charts/studio/values.yaml b/charts/studio/values.yaml
@@ -432,10 +432,6 @@ studioBackend:
     targetCPUUtilizationPercentage: 80
     # targetMemoryUtilizationPercentage: 80
 
-  nameOverride: ""
-
-  fullnameOverride: ""
-
   podAnnotations: {}
 
   podSecurityContext: {}
@@ -483,10 +479,6 @@ studioBeat:
     targetCPUUtilizationPercentage: 80
     # targetMemoryUtilizationPercentage: 80
 
-  nameOverride: ""
-
-  fullnameOverride: ""
-
   podAnnotations: {}
 
   podSecurityContext: {}
@@ -506,6 +498,63 @@ studioBeat:
 
   affinity: {}
 
+studioLeo:
+  # -- Studio: Additional environment variables for leo pods
+  envVars: {}
+  # Example:
+  # envVars:
+  #   DEBUG: "True"
+
+  # -- Studio: The name of an existing Secret that contains sensitive environment variables
+  # passed to leo pods.
+  envFromSecret: ""
+
+  replicaCount: 1
+
+  image:
+    repository: docker.iterative.ai/leo-server
+    pullPolicy: IfNotPresent
+    # Overrides the image tag whose default is the chart appVersion.
+    # tag: "v1.34.0"
+
+  service:
+    type: ClusterIP
+    port: 8181
+
+  resources:
+    requests:
+      cpu: 100m
+      memory: 512Mi
+    limits:
+      cpu: 500m
+      memory: 1Gi
+
+  autoscaling:
+    enabled: false
+    minReplicas: 1
+    maxReplicas: 1
+    targetCPUUtilizationPercentage: 80
+    # targetMemoryUtilizationPercentage: 80
+
+  podAnnotations: {}
+
+  podSecurityContext: {}
+  # fsGroup: 2000
+
+  securityContext: {}
+    # capabilities:
+    #   drop:
+    #   - ALL
+    # readOnlyRootFilesystem: true
+    # runAsNonRoot: true
+  # runAsUser: 1000
+
+  nodeSelector: {}
+
+  tolerations: []
+
+  affinity: {}
+
 studioWorker:
   # -- Studio: Additional environment variables for worker pods
   envVars: {}
@@ -534,10 +583,6 @@ studioWorker:
     targetCPUUtilizationPercentage: 80
     # targetMemoryUtilizationPercentage: 80
 
-  nameOverride: ""
-
-  fullnameOverride: ""
-
   podAnnotations: {}
 
   podSecurityContext: {}