ngskcloud
/
gcgj-dify-1.7.0
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Simplify the formatting of output variables, improve code readability; update test cases to verify the correctness of different output configurations.

Browse Source
pull/20921/head
GuanMu 9 months ago
parent 81027f6077
commit ea3a99efcd
2 changed files with 20 additions and 60 deletions
Show Stats Download Patch File Download Diff File

6
api/core/workflow/graph_engine/graph_engine.py
Unescape Escape View File

@ -198,9 +198,11 @@ class GraphEngine:
].strip()
# Only save the user-defined output variables, not the entire node output
if (item.route_node_state.node_run_result
if (
item.route_node_state.node_run_result
and item.route_node_state.node_run_result.outputs
and "outputs" in item.route_node_state.node_run_result.outputs):
and "outputs" in item.route_node_state.node_run_result.outputs
):
user_outputs = item.route_node_state.node_run_result.outputs.get("outputs", {})
self.graph_runtime_state.outputs.update(user_outputs)
except Exception as e:

62
api/tests/unit_tests/core/workflow/nodes/answer/test_answer.py
Unescape Escape View File

@ -102,16 +102,8 @@ def test_execute_answer_with_outputs():
"type": "answer",
"answer": "Weather: {{#start.weather#}}, Score: {{#start.score#}}",
"outputs": [
{
"variable": "confidence",
"type": "number",
"value_selector": ["start", "score"]
},
{
"variable": "status",
"type": "string",
"value_selector": ["start", "weather"]
}
{"variable": "confidence", "type": "number", "value_selector": ["start", "score"]},
{"variable": "status", "type": "string", "value_selector": ["start", "weather"]},
],
},
"id": "answer",
@ -155,16 +147,8 @@ def test_execute_answer_with_outputs():
"type": "answer",
"answer": "Weather: {{#start.weather#}}, Score: {{#start.score#}}",
"outputs": [
{
"variable": "confidence",
"type": "number",
"value_selector": ["start", "score"]
},
{
"variable": "status",
"type": "string",
"value_selector": ["start", "weather"]
}
{"variable": "confidence", "type": "number", "value_selector": ["start", "score"]},
{"variable": "status", "type": "string", "value_selector": ["start", "weather"]},
],
},
},
@ -205,16 +189,8 @@ def test_execute_answer_with_complex_outputs():
"type": "answer",
"answer": "Analysis complete",
"outputs": [
{
"variable": "scores",
"type": "array[number]",
"value_selector": ["start", "score_list"]
},
{
"variable": "metadata",
"type": "object",
"value_selector": ["start", "meta_info"]
}
{"variable": "scores", "type": "array[number]", "value_selector": ["start", "score_list"]},
{"variable": "metadata", "type": "object", "value_selector": ["start", "meta_info"]},
],
},
"id": "answer",
@ -258,16 +234,8 @@ def test_execute_answer_with_complex_outputs():
"type": "answer",
"answer": "Analysis complete",
"outputs": [
{
"variable": "scores",
"type": "array[number]",
"value_selector": ["start", "score_list"]
},
{
"variable": "metadata",
"type": "object",
"value_selector": ["start", "meta_info"]
}
{"variable": "scores", "type": "array[number]", "value_selector": ["start", "score_list"]},
{"variable": "metadata", "type": "object", "value_selector": ["start", "meta_info"]},
],
},
},
@ -383,11 +351,7 @@ def test_execute_answer_outputs_variable_not_found():
"type": "answer",
"answer": "Test answer",
"outputs": [
{
"variable": "missing_var",
"type": "string",
"value_selector": ["start", "non_existent"]
}
{"variable": "missing_var", "type": "string", "value_selector": ["start", "non_existent"]}
],
},
"id": "answer",
@ -428,13 +392,7 @@ def test_execute_answer_outputs_variable_not_found():
"title": "Missing variable",
"type": "answer",
"answer": "Test answer",
"outputs": [
{
"variable": "missing_var",
"type": "string",
"value_selector": ["start", "non_existent"]
}
],
"outputs": [{"variable": "missing_var", "type": "string", "value_selector": ["start", "non_existent"]}],
},
},
)

Write Preview
Loading…
Cancel
Save