From d1e70299641e5b0614ad7894dc10e2092a75800b Mon Sep 17 00:00:00 2001
From: N0bodycan <49983270+N0bodycan@users.noreply.github.com>
Date: Wed, 10 Dec 2025 18:50:09 +0800
Subject: [PATCH] Convert gen_mindmap to an asynchronous function

---
 api/db/services/dialog_service.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/db/services/dialog_service.py b/api/db/services/dialog_service.py
index cd6a9a4ba..cfa387930 100644
--- a/api/db/services/dialog_service.py
+++ b/api/db/services/dialog_service.py
@@ -887,7 +887,7 @@ async def async_ask(question, kb_ids, tenant_id, chat_llm_name=None, search_conf
     yield decorate_answer(answer)
 
 
-def gen_mindmap(question, kb_ids, tenant_id, search_config={}):
+async def gen_mindmap(question, kb_ids, tenant_id, search_config={}):
     meta_data_filter = search_config.get("meta_data_filter", {})
     doc_ids = search_config.get("doc_ids", [])
     rerank_id = search_config.get("rerank_id", "")
@@ -931,5 +931,5 @@ def gen_mindmap(question, kb_ids, tenant_id, search_config={}):
         rank_feature=label_question(question, kbs),
     )
     mindmap = MindMapExtractor(chat_mdl)
-    mind_map = asyncio.run(mindmap([c["content_with_weight"] for c in ranks["chunks"]]))
+    mind_map = await mindmap([c["content_with_weight"] for c in ranks["chunks"]])
     return mind_map.output