KevinHuSh commited on
Commit
9487bc0
·
1 Parent(s): ebbe7cb

fix bugs (#662)

Browse files

### 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 CHANGED
@@ -38,9 +38,7 @@ from api.db.services.document_service import DocumentService
38
  from api.settings import RetCode
39
  from api.utils.api_utils import get_json_result
40
  from rag.utils.minio_conn import MINIO
41
- from rag.utils.redis_conn import REDIS_CONN
42
  from api.utils.file_utils import filename_type, thumbnail
43
- from rag.settings import SVR_QUEUE_NAME
44
 
45
 
46
  @manager.route('/upload', methods=['POST'])
 
38
  from api.settings import RetCode
39
  from api.utils.api_utils import get_json_result
40
  from rag.utils.minio_conn import MINIO
 
41
  from api.utils.file_utils import filename_type, thumbnail
 
42
 
43
 
44
  @manager.route('/upload', methods=['POST'])
api/db/services/task_service.py CHANGED
@@ -25,13 +25,16 @@ from api.db.services.common_service import CommonService
25
  from api.db.services.document_service import DocumentService
26
  from api.utils import current_timestamp, get_uuid
27
  from deepdoc.parser.excel_parser import RAGFlowExcelParser
28
- from rag.settings import MINIO, SVR_QUEUE_NAME
 
29
  from rag.utils.redis_conn import REDIS_CONN
30
 
31
 
32
  class TaskService(CommonService):
33
  model = Task
34
 
 
 
35
  def get_tasks(cls, task_id):
36
  fields = [
37
  cls.model.id,
 
25
  from api.db.services.document_service import DocumentService
26
  from api.utils import current_timestamp, get_uuid
27
  from deepdoc.parser.excel_parser import RAGFlowExcelParser
28
+ from rag.settings import SVR_QUEUE_NAME
29
+ from rag.utils.minio_conn import MINIO
30
  from rag.utils.redis_conn import REDIS_CONN
31
 
32
 
33
  class TaskService(CommonService):
34
  model = Task
35
 
36
+ @classmethod
37
+ @DB.connection_context()
38
  def get_tasks(cls, task_id):
39
  fields = [
40
  cls.model.id,