pydantic-ai-dynamodb-persistence

Amazon DynamoDB backend for pydantic-ai-persistence — a PydanticAI StepStore (durable events / snapshots / tool-effect ledger) and message-history store, on DynamoDB.

Install

pip install pydantic-ai-dynamodb-persistence
# or via the core extra:
pip install "pydantic-ai-persistence[dynamodb]"

Requires Python 3.10+.

Usage

Durable step persistence

from pydantic_ai import Agent
from pydantic_ai_harness.step_persistence import StepPersistence
from pydantic_ai_dynamodb_persistence import DynamoDBStepStore

store = DynamoDBStepStore(table_name="pai-steps", region_name="us-east-1",
                         max_snapshots_per_run=20)
agent = Agent("openai:gpt-4o", capabilities=[StepPersistence(store=store)])

Message history

from pydantic_ai_dynamodb_persistence import DynamoDBHistoryStore

store = DynamoDBHistoryStore(table_name="pai-history", region_name="us-east-1")
await store.save("conv-1", result.all_messages())
prior = await store.load("conv-1")

API

• DynamoDBStepStore(table_name, *, region_name=None, boto_session=None, endpoint_url=None, max_snapshots_per_run=None)
• DynamoDBHistoryStore(table_name, *, region_name=None, boto_session=None, endpoint_url=None)
• DynamoDBAsyncKV(...) — the underlying AsyncKV (offloads sync boto3 to a thread).

Standard boto3 auth (env / profile / SSO / IAM role). Single table, PK + SK string keys, auto-created (on-demand billing).

Note: PydanticAI's StepStore is a beta/experimental harness feature; its API may still change.

License

MIT