feat: connect entities to node sets

2025-04-22 11:49:45 +02:00 · 2025-04-22 11:49:45 +02:00 · 85e5e69494
commit 85e5e69494
parent 83b20b1e92
1 changed files with 1 additions and 0 deletions
--- 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