使用 Step Functions 呼叫 Amazon Bedrock

Step Functions 可以直接從Amazon States Language（ASL）控制某些 AWS 服務。如需了解詳細資訊，請參閱 使用其他 服務 和 將參數傳遞至服務 API。

Amazon Bedrock服務整合 API

若要AWS 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
    } 
}

如需Step Functions與其他 AWS 服務搭配使用時如何設定IAM權限的相關資訊，請參閱整合式服務的 IAM 政策。

Amazon Bedrock整合的工作狀態定義

以下任務狀態定義顯示了如何在狀態機器Amazon Bedrock中進行集成。此範例顯示 Task 狀態，result_one該狀態會擷取路徑指定之模型呼叫的完整結果。這是以基礎模型的推論參數為基礎。這個例子使用 Cohere 命令大語言模型（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
}