From 7734b81911d79e7f665cca7f19de18ed433fc5ef Mon Sep 17 00:00:00 2001 From: AdityaKalraShorthillsAI <143726711+AdityaKalraShorthillsAI@users.noreply.github.com> Date: Mon, 2 Dec 2024 02:44:47 +0530 Subject: [PATCH] Added embedding_func argument in Neo4JStorage class's __init__ method in neo4j implementation --- lightrag/kg/neo4j_impl.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index 84efae81..5db3016f 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -27,8 +27,12 @@ class Neo4JStorage(BaseGraphStorage): def load_nx_graph(file_name): print("no preloading of graph with neo4j in production") - def __init__(self, namespace, global_config): - super().__init__(namespace=namespace, global_config=global_config) + def __init__(self, namespace, global_config, embedding_func): + super().__init__( + namespace=namespace, + global_config=global_config, + embedding_func=embedding_func, + ) self._driver = None self._driver_lock = asyncio.Lock() URI = os.environ["NEO4J_URI"]