github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-06-09 10:17:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(langchain): add async implementation to todolist and test (#36313)

Browse Source 
add async func as well
This commit is contained in:
Eugene Yurtsev
2026-03-27 15:38:38 -04:00
committed by GitHub
parent 0351588117
commit c7a677bba5
2 changed files with 47 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_todo.py
View File

@@ -649,6 +649,45 @@ def test_single_write_todos_call_allowed() -> None:
assert result is None
async def test_todo_middleware_agent_creation_with_middleware_async() -> None:
"""Test async agent execution with the planning middleware."""
model = FakeToolCallingModel(
tool_calls=[
[
{
"args": {"todos": [{"content": "Task 1", "status": "pending"}]},
"name": "write_todos",
"type": "tool_call",
"id": "test_call",
}
],
[
{
"args": {"todos": [{"content": "Task 1", "status": "in_progress"}]},
"name": "write_todos",
"type": "tool_call",
"id": "test_call",
}
],
[
{
"args": {"todos": [{"content": "Task 1", "status": "completed"}]},
"name": "write_todos",
"type": "tool_call",
"id": "test_call",
}
],
[],
]
)
middleware = TodoListMiddleware()
agent = create_agent(model=model, middleware=[middleware])
result = await agent.ainvoke({"messages": [HumanMessage("Hello")]})
assert result["todos"] == [{"content": "Task 1", "status": "completed"}]
assert len(result["messages"]) == 8
async def test_parallel_write_todos_calls_rejected_async() -> None:
"""Test async version - parallel write_todos calls are rejected with error messages."""
middleware = TodoListMiddleware()