gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Convert gen_mindmap to an asynchronous function

Browse source
This commit is contained in:
N0bodycan 2025-12-10 18:50:09 +08:00 • committed by GitHub
parent ab4b62031f
commit d1e7029964
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
api/db/services/dialog_service.py
View file

@ -887,7 +887,7 @@ async def async_ask(question, kb_ids, tenant_id, chat_llm_name=None, search_conf
yield decorate_answer(answer) 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", {}) meta_data_filter = search_config.get("meta_data_filter", {})
doc_ids = search_config.get("doc_ids", []) doc_ids = search_config.get("doc_ids", [])
rerank_id = search_config.get("rerank_id", "") 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), rank_feature=label_question(question, kbs),
) )
mindmap = MindMapExtractor(chat_mdl) 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 return mind_map.output