from langchain_core.messages import HumanMessage
from langgraph.graph import MessagesState

from agent_factory import AgentFactory
from question_retriever import get_question


def test_agent() -> None:
    # given
    # grocery list
    task_id = "3cef3a44-215e-4aed-8e3b-b1e3f08063b7"
    question = get_question(task_id=task_id)

    agent_factory = AgentFactory()
    agent = agent_factory.get()

    initial_state = MessagesState(
        messages=[
            agent_factory.system_prompt,
            HumanMessage(content=question)
        ]
    )

    final_state = agent.invoke(input=initial_state)
    answer = final_state["messages"][-1].content

    print(answer)