yangdx
commited on
Commit
·
7e48475
1
Parent(s):
c3a363a
Fix linting
Browse files- lightrag/lightrag.py +9 -3
- 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(
|
|
|
|
|
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(
|
|
|
|
|
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(
|
|
|
|
|
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
|
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"]
|