diff --git a/test_llm.py b/test_llm.py new file mode 100644 index 0000000..1a5af55 --- /dev/null +++ b/test_llm.py @@ -0,0 +1,6 @@ +import os +from agent import llm + +def test_llm_type(): + assert llm.__class__.__name__ == "ChatOllama" + assert llm.model == "llama3" \ No newline at end of file