-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Open
Labels
agent engine[Component] This issue is related to Agent Engine deployment[Component] This issue is related to Agent Engine deploymentbot triaged[Bot] This issue is triaged by ADK bot[Bot] This issue is triaged by ADK bot
Description
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
Labels
agent engine[Component] This issue is related to Agent Engine deployment[Component] This issue is related to Agent Engine deploymentbot triaged[Bot] This issue is triaged by ADK bot[Bot] This issue is triaged by ADK bot