from api.schemas.agent_message import MessageCreate
from api.schemas.doi import DoIReport
from hub_core.schemas.agent_message import MessageCreate as CoreMessageCreate
from hub_core.schemas.doi import DoIReport as CoreDoIReport


def test_state_hub_reexports_core_message_schema() -> None:
    assert MessageCreate is CoreMessageCreate


def test_state_hub_reexports_core_doi_schema() -> None:
    assert DoIReport is CoreDoIReport