AI エージェントのツールとして Step Functions ワークフローを使用する

AWS Step Functions コンソールを開き、[Create a state machine (ステートマシンの作成)] を選択します。

空白 を選択し、選択 を選択します。

Code を選択し、定義を以下に置き換えます。

{
  "Comment": "A workflow that an AI agent starts through AgentCore Gateway",
  "StartAt": "Hello",
  "States": {
    "Hello": {
      "Type": "Pass",
      "Result": "Hello from Step Functions",
      "End": true
    }
  }
}

Config を選択します。ステートマシン名には、 と入力しますAgentHelloWorld。タイプを標準のままにします。

Create を選択してステートマシンを作成します。

プロンプトが表示されたら、Step Functions に実行ロールの作成を許可します。 AWS CLI 代わりに を使用する場合は、次の手順に示すように、ロールを自分で作成します。

ステートマシンの詳細ページで、次の例に示すように ARN を書き留めます。

arn:aws:states:region:123456789012:stateMachine:AgentHelloWorld

Step Functions が引き受けることができる実行ロールを作成します。リージョンとアカウント ID を独自の ID に置き換えます。

aws iam create-role \
  --role-name AgentHelloWorldExecutionRole \
  --assume-role-policy-document '{
    "Version": "2012-10-17",
    "Statement": [
      {
        "Effect": "Allow",
        "Principal": { "Service": "states.amazonaws.com" },
        "Action": "sts:AssumeRole",
        "Condition": {
          "ArnLike": {
            "aws:SourceArn": "arn:aws:states:region:123456789012:stateMachine:*"
          },
          "StringEquals": {
            "aws:SourceAccount": "123456789012"
          }
        }
      }
    ]
  }'

コンソールの手順に示されている定義を という名前のファイルに保存しdefinition.json、ステートマシンを作成します。

aws stepfunctions create-state-machine \
  --name AgentHelloWorld \
  --type STANDARD \
  --definition file://definition.json \
  --role-arn "arn:aws:iam::123456789012:role/AgentHelloWorldExecutionRole" \
  --region region

レスポンスには が含まれますstateMachineArn。ステップ 2 と 3 に注意してください。

IAM コンソールを開き、ロール、ロールの作成を選択します。

信頼されたエンティティタイプで、カスタム信頼ポリシーを選択し、次のように入力します。リージョンとアカウント ID を独自の ID に置き換えます。

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Effect": "Allow",
      "Principal": { "Service": "bedrock-agentcore.amazonaws.com" },
      "Action": "sts:AssumeRole",
      "Condition": {
        "StringEquals": {
          "aws:SourceAccount": "123456789012"
        },
        "ArnLike": {
          "aws:SourceArn": "arn:aws:bedrock-agentcore:region:123456789012:gateway/step-functions-gateway-*"
        }
      }
    }
  ]
}

次へを選択し、次に管理ポリシーをアタッチせずに再度次へを選択します。

[ロール名] に「AgentCoreGatewayToStepFunctions」と入力し、[ロールの作成] を選択します。

ロールを開き、アクセス許可の追加、インラインポリシーの作成、JSON の選択、次のポリシーの入力を選択します。リージョンとアカウント ID を独自の ID に置き換えます。

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "StateMachineOperations",
      "Effect": "Allow",
      "Action": [
        "states:StartExecution",
        "states:ListExecutions",
        "states:DescribeStateMachine"
      ],
      "Resource": "arn:aws:states:region:123456789012:stateMachine:AgentHelloWorld"
    },
    {
      "Sid": "ExecutionOperations",
      "Effect": "Allow",
      "Action": [
        "states:DescribeExecution",
        "states:GetExecutionHistory",
        "states:StopExecution"
      ],
      "Resource": "arn:aws:states:region:123456789012:execution:AgentHelloWorld:*"
    }
  ]
}

ポリシーに という名前を付けStepFunctionsConnectorAccess、ポリシーの作成を選択します。

次の例に示すように、ロール ARN を書き留めます。

arn:aws:iam::123456789012:role/AgentCoreGatewayToStepFunctions

を使用してゲートウェイを作成します AWS CLI。ロール ARN をステップ 2 のものに置き換えます。

aws bedrock-agentcore-control create-gateway \
  --name "step-functions-gateway" \
  --protocol-type MCP \
  --authorizer-type AWS_IAM \
  --role-arn "arn:aws:iam::123456789012:role/AgentCoreGatewayToStepFunctions" \
  --region region

レスポンスには、 gatewayIdと が含まれますgatewayUrl。両方に注意してください。

ゲートウェイのステータスが になるまで待ちますREADY。これは数分かかることがあります。GATEWAY_ID を前のステップgatewayIdの に置き換えます。

aws bedrock-agentcore-control get-gateway \
  --gateway-identifier "GATEWAY_ID" \
  --query status --output text --region region

コネクタターゲットを追加します。configurations リストは、ツールになるオペレーションを選択します。StartExecution、、 ListExecutionsDescribeStateMachineはステートマシンに固定されます。DescribeExecution、GetExecutionHistory、 はエージェントからStopExecution実行 ARN を取得します。GATEWAY_ID をゲートウェイ ID に置き換えます。

aws bedrock-agentcore-control create-gateway-target \
  --gateway-identifier "GATEWAY_ID" \
  --name "hello" \
  --target-configuration '{
    "mcp": {
      "connector": {
        "source": { "connectorId": "states" },
        "configurations": [
          {
            "name": "StartExecution",
            "parameterValues": {
              "stateMachineArn": "arn:aws:states:region:123456789012:stateMachine:AgentHelloWorld"
            }
          },
          {
            "name": "ListExecutions",
            "parameterValues": {
              "stateMachineArn": "arn:aws:states:region:123456789012:stateMachine:AgentHelloWorld"
            }
          },
          {
            "name": "DescribeStateMachine",
            "parameterValues": {
              "stateMachineArn": "arn:aws:states:region:123456789012:stateMachine:AgentHelloWorld"
            }
          },
          { "name": "DescribeExecution" },
          { "name": "GetExecutionHistory" },
          { "name": "StopExecution" }
        ]
      }
    }
  }' \
  --credential-provider-configurations '[{"credentialProviderType": "GATEWAY_IAM_ROLE"}]' \
  --region region

ターゲットステータスが になるまで待ちますREADY。これは数分かかることがあります。TARGET_ID をcreate-gateway-targetレスポンスtargetIdの に置き換えます。

aws bedrock-agentcore-control get-gateway-target \
  --gateway-identifier "GATEWAY_ID" --target-id "TARGET_ID" \
  --query status --output text --region region

エージェントに表示されるツールを一覧表示します。

awscurl --service bedrock-agentcore --region region \
  -X POST "GATEWAY_URL" \
  -H "Content-Type: application/json" \
  -d '{"jsonrpc":"2.0","id":1,"method":"tools/list","params":{}}'

レスポンスには、入力スキーマを含む 6 つのツールが一覧表示されます。が のみをhello___StartExecutionアドバタイズすることに注意してくださいinput。ピン留めされた stateMachineArnはエージェントから非表示になり、上書きすることはできません。

実行を開始します。

awscurl --service bedrock-agentcore --region region \
  -X POST "GATEWAY_URL" \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "id": 2,
    "method": "tools/call",
    "params": {
      "name": "hello___StartExecution",
      "arguments": { "input": "{\"who\": \"agent\"}" }
    }
  }'

ツールの結果には、新しい実行の ARN を含む Step Functions レスポンスが含まれます。

{
  "jsonrpc": "2.0",
  "id": 2,
  "result": {
    "isError": false,
    "content": [
      {
        "type": "text",
        "text": "{\"executionArn\":\"arn:aws:states:region:123456789012:execution:AgentHelloWorld:a1b2c3d4-5678-90ab-cdef-EXAMPLE11111\",\"startDate\":1.7597E9}"
      }
    ]
  }
}

次のステップexecutionArnでは、 に注意してください。

実行を記述します。デフォルトでは、レスポンスにはメタデータのみが含まれます。入力と出力を含めるには、 includedData を に設定しますALL_DATA。EXECUTION_ARN をステップ 4 executionArnの に置き換えます。

awscurl --service bedrock-agentcore --region region \
  -X POST "GATEWAY_URL" \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "id": 3,
    "method": "tools/call",
    "params": {
      "name": "hello___DescribeExecution",
      "arguments": {
        "executionArn": "EXECUTION_ARN",
        "includedData": "ALL_DATA"
      }
    }
  }'

このステートマシンの場合、 statusは SUCCEEDEDで、 は outputです"Hello from Step Functions"。実行が失敗した場合、 statusは FAILEDで、レスポンスには失敗状態の フィールドerrorと causeフィールドが含まれます。

イベント履歴を読み、実行が実行した各ステップを確認します。

awscurl --service bedrock-agentcore --region region \
  -X POST "GATEWAY_URL" \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "id": 4,
    "method": "tools/call",
    "params": {
      "name": "hello___GetExecutionHistory",
      "arguments": { "executionArn": "EXECUTION_ARN" }
    }
  }'

ピン留めされたステートマシンの最近の実行を一覧表示し、オプションでステータスでフィルタリングします。

awscurl --service bedrock-agentcore --region region \
  -X POST "GATEWAY_URL" \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "id": 5,
    "method": "tools/call",
    "params": {
      "name": "hello___ListExecutions",
      "arguments": { "statusFilter": "SUCCEEDED" }
    }
  }'

(オプション) AWS Step Functions コンソールで実行を確認します。AgentHelloWorld を選択し、実行を選択します。ゲートウェイによって開始された実行は、ゲートウェイのサービスロールを呼び出し元とする他の実行のように表示されます AWS CloudTrail。

ゲートウェイターゲットを削除し、次にゲートウェイを削除します。

aws bedrock-agentcore-control delete-gateway-target \
  --gateway-identifier "GATEWAY_ID" --target-id "TARGET_ID" --region region
aws bedrock-agentcore-control delete-gateway \
  --gateway-identifier "GATEWAY_ID" --region region

Step Functions コンソールで、AgentHelloWorldステートマシンを削除します。

IAM コンソールで、AgentCoreGatewayToStepFunctionsロールとステートマシンの実行ロール (Step Functions が作成したロール、または を使用したAgentHelloWorldExecutionRole場合は ) を削除します AWS CLI。