Merge remote-tracking branch 'origin/feature/knowledge_lxc' into develop

This commit is contained in:
lixiangcheng1
2026-02-25 19:19:36 +08:00

View File

@@ -1,4 +1,5 @@
import asyncio
from concurrent.futures import ThreadPoolExecutor
import json
import os
import re
@@ -369,7 +370,7 @@ def build_graphrag_for_kb(kb_id: uuid.UUID):
print(f"{datetime.now().strftime('%H:%M:%S')} GraphRAG task result for task {task}:\n{result}\n")
return result
try:
def sync_task():
trio.run(
lambda: _run(
row=task,
@@ -384,6 +385,10 @@ def build_graphrag_for_kb(kb_id: uuid.UUID):
with_community=with_community,
)
)
try:
with ThreadPoolExecutor(max_workers=1) as executor:
future = executor.submit(sync_task)
future.result() # Blocks until the task completes
except Exception as e:
print(f"{datetime.now().strftime('%H:%M:%S')} GraphRAG task failed for task {task}:\n{str(e)}\n")
finally: