Skip to content

Commit 245f3c1

Browse files
authored
chore: enabling batch api pass through to boto3 client methods (#135)
* chore: enabling batch api pass through to boto3 client * fix: linting issue Signed-off-by: Shreyas Pant <69047353+ShreyP88@users.noreply.github.com> --------- Signed-off-by: Shreyas Pant <69047353+ShreyP88@users.noreply.github.com>
1 parent ecba82d commit 245f3c1

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/bedrock_agentcore/memory/session.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,9 @@ def my_llm(user_input: str, memories: List[Dict]) -> str:
8989
- bedrock-agentcore:DeleteMemoryRecord
9090
- bedrock-agentcore:ListActors
9191
- bedrock-agentcore:ListSessions
92+
- bedrock-agentcore:BatchCreateMemoryRecords
93+
- bedrock-agentcore:BatchDeleteMemoryRecords
94+
- bedrock-agentcore:BatchUpdateMemoryRecords
9295
"""
9396

9497
def __init__(
@@ -138,6 +141,9 @@ def __init__(
138141
"get_event",
139142
"delete_event",
140143
"list_events",
144+
"batch_create_memory_records",
145+
"batch_delete_memory_records",
146+
"batch_update_memory_records",
141147
}
142148

143149
def _validate_and_resolve_region(self, region_name: Optional[str], session: Optional[boto3.Session]) -> str:

0 commit comments

Comments
 (0)