removed __Entity__ labels

neo4j-labs · Sep 17, 2024 · dff7535 · dff7535
1 parent 9237847
commit dff7535
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/backend/src/chunkid_entities.py b/backend/src/chunkid_entities.py
@@ -19,10 +19,18 @@ def process_records(records):
                 relationship = element['relationship']
 
                 if start_node['element_id'] not in seen_nodes:
+                    if "labels" in start_node.keys():
+                        labels = set(start_node["labels"])
+                        labels.discard("__Entity__")
+                        start_node["labels"] = list(labels)
                     nodes.append(start_node)
                     seen_nodes.add(start_node['element_id'])
 
                 if end_node['element_id'] not in seen_nodes:
+                    if "labels" in end_node.keys():
+                        labels = set(end_node["labels"])
+                        labels.discard("__Entity__")
+                        end_node["labels"] = list(labels)
                     nodes.append(end_node)
                     seen_nodes.add(end_node['element_id'])
 

diff --git a/backend/src/shared/constants.py b/backend/src/shared/constants.py
@@ -74,7 +74,7 @@
 MATCH (chunk)-[:PART_OF]->(d:Document)
 CALL {WITH chunk
 MATCH (chunk)-[:HAS_ENTITY]->(e) 
-MATCH path=(e)(()-[rels:!HAS_ENTITY&!PART_OF]-()){0,2}(:!Chunk &! Document) 
+MATCH path=(e)(()-[rels:!HAS_ENTITY&!PART_OF]-()){0,2}(:!Chunk &! Document &! `__Community__`) 
 UNWIND rels as r
 RETURN collect(distinct r) as rels
 }