mirror of https://github.com/langgenius/dify.git
test: correct imported name
This commit is contained in:
parent
8332472944
commit
61c79b0013
|
|
@ -498,10 +498,10 @@ def test_layer_system_basic():
|
|||
|
||||
def test_layer_chaining():
|
||||
"""Test chaining multiple layers."""
|
||||
from core.workflow.graph_engine.layers import DebugLoggingLayer, Layer
|
||||
from core.workflow.graph_engine.layers import DebugLoggingLayer, GraphEngineLayer
|
||||
|
||||
# Create a custom test layer
|
||||
class TestLayer(Layer):
|
||||
class TestLayer(GraphEngineLayer):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.events_received = []
|
||||
|
|
@ -560,10 +560,10 @@ def test_layer_chaining():
|
|||
|
||||
def test_layer_error_handling():
|
||||
"""Test that layer errors don't crash the engine."""
|
||||
from core.workflow.graph_engine.layers import Layer
|
||||
from core.workflow.graph_engine.layers import GraphEngineLayer
|
||||
|
||||
# Create a layer that throws errors
|
||||
class FaultyLayer(Layer):
|
||||
class FaultyLayer(GraphEngineLayer):
|
||||
def on_graph_start(self):
|
||||
raise RuntimeError("Intentional error in on_graph_start")
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue