youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-03-23 15:27:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2 transform

Browse Source
This commit is contained in:
jyong
2025-07-16 01:50:37 +08:00
parent 1ad73ccdc8
commit 2012ea3213
23 changed files with 5867 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/controllers/console/datasets/rag_pipeline/rag_pipeline_workflow.py
View File

@ -46,6 +46,7 @@ from services.errors.llm import InvokeRateLimitError
from services.rag_pipeline.pipeline_generate_service import PipelineGenerateService
from services.rag_pipeline.rag_pipeline import RagPipelineService
from services.rag_pipeline.rag_pipeline_manage_service import RagPipelineManageService
from services.rag_pipeline.rag_pipeline_transform_service import RagPipelineTransformService
logger = logging.getLogger(__name__)
@ -946,6 +947,16 @@ class RagPipelineWorkflowLastRunApi(Resource):
if node_exec is None:
raise NotFound("last run not found")
return node_exec
class RagPipelineTransformApi(Resource):
@setup_required
@login_required
@account_initialization_required
def post(self, dataset_id):
dataset_id = str(dataset_id)
rag_pipeline_transform_service = RagPipelineTransformService()
rag_pipeline_transform_service.transform_dataset(dataset_id)
return {"message": "success"}
api.add_resource(
@ -1056,3 +1067,7 @@ api.add_resource(
RagPipelineWorkflowLastRunApi,
"/rag/pipelines/<uuid:pipeline_id>/workflows/draft/nodes/<string:node_id>/last-run",
)
api.add_resource(
RagPipelineTransformApi,
"/rag/pipelines/transform/datasets/<uuid:dataset_id>",
)