AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 22:38:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add logging for missing edge mapping in StreamProcessor (#12008)

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN- 2024-12-23 17:53:36 +08:00 committed by GitHub
parent 425cc1ea85
commit 75bce2822e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/workflow/nodes/answer/base_stream_processor.py
View File

@ -1,3 +1,4 @@
import logging
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from collections.abc import Generator from collections.abc import Generator
@ -5,6 +6,8 @@ from core.workflow.entities.variable_pool import VariablePool
from core.workflow.graph_engine.entities.event import GraphEngineEvent, NodeRunSucceededEvent from core.workflow.graph_engine.entities.event import GraphEngineEvent, NodeRunSucceededEvent
from core.workflow.graph_engine.entities.graph import Graph from core.workflow.graph_engine.entities.graph import Graph
logger = logging.getLogger(__name__)
class StreamProcessor(ABC): class StreamProcessor(ABC):
def __init__(self, graph: Graph, variable_pool: VariablePool) -> None: def __init__(self, graph: Graph, variable_pool: VariablePool) -> None:
@ -31,6 +34,9 @@ class StreamProcessor(ABC):
if run_result.edge_source_handle: if run_result.edge_source_handle:
reachable_node_ids = [] reachable_node_ids = []
unreachable_first_node_ids = [] unreachable_first_node_ids = []
if finished_node_id not in self.graph.edge_mapping:
logger.warning(f"node {finished_node_id} has no edge mapping")
return
for edge in self.graph.edge_mapping[finished_node_id]: for edge in self.graph.edge_mapping[finished_node_id]:
if ( if (
edge.run_condition edge.run_condition