From b8bdf98a4bfd42404af5fb41a7e3e3371597e972 Mon Sep 17 00:00:00 2001
From: Steven Aldinger <stevenaldinger@gmail.com>
Date: Thu, 12 Jun 2025 00:16:06 -0600
Subject: [PATCH] feat(helm): [BerriAI/litellm#11648] support extraContainers
 in migrations-job.yaml (#11649)

---
 deploy/charts/litellm-helm/templates/migrations-job.yaml | 3 +++
 deploy/charts/litellm-helm/values.yaml                   | 1 +
 2 files changed, 4 insertions(+)

diff --git a/deploy/charts/litellm-helm/templates/migrations-job.yaml b/deploy/charts/litellm-helm/templates/migrations-job.yaml
index ba69f0fef8..f00466bc48 100644
--- a/deploy/charts/litellm-helm/templates/migrations-job.yaml
+++ b/deploy/charts/litellm-helm/templates/migrations-job.yaml
@@ -53,6 +53,9 @@ spec:
           volumeMounts:
             {{- toYaml . | nindent 12 }}
           {{- end }}
+      {{- with .Values.migrationJob.extraContainers }}
+        {{- toYaml . | nindent 8 }}
+      {{- end }}
       {{- with .Values.volumes }}
       volumes:
         {{- toYaml . | nindent 8 }}
diff --git a/deploy/charts/litellm-helm/values.yaml b/deploy/charts/litellm-helm/values.yaml
index 213db35a20..8cf58b507f 100644
--- a/deploy/charts/litellm-helm/values.yaml
+++ b/deploy/charts/litellm-helm/values.yaml
@@ -197,6 +197,7 @@ migrationJob:
   disableSchemaUpdate: false # Skip schema migrations for specific environments. When True, the job will exit with code 0.
   annotations: {}
   ttlSecondsAfterFinished: 120
+  extraContainers: []
 
 # Additional environment variables to be added to the deployment as a map of key-value pairs
 envVars: {