diff --git a/.github/workflows/upgrade_automation.yml b/.github/workflows/upgrade_automation.yml
index 94fceadd2..b94163b1c 100644
--- a/.github/workflows/upgrade_automation.yml
+++ b/.github/workflows/upgrade_automation.yml
@@ -1,12 +1,20 @@
-name: Boilerplate Upgrade Automation
+name: Upgrade Automation
 on:
   workflow_dispatch:
-
+    inputs:
+      component:
+        description: "Name of Flyte component"
+        required: true
+        default: "boilerplate"
+        type: choice
+        options:
+          - boilerplate
+          - flyteidl
 jobs:
   trigger-upgrade:
-    name: Boilerplate Upgrade Automation
+    name: ${{ github.event.inputs.component }} Upgrade
     uses: flyteorg/flytetools/.github/workflows/flyte_automation.yml@master
     with:
-      component: boilerplate
+      component: ${{ github.event.inputs.component }}
     secrets:
-      FLYTE_BOT_PAT: ${{ secrets.FLYTE_BOT_PAT }}
\ No newline at end of file
+      FLYTE_BOT_PAT: ${{ secrets.FLYTE_BOT_PAT }}