Skip to content

Cloud Build Worker Pool Integration #2141

@base698

Description

@base698

Is your feature request related to a problem? Please describe.

When deploying Agent Engines because there’s no way to specify a custom Cloud Build worker pool. This prevents deploying in environments that require private worker pools for security, compliance, or network connectivity to private resources.

Describe the solution you’d like

Add an optional worker_pool parameter to agent_engines.create():

from vertexai import agent_engines

remote_app = agent_engines.create(
    agent_engine=root_agent,
    requirements=[
        "google-cloud-aiplatform[adk,agent_engines]",
    ],
    worker_pool="projects/my-project/locations/us-central1/workerPools/my-private-pool"
)

This should accept the full worker pool resource name and pass it to the underlying Cloud Build process.

Describe alternatives you’ve considered

  • Manual Cloud Build configuration (loses Agent Engines integration)

Metadata

Metadata

Assignees

No one assigned

    Labels

    agent engine[Component] This issue is related to Agent Engine deploymentbot triaged[Bot] This issue is triaged by ADK bot

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions