import requests
from langchain_core.messages import HumanMessage
from agent import build_graph

graph = build_graph()
resp = requests.get('https://agents-course-unit4-scoring.hf.space/questions')
questions = resp.json()

# Test Q7
q7 = questions[6]
result = graph.invoke({'messages': [HumanMessage(content=q7['question'])]})
print(f'Q7 answer: {result["messages"][-1].content}')

# Test Q19  
q19 = questions[18]
result = graph.invoke({'messages': [HumanMessage(content=q19['question'])]})
print(f'Q19 answer: {result["messages"][-1].content}')