From ceb8dd8b57400f5467567113d8d1e633a8c82050 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 1 Nov 2024 18:01:02 +0530
Subject: [PATCH] fix build step naming

---
 .github/workflows/ghcr_deploy.yml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/.github/workflows/ghcr_deploy.yml b/.github/workflows/ghcr_deploy.yml
index 587abc8ea..35a211c12 100644
--- a/.github/workflows/ghcr_deploy.yml
+++ b/.github/workflows/ghcr_deploy.yml
@@ -19,6 +19,13 @@ env:
   IMAGE_NAME: ${{ github.repository }}
   CHART_NAME: litellm-helm
 
+# Add workflow input
+inputs:
+  image_tag:
+    description: 'Custom image tag name'
+    required: false
+    default: ''  # Empty default means we'll use PR-based naming
+
 # There is a single job in this workflow. It's configured to run on the latest available version of Ubuntu.
 jobs:
   # print commit hash, tag, and release type
@@ -99,6 +106,8 @@ jobs:
         uses: docker/metadata-action@9ec57ed1fcdbf14dcef7dfbe97b2010124a938b7
         with:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+          tags: |
+            ${{ inputs.image_tag != '' && format('type=raw,value={0}', inputs.image_tag) || 'type=ref,event=pr' }}
       # Configure multi platform Docker builds
       - name: Set up QEMU
         uses: docker/setup-qemu-action@e0e4588fad221d38ee467c0bffd91115366dc0c5