gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Rename method for clarity and improve code readability

Browse source 
- Rename _process_entity_relation_graph to _process_extract_entities
This commit is contained in:
yangdx 2025-09-04 11:48:31 +08:00
parent 9b516a8a53
commit de972f6222
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lightrag/lightrag.py
View file

@ -993,7 +993,7 @@ class LightRAG:
tasks = [ tasks = [
self.chunks_vdb.upsert(inserting_chunks), self.chunks_vdb.upsert(inserting_chunks),
self._process_entity_relation_graph(inserting_chunks), self._process_extract_entities(inserting_chunks),
self.full_docs.upsert(new_docs), self.full_docs.upsert(new_docs),
self.text_chunks.upsert(inserting_chunks), self.text_chunks.upsert(inserting_chunks),
] ]
@ -1582,7 +1582,7 @@ class LightRAG:
# Stage 2: Process entity relation graph (after text_chunks are saved) # Stage 2: Process entity relation graph (after text_chunks are saved)
entity_relation_task = asyncio.create_task( entity_relation_task = asyncio.create_task(
self._process_entity_relation_graph( self._process_extract_entities(
chunks, pipeline_status, pipeline_status_lock chunks, pipeline_status, pipeline_status_lock
) )
) )
@ -1792,7 +1792,7 @@ class LightRAG:
pipeline_status["latest_message"] = log_message pipeline_status["latest_message"] = log_message
pipeline_status["history_messages"].append(log_message) pipeline_status["history_messages"].append(log_message)
async def _process_entity_relation_graph( async def _process_extract_entities(
self, chunk: dict[str, Any], pipeline_status=None, pipeline_status_lock=None self, chunk: dict[str, Any], pipeline_status=None, pipeline_status_lock=None
) -> list: ) -> list:
try: try: