diff --git a/graphgen/models/kg_builder/light_rag_kg_builder.py b/graphgen/models/kg_builder/light_rag_kg_builder.py index 53fe1d66..460dcea0 100644 --- a/graphgen/models/kg_builder/light_rag_kg_builder.py +++ b/graphgen/models/kg_builder/light_rag_kg_builder.py @@ -164,18 +164,9 @@ async def merge_edges( set([dp["source_id"] for dp in edge_data] + source_ids) ) - for insert_id in [src_id, tgt_id]: - if not kg_instance.has_node(insert_id): - kg_instance.upsert_node( - insert_id, - node_data={ - "entity_type": "UNKNOWN", - "entity_name": insert_id, - "description": "", - "source_id": source_id, - "length": self.tokenizer.count_tokens(description), - }, - ) + if not kg_instance.has_node(src_id) or not kg_instance.has_node(tgt_id): + logger.warning("Edge (%s, %s) has missing nodes.", src_id, tgt_id) + return {} description = await self._handle_kg_summary( f"({src_id}, {tgt_id})", description