From 9c673c0f33f4ef8ac049ad86ef6240a78b295823 Mon Sep 17 00:00:00 2001 From: Kevin Green Date: Mon, 24 Nov 2025 18:49:07 -0500 Subject: [PATCH] Add support to specify a storage class for brainstore readers and writers --- braintrust/templates/brainstore-reader-deployment.yaml | 10 ++++++++++ braintrust/templates/brainstore-writer-deployment.yaml | 10 ++++++++++ braintrust/values.yaml | 7 +++++-- 3 files changed, 25 insertions(+), 2 deletions(-) diff --git a/braintrust/templates/brainstore-reader-deployment.yaml b/braintrust/templates/brainstore-reader-deployment.yaml index cc4b002..2fa2d8e 100644 --- a/braintrust/templates/brainstore-reader-deployment.yaml +++ b/braintrust/templates/brainstore-reader-deployment.yaml @@ -134,6 +134,16 @@ spec: resources: requests: storage: {{ required "brainstore.reader.volume.size must be set" .Values.brainstore.reader.volume.size | quote }} + {{- else if .Values.brainstore.reader.volume.ephemeralVolumeStorageClassName }} + ephemeral: + volumeClaimTemplate: + spec: + volumeMode: Filesystem + accessModes: ["ReadWriteOnce"] + storageClassName: {{ .Values.brainstore.reader.volume.ephemeralVolumeStorageClassName }} + resources: + requests: + storage: {{ required "brainstore.reader.volume.size must be set" .Values.brainstore.reader.volume.size | quote }} {{- else }} emptyDir: {} {{- end }} diff --git a/braintrust/templates/brainstore-writer-deployment.yaml b/braintrust/templates/brainstore-writer-deployment.yaml index 3024c47..a05fc9a 100644 --- a/braintrust/templates/brainstore-writer-deployment.yaml +++ b/braintrust/templates/brainstore-writer-deployment.yaml @@ -134,6 +134,16 @@ spec: resources: requests: storage: {{ required "brainstore.writer.volume.size must be set" .Values.brainstore.writer.volume.size | quote }} + {{- else if .Values.brainstore.writer.volume.ephemeralVolumeStorageClassName }} + ephemeral: + volumeClaimTemplate: + spec: + volumeMode: Filesystem + accessModes: ["ReadWriteOnce"] + storageClassName: {{ .Values.brainstore.writer.volume.ephemeralVolumeStorageClassName }} + resources: + requests: + storage: {{ required "brainstore.writer.volume.size must be set" .Values.brainstore.writer.volume.size | quote }} {{- else }} emptyDir: {} {{- end }} diff --git a/braintrust/values.yaml b/braintrust/values.yaml index 307b439..6505535 100644 --- a/braintrust/values.yaml +++ b/braintrust/values.yaml @@ -169,8 +169,10 @@ brainstore: # Optional: Volume configuration for cache storage # When not set, uses default emptyDir: {} (backward compatible) volume: - # Storage size for ephemeral storage requests (used with GKE Autopilot local SSDs) + # Storage size for ephemeral storage requests size: "" + # Storage class for ephemeral storage requests + ephemeralVolumeStorageClassName: "" extraEnvVars: [] nodeSelector: {} tolerations: [] @@ -206,8 +208,9 @@ brainstore: # When not set, uses default emptyDir: {} (backward compatible) volume: # Storage size for ephemeral storage requests - # Used with GKE Autopilot local SSDs and Azure Container Storage CSI size: "" + # Storage class for ephemeral storage requests + ephemeralVolumeStorageClassName: "" extraEnvVars: [] # Example: # - name: MY_ENV_VAR