From 4ffdf68a2099921ef79f44d38b52a2d18362aa76 Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Fri, 6 Jun 2025 16:03:35 +0800 Subject: [PATCH] r2 --- .../rag_pipeline/rag_pipeline_dsl_service.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/api/services/rag_pipeline/rag_pipeline_dsl_service.py b/api/services/rag_pipeline/rag_pipeline_dsl_service.py index 2e1ed57908..8787490555 100644 --- a/api/services/rag_pipeline/rag_pipeline_dsl_service.py +++ b/api/services/rag_pipeline/rag_pipeline_dsl_service.py @@ -197,8 +197,8 @@ class RagPipelineDslService: # Validate and fix DSL version if not data.get("version"): data["version"] = "0.1.0" - if not data.get("kind") or data.get("kind") != "rag-pipeline": - data["kind"] = "rag-pipeline" + if not data.get("kind") or data.get("kind") != "rag_pipeline": + data["kind"] = "rag_pipeline" imported_version = data.get("version", "0.1.0") # check if imported_version is a float-like string @@ -277,8 +277,7 @@ class RagPipelineDslService: dataset_id = None for node in nodes: if node.get("data", {}).get("type") == "knowledge_index": - knowledge_configuration = node.get("data", {}).get("knowledge_configuration", {}) - knowledge_configuration = KnowledgeConfiguration(**knowledge_configuration) + knowledge_configuration = KnowledgeConfiguration(**node.get("data", {})) if ( dataset and pipeline.is_published @@ -412,8 +411,7 @@ class RagPipelineDslService: dataset_id = None for node in nodes: if node.get("data", {}).get("type") == "knowledge_index": - knowledge_configuration = node.get("data", {}).get("knowledge_configuration", {}) - knowledge_configuration = KnowledgeConfiguration(**knowledge_configuration) + knowledge_configuration = KnowledgeConfiguration(**node.get("data", {})) if not dataset: dataset = Dataset( tenant_id=account.current_tenant_id, @@ -644,7 +642,7 @@ class RagPipelineDslService: export_data = { "version": CURRENT_DSL_VERSION, "kind": "rag_pipeline", - "pipeline": { + "rag_pipeline": { "name": pipeline.name, "icon": icon_info.get("icon", "📙") if icon_info else "📙", "icon_type": icon_info.get("icon_type", "emoji") if icon_info else "emoji",