def evaluator_function(step_input: StepInput, session_state):
    return session_state.get("test") == "test_1"

condition_step = Condition(
    name="condition_step",
    evaluator=evaluator_function,
    steps=[display_step],
)

        __
        __