diff --git a/agentex/pyproject.toml b/agentex/pyproject.toml
index 40858d3e..a82d25d7 100644
--- a/agentex/pyproject.toml
+++ b/agentex/pyproject.toml
@@ -38,7 +38,7 @@ dev = [
     "pre-commit>=4.2.0",
     "pytest>=8.3.3,<9",
     "pytest-asyncio>=1.0.0,<2",
-    "agentex-sdk",
+    "agentex-sdk[adk]",
     "pytest-cov>=5.0.0,<6",
     "vulture>=2.14",
     "ruff>=0.3.4",
@@ -58,11 +58,11 @@ docs = [
     "mkdocs-macros-plugin>=1.3.7,<2",
     "mkdocstrings-python>=1.16.12",
     "mkdocs>=1.6.1",
-    "agentex-sdk",
+    "agentex-sdk[adk]",
     "griffe-pydantic>=1.1.4",
     "pymdown-extensions>=10.0,<11",
     "Pygments>=2.19.2,<2.20",
-    "agentex-sdk",
+    "agentex-sdk[adk]",
 ]
 
 [tool.hatch.build.targets.sdist]
diff --git a/pyproject.toml b/pyproject.toml
index a8696390..92a707c7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -6,7 +6,7 @@ authors = [{ name = "Felix Su", email = "felix.su@scale.com" }]
 readme = "README.md"
 requires-python = ">=3.12,<3.13"
 dependencies = [
-    "agentex-sdk>=0.9.4",
+    "agentex-sdk[adk]>=0.9.4",
     "temporalio>=1.18.0",
 ]