Skip to content

Google Cloud Agent Platform express mode for ADK

Supported in ADKPython v0.1.0Java v0.1.0预览版

Google Cloud Agent Platform express mode 为原型设计和开发提供了免费访问层级，使你无需创建完整的 Google Cloud 项目即可使用 Agent Platform 服务。此服务包括对许多强大的 Agent Platform 服务的访问，包括：

你可以使用 Gmail 账号注册 express mode 账户，并接收用于 ADK 的 API 密钥。通过 Google Cloud Console 获取 API 密钥。有关更多信息，请参阅 Agent Platform express mode。

预览版发布

Agent Platform express mode 功能是预览版发布。有关更多信息，请参阅发布阶段说明。

Agent Platform express mode 限制

Agent Platform express mode 项目只有 90 天有效期，并且只有部分服务可用且配额有限。例如，Agent Runtime 实例的数量限制为 10 个，部署到 Agent Runtime 需要付费访问。要删除配额限制并使用所有 Agent Platform 服务，请向你的 express mode 项目添加结算账号。

配置 Agent Runtime 容器

使用 Agent Platform express mode 时，创建一个 AgentEngine 对象以启用 Agent Platform 对智能体组件（如 Session 和 Memory 对象）的管理。通过这种方法，Session 对象将作为 AgentEngine 对象的子项进行处理。在运行智能体之前，请确保你的环境变量已正确设置，如下所示：

GOOGLE_GENAI_USE_VERTEXAI=TRUE
GOOGLE_API_KEY=在此粘贴你的实际_EXPRESS_MODE_API_KEY

接下来，使用 Agent Platform SDK 创建你的 Agent Runtime 实例。

  1. 导入 Agent Platform SDK。

    import vertexai
from vertexai import agent_engines

  2. 使用你的 API 密钥初始化 Agent Platform 客户端并创建一个 Agent Engine 实例。

    # Create Agent Runtime with Gen AI SDK
client = vertexai.Client(
  api_key="你的_API_KEY",
)

agent_engine = client.agent_engines.create(
  config={
    "display_name": "Demo Agent Runtime",
    "description": "Agent Runtime for Session and Memory",
  })

  3. 从响应中获取 Agent Runtime 的名称和 ID，用于记忆和会话。

    APP_ID = agent_engine.api_resource.name.split('/')[-1]

使用 VertexAiSessionService 管理会话

VertexAiSessionService 与 Agent Platform Express 模式 API 密钥兼容。你可以初始化会话对象而无需指定任何项目或位置。

# 需要安装：pip install google-adk[vertexai]
# 并在环境变量中设置：
# GOOGLE_GENAI_USE_VERTEXAI=TRUE
# GOOGLE_API_KEY=在此粘贴你的实际_EXPRESS_MODE_API_KEY

from google.adk.sessions import VertexAiSessionService

# 此服务使用的 APP_ID 应为 Reasoning Engine 的 ID 或名称
APP_ID = "你的-reasoning-engine-id"

    # 使用 express mode 初始化时，无需指定项目和位置
session_service = VertexAiSessionService(agent_engine_id=APP_ID)

# 调用服务方法时使用 APP_ID，例如：
# session = await session_service.create_session(app_name=APP_ID, user_id= ...)

Session Service 配额

针对免费的 express mode 项目，VertexAiSessionService 拥有以下配额限制：

  • 每分钟创建、删除或更新 10 个 Agent Runtime 会话
  • 每分钟向 Agent Runtime 会话追加 30 个事件

使用 VertexAiMemoryBankService 管理记忆

VertexAiMemoryBankService 与 Agent Platform express mode API 密钥兼容。你可以初始化记忆对象而无需指定任何项目或位置。

# 需要安装：pip install google-adk[vertexai]
# 并在环境变量中设置：
# GOOGLE_GENAI_USE_VERTEXAI=TRUE
# GOOGLE_API_KEY=在此粘贴你的实际_EXPRESS_MODE_API_KEY

from google.adk.memory import VertexAiMemoryBankService

# 此服务使用的 APP_ID 应为 Reasoning Engine 的 ID 或名称
APP_ID = "你的-reasoning-engine-id"

# 在使用 express mode 初始化时，无需指定项目和位置
memory_service = VertexAiMemoryBankService(agent_engine_id=APP_ID)

# 从该会话生成记忆，以便智能体可以记住关于用户的相关详细信息
# memory = await memory_service.add_session_to_memory(session)

Memory Service 配额

针对免费的 express mode 项目，VertexAiMemoryBankService 拥有以下配额限制：

  • 每分钟创建、删除或更新 10 个 Agent Runtime 记忆资源
  • 每分钟获取、列出或检索 10 个 Agent Runtime 记忆库

代码示例：带有会话和记忆的天气智能体

此代码示例展示了一个同时利用 VertexAiSessionService 和 VertexAiMemoryBankService 进行上下文管理的天气智能体，它允许你的智能体回忆用户的偏好和对话记录。