Files
graphtest/tests/integration_tests/test_graph.py
William Fu-Hinthorn 4e6582bd91 tmp
2024-09-13 16:30:31 -07:00

15 lines
319 B
Python

import pytest
from langsmith import unit
from agent import graph
@pytest.mark.asyncio
@unit
async def test_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()