From 7f52a7f12881943e570c59fad059a81129f07c8f Mon Sep 17 00:00:00 2001
From: Hammad Bashir <HammadB@users.noreply.github.com>
Date: Mon, 22 Jul 2024 21:30:44 -0700
Subject: [PATCH] [ENH] Support FE replica count (#2561)

## Description of changes

*Summarize the changes made by this PR.*
 - Improvements & Bug fixes
	 - Add replica count in helm chart and bump version
 - New functionality
	 - /

## Test plan
*How are these changes tested?*
Existing tests ensure no functional regressions. This is not explicitly
tested.
- [x] Tests pass locally with `pytest` for python, `yarn test` for js,
`cargo test` for rust

## Documentation Changes
None
---
 k8s/distributed-chroma/Chart.yaml                      | 2 +-
 k8s/distributed-chroma/templates/frontend-service.yaml | 2 +-
 k8s/distributed-chroma/values.yaml                     | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/k8s/distributed-chroma/Chart.yaml b/k8s/distributed-chroma/Chart.yaml
index 198b6dfbe2d..0ad3a1ec1e3 100644
--- a/k8s/distributed-chroma/Chart.yaml
+++ b/k8s/distributed-chroma/Chart.yaml
@@ -16,7 +16,7 @@ apiVersion: v2
 name: distributed-chroma
 description: A helm chart for distributed Chroma
 type: application
-version: 0.1.5
+version: 0.1.6
 appVersion: "0.4.23"
 keywords:
   - chroma
diff --git a/k8s/distributed-chroma/templates/frontend-service.yaml b/k8s/distributed-chroma/templates/frontend-service.yaml
index 81d169c043f..7ed7226f61b 100644
--- a/k8s/distributed-chroma/templates/frontend-service.yaml
+++ b/k8s/distributed-chroma/templates/frontend-service.yaml
@@ -4,7 +4,7 @@ metadata:
   name: frontend-service
   namespace: {{ .Values.namespace }}
 spec:
-  replicas: 2
+  replicas: {{ .Values.frontendService.replicaCount }}
   selector:
     matchLabels:
       app: frontend-service
diff --git a/k8s/distributed-chroma/values.yaml b/k8s/distributed-chroma/values.yaml
index 25316472c94..5f6298cd162 100644
--- a/k8s/distributed-chroma/values.yaml
+++ b/k8s/distributed-chroma/values.yaml
@@ -32,6 +32,7 @@ frontendService:
         value: "http://jaeger:4317"
       - name: CHROMA_OTEL_GRANULARITY
         value: all
+  replicaCount: 2
 sysdb:
   image:
     repository: 'local'