gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
AdityaKalraShorthillsAI
2024-12-02 02:44:47 +05:30
parent f3fe5d3b64
commit 8d5af931d7
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/kg/neo4j_impl.py
View File

@@ -27,8 +27,12 @@ class Neo4JStorage(BaseGraphStorage):
def load_nx_graph(file_name): def load_nx_graph(file_name):
print("no preloading of graph with neo4j in production") print("no preloading of graph with neo4j in production")
def __init__(self, namespace, global_config): def __init__(self, namespace, global_config, embedding_func):
super().__init__(namespace=namespace, global_config=global_config) super().__init__(
namespace=namespace,
global_config=global_config,
embedding_func=embedding_func,
)
self._driver = None self._driver = None
self._driver_lock = asyncio.Lock() self._driver_lock = asyncio.Lock()
URI = os.environ["NEO4J_URI"] URI = os.environ["NEO4J_URI"]