AdityaKalraShorthillsAI commited on
Commit
4bd3001
·
1 Parent(s): 65da904

Added embedding_func argument in Neo4JStorage class's __init__ method in neo4j implementation

Browse files
Files changed (1) hide show
  1. lightrag/kg/neo4j_impl.py +6 -2
lightrag/kg/neo4j_impl.py CHANGED
@@ -27,8 +27,12 @@ class Neo4JStorage(BaseGraphStorage):
27
  def load_nx_graph(file_name):
28
  print("no preloading of graph with neo4j in production")
29
 
30
- def __init__(self, namespace, global_config):
31
- super().__init__(namespace=namespace, global_config=global_config)
 
 
 
 
32
  self._driver = None
33
  self._driver_lock = asyncio.Lock()
34
  URI = os.environ["NEO4J_URI"]
 
27
  def load_nx_graph(file_name):
28
  print("no preloading of graph with neo4j in production")
29
 
30
+ def __init__(self, namespace, global_config, embedding_func):
31
+ super().__init__(
32
+ namespace=namespace,
33
+ global_config=global_config,
34
+ embedding_func=embedding_func,
35
+ )
36
  self._driver = None
37
  self._driver_lock = asyncio.Lock()
38
  URI = os.environ["NEO4J_URI"]