diff --git a/Makefile b/Makefile
index 09185f5d435..a53cde0a026 100644
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,7 @@
 .PHONY: docs build
 
 setup-docs-ci:
-	pip install -e . playwright lazydocs nbformat
+	pip install -e .[docs]
 	playwright install
 
 	cd docs && \
diff --git a/pyproject.toml b/pyproject.toml
index 605ceabb8ed..2b237d75426 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -46,7 +46,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-trace_server = []
+docs = ["playwright", "lazydocs", "nbformat", "nbconvert"]
 anthropic = ["anthropic>=0.18.0"]
 cerebras = ["cerebras-cloud-sdk"]
 cohere = ["cohere>=5.9.1,<5.9.3"]