opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix preview run

This commit is contained in:
jyong 2025-08-28 22:01:49 +08:00
parent 1db04aa729
commit 39080eed10
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/workflow/nodes/datasource/datasource_node.py
View File

@ -21,7 +21,7 @@ from core.variables.segments import ArrayAnySegment
from core.variables.variables import ArrayAnyVariable
from core.workflow.entities.variable_pool import VariablePool, VariableValue
from core.workflow.entities.workflow_node_execution import WorkflowNodeExecutionStatus
from core.workflow.enums import ErrorStrategy, NodeType, SystemVariableKey
from core.workflow.enums import ErrorStrategy, NodeType, SystemVariableKey, NodeExecutionType
from core.workflow.node_events import NodeRunResult, StreamChunkEvent, StreamCompletedEvent
from core.workflow.nodes.base.entities import BaseNodeData, RetryConfig
from core.workflow.nodes.base.node import Node
@ -44,6 +44,7 @@ class DatasourceNode(Node):
_node_data: DatasourceNodeData
node_type = NodeType.DATASOURCE
execution_type = NodeExecutionType.ROOT
def init_node_data(self, data: Mapping[str, Any]) -> None:
self._node_data = DatasourceNodeData.model_validate(data)

3
api/services/rag_pipeline/rag_pipeline.py
View File

@ -41,6 +41,7 @@ from core.workflow.entities.workflow_node_execution import (
)
from core.workflow.enums import ErrorStrategy, NodeType, SystemVariableKey
from core.workflow.errors import WorkflowNodeRunFailedError
from core.workflow.graph_events import NodeRunSucceededEvent, NodeRunFailedEvent
from core.workflow.graph_events.base import GraphNodeEventBase
from core.workflow.node_events.base import NodeRunResult
from core.workflow.node_events.node import StreamCompletedEvent
@ -756,7 +757,7 @@ class RagPipelineService:
node_run_result: NodeRunResult | None = None
for event in generator:
if isinstance(event, StreamCompletedEvent):
if isinstance(event, (NodeRunSucceededEvent, NodeRunFailedEvent)):
node_run_result = event.node_run_result
# sign output files
node_run_result.outputs = WorkflowEntry.handle_special_values(node_run_result.outputs) or {}