使用 Step Functions 调用亚马逊 Bedrock AgentCore 安全带

你可以将 Step Functions 与 Amazon Bedrock AgentCore 集成，从状态机调用安全带。Harness 是一种托管运行时，用于协调模型推理、工具使用和多回合对话。在 Workflow Studio 中，搜索AgentCore InvokeHarness以找到此状态并将其拖到您的工作流程中。

在配置面板中，您可以使用 Quick Create Harness 创建新的安全带和执行角色，也可以选择现有的线束 ARN。使用现有线束时，可以按每次调用覆盖配置——任务状态定义中的值会覆盖线束默认值。有关可用参数，请参阅InvokeHarness《亚马逊 Bedrock AgentCore API 参考》。有关线束执行角色的更多信息，请参阅 Amazon Bedrock AgentCore 开发者指南中的运行时权限。

要了解如何在 Step Functions 中与 AWS 服务集成，请参阅集成 服务和在 Step Functions 中将参数传递给服务 API。

优化的亚马逊 Bedrock AgentCore 安全带 API

支持以下 API：

InvokeHarness

调用安全带来运行 AI 代理，该代理可以使用工具、访问内存和执行多回合对话。

支持的模式：仅限请求响应。

有关完整的请求语法，请参阅InvokeHarness《Amazon Bedrock AgentCore API 参考》。

响应字段

响应语法

{
  "Output": {
    "Message": {
      "Role": "string",
      "Content": [
        {
          "Text": "string"
        }
      ]
    }
  },
  "StopReason": "string",
  "Usage": {
    "InputTokens": long,
    "OutputTokens": long,
    "TotalTokens": long
  },
  "Metrics": {
    "LatencyMs": long
  }
}

Amazon Bedrock AgentCore 集成的任务状态定义

以下示例说明如何定义调用 Amazon Bedro AgentCore ck 工具的任务状态。

该RuntimeSessionId字段用于标识对话会话。在所有调用中使用相同的会话 ID 继续对话。

{
  "Type": "Task",
  "Resource": "arn:aws:states:::bedrockagentcore:invokeHarness",
  "Arguments": {
    "HarnessArn": "arn:aws:bedrock-agentcore:us-east-1:123456789012:harness/my-agent-harness",
    "RuntimeSessionId": "{% $uuid() %}",
    "Messages": [
      {
        "Content": [{ "Text": "{% $states.input.userMessage %}" }],
        "Role": "user"
      }
    ],
    "SystemPrompt": [{ "Text": "You are a helpful customer service agent." }],
    "Model": {
      "BedrockModelConfig": {
        "Temperature": 0.7,
        "ModelId": "global.anthropic.claude-sonnet-4-6"
      }
    },
    "MaxIterations": 75,
    "TimeoutSeconds": 600
  },
  "End": true
}

{
  "Type": "Task",
  "Resource": "arn:aws:states:::bedrockagentcore:invokeHarness",
  "Arguments": {
    "HarnessArn": "arn:aws:bedrock-agentcore:us-east-1:123456789012:harness/order-agent",
    "RuntimeSessionId": "{% $uuid() %}",
    "Messages": [
      {
        "Content": [{ "Text": "What is the status of order #12345?" }],
        "Role": "user"
      }
    ],
    "Tools": [
      {
        "Type": "agentcore_browser",
        "Name": "aws_browser_v1",
        "Config": {
          "AgentCoreBrowser": {
            "BrowserArn": "arn:aws:bedrock-agentcore:us-east-1:aws:browser/aws.browser.v1"
          }
        }
      }
    ],
    "MaxIterations": 10,
    "TimeoutSeconds": 300
  },
  "End": true
}

错误处理

InvokeHarnessAPI 可能会因各种错误而失败，包括限制、验证和拒绝访问错误。有关完整列表，请参阅 Amazon Bedrock AgentCore API 参考中的InvokeHarness错误。

以下示例显示了带有Retry和Catch字段的任务状态，用于处理错误：

{
  "Type": "Task",
  "Resource": "arn:aws:states:::bedrockagentcore:invokeHarness",
  "Arguments": {
    "HarnessArn": "arn:aws:bedrock-agentcore:us-east-1:123456789012:harness/my-harness",
    "Messages": [
      {
        "Content": [{ "Text": "{% $states.input.userMessage %}" }],
        "Role": "user"
      }
    ]
  },
  "Retry": [
    {
      "ErrorEquals": ["BedrockAgentCore.ThrottlingException"],
      "IntervalSeconds": 2,
      "MaxAttempts": 3,
      "BackoffRate": 2.0
    }
  ],
  "Catch": [
    {
      "ErrorEquals": ["BedrockAgentCore.ResourceNotFoundException"],
      "Next": "HandleNotFound"
    },
    {
      "ErrorEquals": ["States.ALL"],
      "Next": "HandleError"
    }
  ],
  "End": true
}

调用 Amazon Bedrock 的 IAM 政策 AgentCore

以下示例模板展示了如何根据状态机定义中的资源 AWS Step Functions 生成 IAM 策略。有关更多信息，请参阅Step Functions 如何为集成服务生成 IAM 策略和探索 Step Functions 中的服务集成模式。

Amazon Bedrock AgentCore 集成的 IAM 策略示例

以下示例展示了如何为您的 Step Functions 执行角色创建 IAM 策略以与 Amazon Bedrock AgentCore 资源进行交互。

在以下策略示例中，使用您自己的值替换占位符值。

用于调用特定工具的 IAM 策略

以下示例策略允许通过 ARN 调用特定的 Amazon Bedrock AgentCore 安全带。

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Sid": "InvokeSpecificHarness",
            "Effect": "Allow",
            "Action": [
                "bedrock-agentcore:InvokeHarness",
                "bedrock-agentcore:InvokeAgentRuntime"
            ],
            "Resource": "arn:aws:bedrock-agentcore:region:accountId:harness/harnessName"
        }
    ]
}

用于调用账户中所有安全带的 IAM 策略

以下示例政策允许在您的账户中调用任何 Amazon Bedrock AgentCore 安全带。我们建议尽可能将范围缩小到特定的安全带 ARN。

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Sid": "InvokeAllHarnesses",
            "Effect": "Allow",
            "Action": [
                "bedrock-agentcore:InvokeHarness",
                "bedrock-agentcore:InvokeAgentRuntime"
            ],
            "Resource": "arn:aws:bedrock-agentcore:region:accountId:harness/*"
        }
    ]
}