from langchain_core.messages import AIMessage

from gaia_agent.agent import GaiaAgent


class FakeChatModel:
    def invoke(self, messages):
        return AIMessage(content="Dummy LLM answer")


def test_agent_returns_llm_answer():
    agent = GaiaAgent(llm=FakeChatModel())

    assert (
        agent(
            "What is the answer?",
            session_id="test-session",
            user_id="test-user",
            task_id="test-task",
        )
        == "Dummy LLM answer"
    )