remove core properties from attribute list (#277)

getzep · Feb 25, 2025 · db0df06 · db0df06
1 parent 52046ff
commit db0df06
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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(