from ai_archive.schema import Conversation, Message


def test_conversation_to_dict():
    convo = Conversation(
        source="chatgpt",
        conversation_id="c1",
        title="Example",
        messages=[Message(message_id="m1", role="user", content="hello")],
    )
    data = convo.to_dict()
    assert data["source"] == "chatgpt"
    assert data["messages"][0]["role"] == "user"