diff --git a/AGENTS.md b/AGENTS.md
index 09ff6e3..8ca86b5 100644
--- a/AGENTS.md
+++ b/AGENTS.md
@@ -495,9 +495,10 @@ const allResults = results.getResults();
 **Python:**
 
 ```python
+from collections.abc import Sequence
 from aws_durable_execution_sdk_python.concurrency import MapConfig, CompletionConfig
 
-def process_item(ctx: DurableContext, item: dict, index: int) -> dict:
+def process_item(ctx: DurableContext, item: dict, index: int, items: Sequence[dict]) -> dict:
     return ctx.step(lambda _: process(item), name=f"process-{index}")
 
 results = context.map(