13 lines
332 B
Python
13 lines
332 B
Python
import pytest
|
|
from agent import graph
|
|
from langsmith import expect, unit
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@unit
|
|
async def test_agent_simple_passthrough() -> None:
|
|
res = await graph.ainvoke(
|
|
{"messages": [{"role": "user", "content": "What's 62 - 19?"}]}
|
|
)
|
|
expect(res["messages"][-1]["content"][0]["text"]).to_contain("43")
|