fix

api/controllers/console/datasets/rag_pipeline/datasource_auth.py

@@ -1,10 +1,7 @@
 from fastapi.encoders import jsonable_encoder
 from flask import make_response, redirect, request
-from flask_login import current_user  # type: ignore
-from flask_restful import (  # type: ignore
-    Resource,  # type: ignore
-    reqparse,
-)
+from flask_login import current_user
+from flask_restx import Resource, reqparse
 from werkzeug.exceptions import Forbidden, NotFound
 
 from configs import dify_config

api/controllers/console/datasets/rag_pipeline/datasource_content_preview.py

@@ -1,4 +1,4 @@
-from flask_restful import (  # type: ignore
+from flask_restx import (  # type: ignore
     Resource,  # type: ignore
     reqparse,
 )

api/controllers/console/datasets/rag_pipeline/rag_pipeline.py

@@ -1,7 +1,7 @@
 import logging
 
 from flask import request
-from flask_restful import Resource, reqparse
+from flask_restx import Resource, reqparse
 from sqlalchemy.orm import Session
 
 from controllers.console import api

api/controllers/console/datasets/rag_pipeline/rag_pipeline_datasets.py

@@ -1,5 +1,5 @@
 from flask_login import current_user  # type: ignore  # type: ignore
-from flask_restful import Resource, marshal, reqparse  # type: ignore
+from flask_restx import Resource, marshal, reqparse  # type: ignore
 from werkzeug.exceptions import Forbidden
 
 import services

api/controllers/console/datasets/rag_pipeline/rag_pipeline_draft_variable.py

@@ -2,7 +2,7 @@ import logging
 from typing import Any, NoReturn
 
 from flask import Response
-from flask_restful import Resource, fields, inputs, marshal, marshal_with, reqparse
+from flask_restx import Resource, fields, inputs, marshal, marshal_with, reqparse
 from sqlalchemy.orm import Session
 from werkzeug.exceptions import Forbidden
 

api/controllers/console/datasets/rag_pipeline/rag_pipeline_import.py

@@ -1,7 +1,7 @@
 from typing import cast
 
 from flask_login import current_user  # type: ignore
-from flask_restful import Resource, marshal_with, reqparse  # type: ignore
+from flask_restx import Resource, marshal_with, reqparse  # type: ignore
 from sqlalchemy.orm import Session
 from werkzeug.exceptions import Forbidden
 

api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py

@@ -3,8 +3,8 @@ import logging
 from typing import cast
 
 from flask import abort, request
-from flask_restful import Resource, inputs, marshal_with, reqparse  # type: ignore  # type: ignore
-from flask_restful.inputs import int_range  # type: ignore
+from flask_restx import Resource, inputs, marshal_with, reqparse  # type: ignore  # type: ignore
+from flask_restx.inputs import int_range  # type: ignore
 from sqlalchemy.orm import Session
 from werkzeug.exceptions import Forbidden, InternalServerError, NotFound