Kevin Hu commited on
Commit
4d0a7c7
·
1 Parent(s): b08b226

fix delete message error (#2153)

Browse files

### What problem does this PR solve?



### Type of change

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

api/apps/conversation_app.py CHANGED
@@ -227,7 +227,7 @@ def delete_msg():
227
  assert conv["message"][i+1]["id"] == req["message_id"]
228
  conv["message"].pop(i)
229
  conv["message"].pop(i)
230
- conv["reference"].pop(i)
231
  break
232
 
233
  ConversationService.update_by_id(conv["id"], conv)
 
227
  assert conv["message"][i+1]["id"] == req["message_id"]
228
  conv["message"].pop(i)
229
  conv["message"].pop(i)
230
+ conv["reference"].pop(max(0, i//2-1))
231
  break
232
 
233
  ConversationService.update_by_id(conv["id"], conv)
api/db/services/task_service.py CHANGED
@@ -42,6 +42,7 @@ class TaskService(CommonService):
42
  cls.model.doc_id,
43
  cls.model.from_page,
44
  cls.model.to_page,
 
45
  Document.kb_id,
46
  Document.parser_id,
47
  Document.parser_config,
 
42
  cls.model.doc_id,
43
  cls.model.from_page,
44
  cls.model.to_page,
45
+ cls.model.retry_count,
46
  Document.kb_id,
47
  Document.parser_id,
48
  Document.parser_config,