From 8cbfaa2c03dfc891cd7effae6eca606bbcfc6fbc Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Tue, 16 Sep 2025 19:13:22 +0800 Subject: [PATCH] dev/reformat --- .../service_api/dataset/rag_pipeline/rag_pipeline_workflow.py | 2 +- api/services/rag_pipeline/pipeline_generate_service.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/controllers/service_api/dataset/rag_pipeline/rag_pipeline_workflow.py b/api/controllers/service_api/dataset/rag_pipeline/rag_pipeline_workflow.py index c386e8f41e..55bfdde009 100644 --- a/api/controllers/service_api/dataset/rag_pipeline/rag_pipeline_workflow.py +++ b/api/controllers/service_api/dataset/rag_pipeline/rag_pipeline_workflow.py @@ -67,7 +67,7 @@ class DatasourceNodeRunApi(DatasetApiResource): """Resource for datasource node run.""" @service_api_ns.doc(shortcut="pipeline_datasource_node_run") - @service_api_ns.doc(description="Run a atasource node for a rag pipeline") + @service_api_ns.doc(description="Run a datasource node for a rag pipeline") @service_api_ns.doc( path={ "dataset_id": "Dataset ID", diff --git a/api/services/rag_pipeline/pipeline_generate_service.py b/api/services/rag_pipeline/pipeline_generate_service.py index 51e03d11c7..563174c528 100644 --- a/api/services/rag_pipeline/pipeline_generate_service.py +++ b/api/services/rag_pipeline/pipeline_generate_service.py @@ -32,9 +32,9 @@ class PipelineGenerateService: """ try: workflow = cls._get_workflow(pipeline, invoke_from) - if args.get("original_document_id"): + if original_document_id := args.get("original_document_id"): # update document status to waiting - cls.update_document_status(args.get("original_document_id", "")) + cls.update_document_status(original_document_id) return PipelineGenerator.convert_to_event_stream( PipelineGenerator().generate( pipeline=pipeline,