2024-09-13 16:28:17 -07:00
|
|
|
import pytest
|
2024-09-13 16:57:34 -07:00
|
|
|
from langsmith import expect, unit
|
2024-09-13 16:28:17 -07:00
|
|
|
|
2024-09-13 17:06:33 -07:00
|
|
|
from agent import graph
|
|
|
|
|
|
2024-09-13 16:28:17 -07:00
|
|
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
|
|
|
@unit
|
2024-09-13 16:30:31 -07:00
|
|
|
async def test_agent_simple_passthrough() -> None:
|
2024-09-13 17:06:33 -07:00
|
|
|
res = await graph.ainvoke({"messages": ["user", "What's 62 - 19?"]})
|
|
|
|
|
expect(str(res["messages"][-1].content)).to_contain("43")
|