diff --git a/pyproject.toml b/pyproject.toml
index 60bc8f02..d83b1ed9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mem0ai"
-version = "0.1.36"
+version = "0.1.37"
 description = "Long-term memory for AI Agents"
 authors = ["Mem0 <founders@mem0.ai>"]
 exclude = [
@@ -28,6 +28,9 @@ langchain-community = "^0.3.1"
 neo4j = "^5.23.1"
 rank-bm25 = "^0.2.2"
 
+[tool.poetry.extras]
+graph = ["langchain-community", "neo4j", "rank-bm25"]
+
 [tool.poetry.group.test.dependencies]
 pytest = "^8.2.2"