chore: add new fields

PiperOrigin-RevId: 913942680

Author: yinghsienwu

vertexai/_genai/types/__init__.py

@@ -1213,9 +1213,6 @@
 from .common import SandboxEnvironmentTemplateResourceRequirements
 from .common import SandboxEnvironmentTemplateResourceRequirementsDict
 from .common import SandboxEnvironmentTemplateResourceRequirementsOrDict
-from .common import SandboxEnvironmentTemplateWarmPoolConfig
-from .common import SandboxEnvironmentTemplateWarmPoolConfigDict
-from .common import SandboxEnvironmentTemplateWarmPoolConfigOrDict
 from .common import SandboxState
 from .common import SavedQuery
 from .common import SavedQueryDict
@@ -1244,6 +1241,9 @@
 from .common import SchemaPromptSpecAppBuilderDataLinkedResourceDict
 from .common import SchemaPromptSpecAppBuilderDataLinkedResourceOrDict
 from .common import SchemaPromptSpecAppBuilderDataOrDict
+from .common import SchemaPromptSpecInteractionData
+from .common import SchemaPromptSpecInteractionDataDict
+from .common import SchemaPromptSpecInteractionDataOrDict
 from .common import SchemaPromptSpecMultimodalPrompt
 from .common import SchemaPromptSpecMultimodalPromptDict
 from .common import SchemaPromptSpecMultimodalPromptOrDict
@@ -1313,6 +1313,7 @@
 from .common import SkillRevision
 from .common import SkillRevisionDict
 from .common import SkillRevisionOrDict
+from .common import SkillSource
 from .common import SkillState
 from .common import State
 from .common import Strategy
@@ -2315,9 +2316,6 @@
     "CreateSandboxEnvironmentTemplateConfig",
     "CreateSandboxEnvironmentTemplateConfigDict",
     "CreateSandboxEnvironmentTemplateConfigOrDict",
-    "SandboxEnvironmentTemplateWarmPoolConfig",
-    "SandboxEnvironmentTemplateWarmPoolConfigDict",
-    "SandboxEnvironmentTemplateWarmPoolConfigOrDict",
     "SandboxEnvironmentTemplate",
     "SandboxEnvironmentTemplateDict",
     "SandboxEnvironmentTemplateOrDict",
@@ -2489,6 +2487,9 @@
     "SchemaPromptSpecPartList",
     "SchemaPromptSpecPartListDict",
     "SchemaPromptSpecPartListOrDict",
+    "SchemaPromptSpecInteractionData",
+    "SchemaPromptSpecInteractionDataDict",
+    "SchemaPromptSpecInteractionDataOrDict",
     "SchemaPromptSpecStructuredPrompt",
     "SchemaPromptSpecStructuredPromptDict",
     "SchemaPromptSpecStructuredPromptOrDict",
@@ -2729,6 +2730,7 @@
     "DefaultContainerCategory",
     "PostSnapshotAction",
     "Framework",
+    "SkillSource",
     "EvaluationItemType",
     "SamplingMethod",
     "EvaluationRunState",

vertexai/_genai/types/common.py

@@ -383,6 +383,17 @@ class Framework(_common.CaseInSensitiveEnum):
     """Angular framework."""
 
 
+class SkillSource(_common.CaseInSensitiveEnum):
+    """Output only. The source of the Skill."""
+
+    SKILL_SOURCE_UNSPECIFIED = "SKILL_SOURCE_UNSPECIFIED"
+    """The skill source is unspecified."""
+    USER = "USER"
+    """The skill is created by a user."""
+    SYSTEM = "SYSTEM"
+    """The skill is a system skill."""
+
+
 class EvaluationItemType(_common.CaseInSensitiveEnum):
     """The type of the EvaluationItem."""
 
@@ -7217,29 +7228,29 @@ class MemoryGenerationTriggerConfigGenerationTriggerRule(_common.BaseModel):
 
     event_count: Optional[int] = Field(
         default=None,
-        description="""Specifies to trigger generation when the event count reaches this limit.""",
+        description="""Optional. Specifies to trigger generation when the event count reaches this limit.""",
     )
     fixed_interval: Optional[str] = Field(
         default=None,
-        description="""Specifies to trigger generation at a fixed interval. The duration must have a minute-level granularity.""",
+        description="""Optional. Specifies to trigger generation at a fixed interval. The duration must have a minute-level granularity.""",
     )
     idle_duration: Optional[str] = Field(
         default=None,
-        description="""Specifies to trigger generation if the stream is inactive for the specified duration after the most recent event. The duration must have a minute-level granularity.""",
+        description="""Optional. Specifies to trigger generation if the stream is inactive for the specified duration after the most recent event. The duration must have a minute-level granularity.""",
     )
 
 
 class MemoryGenerationTriggerConfigGenerationTriggerRuleDict(TypedDict, total=False):
     """Represents the active rule that determines when to flush the buffer."""
 
     event_count: Optional[int]
-    """Specifies to trigger generation when the event count reaches this limit."""
+    """Optional. Specifies to trigger generation when the event count reaches this limit."""
 
     fixed_interval: Optional[str]
-    """Specifies to trigger generation at a fixed interval. The duration must have a minute-level granularity."""
+    """Optional. Specifies to trigger generation at a fixed interval. The duration must have a minute-level granularity."""
 
     idle_duration: Optional[str]
-    """Specifies to trigger generation if the stream is inactive for the specified duration after the most recent event. The duration must have a minute-level granularity."""
+    """Optional. Specifies to trigger generation if the stream is inactive for the specified duration after the most recent event. The duration must have a minute-level granularity."""
 
 
 MemoryGenerationTriggerConfigGenerationTriggerRuleOrDict = Union[
@@ -7701,7 +7712,7 @@ class KeepAliveProbeHttpGet(_common.BaseModel):
 
     path: Optional[str] = Field(
         default=None,
-        description="""Required. Specifies the path of the HTTP GET request (e.g., "/is_busy").""",
+        description="""Required. Specifies the path of the HTTP GET request (e.g., `"/is_busy"`).""",
     )
     port: Optional[int] = Field(
         default=None,
@@ -7713,7 +7724,7 @@ class KeepAliveProbeHttpGetDict(TypedDict, total=False):
     """Specifies the HTTP GET configuration for the probe."""
 
     path: Optional[str]
-    """Required. Specifies the path of the HTTP GET request (e.g., "/is_busy")."""
+    """Required. Specifies the path of the HTTP GET request (e.g., `"/is_busy"`)."""
 
     port: Optional[int]
     """Optional. Specifies the port number on the container to which the request is sent."""
@@ -7847,7 +7858,7 @@ class ReasoningEngineSpecPackageSpec(_common.BaseModel):
     )
     python_version: Optional[str] = Field(
         default=None,
-        description="""Optional. The Python version. Supported values are 3.9, 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, the default value is 3.10.""",
+        description="""Optional. The Python version. Supported values are 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, the default value is 3.10.""",
     )
     requirements_gcs_uri: Optional[str] = Field(
         default=None,
@@ -7865,7 +7876,7 @@ class ReasoningEngineSpecPackageSpecDict(TypedDict, total=False):
     """Optional. The Cloud Storage URI of the pickled python object."""
 
     python_version: Optional[str]
-    """Optional. The Python version. Supported values are 3.9, 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, the default value is 3.10."""
+    """Optional. The Python version. Supported values are 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, the default value is 3.10."""
 
     requirements_gcs_uri: Optional[str]
     """Optional. The Cloud Storage URI of the `requirements.txt` file"""
@@ -8059,7 +8070,7 @@ class ReasoningEngineSpecSourceCodeSpecPythonSpec(_common.BaseModel):
     )
     version: Optional[str] = Field(
         default=None,
-        description="""Optional. The version of Python to use. Support version includes 3.9, 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, default value is 3.10.""",
+        description="""Optional. The version of Python to use. Supported versions include 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, default value is 3.10.""",
     )
 
 
@@ -8076,7 +8087,7 @@ class ReasoningEngineSpecSourceCodeSpecPythonSpecDict(TypedDict, total=False):
     """Optional. The path to the requirements file, relative to the source root. If not specified, defaults to "requirements.txt"."""
 
     version: Optional[str]
-    """Optional. The version of Python to use. Support version includes 3.9, 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, default value is 3.10."""
+    """Optional. The version of Python to use. Supported versions include 3.10, 3.11, 3.12, 3.13, 3.14. If not specified, default value is 3.10."""
 
 
 ReasoningEngineSpecSourceCodeSpecPythonSpecOrDict = Union[
@@ -12474,10 +12485,6 @@ class SandboxEnvironmentConnectionInfo(_common.BaseModel):
         default=None,
         description="""Output only. The internal IP address of the SandboxEnvironment.""",
     )
-    sandbox_hostname: Optional[str] = Field(
-        default=None,
-        description="""Output only. The hostname of the SandboxEnvironment.""",
-    )
     routing_token: Optional[str] = Field(
         default=None,
         description="""Output only. The routing token for the SandboxEnvironment.""",
@@ -12496,9 +12503,6 @@ class SandboxEnvironmentConnectionInfoDict(TypedDict, total=False):
     sandbox_internal_ip: Optional[str]
     """Output only. The internal IP address of the SandboxEnvironment."""
 
-    sandbox_hostname: Optional[str]
-    """Output only. The hostname of the SandboxEnvironment."""
-
     routing_token: Optional[str]
     """Output only. The routing token for the SandboxEnvironment."""
 
@@ -13291,28 +13295,6 @@ class _CreateSandboxEnvironmentTemplateRequestParametersDict(TypedDict, total=Fa
 ]
 
 
-class SandboxEnvironmentTemplateWarmPoolConfig(_common.BaseModel):
-    """Configuration for a warm pool of sandbox instances."""
-
-    target_instance_count: Optional[int] = Field(
-        default=None,
-        description="""Optional. The target number of pre-warmed instances to maintain.""",
-    )
-
-
-class SandboxEnvironmentTemplateWarmPoolConfigDict(TypedDict, total=False):
-    """Configuration for a warm pool of sandbox instances."""
-
-    target_instance_count: Optional[int]
-    """Optional. The target number of pre-warmed instances to maintain."""
-
-
-SandboxEnvironmentTemplateWarmPoolConfigOrDict = Union[
-    SandboxEnvironmentTemplateWarmPoolConfig,
-    SandboxEnvironmentTemplateWarmPoolConfigDict,
-]
-
-
 class SandboxEnvironmentTemplate(_common.BaseModel):
     """A sandbox environment template."""
 
@@ -13363,10 +13345,6 @@ class SandboxEnvironmentTemplate(_common.BaseModel):
         default=None,
         description="""Output only. The timestamp when this SandboxEnvironmentTemplate was most recently updated.""",
     )
-    warm_pool_config: Optional[SandboxEnvironmentTemplateWarmPoolConfig] = Field(
-        default=None,
-        description="""Optional. The configuration for the warm pool of this template.""",
-    )
 
 
 class SandboxEnvironmentTemplateDict(TypedDict, total=False):
@@ -13409,9 +13387,6 @@ class SandboxEnvironmentTemplateDict(TypedDict, total=False):
     update_time: Optional[datetime.datetime]
     """Output only. The timestamp when this SandboxEnvironmentTemplate was most recently updated."""
 
-    warm_pool_config: Optional[SandboxEnvironmentTemplateWarmPoolConfigDict]
-    """Optional. The configuration for the warm pool of this template."""
-
 
 SandboxEnvironmentTemplateOrDict = Union[
     SandboxEnvironmentTemplate, SandboxEnvironmentTemplateDict
@@ -16453,6 +16428,27 @@ class SchemaPromptSpecPartListDict(TypedDict, total=False):
 ]
 
 
+class SchemaPromptSpecInteractionData(_common.BaseModel):
+    """Defines data for an interaction prompt."""
+
+    interaction_ids: Optional[list[str]] = Field(
+        default=None,
+        description="""Optional. Lists interaction IDs associated with the prompt. This maps 1:1 to PromptMessage.contents. If InteractionData is present, every prompt message has an interaction ID.""",
+    )
+
+
+class SchemaPromptSpecInteractionDataDict(TypedDict, total=False):
+    """Defines data for an interaction prompt."""
+
+    interaction_ids: Optional[list[str]]
+    """Optional. Lists interaction IDs associated with the prompt. This maps 1:1 to PromptMessage.contents. If InteractionData is present, every prompt message has an interaction ID."""
+
+
+SchemaPromptSpecInteractionDataOrDict = Union[
+    SchemaPromptSpecInteractionData, SchemaPromptSpecInteractionDataDict
+]
+
+
 class SchemaPromptSpecStructuredPrompt(_common.BaseModel):
     """Represents a structured prompt."""
 
@@ -16489,6 +16485,9 @@ class SchemaPromptSpecStructuredPrompt(_common.BaseModel):
     prompt_message: Optional[SchemaPromptSpecPromptMessage] = Field(
         default=None, description="""The prompt message."""
     )
+    interaction_data: Optional[SchemaPromptSpecInteractionData] = Field(
+        default=None, description="""Data for interaction use case."""
+    )
 
 
 class SchemaPromptSpecStructuredPromptDict(TypedDict, total=False):
@@ -16521,6 +16520,9 @@ class SchemaPromptSpecStructuredPromptDict(TypedDict, total=False):
     prompt_message: Optional[SchemaPromptSpecPromptMessageDict]
     """The prompt message."""
 
+    interaction_data: Optional[SchemaPromptSpecInteractionDataDict]
+    """Data for interaction use case."""
+
 
 SchemaPromptSpecStructuredPromptOrDict = Union[
     SchemaPromptSpecStructuredPrompt, SchemaPromptSpecStructuredPromptDict
@@ -18107,6 +18109,17 @@ class Skill(_common.BaseModel):
     state: Optional[SkillState] = Field(
         default=None, description="""Output only. The state of the Skill."""
     )
+    labels: Optional[dict[str, str]] = Field(
+        default=None,
+        description="""The labels with user-defined metadata to organize Skills.""",
+    )
+    sha256: Optional[str] = Field(
+        default=None,
+        description="""Output only. The SHA256 checksum of the zipped filesystem.""",
+    )
+    skill_source: Optional[SkillSource] = Field(
+        default=None, description="""Output only. The source of the Skill."""
+    )
 
 
 class SkillDict(TypedDict, total=False):
@@ -18142,6 +18155,15 @@ class SkillDict(TypedDict, total=False):
     state: Optional[SkillState]
     """Output only. The state of the Skill."""
 
+    labels: Optional[dict[str, str]]
+    """The labels with user-defined metadata to organize Skills."""
+
+    sha256: Optional[str]
+    """Output only. The SHA256 checksum of the zipped filesystem."""
+
+    skill_source: Optional[SkillSource]
+    """Output only. The source of the Skill."""
+
 
 SkillOrDict = Union[Skill, SkillDict]
 

vertexai/_genai/types/evals.py

@@ -502,6 +502,10 @@ class UserScenario(_common.BaseModel):
         default=None,
         description="""The plan for the conversation, used to drive the multi-turn agent run and generate the simulated agent evaluation dataset.""",
     )
+    test_case_title: Optional[str] = Field(
+        default=None,
+        description="""Represents a short 3-5 word title for eval test case.""",
+    )
 
 
 class UserScenarioDict(TypedDict, total=False):
@@ -513,6 +517,9 @@ class UserScenarioDict(TypedDict, total=False):
     conversation_plan: Optional[str]
     """The plan for the conversation, used to drive the multi-turn agent run and generate the simulated agent evaluation dataset."""
 
+    test_case_title: Optional[str]
+    """Represents a short 3-5 word title for eval test case."""
+
 
 UserScenarioOrDict = Union[UserScenario, UserScenarioDict]
 
@@ -536,6 +543,17 @@ class UserScenarioGenerationConfig(_common.BaseModel):
         default=None,
         description="""Environment context to drive simulation. For example, for a QA agent, this could be the docs queried by the tools.""",
     )
+    environment_data: Optional[str] = Field(
+        default=None, description="""Optional. Environment data in string type."""
+    )
+    simulation_instruction: Optional[str] = Field(
+        default=None,
+        description="""Optional. Simulation instruction to guide the user scenario generation.""",
+    )
+    user_scenario_count: Optional[int] = Field(
+        default=None,
+        description="""Required. The number of user scenarios to generate. The maximum number of scenarios that can be generated is 100.""",
+    )
 
 
 class UserScenarioGenerationConfigDict(TypedDict, total=False):
@@ -553,6 +571,15 @@ class UserScenarioGenerationConfigDict(TypedDict, total=False):
     environment_context: Optional[str]
     """Environment context to drive simulation. For example, for a QA agent, this could be the docs queried by the tools."""
 
+    environment_data: Optional[str]
+    """Optional. Environment data in string type."""
+
+    simulation_instruction: Optional[str]
+    """Optional. Simulation instruction to guide the user scenario generation."""
+
+    user_scenario_count: Optional[int]
+    """Required. The number of user scenarios to generate. The maximum number of scenarios that can be generated is 100."""
+
 
 UserScenarioGenerationConfigOrDict = Union[
     UserScenarioGenerationConfig, UserScenarioGenerationConfigDict