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()

# Q1
q1 = questions[0]
result = graph.invoke({'messages': [HumanMessage(content=q1['question'])]})
print(f"Q1 answer: {result['messages'][-1].content}")
print()

# Q14
q14 = questions[13]
result = graph.invoke({'messages': [HumanMessage(content=q14['question'])]})
print(f"Q14 answer: {result['messages'][-1].content}")