-
Notifications
You must be signed in to change notification settings - Fork 213
feat(langgraph-checkpoint-aws): add Valkey checkpoint saver for AgentCore #738
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat(langgraph-checkpoint-aws): add Valkey checkpoint saver for AgentCore #738
Conversation
Updated AgentCore Valkey saver and helpers to use langgraph-checkpoint 3.0 API. Changes: - Removed jsonplus_serde field (no longer needed in 3.0) - Updated pyproject.toml to require langgraph-checkpoint>=3.0.0 - Removed JsonPlusSerializer imports (use SerializerProtocol instead) - Updated tests to match new API
…d of Protocol Cannot instantiate Protocol classes directly. Use concrete JsonPlusSerializer implementation for default serializer in AgentCoreEventClient. Fixes: 'Cannot instantiate protocol class SerializerProtocol' mypy error
- Add proper try/except for Valkey imports in integration tests - Fix line length issue (E501) in unit tests by splitting long comment - Ensure tests skip gracefully when valkey package is not installed
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@seaofawareness looks good overall, one comment regarding the sample notebook.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM - thanks for all of your work on the Valkey integrations @seaofawareness !
This PR provides AgentCore-compatible checkpoint savers that use Valkey as the storage backend, combining AgentCore session management concepts with Valkey's storage and search capabilities.