[アップデート] Amazon Bedrock AgentCore Runtime に Session Storage が追加されました

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

import os

os.environ["BYPASS_TOOL_CONSENT"] = "true"

from strands import Agent
from strands.session import FileSessionManager
from strands.models import BedrockModel
from strands_tools import file_read, file_write, shell
from bedrock_agentcore.runtime import BedrockAgentCoreApp

app = BedrockAgentCoreApp()
WORKSPACE = "/mnt/workspace"

model = BedrockModel(model_id="us.anthropic.claude-sonnet-4-20250514-v1:0")
tools = [file_read, file_write, shell]

@app.entrypoint
def handle_request(payload):
    session_id = payload.get("session_id", "default")

    session_manager = FileSessionManager(
        session_id=session_id,
        storage_dir=f"{WORKSPACE}/.sessions"
    )

    agent = Agent(
        model=model,
        tools=tools,
        session_manager=session_manager,
        system_prompt="あなたはコーディングアシスタントです。プロジェクトファイルは /mnt/workspace にあります。日本語で応答してください。"
    )

    response = agent(payload.get("prompt"))
    return {"response": response.message["content"][0]["text"]}

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

agentcore configure 
# 基本デフォルトの設定で対話形式で進めていく
agentcore deploy

✓ Agent 'coding-agent' deployed successfully
  Runtime ARN: arn:aws:bedrock-agentcore:us-east-1:123456789012:agent-runtime/abc123def
  Endpoint ARN: arn:aws:bedrock-agentcore:us-east-1:123456789012:agent-runtime-endpoint/xyz789

import boto3

RUNTIME_ID = "abc123def"  # agentcore deploy で取得した Runtime ID に置き換えてください

client = boto3.client("bedrock-agentcore-control", region_name="us-east-1")

# 既存の設定を取得（UpdateAgentRuntime は artifact / roleArn / networkConfiguration が必須）
current = client.get_agent_runtime(agentRuntimeId=RUNTIME_ID)

response = client.update_agent_runtime(
    agentRuntimeId=RUNTIME_ID,
    agentRuntimeArtifact=current["agentRuntimeArtifact"],
    roleArn=current["roleArn"],
    networkConfiguration=current["networkConfiguration"],
    filesystemConfigurations=[
        {
            "sessionStorage": {
                "mountPath": "/mnt/workspace"
            }
        }
    ]
)

print(f"Status: {response['status']}")

Status: UPDATING

import boto3, json

RUNTIME_ID = "abc123def"  # 同じ Runtime ID

client = boto3.client("bedrock-agentcore-control", region_name="us-east-1")

response = client.get_agent_runtime(agentRuntimeId=RUNTIME_ID)
print(json.dumps(response["filesystemConfigurations"], indent=2))

[
  {
    "sessionStorage": {
      "mountPath": "/mnt/workspace"
    }
  }
]

import boto3
import json

client = boto3.client("bedrock-agentcore", region_name="us-east-1")

# agentcore deploy の出力に表示された Runtime ARN に置き換えてください
agent_arn = "arn:aws:bedrock-agentcore:us-east-1:123456789012:agent-runtime/abc123def"
session_id = "session-storage-test-001-jinno-yudai"  # 33文字以上が必要

def invoke(prompt):
    resp = client.invoke_agent_runtime(
        agentRuntimeArn=agent_arn,
        runtimeSessionId=session_id,
        payload=json.dumps({
            "prompt": prompt,
            "session_id": "conv-001"
        }).encode()
    )
    return json.loads(b"".join(resp["response"]))["response"]

result = invoke(
    "/mnt/workspace に calculator.py を作成してください。"
    "add, subtract, multiply の3つの関数を実装してください。"
    "作成後、/mnt/workspace のファイル一覧を表示してください。"
)
print(result)

calculator.pyファイルを正常に作成しました！

作成したファイルの詳細：
- ファイル名: calculator.py
- 場所: /mnt/workspace/calculator.py
- 実装した関数:
  1. add(a, b) - 2つの数値の足し算
  2. subtract(a, b) - 2つの数値の引き算
  3. multiply(a, b) - 2つの数値の掛け算

現在の/mnt/workspaceディレクトリの内容：
- calculator.py (892バイト)
- .sessions/ ディレクトリ

import boto3
import sys

client = boto3.client("bedrock-agentcore", region_name="us-east-1")

agent_arn = "arn:aws:bedrock-agentcore:us-east-1:123456789012:agent-runtime/abc123def"
session_id = "session-storage-test-001-jinno-yudai"

def run_command(command):
    response = client.invoke_agent_runtime_command(
        agentRuntimeArn=agent_arn,
        runtimeSessionId=session_id,
        contentType="application/json",
        accept="application/vnd.amazon.eventstream",
        body={
            "command": f'/bin/bash -c "{command}"',
            "timeout": 60
        }
    )
    for event in response.get("stream", []):
        if "chunk" in event:
            chunk = event["chunk"]
            if "contentDelta" in chunk:
                delta = chunk["contentDelta"]
                if delta.get("stdout"):
                    print(delta["stdout"], end="")
            if "contentStop" in chunk:
                print(f"--- exit code: {chunk['contentStop'].get('exitCode')} ---")

run_command("ls -la /mnt/workspace")
run_command("cat /mnt/workspace/calculator.py")

=== ls -la /mnt/workspace ===
total 5
drwxr-xr-x 3 root root    0 Mar 25 06:04 .
drwxr-xr-x 1 root root 4096 Mar 25 06:04 ..
drwxr-xr-x 3 root root    0 Mar 25 06:04 .sessions
-rw-r--r-- 1 root root  892 Mar 25 06:04 calculator.py

--- exit code: 0 ---

=== cat /mnt/workspace/calculator.py ===
def add(a, b):
    """
    2つの数値を足し算します

    Args:
        a (float): 第一の数値
        b (float): 第二の数値

    Returns:
        float: a + b の結果
    """
    return a + b

def subtract(a, b):
    """
    2つの数値を引き算します

    Args:
        a (float): 第一の数値
        b (float): 第二の数値

    Returns:
        float: a - b の結果
    """
    return a - b

def multiply(a, b):
    """
    2つの数値を掛け算します

    Args:
        a (float): 第一の数値
        b (float): 第二の数値

    Returns:
        float: a * b の結果
    """
    return a * b

if __name__ == "__main__":
    # テスト用のサンプルコード
    print("Calculator Test:")
    print(f"5 + 3 = {add(5, 3)}")
    print(f"10 - 4 = {subtract(10, 4)}")
    print(f"6 * 7 = {multiply(6, 7)}")
--- exit code: 0 ---

client.stop_runtime_session(
    agentRuntimeArn=agent_arn,
    runtimeSessionId=session_id
)
print("Session stopped.")

Session stopped.

=== ls -la /mnt/workspace ===
total 5
drwxr-xr-x 3 root root    0 Mar 25 06:04 .
drwxr-xr-x 1 root root 4096 Mar 25 06:05 ..
drwxr-xr-x 3 root root    0 Mar 25 06:04 .sessions
-rw-r--r-- 1 root root  892 Mar 25 06:04 calculator.py

--- exit code: 0 ---

=== cat /mnt/workspace/calculator.py ===
def add(a, b):
    """
    2つの数値を足し算します

    Args:
        a (float): 第一の数値
        b (float): 第二の数値

    Returns:
        float: a + b の結果
    """
    return a + b

def subtract(a, b):
    """
    2つの数値を引き算します
    ...（省略）
    """
    return a - b

def multiply(a, b):
    """
    2つの数値を掛け算します
    ...（省略）
    """
    return a * b

if __name__ == "__main__":
    # テスト用のサンプルコード
    print("Calculator Test:")
    print(f"5 + 3 = {add(5, 3)}")
    print(f"10 - 4 = {subtract(10, 4)}")
    print(f"6 * 7 = {multiply(6, 7)}")
--- exit code: 0 ---

result = invoke(
    "先ほど作成した calculator に divide 関数を追加してください。"
    "また、test_calculator.py というテストスクリプトも作成してください。"
)
print(result)

完了しました！以下の作業を行いました：

1. calculator.py の更新
- divide関数を追加: 除算機能を実装
- ゼロ除算エラー処理: 適切なエラーハンドリングを追加

2. test_calculator.py の作成
- 包括的なテストスイート: unittestフレームワークを使用
- テストケース:
  - 基本的な四則演算（add, subtract, multiply, divide）
  - ゼロ除算のエラーハンドリング
  - 負の数の計算
  - 小数点の計算
  - 大きな数値の計算

テスト結果: すべてのテストが成功 - 7つのテストケースすべてが通過

現在のファイル構成:
/mnt/workspace/
├── calculator.py (1,489 bytes) - 四則演算関数
└── test_calculator.py (3,758 bytes) - テストスイート

response = client.update_agent_runtime(
    agentRuntimeId=RUNTIME_ID,
    agentRuntimeArtifact=current["agentRuntimeArtifact"],
    roleArn=current["roleArn"],
    networkConfiguration=current["networkConfiguration"],
    filesystemConfigurations=[]
)

botocore.errorfactory.RuntimeClientError: An error occurred (RuntimeClientError)
when calling the InvokeAgentRuntime operation:

"errorType": "PermissionError",
"errorMessage": "[Errno 13] Permission denied: '/mnt/workspace'"