remove core properties from attribute list (#277)
This commit is contained in:
parent
52046ffe9f
commit
db0df06975
1 changed files with 10 additions and 1 deletions
|
|
@ -514,7 +514,7 @@ def get_episodic_node_from_record(record: Any) -> EpisodicNode:
|
||||||
|
|
||||||
|
|
||||||
def get_entity_node_from_record(record: Any) -> EntityNode:
|
def get_entity_node_from_record(record: Any) -> EntityNode:
|
||||||
return EntityNode(
|
entity_node = EntityNode(
|
||||||
uuid=record['uuid'],
|
uuid=record['uuid'],
|
||||||
name=record['name'],
|
name=record['name'],
|
||||||
group_id=record['group_id'],
|
group_id=record['group_id'],
|
||||||
|
|
@ -525,6 +525,15 @@ def get_entity_node_from_record(record: Any) -> EntityNode:
|
||||||
attributes=record['attributes'],
|
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:
|
def get_community_node_from_record(record: Any) -> CommunityNode:
|
||||||
return CommunityNode(
|
return CommunityNode(
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue