From 7adc8c122708e15fd24c5f041451b17b441be053 Mon Sep 17 00:00:00 2001
From: Hardik Shah <hjshah@fb.com>
Date: Wed, 15 Jan 2025 19:32:35 -0800
Subject: [PATCH] add notebook test to nightly build

---
 .github/workflows/publish-to-test-pypi.yml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index 957761235..848c6b2a3 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -199,7 +199,13 @@ jobs:
       - publish-to-testpypi
       - trigger-client-and-models-build
     runs-on: ubuntu-latest
+    env:
+      TOGETHER_API_KEY: ${{ secrets.TOGETHER_API_KEY }}
+      TAVILY_SEARCH_API_KEY: ${{ secrets.TAVILY_SEARCH_API_KEY }}
     steps:
+    - uses: actions/checkout@v4
+      with:
+        persist-credentials: false
     - name: Install the package
       run: |
         max_attempts=6
@@ -228,5 +234,9 @@ jobs:
         llama stack list-apis
         llama stack list-providers inference
         llama stack list-providers telemetry
+    - name: Test Notebook
+      run: |
+        pip install pytest nbval
+        pytest -v -s --nbval-lax ./docs/notebooks/Llama_Stack_Building_AI_Applications.ipynb
 
     # TODO: add trigger for integration test workflow & docker builds