[アップデート] AWS Step FunctionsのワークフローをAgentCore Gateway経由でAIエージェントのツールとして使えるようになりました！

git clone https://github.com/yuu551/agentcore-gateway-stepfunctions-tools.git
cd agentcore-gateway-stepfunctions-tools/cdk
pnpm install

// Step Functionsのコネクタターゲット（L2がないのでL1で書く）
const connectorTarget = (name: string, configurations: agentcore.CfnGatewayTarget.ConnectorConfigurationProperty[]) =>
  new agentcore.CfnGatewayTarget(this, `${name}Target`, {
    gatewayIdentifier: gateway.gatewayId,
    name,
    targetConfiguration: {
      mcp: { connector: { source: { connectorId: "states" }, configurations } },
    },
    credentialProviderConfigurations: [{ credentialProviderType: "GATEWAY_IAM_ROLE" }],
  });

connectorTarget("order", [
  { name: "StartExecution", parameterValues: { stateMachineArn: order.stateMachineArn } },
  { name: "ListExecutions", parameterValues: { stateMachineArn: order.stateMachineArn } },
  { name: "DescribeStateMachine", parameterValues: { stateMachineArn: order.stateMachineArn } },
  { name: "DescribeExecution" },
  { name: "GetExecutionHistory" },
  { name: "StopExecution" },
]);
connectorTarget("stock", [
  { name: "StartSyncExecution", parameterValues: { stateMachineArn: stock.stateMachineArn } },
]);

"GetStock": {
  "Type": "Task",
  "Resource": "arn:aws:states:::dynamodb:getItem",
  "Arguments": {
    "TableName": "${TABLE_NAME}",
    "Key": {
      "item": {
        "S": "{% $states.input.item %}"
      }
    }
  },
  "Output": "{% $exists($states.result.Item) ? {'item': $states.result.Item.item.S, 'stock': $number($states.result.Item.stock.N), 'unitPrice': $number($states.result.Item.unitPrice.N), 'dailySales': $number($states.result.Item.dailySales.N)} : {'item': $states.input.item, 'status': 'NOT_FOUND'} %}",
  "End": true
}

"WaitForApproval": {
  "Type": "Task",
  "Resource": "arn:aws:states:::sns:publish.waitForTaskToken",
  "TimeoutSeconds": 86400,
  "Arguments": {
    "TopicArn": "${TOPIC_ARN}",
    "Subject": "Purchase order approval request",
    "Message": "{% '発注の承認をお願いします。\n\n商品: ' & $states.input.item & '\n数量: ' & $states.input.quantity & '\n合計: ' & $states.input.quantity * $states.input.unitPrice & '円\n理由: ' & $states.input.reason & '\n\n承認・却下はこちらから\n' & '${APPROVAL_URL}?token=' & $encodeUrlComponent($states.context.Task.Token) %}"
  },
  "Output": "{% $merge([$states.input, {'approval': $states.result}]) %}",
  "Next": "IsApproved"
}

import base64
import hmac
import html
import json
import os
from urllib.parse import parse_qs

import boto3

sfn = boto3.client("stepfunctions")
PASSWORD = boto3.client("secretsmanager").get_secret_value(SecretId=os.environ["PASSWORD_SECRET_ARN"])["SecretString"]

PAGE = """<!doctype html><meta charset="utf-8"><title>発注の承認</title>
<p>今回の検証用の簡素なページです。本番用ではありません。</p>
<form method="post"><input type="hidden" name="token" value="{token}">
<button name="action" value="approve">承認する</button>
<button name="action" value="reject">却下する</button></form>"""

def handler(event, context):
    # 店長のパスワードを知っている人だけが開けるようにする（Basic認証）
    if not authorized(event["headers"].get("authorization", "")):
        return {"statusCode": 401, "headers": {"WWW-Authenticate": 'Basic realm="approval"'}, "body": "Unauthorized"}

    # リンクを開いただけでは承認せず、ボタンが押されたら結果を返す
    if event["requestContext"]["http"]["method"] == "GET":
        token = event["queryStringParameters"]["token"]
        return page(PAGE.format(token=html.escape(token)))

    body = event.get("body", "")
    if event.get("isBase64Encoded"):
        body = base64.b64decode(body).decode()
    form = {k: v[0] for k, v in parse_qs(body).items()}
    approved = form["action"] == "approve"
    sfn.send_task_success(taskToken=form["token"], output=json.dumps({"approved": approved}))
    return page("<p>承認しました</p>" if approved else "<p>却下しました</p>")

def authorized(header):
    if not header.startswith("Basic "):
        return False
    try:
        _, _, password = base64.b64decode(header.removeprefix("Basic ")).decode().partition(":")
    except ValueError:
        return False
    return hmac.compare_digest(password, PASSWORD)

def page(body):
    return {"statusCode": 200, "headers": {"Content-Type": "text/html; charset=utf-8"}, "body": body}

pnpm exec cdk deploy -c email=<店長のメールアドレス> --outputs-file outputs.json
cd ..
jq -r '.SfnGatewayToolsStack | "GATEWAY_URL=\(.GatewayUrl)\nSTOCK_ARN=\(.StockArn)"' cdk/outputs.json > .env

aws secretsmanager get-secret-value --secret-id <PasswordSecretArnの値> \
  --query SecretString --output text

uv sync
set -a; source .env; set +a

uv run list_tools.py

order___StartExecution
   {"type": "object", "properties": {"input": {"description": "The string that contains the JSON input data of the execution. ...", "type": "string"}}}
   Starts a state machine execution. Returns the executionArn to use with DescribeExecution, GetExecutionHistory and StopExecution.
stock___StartSyncExecution
   {"type": "object", "properties": {"input": {"description": "The string that contains the JSON input data of the execution. ...", "type": "string"}}}
   Starts a Synchronous Express state machine execution and waits for it to complete. Express workflows only; fails on a Standard state machine.

import json
import os
import re

from mcp_proxy_for_aws.client import aws_iam_streamablehttp_client
from strands.hooks import BeforeToolCallEvent
from strands.tools.mcp import MCPClient

def gateway_client(pattern: str) -> MCPClient:
    # Gatewayのツールのうち、名前がpatternに合うものだけをエージェントに渡す
    return MCPClient(
        lambda: aws_iam_streamablehttp_client(
            endpoint=os.environ["GATEWAY_URL"],
            aws_region="us-east-1",
            aws_service="bedrock-agentcore",
        ),
        tool_filters={"allowed": [re.compile(pattern)]},
    )

def escape_non_ascii(event: BeforeToolCallEvent):
    args = event.tool_use["input"]
    print(f"\n[{event.tool_use['name']}] {json.dumps(args, ensure_ascii=False)}")
    # ワークフローに渡すJSONの日本語を \uXXXX 形式にしてから送る
    if isinstance(args.get("input"), str):
        args["input"] = json.dumps(json.loads(args["input"]))

import sys

from strands import Agent
from strands.hooks import BeforeToolCallEvent

from gateway_tools import escape_non_ascii, gateway_client

SYSTEM_PROMPT = """あなたはスーパーの発注担当です。
- 在庫は stock___StartSyncExecution に {"item": "商品名"} を渡して確認します
- 発注は order___StartExecution に {"item": "商品名", "quantity": 発注数, "unitPrice": 単価, "reason": "発注理由"} を渡して開始します
- 発注を開始したら order___DescribeExecution で結果を確認します
"""

agent = Agent(
    model="us.anthropic.claude-haiku-5-5",
    tools=[gateway_client(r"^(stock|order)___")],
    system_prompt=SYSTEM_PROMPT,
)
agent.hooks.add_callback(BeforeToolCallEvent, escape_non_ascii)
agent(sys.argv[1])

uv run store_agent.py "牛乳の在庫が少ないので、明日の分を発注しておいて"

まず牛乳の在庫を確認します。

[stock___StartSyncExecution] {"input": "{\"item\": \"牛乳\"}"}
牛乳の在庫は3本、1日の販売数は40本、単価は180円でした。明日の販売分として40本を発注します。

[order___StartExecution] {"input": "{\"item\": \"牛乳\", \"quantity\": 40, \"unitPrice\": 180, \"reason\": \"在庫残り3本のため、明日分（1日の販売数40本）を補充\"}"}

[order___DescribeExecution] {"executionArn": "arn:aws:states:us-east-1:123456789012:execution:PurchaseOrderWorkflow:4daf7182-...", "includedData": "ALL_DATA"}
牛乳の発注が完了しました。

- **商品**: 牛乳
- **発注数**: 40本
- **単価**: 180円
- **合計**: 7,200円
- **ステータス**: ORDERED（発注済み）

uv run store_agent.py "米5kgが残りわずかなので、1週間分を発注して"

在庫を確認します。

[stock___StartSyncExecution] {"input": "{\"item\": \"米5kg\"}"}
在庫は2個、1日の販売数は8個、単価は2,980円でした。1週間分は8×7=56個なので、手持ちの2個を引いて54個を発注します。

[order___StartExecution] {"input": "{\"item\": \"米5kg\", \"quantity\": 54, \"unitPrice\": 2980, \"reason\": \"残りわずかのため1週間分を発注（1日販売8個×7日=56個、在庫2個を差し引いて54個）\"}"}

[order___DescribeExecution] {...}
米5kgを54個発注しました。現在、発注処理は実行中です。

- **在庫**: 2個（残りわずか）
- **1日の販売数**: 8個
- **発注数**: 54個（1週間分56個から在庫2個を引いた数）
- **単価**: 2,980円
- **概算金額**: 160,920円（税抜き、54個×2,980円）

uv run store_agent.py "米5kgの発注がまだ終わらないみたい。どこで止まってるか調べて"

発注の実行一覧を確認します。

[order___ListExecutions] {"statusFilter": "RUNNING"}

[order___DescribeExecution] {...}

[order___GetExecutionHistory] {"executionArn": "arn:aws:states:us-east-1:123456789012:execution:PurchaseOrderWorkflow:beb59a2e-..."}
発注は**承認待ち**で止まっています。エラーではなく、承認者の判断を待っている状態です。

**状況**
- 商品: 米5kg、数量54個、単価2,980円、合計160,920円
- 理由: 残りわずかのため1週間分を発注（1日販売8個×7日=56個、在庫2個を差し引いて54個）
- 現在のステップ: `WaitForApproval`（SNSで承認依頼を送信済み）
- 承認依頼メールの送信は 2026-10-30 09:54:31 (GMT) に完了しています。

**止まっている理由**
承認依頼メールに記載の承認・却下リンクから、承認者が判断するのを待っています。承認か却下の操作があるまで次に進めません。

**今後の動き**
- 承認されると発注が進みます。
- 却下されると発注は行われません。
- 24時間（86,400秒）以内に応答がないとタイムアウトします。

uv run store_agent.py "さっきの米5kgの発注はどうなった？"

[order___ListExecutions] {}

[order___DescribeExecution] {...}

[order___DescribeExecution] {...}
米5kgの発注は完了しています（ステータス：SUCCEEDED）。

- **商品**：米5kg
- **発注数**：54個（単価 2,980円）
- **合計金額**：160,920円
- **発注理由**：残りわずかだった1週間分を発注（1日販売数8個×7日＝56個、在庫2個を差し引いて54個）

aws stepfunctions get-execution-history --execution-arn <米5kgの実行ARN> \
  --query 'events[].[id,type,stateEnteredEventDetails.name]' --output text

4	TaskStateEntered	WaitForApproval
5	TaskScheduled	None
6	TaskStarted	None
7	TaskSubmitted	None
8	TaskSucceeded	None
9	TaskStateExited	None
10	ChoiceStateEntered	IsApproved
11	ChoiceStateExited	None
12	PassStateEntered	PlaceOrder
13	PassStateExited	None
14	ExecutionSucceeded	None

uv run call_tool.py order___StartExecution "{\"input\": \"{}\", \"stateMachineArn\": \"$STOCK_ARN\"}"

isError: True
IllegalArgumentException - Invalid request parameters: cannot set parameter(s): [/stateMachineArn]

// Cedarのポリシー
const gatewayEntity = `AgentCore::Gateway::"${gateway.gatewayArn}"`;
const cedar = (id: string, statement: string, ignoreFindings = false) =>
  new agentcore.Policy(this, id, {
    policyEngine,
    statement: agentcore.PolicyStatement.fromCedar(statement),
    validationMode: ignoreFindings ? agentcore.PolicyValidationMode.IGNORE_ALL_FINDINGS : undefined,
  });
// Cedarはデフォルト拒否なので、このGatewayのツールをIAMの呼び出し元に許可しておく
cedar("AllowGatewayTools", `permit(principal is AgentCore::IamEntity, action, resource == ${gatewayEntity});`, true);
// 発注の取り消しは店長のロールだけ（ポリシーはツールと突き合わせて検証されるので、ターゲットの後に作る）
cedar("OnlyManagerCanStop", `forbid(principal is AgentCore::IamEntity, action == AgentCore::Action::"order___StopExecution", resource == ${gatewayEntity})
  unless { principal.id like "*sfn-gateway-tools-manager*" };`).node.addDependency(orderTarget);

uv run store_agent.py "さっきの米5kgの発注、やっぱり取り消して"

[order___ListExecutions] {"statusFilter": "RUNNING"}

[order___DescribeExecution] {...}
米5kgの発注（54袋、単価2,980円、合計160,920円）はまだ実行中で、取り消せていません。

私の持っているツールには、実行中の発注を停止する機能がありません。

Tool Execution Denied: Tool call not allowed due to policy enforcement [Policy evaluation denied due to SfnGatewayToolsStackOnlyManagerCanStopFD3C1D49-...]

# サブシェルで実行し、終わったら元の認証情報に戻す
(
  CREDS=$(aws sts assume-role --role-arn <ManagerRoleArnの値> --role-session-name manager --query Credentials)
  export AWS_ACCESS_KEY_ID=$(echo $CREDS | jq -r .AccessKeyId)
  export AWS_SECRET_ACCESS_KEY=$(echo $CREDS | jq -r .SecretAccessKey)
  export AWS_SESSION_TOKEN=$(echo $CREDS | jq -r .SessionToken)
  uv run call_tool.py order___StopExecution '{"executionArn": "<米5kgの実行ARN>"}'
)

isError: False
{"stopDate":1.793360833268E9}

forbid(principal is AgentCore::IamEntity, action == AgentCore::Action::"order___StartExecution", resource == AgentCore::Gateway::"<GatewayのARN>")
  unless { context has input && context.input has input && context.input.input like "*\"reason\"*" };

Tool Execution Denied: Tool call not allowed due to policy enforcement [Authorization denied due to policy evaluation errors:
Value does not match expected type]

cd cdk
pnpm exec cdk destroy -c email=<店長のメールアドレス>