From be830741310f6bcff9c13494dcb9586f537f2299 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Mon, 26 May 2025 10:59:01 +0800
Subject: [PATCH] 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)
---
 rag/svr/task_executor.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py
index 4e9eb3d2a..1067488aa 100644
--- a/rag/svr/task_executor.py
+++ b/rag/svr/task_executor.py
@@ -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