15 lines
331 B
Python
15 lines
331 B
Python
|
|
import pytest
|
||
|
|
from langsmith import unit
|
||
|
|
|
||
|
|
from react_agent import graph
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
@unit
|
||
|
|
async def test_react_agent_simple_passthrough() -> None:
|
||
|
|
res = await graph.ainvoke(
|
||
|
|
{"messages": [("user", "Who is the founder of LangChain?")]}
|
||
|
|
)
|
||
|
|
|
||
|
|
assert "harrison" in str(res["messages"][-1].content).lower()
|