From 5e7aaf2c41ce8839961f1326132d7f6da69f67de Mon Sep 17 00:00:00 2001 From: Stephen Hu Date: Mon, 28 Jul 2025 17:32:12 +0800 Subject: [PATCH] Fix:When deleting a knowledge base that is currently performing a parsing task, the parsing queue will not be deleted! (#9018) ### What problem does this PR solve? https://github.com/infiniflow/ragflow/issues/8995 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --------- Co-authored-by: Kevin Hu --- api/apps/sdk/dataset.py | 4 +--- api/db/services/document_service.py | 2 ++ 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/apps/sdk/dataset.py b/api/apps/sdk/dataset.py index cfaca4674..7b25f1d8b 100644 --- a/api/apps/sdk/dataset.py +++ b/api/apps/sdk/dataset.py @@ -18,10 +18,8 @@ import logging import os import json - from flask import request from peewee import OperationalError - from api import settings from api.db import FileSource, StatusEnum from api.db.db_models import File @@ -526,4 +524,4 @@ def delete_knowledge_graph(tenant_id,dataset_id): _, kb = KnowledgebaseService.get_by_id(dataset_id) settings.docStoreConn.delete({"knowledge_graph_kwd": ["graph", "subgraph", "entity", "relation"]}, search.index_name(kb.tenant_id), dataset_id) - return get_result(data=True) \ No newline at end of file + return get_result(data=True) diff --git a/api/db/services/document_service.py b/api/db/services/document_service.py index 6d8e4b691..ec8e5f64a 100644 --- a/api/db/services/document_service.py +++ b/api/db/services/document_service.py @@ -229,8 +229,10 @@ class DocumentService(CommonService): @classmethod @DB.connection_context() def remove_document(cls, doc, tenant_id): + from api.db.services.task_service import TaskService cls.clear_chunk_num(doc.id) try: + TaskService.filter_delete(Task.doc_id == doc.id) page = 0 page_size = 1000 all_chunk_ids = []