Merge pull request trustyai-explainability#45 from m-misiura/lls_test_upstream

Run unit tests with both lls stable and main

Author: m-misiura

.github/workflows/test-unit.yml

@@ -1,28 +1,48 @@
 name: Run Unit Tests
 
 on:
+  push:
+    branches: [main, lls_test_upstream]
   pull_request:
     branches: [main]
 
 jobs:
   test:
-    name: Run Tests
+    name: "Test with ${{ matrix.test-name }}"
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        include:
+          - python-version: "3.12"
+            llama-stack-source: "stable"
+            test-name: "llama-stack stable (pyproject.toml)"
+            install-cmd: "python -m pip install -e \".[test]\""
+          - python-version: "3.12"
+            llama-stack-source: "main"
+            test-name: "llama-stack main branch (upstream)"
+            install-cmd: "python -m pip install git+https://github.com/llamastack/llama-stack.git@main && python -m pip install -e \".[test]\""
+      fail-fast: false
 
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
 
-      - name: Set up Python
+      - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v4
         with:
-          python-version: '3.12'
+          python-version: ${{ matrix.python-version }}
 
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install -e ".[test]"
+          ${{ matrix.install-cmd }}
 
       - name: Run tests
         run: |
-          python -m pytest tests/unit/ -v
+          python -m pytest tests/unit/ -v
+
+      - name: Display test environment info
+        run: |
+          echo "Testing with: ${{ matrix.test-name }}"
+          python -c "import llama_stack; print(f'llama-stack location: {llama_stack.__file__}')"
+          python -m pip show llama-stack