From 6daf23a22b342d151089bbb3739fca1009cb8855 Mon Sep 17 00:00:00 2001
From: ChristianZaccaria <christian.zaccaria.cz@gmail.com>
Date: Mon, 7 Jul 2025 13:52:10 +0100
Subject: [PATCH] Exclude non-essential files from coverage report

---
 .coveragerc           | 5 +++++
 scripts/unit-tests.sh | 5 ++++-
 uv.lock               | 2 ++
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/.coveragerc b/.coveragerc
index e16c2e461..e9bb3b68b 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -4,3 +4,8 @@ omit =
     */llama_stack/providers/*
     */llama_stack/templates/*
     .venv/*
+    */llama_stack/cli/scripts/*
+    */llama_stack/ui/*
+    */llama_stack/strong_typing/*
+    */llama_stack/env.py
+    */__init__.py
diff --git a/scripts/unit-tests.sh b/scripts/unit-tests.sh
index 5f3071172..458cd383d 100755
--- a/scripts/unit-tests.sh
+++ b/scripts/unit-tests.sh
@@ -16,6 +16,9 @@ if [ $FOUND_PYTHON -ne 0 ]; then
      uv python install "$PYTHON_VERSION"
 fi
 
+# Run unit tests with coverage
 uv run --python "$PYTHON_VERSION" --with-editable . --group unit \
-    coverage run --source=llama_stack -m pytest -s -v tests/unit/ "$@" && \
+    coverage run --source=llama_stack -m pytest -s -v tests/unit/ "$@"
+
+# Generate HTML coverage report
 uv run --python "$PYTHON_VERSION" coverage html -d htmlcov-$PYTHON_VERSION
diff --git a/uv.lock b/uv.lock
index 666cdf21f..20d34ee8d 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1372,6 +1372,7 @@ unit = [
     { name = "aiosqlite" },
     { name = "blobfile" },
     { name = "chardet" },
+    { name = "coverage" },
     { name = "faiss-cpu" },
     { name = "litellm" },
     { name = "mcp" },
@@ -1480,6 +1481,7 @@ unit = [
     { name = "aiosqlite" },
     { name = "blobfile" },
     { name = "chardet" },
+    { name = "coverage" },
     { name = "faiss-cpu" },
     { name = "litellm" },
     { name = "mcp" },