diff --git a/setup.py b/setup.py
index 650d785c1..b718ab726 100644
--- a/setup.py
+++ b/setup.py
@@ -66,6 +66,7 @@ setuptools.setup(
         "graph": ["networkx", "matplotlib"],
         "websurfer": ["beautifulsoup4", "markdownify", "pdfminer.six", "pathvalidate"],
         "redis": ["redis"],
+        "cosmosdb": ["azure-cosmos>=4.2.0"],
         "websockets": ["websockets>=12.0,<13"],
         "jupyter-executor": jupyter_executor,
         "types": ["mypy==1.9.0", "pytest>=6.1.1,<8"] + jupyter_executor,