panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Loop node exception handling (#4092)

v3.2
shaohuzhang1 2025-09-23 17:22:17 +08:00 committed by GitHub
parent ccb43a05a5
commit 80da909dc3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/application/flow/step_node/loop_node/impl/base_loop_node.py
View File

@ -9,6 +9,8 @@
import time import time
from typing import Dict, List from typing import Dict, List
from django.utils.translation import gettext as _
from application.flow.common import Answer from application.flow.common import Answer
from application.flow.i_step_node import NodeResult, WorkFlowPostHandler, INode from application.flow.i_step_node import NodeResult, WorkFlowPostHandler, INode
from application.flow.step_node.loop_node.i_loop_node import ILoopNode from application.flow.step_node.loop_node.i_loop_node import ILoopNode
@ -16,7 +18,6 @@ from application.flow.tools import Reasoning
from application.models import ChatRecord from application.models import ChatRecord
from common.handle.impl.response.loop_to_response import LoopToResponse from common.handle.impl.response.loop_to_response import LoopToResponse
from maxkb.const import CONFIG from maxkb.const import CONFIG
from django.utils.translation import gettext as _
max_loop_count = int(CONFIG.get("WORKFLOW_LOOP_NODE_MAX_LOOP_COUNT", 500)) max_loop_count = int(CONFIG.get("WORKFLOW_LOOP_NODE_MAX_LOOP_COUNT", 500))
@ -173,7 +174,17 @@ def loop(workflow_manage_new_instance, node: INode, generate_loop):
answer += content_chunk answer += content_chunk
yield chunk yield chunk
if chunk.get('node_status', "SUCCESS") == 'ERROR': if chunk.get('node_status', "SUCCESS") == 'ERROR':
raise Exception(chunk.get('content')) insert_or_replace(loop_node_data, index, instance.get_runtime_details())
insert_or_replace(loop_answer_data, index,
get_answer_list(instance, child_node_node_dict, node.runtime_node_id))
node.context['is_interrupt_exec'] = is_interrupt_exec
node.context['loop_node_data'] = loop_node_data
node.context['loop_answer_data'] = loop_answer_data
node.context["index"] = current_index
node.context["item"] = current_index
node.status = 500
node.err_message = chunk.get('content')
return
node_type = chunk.get('node_type') node_type = chunk.get('node_type')
if node_type == 'form-node': if node_type == 'form-node':
break_outer = True break_outer = True
@ -182,7 +193,6 @@ def loop(workflow_manage_new_instance, node: INode, generate_loop):
start_node_data = None start_node_data = None
chat_record = None chat_record = None
child_node = None child_node = None
loop_global_data = instance.context
insert_or_replace(loop_node_data, index, instance.get_runtime_details()) insert_or_replace(loop_node_data, index, instance.get_runtime_details())
insert_or_replace(loop_answer_data, index, insert_or_replace(loop_answer_data, index,
get_answer_list(instance, child_node_node_dict, node.runtime_node_id)) get_answer_list(instance, child_node_node_dict, node.runtime_node_id))