remove core properties from attribute list (#277)

This commit is contained in:
Preston Rasmussen 2025-02-25 15:55:06 -05:00 committed by GitHub
parent 52046ffe9f
commit db0df06975
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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(