fix: skip processing for missing source_node_id in node_id_config_map…

…ping Signed-off-by: -LAN- <[email protected]>
langgenius · Dec 23, 2024 · 56dfee9 · 56dfee9
1 parent c1aa55f
commit 56dfee9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/core/workflow/nodes/answer/answer_stream_generate_router.py b/api/core/workflow/nodes/answer/answer_stream_generate_router.py
@@ -147,6 +147,9 @@ def _recursive_fetch_answer_dependencies(
         reverse_edges = reverse_edge_mapping.get(current_node_id, [])
         for edge in reverse_edges:
             source_node_id = edge.source_node_id
+            # found sometimes source_node_id is not in node_id_config_mapping, temporary fix by skip it
+            if source_node_id not in node_id_config_mapping:
+                continue
             source_node_type = node_id_config_mapping[source_node_id].get("data", {}).get("type")
             source_node_data = node_id_config_mapping[source_node_id].get("data", {})
             if (