youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix preview run

Browse Source
This commit is contained in:
jyong
2025-08-28 15:39:22 +08:00
parent 4403a26f37
commit a9f7ee029e
5 changed files with 34 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/workflow/nodes/base/node.py
View File

@ -117,6 +117,12 @@ class Node:
if isinstance(self, ToolNode):
start_event.provider_id = getattr(self.get_base_node_data(), "provider_id", "")
start_event.provider_type = getattr(self.get_base_node_data(), "provider_type", "")
from core.workflow.nodes.datasource.datasource_node import DatasourceNode
if isinstance(self, DatasourceNode):
start_event.provider_id = f"{getattr(self.get_base_node_data(), 'plugin_id', '')}/{getattr(self.get_base_node_data(), 'provider_name', '')}"
start_event.provider_type = getattr(self.get_base_node_data(), "provider_type", "")
from typing import cast

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

@ -314,7 +314,7 @@ class DatasourceNode(Node):
*,
graph_config: Mapping[str, Any],
node_id: str,
node_data: DatasourceNodeData,
node_data: Mapping[str, Any],
) -> Mapping[str, Sequence[str]]:
"""
Extract variable selector to variable mapping
@ -323,10 +323,11 @@ class DatasourceNode(Node):
:param node_data: node data
:return:
"""
typed_node_data = DatasourceNodeData.model_validate(node_data)
result = {}
if node_data.datasource_parameters:
for parameter_name in node_data.datasource_parameters:
input = node_data.datasource_parameters[parameter_name]
if typed_node_data.datasource_parameters:
for parameter_name in typed_node_data.datasource_parameters:
input = typed_node_data.datasource_parameters[parameter_name]
if input.type == "mixed":
assert isinstance(input.value, str)
selectors = VariableTemplateParser(input.value).extract_variable_selectors()
@ -430,7 +431,7 @@ class DatasourceNode(Node):
)
elif message.type == DatasourceMessage.MessageType.JSON:
assert isinstance(message.message, DatasourceMessage.JsonMessage)
if self._node_type == NodeType.AGENT:
if self.node_type == NodeType.AGENT:
msg_metadata = message.message.json_object.pop("execution_metadata", {})
agent_execution_metadata = {
key: value