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

refactor(graph_engine): Move setup methods into __init__

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN-
2025-09-01 12:08:03 +08:00
parent 0fdb1b2bc9
commit 88622f70fb
3 changed files with 68 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/tests/unit_tests/core/workflow/graph_engine/test_conditional_streaming_vs_template_workflow.py
View File

@ -100,7 +100,7 @@ def test_streaming_output_with_blocking_equals_one():
)
# Check that NodeRunStreamChunkEvent contains 'query' should has same id with Start NodeRunStartedEvent
start_node_id = engine.graph.root_node.id
start_node_id = graph.root_node.id
start_events = [e for e in events if isinstance(e, NodeRunStartedEvent) and e.node_id == start_node_id]
assert len(start_events) == 1, f"Expected 1 start event for node {start_node_id}, but got {len(start_events)}"
start_event = start_events[0]
@ -210,7 +210,7 @@ def test_streaming_output_with_blocking_not_equals_one():
assert isinstance(chunk_event.chunk, str), f"Expected chunk to be string, but got {type(chunk_event.chunk)}"
# Check that NodeRunStreamChunkEvent contains 'query' should has same id with Start NodeRunStartedEvent
start_node_id = engine.graph.root_node.id
start_node_id = graph.root_node.id
start_events = [e for e in events if isinstance(e, NodeRunStartedEvent) and e.node_id == start_node_id]
assert len(start_events) == 1, f"Expected 1 start event for node {start_node_id}, but got {len(start_events)}"
start_event = start_events[0]