-
Notifications
You must be signed in to change notification settings - Fork 383
Description
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
andclient.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