diff --git a/pyproject.toml b/pyproject.toml
index a616526166..82c5f0c40a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -126,6 +126,10 @@ visualization = [
     "rerun-sdk>=0.20.0",
 ]
 
+viewer = [
+    "dimos-viewer>=0.1.0",
+]
+
 agents = [
     "langchain==1.2.3",
     "langchain-chroma>=1,<2",