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,
|