fix(tests): include shared dependencies in the unit test environment

Install the shared test dependency group alongside unit-only extras so the supported unit test command can collect provider and library client tests in a clean environment. Update the unit dependency list and test docs to match the validated workflow.

Signed-off-by: Sumanth Kamenani <skamenan@redhat.com>

Author: skamenan7

pyproject.toml

@@ -115,12 +115,14 @@ type_checking = [
     "ollama",
     "llama-stack-client==0.4.6",
 ]
-# These are the dependencies required for running unit tests.
+# These are additional dependencies required for running unit tests beyond the
+# shared `test` dependency group.
 unit = [
     "anthropic",
     "databricks-sdk",
     "sqlite-vec",
     "ollama",
+    "llama-stack-client==0.4.6",
     "aiosqlite",
     "aiohttp",
     "psycopg2-binary>=2.9.0",
@@ -129,17 +131,21 @@ unit = [
     "chardet",
     "sqlalchemy",
     "sqlalchemy[asyncio]>=2.0.41",
+    "torch>=2.6.0",
+    "torchvision>=0.21.0",
+    "fairscale",
     "blobfile",
     "faiss-cpu",
     "litellm",
     "together",
     "coverage",
     "moto[s3]>=5.1.10",
 ]
-# 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
-# separately. If you are using "uv" to execute your tests, you can use the "--group" flag to specify extra
-# dependencies.
+# These are the core dependencies shared across test environments. Integration
+# tests use this group directly, and the unit test wrapper installs it alongside
+# the `unit` group. If a provider requires additional dependencies, please add
+# them to your environment separately. If you are using "uv" to execute your
+# tests, you can use the "--group" flag to specify extra dependencies.
 test = [
     "aiosqlite",
     "aiohttp",

scripts/unit-tests.sh

@@ -25,6 +25,7 @@ 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 \
+# Run unit tests with coverage. The shared `test` group provides core provider
+# dependencies, and `unit` layers on unit-only extras.
+uv run --python "$PYTHON_VERSION" --with-editable . --group test --group unit \
     coverage run --source=src/llama_stack -m pytest -s -v tests/unit/ "$@"

tests/README.md

@@ -26,7 +26,7 @@ This gives you:
 
 You can run the unit tests with:
 ```bash
-uv run --group unit pytest -sv tests/unit/
+uv run --group test --group unit pytest -sv tests/unit/
 ```
 
 For running integration tests, you must provide a few things: