File size: 676 Bytes
1b8aef5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
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)
|