diff --git a/pyproject.toml b/pyproject.toml
index 0b5c7f6df..8d8137233 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -67,6 +67,7 @@ unit = [
     "aiosqlite",
     "aiohttp",
     "pypdf",
+    "mcp",
     "chardet",
     "qdrant-client",
     "opentelemetry-exporter-otlp-proto-http",
diff --git a/uv.lock b/uv.lock
index f2c9d59c1..f6dae2944 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1544,6 +1544,7 @@ unit = [
     { name = "aiohttp" },
     { name = "aiosqlite" },
     { name = "chardet" },
+    { name = "mcp" },
     { name = "openai" },
     { name = "opentelemetry-exporter-otlp-proto-http" },
     { name = "pypdf" },
@@ -1576,6 +1577,7 @@ requires-dist = [
     { name = "llama-stack-client", specifier = ">=0.2.7" },
     { name = "llama-stack-client", marker = "extra == 'ui'", specifier = ">=0.2.7" },
     { name = "mcp", marker = "extra == 'test'" },
+    { name = "mcp", marker = "extra == 'unit'" },
     { name = "myst-parser", marker = "extra == 'docs'" },
     { name = "nbval", marker = "extra == 'dev'" },
     { name = "openai", specifier = ">=1.66" },