Cross-account acceso a la memoria

Amazon Bedrock AgentCore Memory admite el acceso entre cuentas, lo que le permite crear arquitecturas de varias cuentas en las que los recursos de memoria y los agentes de consumo abarquen varias cuentas. AWS Cross-account el acceso cubre dos escenarios:

Requisitos previos

Antes de configurar el acceso a la memoria entre cuentas, compruebe que dispone de lo siguiente:

Cross-account acceso al plano de datos

Puede permitir que los directores de otra cuenta llamen a las API del plano de datos de memoria directamente desde su recurso de memoria. Esto se configura adjuntando una política basada en recursos a la memoria mediante la API. PutResourcePolicy Para obtener más información sobre las políticas basadas en recursos, consulte las Resource-based políticas de Amazon Bedrock. AgentCore

Funcionamiento

Acciones admitidas

Puedes conceder acceso entre cuentas para cualquier acción del plano de datos de memoria. En la siguiente tabla se enumeran las acciones disponibles:

Adjunte una política basada en recursos a una memoria

Otorga una sola acción a otra cuenta

El siguiente ejemplo otorga permiso a la cuenta B (<account-B-id>) para invocar BatchCreateMemoryRecords un recurso de memoria de la cuenta A (<account-A-id>):

aws bedrock-agentcore-control put-resource-policy \
    --region us-east-1 \
    --resource-arn "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>" \
    --policy '{
        "Version": "2012-10-17",
        "Statement": [
            {
                "Sid": "AllowCrossAccountBatchCreate",
                "Effect": "Allow",
                "Principal": {
                    "AWS": "arn:aws:iam::<account-B-id>:root"
                },
                "Action": "bedrock-agentcore:BatchCreateMemoryRecords",
                "Resource": "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>"
            }
        ]
    }'

Otorga varias acciones a otra cuenta

El siguiente ejemplo otorga a la cuenta B pleno acceso de lectura y escritura a los registros y eventos de la memoria:

aws bedrock-agentcore-control put-resource-policy \
    --region us-east-1 \
    --resource-arn "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>" \
    --policy '{
        "Version": "2012-10-17",
        "Statement": [
            {
                "Sid": "AllowCrossAccountMemoryReadWrite",
                "Effect": "Allow",
                "Principal": {
                    "AWS": "arn:aws:iam::<account-B-id>:root"
                },
                "Action": [
                    "bedrock-agentcore:CreateEvent",
                    "bedrock-agentcore:GetEvent",
                    "bedrock-agentcore:ListEvents",
                    "bedrock-agentcore:ListActors",
                    "bedrock-agentcore:ListSessions",
                    "bedrock-agentcore:BatchCreateMemoryRecords",
                    "bedrock-agentcore:BatchUpdateMemoryRecords",
                    "bedrock-agentcore:BatchDeleteMemoryRecords",
                    "bedrock-agentcore:ListMemoryRecords",
                    "bedrock-agentcore:RetrieveMemoryRecords",
                    "bedrock-agentcore:GetMemoryRecord",
                    "bedrock-agentcore:DeleteMemoryRecord"
                ],
                "Resource": "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>"
            }
        ]
    }'

Conceda acceso a una función de IAM específica

Para seguir el principio del privilegio mínimo, concede acceso a un rol específico en lugar de a toda la cuenta:

aws bedrock-agentcore-control put-resource-policy \
    --region us-east-1 \
    --resource-arn "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>" \
    --policy '{
        "Version": "2012-10-17",
        "Statement": [
            {
                "Sid": "AllowSpecificRoleReadAccess",
                "Effect": "Allow",
                "Principal": {
                    "AWS": "arn:aws:iam::<account-B-id>:role/AgentMemoryReaderRole"
                },
                "Action": [
                    "bedrock-agentcore:RetrieveMemoryRecords",
                    "bedrock-agentcore:ListMemoryRecords",
                    "bedrock-agentcore:GetMemoryRecord"
                ],
                "Resource": "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>"
            }
        ]
    }'

Elimine una política basada en recursos

Para revocar el acceso al plano de datos entre cuentas, elimine la política basada en recursos:

aws bedrock-agentcore-control delete-resource-policy \
    --region us-east-1 \
    --resource-arn "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>"

Tras eliminar la política, cualquier solicitud posterior entre cuentas devolverá un. AccessDeniedException

Llame a las API del plano de datos desde la cuenta B

Una vez adjunta la política basada en recursos, un director de la cuenta B puede llamar a las API del plano de datos especificando el ARN de memoria completa como. memory-id

Cree registros de memoria a partir de la cuenta B:

aws bedrock-agentcore batch-create-memory-records \
    --region us-east-1 \
    --memory-id "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>" \
    --records '[
      {
        "requestIdentifier": "rec-001",
        "content": {"text": "User prefers morning meeting times before 10 AM"},
        "namespaces": ["preferences/user-123"],
        "timestamp": "1729525989"
      }
    ]'

Recupere los registros de memoria de la cuenta B:

aws bedrock-agentcore retrieve-memory-records \
    --region us-east-1 \
    --memory-id "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>" \
    --namespace "preferences/user-123" \
    --search-criteria '{"searchQuery": "meeting preferences"}' \
    --max-results 10

Crea eventos desde la cuenta B:

aws bedrock-agentcore create-event \
    --region us-east-1 \
    --memory-id "arn:aws:bedrock-agentcore:us-east-1:<account-A-id>:memory/<memory-id>" \
    --actor-id "agent-in-account-b" \
    --session-id "cross-account-session-1" \
    --event-timestamp "$(date -u +"%Y-%m-%dT%H:%M:%S.%3NZ")" \
    --payload '[
      {
        "conversational": {
          "content": {"text": "Schedule my standup for 9 AM tomorrow"},
          "role": "USER"
        }
      }
    ]'

Cross-account destinos de entrega

Cuando crea una memoria con una estrategia personalizada (autogestionada) o una configuración de entrega de transmisiones, Amazon Bedrock AgentCore utiliza una función de ejecución de memoria para entregar las cargas útiles a los recursos de destino. Cuando esos recursos residen en una cuenta diferente, debe configurar ambas partes: la función de ejecución de la cuenta A necesita permisos para acceder a los recursos, y los recursos de la cuenta B necesitan políticas que permitan el acceso desde la cuenta A.

Funcionamiento

Configure la función de ejecución de la memoria

Cree la función de ejecución de memoria en la cuenta A. La función requiere una política de confianza que permita AgentCore a Amazon Bedrock asumirla y una política de permisos que conceda acceso a los recursos de destino en la cuenta B.

Política de confianza:

{
"Version": "2012-10-17",
  "Statement": [
    {
      "Effect": "Allow",
      "Principal": {
        "Service": "bedrock-agentcore.amazonaws.com"
      },
      "Action": "sts:AssumeRole",
      "Condition": {
        "ArnLike": {
          "aws:SourceArn": "arn:aws:bedrock-agentcore:<region>:<account-A-id>:memory/*"
        }
      }
    }
  ]
}

Política de permisos (incluya únicamente las declaraciones pertinentes a los recursos que esté utilizando):

{
"Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "AllowS3Access",
      "Effect": "Allow",
      "Action": [
        "s3:PutObject",
        "s3:GetObject"
      ],
      "Resource": "arn:aws:s3:::<bucket-name-in-account-B>/*"
    },
    {
      "Sid": "AllowSNSPublish",
      "Effect": "Allow",
      "Action": "sns:Publish",
      "Resource": "arn:aws:sns:<region>:<account-B-id>:<topic-name>"
    },
    {
      "Sid": "AllowKinesisAccess",
      "Effect": "Allow",
      "Action": [
        "kinesis:PutRecords",
        "kinesis:DescribeStream"
      ],
      "Resource": "arn:aws:kinesis:<region>:<account-B-id>:stream/<stream-name>"
    }
  ]
}

Configure la política de bucket de S3 en la cuenta B

Agrega la siguiente política basada en recursos al depósito de S3 de la cuenta B para permitir que la función de ejecución de memoria de la cuenta A entregue cargas útiles:

{
"Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "AllowMemoryExecutionRoleAccess",
      "Effect": "Allow",
      "Principal": {
        "AWS": "arn:aws:iam::<account-A-id>:role/<memory-execution-role-name>"
      },
      "Action": [
        "s3:PutObject",
        "s3:GetObject"
      ],
      "Resource": "arn:aws:s3:::<bucket-name>/*"
    }
  ]
}

Configure la política temática de SNS en la cuenta B

Agregue la siguiente política basada en recursos al tema de SNS en la cuenta B para permitir que la función de ejecución de memoria de la cuenta A publique las notificaciones:

{
"Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "AllowMemoryExecutionRolePublish",
      "Effect": "Allow",
      "Principal": {
        "AWS": "arn:aws:iam::<account-A-id>:role/<memory-execution-role-name>"
      },
      "Action": "sns:Publish",
      "Resource": "arn:aws:sns:<region>:<account-B-id>:<topic-name>"
    }
  ]
}

Configurar la política de transmisión de datos de Kinesis en la cuenta B

Añada la siguiente política basada en recursos al flujo de datos de Kinesis en la cuenta B para permitir que la función de ejecución de memoria de la cuenta A transmita eventos:

{
"Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "AllowMemoryExecutionRolePutRecords",
      "Effect": "Allow",
      "Principal": {
        "AWS": "arn:aws:iam::<account-A-id>:role/<memory-execution-role-name>"
      },
      "Action": [
        "kinesis:PutRecords",
        "kinesis:DescribeStream"
      ],
      "Resource": "arn:aws:kinesis:<region>:<account-B-id>:stream/<stream-name>"
    }
  ]
}

Cree una memoria con S3 y SNS multicuenta

Tras configurar la función de ejecución y las políticas de recursos, cree una memoria en la cuenta A que haga referencia a los recursos entre cuentas de la cuenta B:

aws bedrock-agentcore-control create-memory \
    --region us-east-1 \
    --name "cross-account-memory" \
    --description "Memory with cross-account S3 and SNS delivery" \
    --event-expiry-duration 30 \
    --memory-execution-role-arn "arn:aws:iam::<account-A-id>:role/MemoryCrossAccountRole" \
    --memory-strategies '[
      {
        "customMemoryStrategy": {
          "name": "cross_account_strategy",
          "configuration": {
            "selfManagedConfiguration": {
              "triggerConditions": [
                {"messageBasedTrigger": {"messageCount": 5}}
              ],
              "invocationConfiguration": {
                "topicArn": "arn:aws:sns:us-east-1:<account-B-id>:memory-notifications",
                "payloadDeliveryBucketName": "<bucket-name>"
              },
              "historicalContextWindowSize": 10
            }
          }
        }
      }
    ]'

Cree un recuerdo con la transmisión multicuenta de Kinesis

aws bedrock-agentcore-control create-memory \
    --region us-east-1 \
    --name "cross-account-streaming-memory" \
    --description "Memory with cross-account Kinesis streaming" \
    --event-expiry-duration 30 \
    --memory-execution-role-arn "arn:aws:iam::<account-A-id>:role/MemoryCrossAccountRole" \
    --stream-delivery-resources '{
      "resources": [
        {
          "kinesis": {
            "dataStreamArn": "arn:aws:kinesis:us-east-1:<account-B-id>:stream/memory-record-stream",
            "contentConfigurations": [
              {
                "type": "MEMORY_RECORDS",
                "level": "FULL_CONTENT"
              }
            ]
          }
        }
      ]
    }'

Prácticas recomendadas