mirror of
https://github.com/langgenius/dify.git
synced 2026-11-13 14:51:39 +08:00
Charles Yao
231e423431
Track backend integration, Ruff cleanup and migration reconciliation in ENG-1271. Retain required copy contracts without implementing frontend interactions.
84 lines
3.6 KiB
Python
84 lines
3.6 KiB
Python
"""Translate a WorkflowGenerator graph into Dify Builder MutationIntents.
|
|
|
|
The generator emits ReactFlow nodes ({id, type, data:{type:<real>,...}}) and
|
|
edges. The node-level "type" is the ReactFlow component key -- "custom" for
|
|
ordinary nodes, but "custom-iteration-start" / "custom-loop-start" for the
|
|
container start markers -- and it is carried through as `flow_type`, because
|
|
a marker persisted as a generic "custom" node has no canvas component and
|
|
crashes the editor (React error #130, ESQ1-288). We emit one create_node per
|
|
node (config = node.data minus the redundant "type" key -- apply_create_node
|
|
re-sets it) followed by one connect per edge, so every connect endpoint
|
|
already exists at apply time. Node ids are preserved verbatim -- including
|
|
the generator's start node id -- so intra-node references (variable
|
|
selectors, prompt {{#id.var#}}) stay valid; reconciling the generator's start
|
|
node with the draft's existing start is the handler's job, not this
|
|
translation step."""
|
|
|
|
from typing import Any
|
|
|
|
from core.dify_builder.models import MutationIntent
|
|
|
|
# The node-level ReactFlow type every ordinary node carries. Only a
|
|
# non-generic value is worth putting on an intent -- `_build_node` already
|
|
# defaults to this one.
|
|
_GENERIC_FLOW_TYPE = "custom"
|
|
|
|
|
|
def to_intents(graph: dict[str, Any]) -> list[MutationIntent]:
|
|
nodes = graph.get("nodes") or []
|
|
edges = graph.get("edges") or []
|
|
|
|
creates: list[MutationIntent] = []
|
|
for node in nodes:
|
|
data = node.get("data") or {}
|
|
node_type = data.get("type")
|
|
node_id = node.get("id")
|
|
if not node_type or not node_id:
|
|
continue
|
|
if node_type == "agent" and str(data.get("version")) == "2":
|
|
from core.workflow.nodes.agent_v2.entities import DifyAgentNodeData
|
|
from models.agent_config_entities import WorkflowNodeJobConfig
|
|
|
|
DifyAgentNodeData.model_validate(data)
|
|
WorkflowNodeJobConfig.model_validate(
|
|
{
|
|
"declared_outputs": data.get("agent_declared_outputs") or [],
|
|
"output_routes": data.get("agent_output_routes") or {},
|
|
}
|
|
)
|
|
config = {k: v for k, v in data.items() if k != "type"}
|
|
args: dict[str, Any] = {
|
|
"node_type": str(node_type),
|
|
"config": config,
|
|
"node_id": str(node_id),
|
|
}
|
|
# The ReactFlow wrapper carries nesting and layout OUTSIDE `data`, so
|
|
# reading only id+data dropped them on every build: children landed
|
|
# un-nested and the generator's layout was replaced by defaults.
|
|
parent_id = node.get("parentId")
|
|
if parent_id:
|
|
args["parent_id"] = str(parent_id)
|
|
position = node.get("position")
|
|
if isinstance(position, dict):
|
|
args["position"] = dict(position)
|
|
flow_type = node.get("type")
|
|
if flow_type and flow_type != _GENERIC_FLOW_TYPE:
|
|
args["flow_type"] = str(flow_type)
|
|
creates.append(MutationIntent(op="create_node", args=args))
|
|
|
|
connects: list[MutationIntent] = []
|
|
for edge in edges:
|
|
src, tgt = edge.get("source"), edge.get("target")
|
|
if not src or not tgt:
|
|
continue
|
|
args: dict[str, Any] = {"from_node": str(src), "to_node": str(tgt)}
|
|
source_handle = edge.get("sourceHandle")
|
|
if source_handle:
|
|
args["source_handle"] = source_handle
|
|
target_handle = edge.get("targetHandle")
|
|
if target_handle:
|
|
args["target_handle"] = target_handle
|
|
connects.append(MutationIntent(op="connect", args=args))
|
|
|
|
return creates + connects
|