File tree Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Original file line number Diff line number Diff line change @@ -514,7 +514,7 @@ def get_episodic_node_from_record(record: Any) -> EpisodicNode:
514
514
515
515
516
516
def get_entity_node_from_record (record : Any ) -> EntityNode :
517
- return EntityNode (
517
+ entity_node = EntityNode (
518
518
uuid = record ['uuid' ],
519
519
name = record ['name' ],
520
520
group_id = record ['group_id' ],
@@ -525,6 +525,15 @@ def get_entity_node_from_record(record: Any) -> EntityNode:
525
525
attributes = record ['attributes' ],
526
526
)
527
527
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
+
528
537
529
538
def get_community_node_from_record (record : Any ) -> CommunityNode :
530
539
return CommunityNode (
You can’t perform that action at this time.
0 commit comments