fix: backwards invoke nodes

This commit is contained in:
Yeuoly 2025-01-07 20:52:25 +08:00
parent c3440a27fb
commit fb7b2c8ff3

View File

@ -2,7 +2,7 @@ import logging
import time import time
import uuid import uuid
from collections.abc import Generator, Mapping, Sequence from collections.abc import Generator, Mapping, Sequence
from typing import Any, Optional from typing import Any, Optional, cast
from configs import dify_config from configs import dify_config
from core.app.apps.base_app_queue_manager import GenerateTaskStoppedError from core.app.apps.base_app_queue_manager import GenerateTaskStoppedError
@ -237,7 +237,7 @@ class WorkflowEntry:
if node_type not in {NodeType.PARAMETER_EXTRACTOR, NodeType.QUESTION_CLASSIFIER}: if node_type not in {NodeType.PARAMETER_EXTRACTOR, NodeType.QUESTION_CLASSIFIER}:
raise ValueError(f"Node type {node_type} not supported") raise ValueError(f"Node type {node_type} not supported")
node_cls = node_type_classes_mapping.get(node_type) node_cls = NODE_TYPE_CLASSES_MAPPING[node_type]["1"]
if not node_cls: if not node_cls:
raise ValueError(f"Node class not found for node type {node_type}") raise ValueError(f"Node class not found for node type {node_type}")