feat: add new fields

PiperOrigin-RevId: 913942680

Author: yinghsienwu

agentplatform/_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
@@ -1314,6 +1314,7 @@
 from .common import SkillRevisionDict
 from .common import SkillRevisionOrDict
 from .common import SkillRevisionState
+from .common import SkillSource
 from .common import SkillState
 from .common import State
 from .common import Strategy
@@ -2316,9 +2317,6 @@
     "CreateSandboxEnvironmentTemplateConfig",
     "CreateSandboxEnvironmentTemplateConfigDict",
     "CreateSandboxEnvironmentTemplateConfigOrDict",
-    "SandboxEnvironmentTemplateWarmPoolConfig",
-    "SandboxEnvironmentTemplateWarmPoolConfigDict",
-    "SandboxEnvironmentTemplateWarmPoolConfigOrDict",
     "SandboxEnvironmentTemplate",
     "SandboxEnvironmentTemplateDict",
     "SandboxEnvironmentTemplateOrDict",
@@ -2490,6 +2488,9 @@
     "SchemaPromptSpecPartList",
     "SchemaPromptSpecPartListDict",
     "SchemaPromptSpecPartListOrDict",
+    "SchemaPromptSpecInteractionData",
+    "SchemaPromptSpecInteractionDataDict",
+    "SchemaPromptSpecInteractionDataOrDict",
     "SchemaPromptSpecStructuredPrompt",
     "SchemaPromptSpecStructuredPromptDict",
     "SchemaPromptSpecStructuredPromptOrDict",
@@ -2731,6 +2732,7 @@
     "PostSnapshotAction",
     "Framework",
     "SkillState",
+    "SkillSource",
     "EvaluationItemType",
     "SamplingMethod",
     "EvaluationRunState",

agentplatform/_genai/types/common.py

@@ -398,6 +398,17 @@ class SkillState(_common.CaseInSensitiveEnum):
     """The Skill is being deleted."""
 
 
+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."""
 
@@ -7232,29 +7243,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[
@@ -7716,7 +7727,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,
@@ -7728,7 +7739,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."""
@@ -7862,7 +7873,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,
@@ -7880,7 +7891,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"""
@@ -8074,7 +8085,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.""",
     )
 
 
@@ -8091,7 +8102,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[
@@ -12489,10 +12500,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.""",
@@ -12511,9 +12518,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."""
 
@@ -13306,28 +13310,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."""
 
@@ -13378,10 +13360,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):
@@ -13424,9 +13402,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
@@ -16468,6 +16443,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."""
 
@@ -16504,6 +16500,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):
@@ -16536,6 +16535,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
@@ -18122,6 +18124,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):
@@ -18157,6 +18170,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]