diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index 1c6616a94..a5a88caa5 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -902,6 +902,7 @@ async def do_handle_task(task): task_embedding_id = task["embd_id"] task_language = task["language"] task_llm_id = task["parser_config"].get("llm_id") or task["llm_id"] + task["llm_id"] = task_llm_id task_dataset_id = task["kb_id"] task_doc_id = task["doc_id"] task_document_name = task["name"]