Skip to content

AgentEngineConfig validation fails for service_account #5698

@jbolilia

Description

@jbolilia

Thanks for stopping by to let us know something could be better!

PLEASE READ: If you have a support contract with Google, please create an issue in the support console instead of filing on GitHub. This will ensure a timely response.

Is your feature request related to a problem? Please describe.
Inconsistent experience between agent_engines.create() and client.agent_engines.create() (or update). Currently unable to assign service account when using the vertexai.Client()

A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
The agent_engines approach doesn't support using context_spec which is needed for memory bank updates.

Describe the solution you'd like
Add the service_account as an arg for AgentEngineConfig and ReasoningEngineSpec

A clear and concise description of what you want to happen.

  • Add the changes to types for both the pydantic models and typedDict
  • Add the new config to the client.agent_engines.create and client.agent_engines.update arguments
  • Add handling if service_account is provided

A clear and concise description of any alternative solutions or features you've considered.

  • Tried to use the agent_engines.create() client but unable to use the context_spec inputs.
2025-08-12 08:08:08,480 - ERROR - ❌ Error during Agent Engine creation: 1 validation error for AgentEngineConfig
service_account
  Extra inputs are not permitted [type=extra_forbidden, input_value='[email protected]', input_type=str]
    For further information visit https://errors.pydantic.dev/2.11/v/extra_forbidden
Traceback (most recent call last):
  File "/path/to/my-agent/utils/managers.py", line 312, in create_or_update_agent
    remote_app = self.client.agent_engines.create(**params)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/path/to/venv/lib/python3.12/site-packages/vertexai/_genai/agent_engines.py", line 2704, in create
    config = types.AgentEngineConfig.model_validate(config)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/path/to/venv/lib/python3.12/site-packages/pydantic/main.py", line 705, in model_validate
    return cls.__pydantic_validator__.validate_python(
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
pydantic_core._pydantic_core.ValidationError: 1 validation error for AgentEngineConfig
service_account
  Extra inputs are not permitted [type=extra_forbidden, input_value='[email protected]', input_type=str]
    For further information visit https://errors.pydantic.dev/2.11/v/extra_forbidden

Metadata

Metadata

Assignees

Labels

api: vertex-aiIssues related to the googleapis/python-aiplatform API.

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions