File size: 545 Bytes
b70c4a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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}')
|