From ec1412d5dff6ea5f830da266d7d2f874ff093c78 Mon Sep 17 00:00:00 2001
From: Sixian Yi <sxyi@meta.com>
Date: Thu, 23 Jan 2025 00:53:49 -0800
Subject: [PATCH] fix

---
 .github/workflows/tests.yml | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 2157fb6a99..9832c07d9d 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -14,20 +14,30 @@ jobs:
       TAVILY_SEARCH_API_KEY: ${{ secrets.TAVILY_SEARCH_API_KEY }}
     strategy:
       matrix:
-        provider: [fireworks, ollama]
+        provider: [fireworks, together]
     steps:
       - uses: actions/checkout@v4
+
       - name: Echo branch name
         run: echo "Running on branch ${{ github.ref }}"
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt pytest
           pip install -e .
+
       - name: Build providers
         run: |
           llama stack build --template ${{ matrix.provider }} --image-type venv
+
+      - name: Install local llama-stack-client and llama-models
+        run: |
+          pip install git+https://github.com/meta-llama/llama-stack-client-python.git
+          pip install git+https://github.com/meta-llama/llama-models.git
+
       - name: Run client-sdk test
         working-directory: "${{ github.workspace }}"
-        run:
+        run: |
+          export INFERENCE_MODEL=meta-llama/Llama-3.1-8B-Instruct
           LLAMA_STACK_CONFIG=./llama_stack/templates/${{ matrix.provider }}/run.yaml pytest ./tests/client-sdk/inference/test_inference.py