From a5aed2412f0e8aecc50b10563fb783163958595a Mon Sep 17 00:00:00 2001 From: KevinHuSh Date: Tue, 7 May 2024 16:41:56 +0800 Subject: [PATCH] fix bugs (#662) ### What problem does this PR solve? Fix import error for task_service.py ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/apps/document_app.py | 2 -- api/db/services/task_service.py | 5 ++++- 2 files changed, 4 insertions(+), 3 deletions(-) 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,