From 85e5e694948d9aafa323733e4e5ea80847e2a212 Mon Sep 17 00:00:00 2001
From: lxobr <122801072+lxobr@users.noreply.github.com>
Date: Tue, 22 Apr 2025 11:49:45 +0200
Subject: [PATCH] feat: connect entities to node sets

---
 cognee/modules/graph/utils/expand_with_nodes_and_edges.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cognee/modules/graph/utils/expand_with_nodes_and_edges.py b/cognee/modules/graph/utils/expand_with_nodes_and_edges.py
index 775930638..e1f61d115 100644
--- a/cognee/modules/graph/utils/expand_with_nodes_and_edges.py
+++ b/cognee/modules/graph/utils/expand_with_nodes_and_edges.py
@@ -144,6 +144,7 @@ def expand_with_nodes_and_edges(
                     is_a=type_node,
                     description=node.description,
                     ontology_valid=ontology_validated_source_ent,
+                    belongs_to_set=data_chunk.belongs_to_set,
                 )
 
                 added_nodes_map[entity_node_key] = entity_node