Kevin Hu commited on
Commit
80378de
·
1 Parent(s): 281a904

Fix field missing issue. (#3931)

Browse files

### What problem does this PR solve?

#3905
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. rag/svr/task_executor.py +1 -1
rag/svr/task_executor.py CHANGED
@@ -273,7 +273,7 @@ def embedding(docs, mdl, parser_config=None, callback=None):
273
  batch_size = 16
274
  tts, cnts = [], []
275
  for d in docs:
276
- tts.append(rmSpace(d["title_tks"]))
277
  c = "\n".join(d.get("question_kwd", []))
278
  if not c:
279
  c = d["content_with_weight"]
 
273
  batch_size = 16
274
  tts, cnts = [], []
275
  for d in docs:
276
+ tts.append(rmSpace(d.get("docnm_kwd", "Title")))
277
  c = "\n".join(d.get("question_kwd", []))
278
  if not c:
279
  c = d["content_with_weight"]