chore: expose DynamoDBSaver from root path (#746)

Simplified the import path for `DynamoDBSaver` by exposing it at the
package root level, improving developer experience and consistency with
other checkpoint savers.

---------

Co-authored-by: Sunil Yadav <[email protected]>
Co-authored-by: Michael Chin <[email protected]>

Author: 3 people

libs/langgraph-checkpoint-aws/README.md

@@ -190,7 +190,7 @@ with ValkeyCache.from_conn_string(
 
 ```python
 from langgraph.graph import StateGraph
-from langgraph_checkpoint_aws.checkpoint.dynamodb import DynamoDBSaver
+from langgraph_checkpoint_aws import DynamoDBSaver
 
 # Basic usage with DynamoDB only
 checkpointer = DynamoDBSaver(
@@ -376,7 +376,7 @@ All components support async operations:
 ```python
 from langgraph_checkpoint_aws.async_saver import AsyncBedrockSessionSaver
 from langgraph_checkpoint_aws import AsyncValkeySaver
-from langgraph_checkpoint_aws.checkpoint.dynamodb import DynamoDBSaver
+from langgraph_checkpoint_aws import DynamoDBSaver
 
 # Async Bedrock usage
 session_saver = AsyncBedrockSessionSaver(region_name="us-west-2")

libs/langgraph-checkpoint-aws/docs/dynamodb/DynamoDBSaver.md

@@ -43,7 +43,7 @@ pip install langgraph-checkpoint-aws
 ## Quick Start
 
 ```python
-from langgraph_checkpoint_aws.checkpoint.dynamodb import DynamoDBSaver
+from langgraph_checkpoint_aws import DynamoDBSaver
 from langgraph.graph import StateGraph
 
 # Initialize checkpointer

libs/langgraph-checkpoint-aws/langgraph_checkpoint_aws/__init__.py

@@ -12,6 +12,9 @@
 from langgraph_checkpoint_aws.agentcore.store import (
     AgentCoreMemoryStore,
 )
+from langgraph_checkpoint_aws.checkpoint.dynamodb import (
+    DynamoDBSaver,
+)
 
 # Conditional imports for Valkey functionality
 try:
@@ -98,6 +101,7 @@ def _missing_dependencies_error(*args: Any, **kwargs: Any) -> Any:
     "ValkeyValidationError",
     "ValkeySaver",
     "ValkeyCache",
+    "DynamoDBSaver",
     "SDK_USER_AGENT",
     "valkey_available",
 ]

libs/langgraph-checkpoint-aws/tests/integration_tests/checkpoint/dynamodb/test_langgraph_dynamodb_integration.py

@@ -19,7 +19,7 @@
 from langgraph.graph.message import add_messages
 from langgraph.types import Command
 
-from langgraph_checkpoint_aws.checkpoint.dynamodb import DynamoDBSaver
+from langgraph_checkpoint_aws import DynamoDBSaver
 
 logger = logging.getLogger(__name__)
 

libs/langgraph-checkpoint-aws/tests/unit_tests/checkpoint/dynamodb/test_saver.py

@@ -15,7 +15,7 @@
     TEST_TTL_SECONDS,
 )
 
-from langgraph_checkpoint_aws.checkpoint.dynamodb.saver import DynamoDBSaver
+from langgraph_checkpoint_aws import DynamoDBSaver
 
 
 class TestDynamoDBSaverInitialization:

samples/memory/dynamodb_saver.ipynb

@@ -70,7 +70,7 @@
     "from langgraph.graph.message import add_messages\n",
     "\n",
     "# Import DynamoDB saver\n",
-    "from langgraph_checkpoint_aws.checkpoint.dynamodb import DynamoDBSaver\n",
+    "from langgraph_checkpoint_aws import DynamoDBSaver\n",
     "import boto3\n",
     "\n",
     "print(\"✅ All dependencies imported successfully!\")\n",
@@ -273,7 +273,7 @@
     "    temperature=0.7,\n",
     "    max_tokens=2048,\n",
     "    region_name=REGION_NAME,\n",
-    "    client=session.client('bedrock-runtime')\n",
+    "    client=boto_session.client('bedrock-runtime')\n",
     ")\n",
     "\n",
     "print(\"✅ Language model initialized (Claude 4 Sonnet)\")\n",