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