使用 IAM Conditions 控制对记忆库的访问权限

记忆库可让您根据记忆和记忆修订版本资源的范围属性控制对这些资源的访问权限。您可以使用 IAM Conditions 定义精细的访问权限控制，例如根据通用表达式语言中定义的条件限制特定用户或群组的访问权限。

本文档介绍了如何为项目中的记忆和修订版本创建和应用条件访问权限政策。

如需简要了解权限，请参阅查看必要的角色。

概览

借助 IAM Conditions，您可以仅在满足指定条件时才授予对记忆和记忆修订版本资源的访问权限。您可以使用 API 属性 "aiplatform.googleapis.com/memoryScope"（其中包含用通用表达式语言编写的表达式），根据记忆资源中的 scope 字段来控制对记忆的访问权限。范围是一个任意字典，在创建或生成记忆时提供，例如 {'user_id': '123'}，可让您整理哪些记忆属于哪个群组。

这些条件式 Identity and Access Management 政策在项目级层创建，适用于项目中的所有记忆。您可以将 IAM 条件应用于各种主账号，包括项目的用户和服务账号。

IAM Conditions 条件可用于同时授予多个相关 Memory Bank 资源（包括尚不存在的资源）的 Identity and Access Management (IAM) 访问权限。您可以限制对记忆和修订版本的访问权限，以便用户只能访问自己的信息，或者开发者只能查看某些记忆库资源，而无需获得特殊权限。

准备工作

如需为记忆和记忆修订版本设置条件 IAM 政策，请执行以下操作：

• 查看 IAM Conditions：熟悉 IAM Conditions 概览。
• 确定必要的角色：确定哪些专用 Memory Bank IAM 角色适合您的应用场景，以确保遵循最小权限原则。

• 确定受影响的主账号：确定组织中哪些人应获得哪些权限。例如，应该考虑以下事项：

  • 开发者是否应该能够查看所有记忆。
  • 项目管理员是否应该能够查看所有回忆。
  • 某些代理身份是否只能访问某些记忆。
• 授予 IAM 角色：确保您拥有执行本文档中的任务所需的必要权限。

  如需获得将 IAM Condition 应用于 Agent Platform Memory Bank 资源所需的权限，请让管理员向您授予以下 IAM 角色：

  • 对于项目：Project IAM Admin (`roles/resourcemanager.projectIamAdmin`)

  如需详细了解如何授予角色，请参阅管理对项目、文件夹和组织的访问权限。

  这些预定义角色包含将 IAM Conditions 应用于 Agent Platform Memory Bank 资源所需的权限。如需查看所需的确切权限，请展开所需权限部分：

  所需权限

  您需要具备以下权限才能将 IAM Condition 应用于 Agent Platform 记忆库资源：

  • 在项目级层设置条件 IAM 访问权限：`resourcemanager.projects.setIamPolicy`

  您也可以使用自定义角色或其他预定义角色来获取这些权限。

  如果您计划在整个组织中使用 IAM Conditions 条件，则还需要管理组织政策的权限。

为记忆功能创建条件访问权限

通过在项目级层向 IAM 政策绑定添加条件，可以授予对记忆的有条件访问权限。该条件使用 api.getAttribute('aiplatform.googleapis.com/memoryScope', {}) 函数来检查内存资源的作用域映射。您可以在创建或生成记忆时定义范围。

如需查看有关创建具有条件的 IAM 政策的详细指南，请参阅允许政策中的条件。

如需向主账号授予单个角色，请执行以下操作：

resource "google_project_iam_member" "example" {
  project    = "PROJECT_ID"
  role       = "ROLE"
  member     = "MEMBER"
  condition {
    title       = "Memory Access Condition"
    description = "IAM condition for Memory Bank"
    expression  = "CONDITION"
  }
}

范围级权限的最佳实践

为记忆库构建条件时，请遵循以下最佳实践：

1. 使用专门的记忆库角色：记忆库 IAM Condition 应仅用于适用于记忆和记忆修订版本的角色。您可以使用 aiplatform.googleapis.com/memoryViewer、aiplatform.googleapis.com/memoryEditor 和 aiplatform.googleapis.com/memoryUser 等专用角色来防止过度授权。如需了解详情，请参阅专用记忆库 IAM 角色。

2. 使用肯定条件：我们建议在 aiplatform.googleapis.com/memoryScope 上使用肯定条件（例如检查是否相等或是否存在键值对），以提高准确性。由于不受支持的类型和服务由空范围表示，因此否定条件（例如检查不相等）可能会无意中匹配大量资源，这可能会过于宽松，并导致意外授予权限。

3. 尽可能缩短条件：我们建议在 IAM Condition 表达式中使用最短、最简单的逻辑，尤其是在您计划使用大量条件时。IAM 允许政策有大小限制，简化条件可避免达到这些 IAM 限制。如需了解详情，请参阅限制。例如，您可以省略检查表达式 'user_id' in api.getAttribute('aiplatform.googleapis.com/memoryScope', {}) && api.getAttribute('aiplatform.googleapis.com/memoryScope', {})['user_id'] == 'userA' 中是否存在键，而改用较短的语句 api.getAttribute('aiplatform.googleapis.com/memoryScope', {})['user_id'] == 'userA'。如果缺少 user_id 键，则较长表达式的求值结果为 false，而较短表达式会导致错误。由于 IAM Condition 表达式中的错误会导致访问被拒，因此两个表达式会产生相同的结果。

专用记忆库 IAM 角色

使用 IAM Condition 时，请务必防止过于宽松的 IAM 政策。下表列出了在为 Memory Bank API 授予条件式 IAM 角色时可以使用的专用角色：

将 IAM Conditions 与记忆库搭配使用

本部分介绍了将 IAM Conditions 与 Memory Bank 搭配使用的以下示例：

1. 授予对范围完全匹配的回忆的读取权限。

2. 向范围包含特定键值对的回忆授予写入权限。

3. 授予对范围包含特定键的记忆的完整访问权限。

4. 向范围包含特定前缀的记忆授予完整访问权限。

5. 向具有一组允许值的键范围授予对记忆的完整访问权限。

授予对范围完全匹配的记忆的读取权限

以下条件仅向个人 userA@gmail.com 授予对具有确切范围 {"userId": "userA"} 的记忆的查看权限。

这意味着，只要范围正好是 {"userId": "userA"}，该成员就能够获取和检索记忆，以及列出和获取这些记忆的修订版本。用户无权访问具有 {'userId': 'userA', 'source': 'ADK'} 等范围的记忆。

{
  "members": ["user:userA@gmail.com"],
  "role": "roles/aiplatform.memoryViewer",
  "condition": {
    "title": "Memory Access Condition",
    "expression": "api.getAttribute('aiplatform.googleapis.com/memoryScope', {}) == {'userId': 'userA'}"
  }
}

向范围包含特定键值对的回忆授予写入权限

以下条件向个人 developerA@corp.com 授予对包含键值对 'userId': 'userA' 的任何记忆的编辑权限。

这意味着用户能够创建、更新、删除和生成记忆，以及创建和回滚这些记忆的修订版本，其范围包括 {'userId': 'userA'} 和 {'userId': 'userA', 'source': 'ADK'}。

{
  "members": ["user:developerA@corp.com"],
  "role": "roles/aiplatform.memoryEditor",
  "condition": {
    "title": "Memory Access Condition",
    "expression": "api.getAttribute('aiplatform.googleapis.com/memoryScope', {})['userId'] == 'userA'"
  }
}

向范围包含特定键的记忆授予完整访问权限

以下条件向群组 group:engineering@corp.com 用户（查看者和编辑者）授予对具有键 'admin_override' 或 'public_access_flag' 的记忆的访问权限。

这意味着该群组的成员对具有 {'admin_override': 'true'}、{'admin_override': 'true', 'public_access_flag': 'false'} 和 {'userId': 'userA', 'public_access_flag': 'false'} 等范围的记忆具有完整的读写权限。

{
  "members": ["group:engineering@corp.com"],
  "role": "roles/aiplatform.memoryUser",
  "condition": {
    "title": "Memory Access Condition",
    "expression": "('admin_override' in api.getAttribute('aiplatform.googleapis.com/memoryScope', {})) || ('public_access_flag' in api.getAttribute('aiplatform.googleapis.com/memoryScope', {}))"
  }
}

授予对范围包含特定前缀的记忆的完整访问权限

以下条件向群组 group:engineering@corp.com 用户（查看者和编辑者）授予对具有键 'userId' 且值以 'user' 开头的记忆的访问权限。您可以使用 'startsWith' 进行前缀检查，使用 'endsWith' 进行后缀检查。

这意味着该群组的成员对范围为 {'userId': 'userA'}、{'userId': 'userB', 'public_access_flag': 'false'} 等的记忆具有完整的读写权限。

{
  "members": ["group:engineering@corp.com"],
  "role": "roles/aiplatform.memoryUser",
  "condition": {
    "title": "Memory Access Condition",
    "expression": "api.getAttribute('aiplatform.googleapis.com/memoryScope', {})['userId'].startsWith('user')"
  }
}

授予对具有一组允许值的键范围内的记忆的完整访问权限

以下条件向群组 group:engineering@corp.com 用户（查看者和编辑者）授予对具有键 'userId' 且值为 'userA' 或 'userB' 的记忆的访问权限。

这意味着该群组的成员对范围为 {'userId': 'userA'}、{'userId': 'userB', 'public_access_flag': 'false'} 等的记忆具有完整的读写权限。

{
  "members": ["group:engineering@corp.com"],
  "role": "roles/aiplatform.memoryUser",
  "condition": {
    "title": "Memory Access Condition",
    "expression": "api.getAttribute('aiplatform.googleapis.com/memoryScope', {})['userId'] in ['userA', 'userB']"
  }
}

限制

• 主账号限制：IAM 政策最多只能包含 1,500 个唯一主账号。您可以使用 Google 群组来管理此限制。如需了解详情，请参阅针对所有主账号的限制。

• API 支持：针对多个范围运行的方法（例如 ListMemories 和 PurgeMemories）不支持 IAM Condition。如需授予这些权限，您必须向主账号授予无条件角色，例如 aiplatform.googleapis.com/memoryViewer、aiplatform.googleapis.com/memoryUser 或相关的无条件 Gemini Enterprise Agent Platform 角色。