From c7a677bba511d7f3f5c36f9384ae6ba150866e04 Mon Sep 17 00:00:00 2001
From: Eugene Yurtsev <eyurtsev@gmail.com>
Date: Fri, 27 Mar 2026 15:38:38 -0400
Subject: [PATCH] chore(langchain): add async implementation to todolist and
 test (#36313)

add async func as well
---
 .../langchain/agents/middleware/todo.py       |  8 ++++
 .../middleware/implementations/test_todo.py   | 39 +++++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/libs/langchain_v1/langchain/agents/middleware/todo.py b/libs/langchain_v1/langchain/agents/middleware/todo.py
index 21719e3ba5e..813e059c1df 100644
--- a/libs/langchain_v1/langchain/agents/middleware/todo.py
+++ b/libs/langchain_v1/langchain/agents/middleware/todo.py
@@ -152,6 +152,13 @@ def _write_todos(
     )
 
 
+async def _awrite_todos(
+    runtime: ToolRuntime[ContextT, PlanningState[ResponseT]], todos: list[Todo]
+) -> Command[Any]:
+    """Create and manage a structured task list for your current work session."""
+    return _write_todos(runtime, todos)
+
+
 class TodoListMiddleware(AgentMiddleware[PlanningState[ResponseT], ContextT, ResponseT]):
     """Middleware that provides todo list management capabilities to agents.
 
@@ -203,6 +210,7 @@ class TodoListMiddleware(AgentMiddleware[PlanningState[ResponseT], ContextT, Res
                 name="write_todos",
                 description=tool_description,
                 func=_write_todos,
+                coroutine=_awrite_todos,
                 args_schema=WriteTodosInput,
                 infer_schema=False,
             )
diff --git a/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_todo.py b/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_todo.py
index eb37700f4a6..6da56153256 100644
--- a/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_todo.py
+++ b/libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_todo.py
@@ -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()