fix: issue langgenius#12068 by test is answer in the ids

Signed-off-by: yihong0618 <[email protected]>
yihong0618 · Dec 26, 2024 · b6d2a43 · b6d2a43
1 parent 84ac004
commit b6d2a43
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/api/core/workflow/nodes/answer/base_stream_processor.py b/api/core/workflow/nodes/answer/base_stream_processor.py
@@ -48,9 +48,11 @@ def _remove_unreachable_nodes(self, event: NodeRunSucceededEvent | NodeRunExcept
                     # we remove the node maybe shortcut the answer node, so comment this code for now
                     # there is not effect on the answer node and the workflow, when we have a better solution
                     # we can open this code. Issues: #11542 #9560 #10638 #10564
-
-                    # reachable_node_ids.extend(self._fetch_node_ids_in_reachable_branch(edge.target_node_id))
-                    continue
+                    ids = self._fetch_node_ids_in_reachable_branch(edge.target_node_id)
+                    if "answer" in ids:
+                        continue
+                    else:
+                        reachable_node_ids.extend(ids)
                 else:
                     unreachable_first_node_ids.append(edge.target_node_id)