DynamoDB aput_writes runs many writes sequentially #766

import random
from langgraph_checkpoint_aws import DynamoDBSaver
import asyncio
from time import time

TABLE_NAME = "test-langgraph" # Replace with your DynamoDB table name
REGION_NAME = "eu-central-1" # Replace with your AWS region

thread_id = str(random.randint(1, 1_000_000))

inputs_aput_writes = {
    "config": {
        "tags": [],
        "metadata": {
            "thread_id": thread_id,
            "user_id": "mock-id"
        },
        "callbacks": None,
        "recursion_limit": 25,
        "configurable": {
            "checkpoint_ns": "",
            "thread_id": thread_id,
            "user_id": "mock-id",
            "checkpoint_id": "1f0c530b-1ed7-6d1e-bfff-a40ab47d5463"
        }
    },
    "writes": [
        ('user_id', 'mock-id'),
        ('language', 'en'),
        ('thread_id', thread_id),
        ('timestamp', '251119_101538082238'),
        ('data', 'good data'),
        ('branch:to:parse_listing_qualifications', None)
    ],
    "task_id": "576d569d-52df-e0bf-c0ad-5606841615f8",
    "task_path": '~__pregel_pull, __start__'
}


dynamo_db_saver = DynamoDBSaver(
    table_name=TABLE_NAME,
    region_name=REGION_NAME,
)

async def main():
    t0 = time()
    await dynamo_db_saver.aput_writes(**inputs_aput_writes)

    print("Time to run aput_writes: {:.4f} seconds".format(time() - t0))
    
if __name__ == "__main__":

    asyncio.run(main())