From 7cdd8b94d64dae1d1f5321e368108c40f73551e8 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Thu, 9 Jan 2025 18:51:29 -0800
Subject: [PATCH] trigger models build

---
 .github/workflows/publish-to-test-pypi.yml | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index 4cf47e304..06fa8d14e 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -12,8 +12,8 @@ on:
   #   - cron: "0 0 * * *"  # Run every day at midnight
 
 jobs:
-  trigger-client-build:
-    name: Trigger llama-stack-client build
+  trigger-client-and-models-build:
+    name: Trigger llama-stack-client and llama-models build
     runs-on: ubuntu-latest
     steps:
     - name: Get date
@@ -37,6 +37,13 @@ jobs:
         repository: meta-llama/llama-stack-client-python
         event-type: build-client-package
         client-payload: '{"source": "llama-stack-nightly", "version": "${{ steps.version.outputs.version }}"}'
+    - name: Trigger llama-models workflow
+      uses: peter-evans/repository-dispatch@v2
+      with:
+        token: ${{ secrets.PAT_TOKEN }}
+        repository: meta-llama/llama-models
+        event-type: build-models-package
+        client-payload: '{"source": "llama-stack-nightly", "version": "${{ steps.version.outputs.version }}"}'
     outputs:
       version: ${{ steps.version.outputs.version }}