使用 Step Functions 调用 Amazon Bedrock

Step Functions 可以直接从 Amazon States Language (ASL) 控制某些 Amazon 服务。要了解更多信息，请参阅使用其他服务和将参数传递给服务 API。

Amazon Bedrock 服务集成 API

要将 Amazon Step Functions 与 Amazon Bedrock 集成，您可以使用以下 API。这些 API 与相应的 Amazon Bedrock API 相似，但在传递的请求字段方面有所不同。

下表描述了每个服务集成 API 及其相应 Amazon Bedrock API 之间的差异。

{
    "ModelId": String,  // required
    "Accept": String,  // default: application/json
    "ContentType": String,  // default: application/json
    "Input": {  // not from Bedrock API
        "S3Uri": String
    },  
    "Output": {  // not from Bedrock API
        "S3Uri": String
    } 
}

有关在与其他 Amazon 服务Step Functions一起使用时如何配置IAM权限的信息，请参阅集成服务的 IAM 策略。

Amazon Bedrock 集成的 Task 状态定义

以下 Task 状态定义显示了如何在状态机中与 Amazon Bedrock 集成。此示例显示了一个 Task 状态，它提取了路径指定的模型调用的完整结果，result_one。这基于基础模型的推理参数。此示例使用 Cohere Command 大型语言模型 (LLM)。

{
  "Type": "Task",
  "Resource": "arn:aws:states:::bedrock:invokeModel",
  "Parameters": {
    "ModelId": "cohere.command-text-v14",
    "Body": {
      "prompt.$": "$.prompt_one",
      "max_tokens": 250
    },
    "ContentType": "application/json",
    "Accept": "*/*"
  },
  "ResultPath": "$.result_one",
  "ResultSelector": {
    "result_one.$": "$.Body.generations[0].text"
  },
  "End": true
}