Fix: assistant deleteion issue. (#6906)

### What problem does this PR solve?

#6875

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-04-09 20:29:40 +08:00
committed by GitHub
parent 22758a2763
commit 3bb1e012e6
3 changed files with 19 additions and 12 deletions

View File

@ -26,6 +26,7 @@ from api.utils import get_uuid
from api.utils.api_utils import get_error_data_result, token_required, get_result, check_duplicate_ids
@manager.route('/chats', methods=['POST']) # noqa: F821
@token_required
def create(tenant_id):
@ -265,18 +266,17 @@ def delete(tenant_id):
id_list.append(dia.id)
else:
id_list = ids
unique_id_list, duplicate_messages = check_duplicate_ids(id_list, "chat")
id_list = unique_id_list
for id in id_list:
unique_id_list, duplicate_messages = check_duplicate_ids(id_list, "assistant")
for id in unique_id_list:
if not DialogService.query(tenant_id=tenant_id, id=id, status=StatusEnum.VALID.value):
errors.append(f"You don't own the chat {id}")
errors.append(f"Assistant({id}) not found.")
continue
temp_dict = {"status": StatusEnum.INVALID.value}
DialogService.update_by_id(id, temp_dict)
success_count += 1
if errors:
if success_count > 0:
return get_result(
@ -298,6 +298,7 @@ def delete(tenant_id):
return get_result()
@manager.route('/chats', methods=['GET']) # noqa: F821
@token_required
def list_chat(tenant_id):