from langchain_classic.schema.agent import __all__

EXPECTED_ALL = ["AgentAction", "AgentActionMessageLog", "AgentFinish"]


def test_all_imports() -> None:
    assert set(__all__) == set(EXPECTED_ALL)