From fc1ac3a962adf19883e046403d2edf02de749e48 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 29 Aug 2024 14:07:14 +0800 Subject: [PATCH] fix delete message error (#2153) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/conversation_app.py | 2 +- api/db/services/task_service.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/api/apps/conversation_app.py b/api/apps/conversation_app.py index 617a9f4cb..6ff490b77 100644 --- a/api/apps/conversation_app.py +++ b/api/apps/conversation_app.py @@ -227,7 +227,7 @@ def delete_msg(): assert conv["message"][i+1]["id"] == req["message_id"] conv["message"].pop(i) conv["message"].pop(i) - conv["reference"].pop(i) + conv["reference"].pop(max(0, i//2-1)) break ConversationService.update_by_id(conv["id"], conv) diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py index 40d80f7e4..a8dc09380 100644 --- a/api/db/services/task_service.py +++ b/api/db/services/task_service.py @@ -42,6 +42,7 @@ class TaskService(CommonService): cls.model.doc_id, cls.model.from_page, cls.model.to_page, + cls.model.retry_count, Document.kb_id, Document.parser_id, Document.parser_config,