Skip to content

Commit db0df06

Browse files
authored
remove core properties from attribute list (#277)
1 parent 52046ff commit db0df06

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

graphiti_core/nodes.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -514,7 +514,7 @@ def get_episodic_node_from_record(record: Any) -> EpisodicNode:
514514

515515

516516
def get_entity_node_from_record(record: Any) -> EntityNode:
517-
return EntityNode(
517+
entity_node = EntityNode(
518518
uuid=record['uuid'],
519519
name=record['name'],
520520
group_id=record['group_id'],
@@ -525,6 +525,15 @@ def get_entity_node_from_record(record: Any) -> EntityNode:
525525
attributes=record['attributes'],
526526
)
527527

528+
del entity_node.attributes['uuid']
529+
del entity_node.attributes['name']
530+
del entity_node.attributes['group_id']
531+
del entity_node.attributes['name_embedding']
532+
del entity_node.attributes['summary']
533+
del entity_node.attributes['created_at']
534+
535+
return entity_node
536+
528537

529538
def get_community_node_from_record(record: Any) -> CommunityNode:
530539
return CommunityNode(

0 commit comments

Comments
 (0)