Skip to content

Commit be779ad

Browse files
sararobcopybara-github
authored andcommitted
chore: add agent_server_mode to types
PiperOrigin-RevId: 799785203
1 parent eb3fe6a commit be779ad

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

vertexai/_genai/types.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -204,6 +204,17 @@ class JobState(_common.CaseInSensitiveEnum):
204204
"""The job is partially succeeded, some results may be missing due to errors."""
205205

206206

207+
class AgentServerMode(_common.CaseInSensitiveEnum):
208+
"""The agent server mode."""
209+
210+
AGENT_SERVER_MODE_UNSPECIFIED = "AGENT_SERVER_MODE_UNSPECIFIED"
211+
"""Unspecified agent server mode. Do not use."""
212+
STABLE = "STABLE"
213+
"""Stable agent server mode. This mode has everything stable and well-tested features agent engine offers."""
214+
EXPERIMENTAL = "EXPERIMENTAL"
215+
"""Experimental agent server mode. This mode contains experimental features."""
216+
217+
207218
class ManagedTopicEnum(_common.CaseInSensitiveEnum):
208219
"""Required. The managed topic."""
209220

@@ -3355,6 +3366,9 @@ class ReasoningEngineSpecDeploymentSpec(_common.BaseModel):
33553366
default=None,
33563367
description="""Optional. Environment variables where the value is a secret in Cloud Secret Manager. To use this feature, add 'Secret Manager Secret Accessor' role (roles/secretmanager.secretAccessor) to AI Platform Reasoning Engine Service Agent.""",
33573368
)
3369+
agent_server_mode: Optional[AgentServerMode] = Field(
3370+
default=None, description="""The agent server mode."""
3371+
)
33583372
psc_interface_config: Optional[PscInterfaceConfig] = Field(
33593373
default=None, description="""Optional. Configuration for PSC-I."""
33603374
)
@@ -3385,6 +3399,9 @@ class ReasoningEngineSpecDeploymentSpecDict(TypedDict, total=False):
33853399
secret_env: Optional[list[SecretEnvVarDict]]
33863400
"""Optional. Environment variables where the value is a secret in Cloud Secret Manager. To use this feature, add 'Secret Manager Secret Accessor' role (roles/secretmanager.secretAccessor) to AI Platform Reasoning Engine Service Agent."""
33873401

3402+
agent_server_mode: Optional[AgentServerMode]
3403+
"""The agent server mode."""
3404+
33883405
psc_interface_config: Optional[PscInterfaceConfigDict]
33893406
"""Optional. Configuration for PSC-I."""
33903407

0 commit comments

Comments
 (0)