AdityaKalraShorthillsAI
commited on
Commit
·
4bd3001
1
Parent(s):
65da904
Added embedding_func argument in Neo4JStorage class's __init__ method in neo4j implementation
Browse files
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__(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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"]
|