diff --git a/.devcontainer/setup.sh b/.devcontainer/setup.sh
index 0b137d3858..59262f75df 100755
--- a/.devcontainer/setup.sh
+++ b/.devcontainer/setup.sh
@@ -1,8 +1,13 @@
 # update pip
 pip install --upgrade pip
 
+# install pre-release of quarto for docs
+wget https://github.com/quarto-dev/quarto-cli/releases/download/v1.7.7/quarto-1.7.7-linux-amd64.deb -O quarto.deb && \
+    sudo dpkg -i quarto.deb && \
+    rm quarto.deb
+
 # install dev packages
-pip install -e ".[test,teachable,lmm,retrievechat,mathchat,blendsearch]"
+pip install -e ".[dev]"
 
 # install pre-commit hook if not installed already
 pre-commit install
diff --git a/pyproject.toml b/pyproject.toml
index 560fe7cb85..35280252ce 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -187,6 +187,19 @@ test = [
     "fastapi>=0.115.0,<1",
 ]
 
+docs = [
+    "pydoc-markdown",
+    "pyyaml",
+    "termcolor",
+    "nbclient",
+]
+
+dev = [
+    "pyautogen[test]",
+    "pyautogen[docs]",
+    "pyautogen[types]",
+]
+
 [project.urls]
 Homepage = "https://ag2.ai/"
 Documentation = "https://docs.ag2.ai/docs/Home"