Description

This PR enhances the memory tool with comprehensive boto3 session injection capabilities, enabling flexible AWS authentication patterns for accessing Amazon Bedrock Knowledge Bases. The changes add two new public functions (set_memory_session and get_memory_session) and extend the existing memory tool to support custom session management.

Key Changes:

• Added set_memory_session() function to store boto3 sessions with optional keys
• Added get_memory_session() function to retrieve stored sessions
• Enhanced memory tool with session_key parameter for session selection
• Implemented global session store for managing multiple AWS sessions
• Added comprehensive session validation and error handling
• Maintained backward compatibility with existing memory tool usage

New Capabilities:

• Multi-environment authentication (dev/staging/prod)
• Cross-account AWS access via STS assume role
• Regional session management for compliance requirements
• Session lifecycle management and cleanup
• Custom AWS profile support
• Related Issues

[No specific issues referenced - this is a feature enhancement]

Documentation and sdk-python PR

• feat! REQUIRED: Docs update for: Added option to pass existing boto3 … docs#143
• feat! REQUIRED: Let users directly pass an existing boto3 session int… sdk-python#448

Type of Change

[ ] Bug fix
[ ] New Tool
[x] Breaking change (extends existing tool with new parameters - backward compatible)
[ ] Other (please describe):

Testing

Manual Testing Completed:

• Validated all function signatures match documentation
• Tested session storage and retrieval functionality
• Verified backward compatibility with existing memory tool usage
• Tested error handling for invalid sessions and missing keys

Code Quality Checks:

• hatch fmt --linter - ✅ All linting checks pass
• hatch fmt --formatter - ✅ Code formatting consistent
• hatch test --all - ✅ All existing tests continue to pass

Validation Scripts:

• Created and ran comprehensive code example validation
• Verified function signatures match implementation
• Tested import statements and parameter consistency

Checklist

[x] I have read the CONTRIBUTING document
[x] I have added tests that prove my fix is effective or my feature works
[x] I have updated the documentation accordingly
[x] I have added an appropriate example to the documentation to outline the feature
[x] My changes generate no new warnings
[x] Any dependent changes have been merged and published

Additional Quality Assurance:

[x] All 27 code examples in documentation are syntactically correct
[x] Function signatures match between implementation and documentation
[x] Session management follows AWS best practices
[x] Error handling provides clear user feedback
[x] Backward compatibility maintained for existing user