Add coverage dependency to 'unit' group in pyproject.toml file

.github/workflows/unit-tests.yml

@@ -51,7 +51,7 @@ jobs:
 
       - name: Check coverage
         run: |
-          coverage=$(uv run --python $PYTHON_VERSION coverage report -m | tail -1 | tail -c 4 | head -c 2)
+          coverage=$(uv run --python ${{ matrix.python }} --group unit coverage report -m | tail -1 | tail -c 4 | head -c 2)
           echo "Total coverage: ${coverage}%"
           
           # Check if coverage is below the threshold
@@ -59,5 +59,5 @@ jobs:
               echo "Coverage failed at ${coverage}%, expected at least 40%"
               exit 1
           else
-              echo "Coverage passed, ${coverage}%"
+              echo "Coverage check passed with ${coverage}%"
           fi

pyproject.toml

@@ -90,6 +90,7 @@ unit = [
     "pymilvus>=2.5.12",
     "litellm",
     "together",
+    "coverage",
 ]
 # These are the core dependencies required for running integration tests. They are shared across all
 # providers. If a provider requires additional dependencies, please add them to your environment

scripts/unit-tests.sh

@@ -16,9 +16,6 @@ if [ $FOUND_PYTHON -ne 0 ]; then
      uv python install "$PYTHON_VERSION"
 fi
 
-# Install coverage if not installed
-uv pip install coverage >/dev/null 2>&1
-
 uv run --python "$PYTHON_VERSION" --with-editable . --group unit \
     coverage run --source=llama_stack -m pytest -s -v tests/unit/ "$@" && \
 uv run --python "$PYTHON_VERSION" coverage html -d htmlcov-$PYTHON_VERSION