浏览代码

fix: common prerequisite node workflow remove reachable node that failed to streaming llm… (#19552)

Co-authored-by: zhangshibo <zhangshibo@didiglobal.com>
tags/1.4.0
SSebo 5 个月前
父节点
当前提交
57b3912227
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5
    0
      api/core/workflow/nodes/answer/base_stream_processor.py

+ 5
- 0
api/core/workflow/nodes/answer/base_stream_processor.py 查看文件

@@ -95,7 +95,12 @@ class StreamProcessor(ABC):
if node_id not in self.rest_node_ids:
return

if node_id in reachable_node_ids:
return

self.rest_node_ids.remove(node_id)
self.rest_node_ids.extend(set(reachable_node_ids) - set(self.rest_node_ids))

for edge in self.graph.edge_mapping.get(node_id, []):
if edge.target_node_id in reachable_node_ids:
continue

正在加载...
取消
保存