From 97250c55f03e22ce84caafad833b3e9101b58f4c Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Fri, 13 Mar 2026 11:10:46 -0700
Subject: [PATCH] Revert "tmp: add custom version input to publish_enterprise
 workflow"

This reverts commit 60bb1a22836e5c6bf09cbb51e5d8067a7a436b68.
---
 .github/workflows/publish_enterprise.yml | 53 +++++++++++++++++++-----
 1 file changed, 42 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/publish_enterprise.yml b/.github/workflows/publish_enterprise.yml
index c94f74cfccc..459a233cb71 100644
--- a/.github/workflows/publish_enterprise.yml
+++ b/.github/workflows/publish_enterprise.yml
@@ -4,18 +4,14 @@ on:
   workflow_dispatch:
     inputs:
       bump:
-        description: "Version bump type (ignored when 'version' is set)"
-        required: false
+        description: "Version bump type"
+        required: true
         default: "patch"
         type: choice
         options:
           - patch
           - minor
           - major
-      version:
-        description: "Exact version to publish (e.g. 0.1.34.dev1). Overrides bump."
-        required: false
-        default: ""
 
 jobs:
   publish:
@@ -42,18 +38,53 @@ jobs:
         id: bump
         run: |
           OLD=$(poetry version -s)
-          if [ -n "${{ github.event.inputs.version }}" ]; then
-            poetry version "${{ github.event.inputs.version }}"
-          else
-            poetry version ${{ github.event.inputs.bump }}
-          fi
+          poetry version ${{ github.event.inputs.bump }}
           NEW=$(poetry version -s)
           echo "old=$OLD" >> $GITHUB_OUTPUT
           echo "new=$NEW" >> $GITHUB_OUTPUT
 
+      - name: Update version refs in root pyproject.toml and requirements.txt
+        run: |
+          OLD=${{ steps.bump.outputs.old }}
+          NEW=${{ steps.bump.outputs.new }}
+          sed -i "s/litellm-enterprise = {version = \"${OLD}\"/litellm-enterprise = {version = \"${NEW}\"/" ../pyproject.toml
+          sed -i "s/litellm-enterprise==${OLD}/litellm-enterprise==${NEW}/" ../requirements.txt
+
+      - name: Update poetry.lock
+        working-directory: .
+        run: poetry lock
+
       - name: Build
         run: poetry build
 
+      - name: Commit version bump and create PR
+        id: create-pr
+        run: |
+          git config user.name "github-actions[bot]"
+          git config user.email "github-actions[bot]@users.noreply.github.com"
+          cd ..
+          BRANCH="bump/enterprise-${{ steps.bump.outputs.new }}"
+          git checkout -b "$BRANCH"
+          git add enterprise/pyproject.toml pyproject.toml requirements.txt poetry.lock
+          git commit -m "bump: litellm-enterprise ${{ steps.bump.outputs.old }} → ${{ steps.bump.outputs.new }}"
+          git push origin "$BRANCH" --force
+          gh pr create \
+            --title "bump: litellm-enterprise ${{ steps.bump.outputs.old }} → ${{ steps.bump.outputs.new }}" \
+            --body "Version bump for litellm-enterprise. Merge to update main." \
+            --head "$BRANCH" \
+            --base main \
+          || true
+          PR_URL=$(gh pr list --head "$BRANCH" --json url -q '.[0].url')
+          echo "pr_url=$PR_URL" >> $GITHUB_OUTPUT
+        env:
+          GH_TOKEN: ${{ github.token }}
+
+      - name: Enable auto-merge
+        run: |
+          gh pr merge "${{ steps.create-pr.outputs.pr_url }}" --auto --squash
+        env:
+          GH_TOKEN: ${{ github.token }}
+
       - name: Publish to PyPI
         env:
           TWINE_USERNAME: __token__