diff --git a/api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py b/api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py index 81e344250d..de55f3a48d 100644 --- a/api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py +++ b/api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py @@ -5,6 +5,7 @@ from uuid import UUID from flask import abort, request from flask_restx import Resource, marshal_with # type: ignore +from flask_restx import reqparse from pydantic import BaseModel, Field from sqlalchemy.orm import Session from werkzeug.exceptions import Forbidden, InternalServerError, NotFound @@ -968,10 +969,6 @@ class RagPipelineDatasourceVariableApi(Resource): ) return workflow_node_execution - -from flask_restx import reqparse - - @console_ns.route("/rag/pipelines/recommended-plugins") class RagPipelineRecommendedPluginApi(Resource): @setup_required