diff --git a/api/services/rag_pipeline/pipeline_template/customized/customized_retrieval.py b/api/services/rag_pipeline/pipeline_template/customized/customized_retrieval.py index 5d0130bb8e..bb4db81a70 100644 --- a/api/services/rag_pipeline/pipeline_template/customized/customized_retrieval.py +++ b/api/services/rag_pipeline/pipeline_template/customized/customized_retrieval.py @@ -73,6 +73,6 @@ class CustomizedPipelineTemplateRetrieval(PipelineTemplateRetrievalBase): "icon_info": pipeline_template.icon, "description": pipeline_template.description, "chunk_structure": pipeline_template.chunk_structure, - "export_data": yaml.safe_load(pipeline_template.yaml_content), + "export_data": pipeline_template.yaml_content, "created_by": pipeline_template.created_user_name, } diff --git a/api/services/rag_pipeline/pipeline_template/database/database_retrieval.py b/api/services/rag_pipeline/pipeline_template/database/database_retrieval.py index 91bc9f6cdd..c296111f46 100644 --- a/api/services/rag_pipeline/pipeline_template/database/database_retrieval.py +++ b/api/services/rag_pipeline/pipeline_template/database/database_retrieval.py @@ -72,7 +72,7 @@ class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase): "name": pipeline_template.name, "icon_info": pipeline_template.icon, "chunk_structure": pipeline_template.chunk_structure, - "export_data": yaml.safe_load(pipeline_template.yaml_content), + "export_data": pipeline_template.yaml_content, "created_by": pipeline_template.created_user_name, "description": pipeline_template.description, } diff --git a/api/services/rag_pipeline/rag_pipeline.py b/api/services/rag_pipeline/rag_pipeline.py index f2ffce99b3..429c53d829 100644 --- a/api/services/rag_pipeline/rag_pipeline.py +++ b/api/services/rag_pipeline/rag_pipeline.py @@ -474,9 +474,9 @@ class RagPipelineService: provider_type=datasource_runtime.datasource_provider_type(), ) return { - "result": [result.model_dump() for result in website_crawl_result.result], - "job_id": website_crawl_result.job_id, - "status": website_crawl_result.status, + "result": [result for result in website_crawl_result.result], + "job_id": website_crawl_result.result.job_id, + "status": website_crawl_result.result.status, "provider_type": datasource_node_data.get("provider_type"), } case _: