diff --git a/pyproject.toml b/pyproject.toml
index b8de65781..14b942c38 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -104,6 +104,7 @@ type_checking = [
     "lm-format-enforcer",
     "mcp",
     "ollama",
+    "llama-stack-client>=0.3.0",
 ]
 # These are the dependencies required for running unit tests.
 unit = [
diff --git a/uv.lock b/uv.lock
index db154122a..9d35e9c08 100644
--- a/uv.lock
+++ b/uv.lock
@@ -2043,6 +2043,7 @@ type-checking = [
     { name = "datasets" },
     { name = "fairscale" },
     { name = "faiss-cpu" },
+    { name = "llama-stack-client" },
     { name = "lm-format-enforcer" },
     { name = "mcp" },
     { name = "nest-asyncio" },
@@ -2188,6 +2189,7 @@ type-checking = [
     { name = "datasets" },
     { name = "fairscale" },
     { name = "faiss-cpu" },
+    { name = "llama-stack-client", specifier = ">=0.3.0" },
     { name = "lm-format-enforcer" },
     { name = "mcp" },
     { name = "nest-asyncio" },