fix: backwards invoke nodes

pull/12538/head
Yeuoly 1 year ago
parent c3440a27fb
commit fb7b2c8ff3

@ -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}")

Loading…
Cancel
Save