Update graph doc for installation (#1959)

2024-10-15 20:43:41 +05:30
parent bd5ce7c6d2
commit 2b262a65b2
2 changed files with 19 additions and 2 deletions
--- a/mem0/memory/graph_memory.py
+++ b/mem0/memory/graph_memory.py
@@ -1,7 +1,14 @@
 import logging

-from langchain_community.graphs import Neo4jGraph
-from rank_bm25 import BM25Okapi
+try:
+    from langchain_community.graphs import Neo4jGraph
+except ImportError:
+    raise ImportError("langchain_community is not installed. Please install it using pip install langchain-community")
+
+try:
+    from rank_bm25 import BM25Okapi
+except ImportError:
+    raise ImportError("rank_bm25 is not installed. Please install it using pip install rank-bm25")

 from mem0.graphs.tools import (
    ADD_MEMORY_STRUCT_TOOL_GRAPH,