自訂模型匯入的程式碼範例

建立 Amazon S3 政策以存取內含模型檔案的 Amazon S3 儲存貯體

1. 導覽至位於 https://console.aws.amazon.com/iam 的 IAM 主控台，然後從左側導覽窗格中選擇政策。

2. 選取建立政策，然後選擇 JSON 以開啟政策編輯器。

3. 貼上下列政策，將 ${model-file-bucket} 取代為儲存貯體名稱，然後選取下一步。

   JSON

   JSON

   {
       "Version":"2012-10-17",		 	 	 
       "Statement": [
           {
               "Effect": "Allow",
               "Action": [
                   "s3:GetObject",
                   "s3:ListBucket"
               ],
               "Resource": [
                   "arn:aws:s3:::${model-file-bucket}",
                   "arn:aws:s3:::${model-file-bucket}/*"
               ]
             }
           ]
         }
   

4. 將政策命名為 S3BucketPolicy，然後選取建立政策。

建立 IAM 角色，並連接該政策。

1. 在左側導覽窗格中，選擇角色，然後選取建立角色。

2. 選取自訂信任政策，貼上下列政策，然後選取下一步。

   JSON

   JSON

   {
       "Version":"2012-10-17",		 	 	 
       "Statement": [
           {
               "Effect": "Allow",
               "Principal": {
                   "Service": "bedrock.amazonaws.com"
               },
               "Action": "sts:AssumeRole"
           }
       ] 
   }
   

3. 搜尋您建立的 S3BucketPolicy 政策、選取核取方塊，然後選擇下一步。

4. 將角色命名為 MyImportModelRole，然後選取建立角色。

建立名為 BedrockTrust.json 的檔案，並將下列政策貼入其中。

建立另一個名為 S3BucketPolicy.json 的檔案，並將下列政策貼入其中，以儲存貯體名稱取代 ${model-file-bucket}。

在終端中，瀏覽至包含您建立之政策的資料夾。

提出 CreateRole 請求，以建立名為 MyImportModelRole 的 IAM 角色，以及連接您建立的 BedrockTrust.json 信任政策。

aws iam create-role \
    --role-name MyImportModelRole \
    --assume-role-policy-document file://BedrockTrust.json

提出 CreatePolicy 請求，使用您建立的 S3BucketPolicy.json 檔案建立 S3 資料存取政策。回應會傳回政策的 Arn。

aws iam create-policy \
    --policy-name S3BucketPolicy \
    --policy-document file://S3BucketPolicy.json

提出 AttachRolePolicy 請求，將 S3 資料存取政策連接至角色，在上一個步驟的回應中將 policy-arn 取代為 ARN：

aws iam attach-role-policy \
    --role-name MyImportModelRole \
    --policy-arn ${policy-arn}

執行下列程式碼來提出 CreateRole 請求，以建立名為 MyImportModel 的 IAM 角色，並提出 CreatePolicy 請求，以建立名為 S3BucketPolicy 的 S3 資料存取政策。對於 S3 資料存取政策，請將 ${model-file-bucket} 取代為 S3 儲存貯體名稱。

import boto3
import json

iam = boto3.client("iam")

iam.create_role(
    RoleName="MyImportModelRole",
    AssumeRolePolicyDocument=json.dumps({
        "Version": "2012-10-17",		 	 	 
        "Statement": [
            {
                "Effect": "Allow",
                "Principal": {
                    "Service": "bedrock.amazonaws.com"
                },
                "Action": "sts:AssumeRole"
            }
        ] 
    })
)

iam.create_policy(
    PolicyName="S3BucketPolicy",
    PolicyDocument=json.dumps({
        "Version": "2012-10-17",		 	 	 
        "Statement": [
            {
                "Effect": "Allow",
                "Action": [
                    "s3:GetObject",
                    "s3:ListBucket"
                ],
                "Resource": [
                    "arn:aws:s3:::${training-bucket}",
                    "arn:aws:s3:::${training-bucket}/*"
                ]
            }
          ]
       })
     )

回應中會傳回 Arn。執行下列程式碼片段來提出 AttachRolePolicy 請求，以傳回的 Arn 取代 ${policy-arn}。

iam.attach_role_policy(
    RoleName="MyImportModelRole",
    PolicyArn="${policy-arn}"
)