diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index ca324b6e..afefb1f6 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -1,12 +1,16 @@
 name: Publish Python 🐍 distribution 📦 to TestPyPI
 
 on:
+  repository_dispatch:  # on trigger from llama-stack
+    types: [build-client-package]
+
   workflow_dispatch:  # Keep manual trigger
     inputs:
       rc_version:
         description: 'RC version number (e.g., 1, 2, 3)'
         required: true
         type: string
+
   schedule:
     - cron: "0 0 * * *"  # Run every day at midnight
 
@@ -33,6 +37,11 @@ jobs:
       run: |
         sed -i 's/version = "\([^"]*\)"/version = "\1rc${{ inputs.rc_version }}"/' pyproject.toml
         sed -i 's/__version__ = "\([^"]*\)"/__version__ = "\1rc${{ inputs.rc_version }}"/' src/llama_stack_client/_version.py
+    - name: Update version for repository_dispatch
+      if: github.event_name == 'repository_dispatch' && github.event.client_payload.source == 'llama-stack-nightly'
+      run: |
+        sed -i 's/version = "\([^"]*\)"/version = "\1${{ github.event.client_payload.version }}"/' pyproject.toml
+        sed -i 's/__version__ = "\([^"]*\)"/__version__ = "\1${{ github.event.client_payload.version }}"/' src/llama_stack_client/_version.py
     - name: Set up Python
       uses: actions/setup-python@v5
       with: