Fix: restore task limiter. (#7844)

### What problem does this PR solve?

Close #7828

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-05-26 10:59:01 +08:00 committed by GitHub
parent 1f756947da
commit be83074131
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -550,6 +550,7 @@ async def do_handle_task(task):
with_community = graphrag_conf.get("community", False)
await run_graphrag(task, task_language, with_resolution, with_community, chat_model, embedding_model, progress_callback)
progress_callback(prog=1.0, msg="Knowledge Graph done ({:.2f}s)".format(timer() - start_ts))
task_limiter = trio.CapacityLimiter(MAX_CONCURRENT_TASKS)
return
else:
# Standard chunking methods