From 9c75989d6b73e5ecfd12b577469155f84ae45663 Mon Sep 17 00:00:00 2001
From: hammadb <hammad@trychroma.com>
Date: Tue, 3 Dec 2024 12:21:03 -0800
Subject: [PATCH] [ENH] K8s support query replicas

---
 k8s/distributed-chroma/templates/query-service.yaml | 2 +-
 k8s/distributed-chroma/values.yaml                  | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/k8s/distributed-chroma/templates/query-service.yaml b/k8s/distributed-chroma/templates/query-service.yaml
index 5bd5b241de3..2a4e1bce397 100644
--- a/k8s/distributed-chroma/templates/query-service.yaml
+++ b/k8s/distributed-chroma/templates/query-service.yaml
@@ -37,7 +37,7 @@ metadata:
   namespace: {{ .Values.namespace }}
 spec:
   serviceName: query-service
-  replicas: 2
+  replicas: {{ .Values.queryService.replicaCount }}
   selector:
     matchLabels:
       app: query-service
diff --git a/k8s/distributed-chroma/values.yaml b/k8s/distributed-chroma/values.yaml
index 24bd60839dd..bac3a498c69 100644
--- a/k8s/distributed-chroma/values.yaml
+++ b/k8s/distributed-chroma/values.yaml
@@ -71,6 +71,7 @@ queryService:
   cache:
     hostPath: '/local/cache/chroma'
     mountPath: '/cache/'
+  replicaCount: 2
 
 
 compactionService: