From 14dc3e86427ecac1f6bd67c9d09c5b8e54d2753a Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Wed, 11 Jun 2025 18:03:21 +0800
Subject: [PATCH] r2

---
 api/services/rag_pipeline/rag_pipeline.py             | 6 +++---
 api/services/rag_pipeline/rag_pipeline_dsl_service.py | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/api/services/rag_pipeline/rag_pipeline.py b/api/services/rag_pipeline/rag_pipeline.py
index 2ed44c4a36..df9fea805c 100644
--- a/api/services/rag_pipeline/rag_pipeline.py
+++ b/api/services/rag_pipeline/rag_pipeline.py
@@ -50,7 +50,7 @@ from models.workflow import (
     Workflow,
     WorkflowNodeExecutionTriggeredFrom,
     WorkflowRun,
-    WorkflowType,
+    WorkflowType, WorkflowNodeExecutionModel,
 )
 from services.dataset_service import DatasetService
 from services.datasource_provider_service import DatasourceProviderService
@@ -911,7 +911,7 @@ class RagPipelineService:
         pipeline: Pipeline,
         run_id: str,
         user: Account | EndUser,
-    ) -> list[WorkflowNodeExecution]:
+    ) -> list[WorkflowNodeExecutionModel]:
         """
         Get workflow run node execution list
         """
@@ -930,7 +930,7 @@ class RagPipelineService:
 
         # Use the repository to get the node executions with ordering
         order_config = OrderConfig(order_by=["index"], order_direction="desc")
-        node_executions = repository.get_by_workflow_run(
+        node_executions = repository.get_db_models_by_workflow_run(
             workflow_run_id=run_id,
             order_config=order_config,
             triggered_from=WorkflowNodeExecutionTriggeredFrom.RAG_PIPELINE_RUN,
diff --git a/api/services/rag_pipeline/rag_pipeline_dsl_service.py b/api/services/rag_pipeline/rag_pipeline_dsl_service.py
index 8787490555..fb311482d8 100644
--- a/api/services/rag_pipeline/rag_pipeline_dsl_service.py
+++ b/api/services/rag_pipeline/rag_pipeline_dsl_service.py
@@ -276,7 +276,7 @@ class RagPipelineDslService:
             nodes = graph.get("nodes", [])
             dataset_id = None
             for node in nodes:
-                if node.get("data", {}).get("type") == "knowledge_index":
+                if node.get("data", {}).get("type") == "knowledge-index":
                     knowledge_configuration = KnowledgeConfiguration(**node.get("data", {}))
                     if (
                         dataset