[小ネタ] Amazon Bedrock AgentCore Memory がリソースベースポリシーに対応していたので試してみた

uv init
uv add bedrock-agentcore-starter-toolkit strands-agents strands-agents-tools bedrock-agentcore

import os
from bedrock_agentcore.runtime import BedrockAgentCoreApp
from bedrock_agentcore.memory.integrations.strands.config import AgentCoreMemoryConfig
from bedrock_agentcore.memory.integrations.strands.session_manager import AgentCoreMemorySessionManager
from strands import Agent
from strands.models.bedrock import BedrockModel

app = BedrockAgentCoreApp()

MEMORY_ID = os.environ.get("BEDROCK_AGENTCORE_MEMORY_ID")

@app.entrypoint
def invoke(payload, context):
    config = AgentCoreMemoryConfig(
        memory_id=MEMORY_ID,
        session_id=context.session_id or "default",
        actor_id=payload.get("actorId", "user"),
    )
    session_manager = AgentCoreMemorySessionManager(
        agentcore_memory_config=config, region_name="us-east-1"
    )
    agent = Agent(
        model=BedrockModel(model_id="us.anthropic.claude-sonnet-4-5-20250929-v1:0", region_name="us-east-1"),
        system_prompt="あなたは親切なアシスタントです。",
        session_manager=session_manager,
    )
    response = agent(payload.get("prompt", "こんにちは"))
    return {"response": str(response)}

if __name__ == "__main__":
    app.run()

uv run agentcore configure --entrypoint agent.py
# デフォルトの設定で進めていく
# Memory だけは有効化する
Memory Configuration
Tip: Use --disable-memory flag to skip memory entirely

Existing memory resources found:
  1. xxx
     ID: xxx

Options:
  • Enter a number to use existing memory
  • Press Enter to create new memory
  • Type 's' to skip memory setup
Your choice:
✓ Short-term memory will be enabled (default)
  • Stores conversations within sessions
  • Provides immediate context recall

# 設定が完了したらデプロイ
uv run agentcore deploy

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "DenyNonAgentRuntimeAccess",
      "Effect": "Deny",
      "Principal": "*",
      "Action": [
        "bedrock-agentcore:CreateEvent",
        "bedrock-agentcore:GetEvent",
        "bedrock-agentcore:ListEvents",
        "bedrock-agentcore:DeleteEvent",
        "bedrock-agentcore:RetrieveMemoryRecords",
        "bedrock-agentcore:ListMemoryRecords",
        "bedrock-agentcore:GetMemoryRecord",
        "bedrock-agentcore:DeleteMemoryRecord"
      ],
      "Resource": "arn:aws:bedrock-agentcore:us-east-1:<アカウントID>:memory/<メモリID>",
      "Condition": {
        "StringNotEquals": {
          "aws:PrincipalArn": "arn:aws:iam::<アカウントID>:role/<サービスロール名>"
        }
      }
    }
  ]
}

import boto3

REGION = "us-east-1"
MEMORY_ID = "<メモリID>"

client = boto3.client("bedrock-agentcore", region_name=REGION)

try:
    response = client.list_events(
        memoryId=MEMORY_ID,
        actorId="test-actor",
        sessionId="test-session",
    )
    events = response.get("events", [])
    print(f"アクセス成功: {len(events)} 件のイベントを取得")
except client.exceptions.AccessDeniedException:
    print("アクセス拒否されました")
except Exception as e:
    print(f"エラー: {type(e).__name__}: {e}")

uv run test_access.py
アクセス拒否されました

uv run agentcore invoke '{"prompt": "こんにちは、京都旅行について教えてください。"}' --session-id testsess-ion0-0000-0000-000000000006

Response:
こんにちは！京都旅行についてお手伝いさせていただきます。

京都は日本を代表する観光地で、見どころがたくさんあります。
具体的にどのような情報をお探しですか？

主な観光スポット:
- 清水寺、金閣寺、銀閣寺などの歴史的寺社
- 伏見稲荷大社の千本鳥居
- 嵐山の竹林や渡月橋
- 祇園の古い街並み

季節のおすすめ:
- 春：桜の名所（哲学の道、円山公園など）
- 秋：紅葉（永観堂、東福寺など）
- 夏：祇園祭（7月）
- 冬：雪景色の金閣寺

その他:
- 京料理や抹茶スイーツ
- 着物レンタル体験
- 宿泊施設（旅館、ホテル）

いつ頃の旅行を予定されていますか？また、特に興味のあること
はありますか？より具体的なアドバイスをさせていただきます！

uv run agentcore invoke '{"prompt": "さっき何について話しましたっけ？"}' --session-id testsess-ion0-0000-0000-000000000006

Response:
京都旅行についてお話ししました！

私が京都の主な観光スポット（清水寺、金閣寺、伏見稲荷大社、
嵐山など）、季節ごとのおすすめ、京料理や着物レンタル体験な
どについて情報を提供させていただきました。

そして、あなたの旅行の時期や特に興味のあることについて質問
させていただいたところです。

何か具体的に知りたいことはありますか？