diff --git a/api/apps/document_app.py b/api/apps/document_app.py
index 1ae6fef87..2d129ed66 100644
--- a/api/apps/document_app.py
+++ b/api/apps/document_app.py
@@ -38,9 +38,7 @@ from api.db.services.document_service import DocumentService
 from api.settings import RetCode
 from api.utils.api_utils import get_json_result
 from rag.utils.minio_conn import MINIO
-from rag.utils.redis_conn import REDIS_CONN
 from api.utils.file_utils import filename_type, thumbnail
-from rag.settings import SVR_QUEUE_NAME
 
 
 @manager.route('/upload', methods=['POST'])
diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py
index 6d109e2dc..12707768b 100644
--- a/api/db/services/task_service.py
+++ b/api/db/services/task_service.py
@@ -25,13 +25,16 @@ from api.db.services.common_service import CommonService
 from api.db.services.document_service import DocumentService
 from api.utils import current_timestamp, get_uuid
 from deepdoc.parser.excel_parser import RAGFlowExcelParser
-from rag.settings import MINIO, SVR_QUEUE_NAME
+from rag.settings import SVR_QUEUE_NAME
+from rag.utils.minio_conn import MINIO
 from rag.utils.redis_conn import REDIS_CONN
 
 
 class TaskService(CommonService):
     model = Task
 
+    @classmethod
+    @DB.connection_context()
     def get_tasks(cls, task_id):
         fields = [
             cls.model.id,