mirror of
https://github.com/getzep/graphiti.git
synced 2025-12-27 07:03:47 +00:00
remove core properties from attribute list (#277)
This commit is contained in:
parent
52046ffe9f
commit
db0df06975
@ -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(
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user