update

tests/integration_tests/test_graph.py

@@ -1,12 +1,11 @@
 import pytest
-from agent import graph
 from langsmith import expect, unit
 
+from agent import graph
+
 
 @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")
+    res = await graph.ainvoke({"messages": ["user", "What's 62 - 19?"]})
+    expect(str(res["messages"][-1].content)).to_contain("43")