diff --git a/src/bedrock_agentcore/memory/session.py b/src/bedrock_agentcore/memory/session.py index 2837610..0e266f8 100644 --- a/src/bedrock_agentcore/memory/session.py +++ b/src/bedrock_agentcore/memory/session.py @@ -89,6 +89,9 @@ def my_llm(user_input: str, memories: List[Dict]) -> str: - bedrock-agentcore:DeleteMemoryRecord - bedrock-agentcore:ListActors - bedrock-agentcore:ListSessions + - bedrock-agentcore:BatchCreateMemoryRecords + - bedrock-agentcore:BatchDeleteMemoryRecords + - bedrock-agentcore:BatchUpdateMemoryRecords """ def __init__( @@ -138,6 +141,9 @@ def __init__( "get_event", "delete_event", "list_events", + "batch_create_memory_records", + "batch_delete_memory_records", + "batch_update_memory_records", } def _validate_and_resolve_region(self, region_name: Optional[str], session: Optional[boto3.Session]) -> str: