From 6bdf693b85c4c207bb967472738e1f9be3609203 Mon Sep 17 00:00:00 2001 From: Richard <164130786@qq.com> Date: Fri, 15 Nov 2024 13:11:43 +0800 Subject: [PATCH] fix neo4j bug --- lightrag/kg/neo4j_impl.py | 1 + lightrag/lightrag.py | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index e6b33a9b..32bfbe2e 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/lightrag/kg/neo4j_impl.py @@ -214,6 +214,7 @@ class Neo4JStorage(BaseGraphStorage): neo4jExceptions.ServiceUnavailable, neo4jExceptions.TransientError, neo4jExceptions.WriteServiceUnavailable, + neo4jExceptions.ClientError ) ), ) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 67337098..ce27e76d 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -174,8 +174,7 @@ class LightRAG: ) self.chunk_entity_relation_graph = self.graph_storage_cls( namespace="chunk_entity_relation", - global_config=asdict(self), - embedding_func=self.embedding_func, + global_config=asdict(self) ) #### # add embedding func by walter over