From db0df0697559c7cc9f3babc43e1f5bb4445b9e99 Mon Sep 17 00:00:00 2001 From: Preston Rasmussen <109292228+prasmussen15@users.noreply.github.com> Date: Tue, 25 Feb 2025 15:55:06 -0500 Subject: [PATCH] remove core properties from attribute list (#277) --- graphiti_core/nodes.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/graphiti_core/nodes.py b/graphiti_core/nodes.py index 508a0d4a..3341f857 100644 --- a/graphiti_core/nodes.py +++ b/graphiti_core/nodes.py @@ -514,7 +514,7 @@ def get_episodic_node_from_record(record: Any) -> EpisodicNode: def get_entity_node_from_record(record: Any) -> EntityNode: - return EntityNode( + entity_node = EntityNode( uuid=record['uuid'], name=record['name'], group_id=record['group_id'], @@ -525,6 +525,15 @@ def get_entity_node_from_record(record: Any) -> EntityNode: attributes=record['attributes'], ) + del entity_node.attributes['uuid'] + del entity_node.attributes['name'] + del entity_node.attributes['group_id'] + del entity_node.attributes['name_embedding'] + del entity_node.attributes['summary'] + del entity_node.attributes['created_at'] + + return entity_node + def get_community_node_from_record(record: Any) -> CommunityNode: return CommunityNode(