yangdx commited on
Commit
7e48475
·
1 Parent(s): c3a363a

Fix linting

Browse files
Files changed (2) hide show
  1. lightrag/lightrag.py +9 -3
  2. lightrag/operate.py +1 -1
lightrag/lightrag.py CHANGED
@@ -887,7 +887,9 @@ class LightRAG:
887
  self.chunks_vdb.upsert(chunks)
888
  )
889
  entity_relation_task = asyncio.create_task(
890
- self._process_entity_relation_graph(chunks, pipeline_status, pipeline_status_lock)
 
 
891
  )
892
  full_docs_task = asyncio.create_task(
893
  self.full_docs.upsert(
@@ -1002,7 +1004,9 @@ class LightRAG:
1002
  pipeline_status["latest_message"] = log_message
1003
  pipeline_status["history_messages"].append(log_message)
1004
 
1005
- async def _process_entity_relation_graph(self, chunk: dict[str, Any], pipeline_status=None, pipeline_status_lock=None) -> None:
 
 
1006
  try:
1007
  await extract_entities(
1008
  chunk,
@@ -1018,7 +1022,9 @@ class LightRAG:
1018
  logger.error("Failed to extract entities and relationships")
1019
  raise e
1020
 
1021
- async def _insert_done(self, pipeline_status=None, pipeline_status_lock=None) -> None:
 
 
1022
  tasks = [
1023
  cast(StorageNameSpace, storage_inst).index_done_callback()
1024
  for storage_inst in [ # type: ignore
 
887
  self.chunks_vdb.upsert(chunks)
888
  )
889
  entity_relation_task = asyncio.create_task(
890
+ self._process_entity_relation_graph(
891
+ chunks, pipeline_status, pipeline_status_lock
892
+ )
893
  )
894
  full_docs_task = asyncio.create_task(
895
  self.full_docs.upsert(
 
1004
  pipeline_status["latest_message"] = log_message
1005
  pipeline_status["history_messages"].append(log_message)
1006
 
1007
+ async def _process_entity_relation_graph(
1008
+ self, chunk: dict[str, Any], pipeline_status=None, pipeline_status_lock=None
1009
+ ) -> None:
1010
  try:
1011
  await extract_entities(
1012
  chunk,
 
1022
  logger.error("Failed to extract entities and relationships")
1023
  raise e
1024
 
1025
+ async def _insert_done(
1026
+ self, pipeline_status=None, pipeline_status_lock=None
1027
+ ) -> None:
1028
  tasks = [
1029
  cast(StorageNameSpace, storage_inst).index_done_callback()
1030
  for storage_inst in [ # type: ignore
lightrag/operate.py CHANGED
@@ -341,7 +341,7 @@ async def extract_entities(
341
  relationships_vdb: BaseVectorStorage,
342
  global_config: dict[str, str],
343
  pipeline_status: dict = None,
344
- pipeline_status_lock = None,
345
  llm_response_cache: BaseKVStorage | None = None,
346
  ) -> None:
347
  use_llm_func: callable = global_config["llm_model_func"]
 
341
  relationships_vdb: BaseVectorStorage,
342
  global_config: dict[str, str],
343
  pipeline_status: dict = None,
344
+ pipeline_status_lock=None,
345
  llm_response_cache: BaseKVStorage | None = None,
346
  ) -> None:
347
  use_llm_func: callable = global_config["llm_model_func"]