import os
from agent import llm

def test_llm_type():
    assert llm.__class__.__name__ == "ChatOllama"
    assert llm.model == "llama3"