diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index aed8b7fa4..a8061a817 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -273,7 +273,7 @@ def embedding(docs, mdl, parser_config=None, callback=None): batch_size = 16 tts, cnts = [], [] for d in docs: - tts.append(rmSpace(d["title_tks"])) + tts.append(rmSpace(d.get("docnm_kwd", "Title"))) c = "\n".join(d.get("question_kwd", [])) if not c: c = d["content_with_weight"]