Browse Source

fix: backwards invoke nodes

Yeuoly 3 months ago
parent
commit
fb7b2c8ff3
1 changed files with 2 additions and 2 deletions
  1. 2 2
      api/core/workflow/workflow_entry.py

+ 2 - 2
api/core/workflow/workflow_entry.py

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