google-cloud-agentplatform 2.1.0__tar.gz → 2.2.0__tar.gz

{google_cloud_agentplatform-2.1.0/google_cloud_agentplatform.egg-info → google_cloud_agentplatform-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-agentplatform
-Version: 2.1.0
+Version: 2.2.0
 Summary: Google Cloud Agent Platform Client Library
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/__init__.py

@@ -54,8 +54,17 @@ def __getattr__(name):  # type: ignore[no-untyped-def]
         global _genai_types
         if _genai_types is None:
             _genai_types = importlib.import_module("._genai.types", __name__)
-        if "vertexai.types" not in sys.modules:
-            sys.modules["vertexai.types"] = _genai_types
+        # `types` is an alias for `._genai.types` rather than a real submodule,
+        # so register it to keep
+        # `from agentplatform.types import TypeName`
+        # working without a prior attribute access. Spell it the google3 way:
+        # Copybara rewrites that prefix in both directions, and the external
+        # spelling fails its reversibility check. This key was `vertexai.types`,
+        # which both misnamed this package's own alias and shadowed the real
+        # `vertexai.types` module.
+        types_module_name = f"{__name__}.types"
+        if types_module_name not in sys.modules:
+            sys.modules[types_module_name] = _genai_types
         return _genai_types
 
     raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/_genai/_evals_common.py

@@ -15,7 +15,6 @@
 """Common utilities for evals."""
 
 import asyncio
-import base64
 import collections
 import concurrent.futures
 import contextlib
@@ -38,6 +37,11 @@ from google.genai._gaos.types.interactions import functioncallstep
 from google.genai._gaos.types.interactions import functionresultstep
 from google.genai._gaos.types.interactions import modeloutputstep
 from google.genai._gaos.types.interactions import userinputstep
+
+try:
+    from google.genai._gaos.utils.serializers import ALLOW_UNKNOWN_UNION_VARIANTS
+except (ImportError, AttributeError):
+    ALLOW_UNKNOWN_UNION_VARIANTS = "speakeasy_allow_unknown_union_variants"
 from google.genai.models import Models
 import pandas as pd
 from tqdm import tqdm
@@ -776,7 +780,16 @@ def _interaction_dict_to_agent_data(
     Returns:
         An AgentData object with one or more ConversationTurns.
     """
-    typed_interaction = interaction_types.Interaction.model_validate(interaction)
+    # A server response, so unknown step types must degrade rather than raise.
+    # `Step` is an open discriminated union: parse_open_union only falls back to
+    # the Unknown variant when the validation context carries
+    # ALLOW_UNKNOWN_UNION_VARIANTS, and raises without it so that a
+    # user-constructed request payload surfaces its mistakes locally. This
+    # payload comes off the Interactions API, so it is on the tolerant side of
+    # that line. _interaction_steps_to_events already drops steps it cannot map.
+    typed_interaction = interaction_types.Interaction.model_validate(
+        interaction, context={ALLOW_UNKNOWN_UNION_VARIANTS: True}
+    )
     all_events = _interaction_steps_to_events(typed_interaction.steps or [])
 
     # Group events into turns. Each UserInputStep starts a new turn.
@@ -1690,8 +1703,12 @@ def _resolve_interactions_to_eval_cases(
                 break
             interaction_dict = json.loads(response.body)
             try:
+                # Also a server response -- see _interaction_dict_to_agent_data.
+                # Without the context a single step type this client does not
+                # know about raised, and the except below turned that into a
+                # `break`, silently truncating the interaction history.
                 typed_interaction = interaction_types.Interaction.model_validate(
-                    interaction_dict
+                    interaction_dict, context={ALLOW_UNKNOWN_UNION_VARIANTS: True}
                 )
             except Exception as e:
                 logger.warning("Failed to validate interaction model: %s", e)
@@ -4078,32 +4095,6 @@ async def _convert_evaluation_run_results_async(
     return eval_result, eval_item_map
 
 
-def _object_to_dict(obj: Any) -> Union[dict[str, Any], Any]:
-    """Converts an object to a dictionary."""
-    if obj is None:
-        return obj
-    if isinstance(obj, (int, float, str, bool)):
-        return obj
-    if isinstance(obj, datetime.datetime):
-        return obj.isoformat()
-    if isinstance(obj, bytes):
-        return base64.b64encode(obj).decode("utf-8")
-    if isinstance(obj, (list, tuple)):
-        return [_object_to_dict(item) for item in obj]
-    if isinstance(obj, dict):
-        return {k: _object_to_dict(v) for k, v in obj.items()}
-
-    if not hasattr(obj, "__dict__"):
-        return obj  # Not an object with attributes, return as is (e.g., set)
-
-    result: dict[str, Any] = {}
-    for key, value in obj.__dict__.items():
-        if value is None:
-            continue
-        result[key] = _object_to_dict(value)
-    return result
-
-
 def _get_content(row: dict[str, Any], column: str) -> Optional[genai_types.Content]:
     if isinstance(row[column], str):
         return genai_types.Content(
@@ -4248,7 +4239,9 @@ def _create_evaluation_set_from_dataframe(
     eval_items = []
     for eval_item_request in eval_item_requests:
         gcs_uri = gcs_utils.upload_json_to_prefix(
-            data=_object_to_dict(eval_item_request),
+            data=eval_item_request.model_dump(
+                mode="json", by_alias=True, exclude_none=True
+            ),
             gcs_dest_prefix=gcs_dest_prefix,
             filename_prefix="request",
         )

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/_genai/_gcs_utils.py

@@ -17,13 +17,15 @@ from importlib.metadata import version as get_version
 import io
 import json
 import logging
-from typing import Any, Union
+from typing import Any, TYPE_CHECKING, Union
 import uuid
 
 from google.cloud import storage  # type: ignore[attr-defined]
 from google.genai._api_client import BaseApiClient
 from packaging.version import Version
-import pandas as pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 logger = logging.getLogger(__name__)
@@ -211,6 +213,14 @@ class GcsUtils:
         self, gcs_filepath: str, file_type: str
     ) -> "pd.DataFrame":
         """Reads a file from Google Cloud Storage into a Pandas DataFrame."""
+        try:
+            import pandas as pd  # pylint: disable=g-import-not-at-top
+        except ImportError as e:
+            raise ImportError(
+                "Reading a Cloud Storage file into a DataFrame requires 'pandas'. "
+                "Please install it using pip install "
+                "google-cloud-aiplatform[evaluation]"
+            ) from e
         file_contents = self.read_file_contents(gcs_filepath)
         if file_type == "csv":
             return pd.read_csv(io.StringIO(file_contents), encoding="utf-8")

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/_genai/evals.py

@@ -215,6 +215,13 @@ def _CustomCodeExecutionSpec_from_vertex(
             getv(from_object, ["evaluation_function"]),
         )
 
+    if getv(from_object, ["codeExecutionRegion"]) is not None:
+        setv(
+            to_object,
+            ["code_execution_region"],
+            getv(from_object, ["codeExecutionRegion"]),
+        )
+
     return to_object
 
 
@@ -237,6 +244,13 @@ def _CustomCodeExecutionSpec_to_vertex(
             getv(from_object, ["remote_custom_function"]),
         )
 
+    if getv(from_object, ["code_execution_region"]) is not None:
+        setv(
+            to_object,
+            ["codeExecutionRegion"],
+            getv(from_object, ["code_execution_region"]),
+        )
+
     return to_object
 
 

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/_genai/sandboxes.py

@@ -26,7 +26,6 @@ from typing import Any, Iterator, Optional, Union
 from urllib.parse import urlencode
 
 from google import auth as google_auth
-from google import genai
 from google.auth.transport import requests as google_auth_requests
 from google.genai import _api_module
 from google.genai import _common
@@ -34,6 +33,7 @@ from google.genai import types as genai_types
 from google.genai._common import get_value_by_path as getv
 from google.genai._common import set_value_by_path as setv
 from google.genai.pagers import Pager
+import requests
 
 from . import _runtimes_utils
 from . import types
@@ -1362,6 +1362,7 @@ class Sandboxes(_api_module.BaseModule):
         query_params: Optional[dict[str, object]] = None,
         headers: Optional[dict[str, str]] = None,
         request_dict: Optional[dict[str, object]] = None,
+        psc_endpoint: Optional[str] = None,
     ) -> genai_types.HttpResponse:
         """Sends a command to the sandbox.
 
@@ -1382,6 +1383,16 @@ class Sandboxes(_api_module.BaseModule):
                 Optional. The headers to include in the command.
             request_dict (dict[str, object]):
                 Optional. The request body to include in the command.
+            psc_endpoint (str):
+                Optional. Host (RFC1918 IP or resolvable internal hostname) of the
+                customer-side PSC endpoint forwarding rule that targets
+                ``connection_info.service_attachment``. Needed for VPC-SC sandboxes
+                (``ingressControlConfig.enablePrivateServiceConnect=true``) where
+                ``connection_info.load_balancer_hostname`` and ``load_balancer_ip``
+                are both unset because ingress goes through PSC in the customer's
+                VPC rather than a Google-managed load balancer. When provided, it
+                is used as the data-plane host in place of the load balancer
+                address.
 
         Returns:
             genai_types.HttpResponse: The response from the sandbox.
@@ -1391,12 +1402,21 @@ class Sandboxes(_api_module.BaseModule):
         connection_info = sandbox_environment.connection_info
         if not connection_info:
             raise ValueError("Connection info is not available.")
-        if connection_info.load_balancer_hostname:
+        if psc_endpoint:
+            # VPC-SC path: caller has provisioned a PSC endpoint against
+            # connection_info.service_attachment and passes its host here.
+            endpoint = "https://" + psc_endpoint
+        elif connection_info.load_balancer_hostname:
             endpoint = "https://" + connection_info.load_balancer_hostname
         elif connection_info.load_balancer_ip:
             endpoint = "http://" + connection_info.load_balancer_ip
         else:
-            raise ValueError("Load balancer hostname or ip is not available.")
+            raise ValueError(
+                "No data-plane endpoint available. Non-VPC-SC sandboxes populate"
+                " connection_info.load_balancer_hostname; VPC-SC sandboxes require"
+                " the caller to pass psc_endpoint (the host of the forwarding"
+                " rule targeting connection_info.service_attachment)."
+            )
 
         routing_token = connection_info.routing_token
         if not routing_token:
@@ -1408,15 +1428,27 @@ class Sandboxes(_api_module.BaseModule):
         headers["Authorization"] = f"Bearer {access_token}"
         headers["X-Sandbox-Routing-Token"] = routing_token
         headers["X-Sandbox-Port"] = port
-        endpoint = endpoint + path if path.startswith("/") else endpoint + "/" + path
-        http_options = genai_types.HttpOptions(headers=headers, base_url=endpoint)
-        http_client = genai.Client(vertexai=True, http_options=http_options)
-        # Full path is constructed in this function. The passed in path into request
-        # function will not be used.
-        response = http_client._api_client.request(http_method, path, request_dict)
+        url = endpoint + path if path.startswith("/") else endpoint + "/" + path
+        # The sandbox data plane authenticates via the Authorization,
+        # X-Sandbox-Routing-Token and X-Sandbox-Port headers set above, not via
+        # ADC. Routing this through genai.Client(vertexai=True) builds an
+        # authorized session with no credentials, whose request() then fails with
+        # "'NoneType' object has no attribute 'before_request'". Issue the request
+        # directly instead, mirroring generate_access_token above which also talks
+        # to a non-aiplatform endpoint with plain requests.
+        body = None
+        if request_dict:
+            body = json.dumps(request_dict)
+            headers.setdefault("Content-Type", "application/json")
+        response = requests.request(
+            http_method,
+            url,
+            headers=headers,
+            data=body,
+        )
         return genai_types.HttpResponse(
-            headers=response.headers,
-            body=response.body,
+            headers=dict(response.headers),
+            body=response.text,
         )
 
     def generate_browser_ws_headers(
@@ -1425,6 +1457,7 @@ class Sandboxes(_api_module.BaseModule):
         service_account_email: str,
         port: str = "8080",
         timeout: int = 3600,
+        psc_endpoint: Optional[str] = None,
     ) -> tuple[str, dict[str, str]]:
         """Generates the websocket upgrade headers for the browser.
 
@@ -1438,6 +1471,15 @@ class Sandboxes(_api_module.BaseModule):
                 Defaults to "8080". This should be one of the ports specified during template creation.
             timeout (int):
                 Optional. The timeout in seconds for the token. Defaults to 3600.
+            psc_endpoint (str):
+                Optional. Host (RFC1918 IP or resolvable internal hostname) of the
+                customer-side PSC endpoint forwarding rule that targets
+                ``connection_info.service_attachment``. Needed for VPC-SC
+                sandboxes; see ``send_command`` for details. When provided, it is
+                used as the websocket host in place of the load balancer address,
+                and is propagated to the internal ``send_command`` call that
+                fetches the CDP endpoint.
+
         Returns:
             tuple[str, dict[str, str]]: A tuple containing the websocket URL and
             the headers for websocket upgrade.
@@ -1446,12 +1488,19 @@ class Sandboxes(_api_module.BaseModule):
             raise ValueError("Connection info is not available.")
 
         connection_info = sandbox_environment.connection_info
-        if connection_info.load_balancer_hostname:
+        if psc_endpoint:
+            ws_base_url = "wss://" + psc_endpoint
+        elif connection_info.load_balancer_hostname:
             ws_base_url = "wss://" + connection_info.load_balancer_hostname
         elif connection_info.load_balancer_ip:
             ws_base_url = "ws://" + connection_info.load_balancer_ip
         else:
-            raise ValueError("Load balancer hostname or ip is not available.")
+            raise ValueError(
+                "No data-plane endpoint available. Non-VPC-SC sandboxes populate"
+                " connection_info.load_balancer_hostname; VPC-SC sandboxes require"
+                " the caller to pass psc_endpoint (the host of the forwarding"
+                " rule targeting connection_info.service_attachment)."
+            )
 
         http_access_token = self.generate_access_token(service_account_email, timeout)
         response = self.send_command(
@@ -1460,6 +1509,7 @@ class Sandboxes(_api_module.BaseModule):
             sandbox_environment=sandbox_environment,
             port=port,
             path="/cdp_ws_endpoint",
+            psc_endpoint=psc_endpoint,
         )
         if not response:
             raise ValueError("Failed to get the websocket endpoint.")

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/_genai/types/common.py

@@ -328,6 +328,8 @@ class SandboxState(_common.CaseInSensitiveEnum):
     """Sandbox runtime is pausing."""
     STATE_RESUMING = "STATE_RESUMING"
     """Sandbox runtime is resuming."""
+    STATE_STOPPING = "STATE_STOPPING"
+    """Sandbox runtime is stopping."""
 
 
 class Protocol(_common.CaseInSensitiveEnum):
@@ -1705,6 +1707,10 @@ class CustomCodeExecutionSpec(_common.BaseModel):
   Instance is the evaluation instance, any fields populated in the instance
   are available to the function as instance[field_name].""",
     )
+    code_execution_region: Optional[str] = Field(
+        default=None,
+        description="""Optional. The region to use for code execution. If set, the Code Execution Sandbox will be invoked in the specified region regardless of the request's originating region. Must be a region where the Code Execution Sandbox is available. Supported regions: northamerica-northeast1, southamerica-east1, us-central1, us-east1, us-east4, us-west1, us-west4, europe-central2, europe-north1, europe-southwest1, europe-west1, europe-west2, europe-west3, europe-west4, europe-west6, europe-west8, europe-west9, me-west1, asia-east1, asia-east2, asia-northeast1, asia-northeast3, asia-south1, asia-south2, asia-southeast1, australia-southeast2. If unset, the request's originating region is used; requests from regions where the sandbox is unavailable will fail with UNIMPLEMENTED.""",
+    )
 
 
 class CustomCodeExecutionSpecDict(TypedDict, total=False):
@@ -1724,6 +1730,9 @@ class CustomCodeExecutionSpecDict(TypedDict, total=False):
   Instance is the evaluation instance, any fields populated in the instance
   are available to the function as instance[field_name]."""
 
+    code_execution_region: Optional[str]
+    """Optional. The region to use for code execution. If set, the Code Execution Sandbox will be invoked in the specified region regardless of the request's originating region. Must be a region where the Code Execution Sandbox is available. Supported regions: northamerica-northeast1, southamerica-east1, us-central1, us-east1, us-east4, us-west1, us-west4, europe-central2, europe-north1, europe-southwest1, europe-west1, europe-west2, europe-west3, europe-west4, europe-west6, europe-west8, europe-west9, me-west1, asia-east1, asia-east2, asia-northeast1, asia-northeast3, asia-south1, asia-south2, asia-southeast1, australia-southeast2. If unset, the request's originating region is used; requests from regions where the sandbox is unavailable will fail with UNIMPLEMENTED."""
+
 
 CustomCodeExecutionSpecOrDict = Union[
     CustomCodeExecutionSpec, CustomCodeExecutionSpecDict
@@ -6665,7 +6674,7 @@ class ReservationAffinity(_common.BaseModel):
     )
     values: Optional[list[str]] = Field(
         default=None,
-        description="""Optional. Corresponds to the label values of a reservation resource. This must be the full resource name of the reservation or reservation block.""",
+        description="""Optional. Corresponds to the label values of a reservation resource. This must be the resource name of the reservation, reservation block, or reservation sub- block.""",
     )
 
 
@@ -6679,7 +6688,7 @@ class ReservationAffinityDict(TypedDict, total=False):
     """Required. Specifies the reservation affinity type."""
 
     values: Optional[list[str]]
-    """Optional. Corresponds to the label values of a reservation resource. This must be the full resource name of the reservation or reservation block."""
+    """Optional. Corresponds to the label values of a reservation resource. This must be the resource name of the reservation, reservation block, or reservation sub- block."""
 
 
 ReservationAffinityOrDict = Union[ReservationAffinity, ReservationAffinityDict]
@@ -11538,6 +11547,9 @@ class Memory(_common.BaseModel):
         default=None,
         description="""Optional. Represents the structured content of the memory.""",
     )
+    context: Optional[str] = Field(
+        default=None, description="""Optional. Represents the context of the memory."""
+    )
 
 
 class MemoryDict(TypedDict, total=False):
@@ -11594,6 +11606,9 @@ class MemoryDict(TypedDict, total=False):
     structured_content: Optional[MemoryStructuredContentDict]
     """Optional. Represents the structured content of the memory."""
 
+    context: Optional[str]
+    """Optional. Represents the context of the memory."""
+
 
 MemoryOrDict = Union[Memory, MemoryDict]
 
@@ -13281,6 +13296,10 @@ class MemoryRevision(_common.BaseModel):
         default=None,
         description="""Output only. Represents the structured value of the memory at the time of revision creation.""",
     )
+    context: Optional[str] = Field(
+        default=None,
+        description="""Output only. Represents the context of the Memory Revision. The context may include context from both the historical revisions and the extracted content.""",
+    )
 
 
 class MemoryRevisionDict(TypedDict, total=False):
@@ -13307,6 +13326,9 @@ class MemoryRevisionDict(TypedDict, total=False):
     structured_data: Optional[dict[str, Any]]
     """Output only. Represents the structured value of the memory at the time of revision creation."""
 
+    context: Optional[str]
+    """Output only. Represents the context of the Memory Revision. The context may include context from both the historical revisions and the extracted content."""
+
 
 MemoryRevisionOrDict = Union[MemoryRevision, MemoryRevisionDict]
 
@@ -16564,13 +16586,13 @@ SandboxEnvironmentSpecComputerUseEnvironmentOrDict = Union[
 
 
 class SandboxEnvironmentSpecShellEnvironment(_common.BaseModel):
-    """The shell environment with customized settings."""
+    """The shell environment."""
 
     pass
 
 
 class SandboxEnvironmentSpecShellEnvironmentDict(TypedDict, total=False):
-    """The shell environment with customized settings."""
+    """The shell environment."""
 
     pass
 
@@ -16590,7 +16612,12 @@ class SandboxEnvironmentSpec(_common.BaseModel):
         Field(default=None, description="""Optional. The computer use environment.""")
     )
     shell_environment: Optional[SandboxEnvironmentSpecShellEnvironment] = Field(
-        default=None, description="""Optional. The shell environment."""
+        default=None,
+        description="""Optional. The shell environment for executing shell commands and scripts.""",
+    )
+    use_gke_td: Optional[bool] = Field(
+        default=None,
+        description="""Optional. Immutable. Whether to provision the SandboxEnvironment via the GKE TD pool. Immutable.""",
     )
 
 
@@ -16606,7 +16633,10 @@ class SandboxEnvironmentSpecDict(TypedDict, total=False):
     """Optional. The computer use environment."""
 
     shell_environment: Optional[SandboxEnvironmentSpecShellEnvironmentDict]
-    """Optional. The shell environment."""
+    """Optional. The shell environment for executing shell commands and scripts."""
+
+    use_gke_td: Optional[bool]
+    """Optional. Immutable. Whether to provision the SandboxEnvironment via the GKE TD pool. Immutable."""
 
 
 SandboxEnvironmentSpecOrDict = Union[SandboxEnvironmentSpec, SandboxEnvironmentSpecDict]
@@ -17551,11 +17581,11 @@ class SandboxEnvironmentTemplateEgressControlConfigDnsPeeringConfig(_common.Base
     )
     target_network: Optional[str] = Field(
         default=None,
-        description="""Required. The VPC network name in the target_project where the DNS zone specified by 'domain' is visible.""",
+        description="""Required. The VPC network name in the target_project where the DNS zone specified by `domain` is visible.""",
     )
     target_project: Optional[str] = Field(
         default=None,
-        description="""Required. The project ID hosting the Cloud DNS managed zone that contains the 'domain'. The Vertex AI Service Agent requires the dns.peer role on this project.""",
+        description="""Required. The project ID hosting the Cloud DNS managed zone that contains the `domain`. The Vertex AI Service Agent requires the dns.peer role on this project.""",
     )
 
 
@@ -17568,10 +17598,10 @@ class SandboxEnvironmentTemplateEgressControlConfigDnsPeeringConfigDict(
     """Required. The DNS name suffix of the zone being peered to, e.g., "my-internal-domain.corp.". Must end with a dot."""
 
     target_network: Optional[str]
-    """Required. The VPC network name in the target_project where the DNS zone specified by 'domain' is visible."""
+    """Required. The VPC network name in the target_project where the DNS zone specified by `domain` is visible."""
 
     target_project: Optional[str]
-    """Required. The project ID hosting the Cloud DNS managed zone that contains the 'domain'. The Vertex AI Service Agent requires the dns.peer role on this project."""
+    """Required. The project ID hosting the Cloud DNS managed zone that contains the `domain`. The Vertex AI Service Agent requires the dns.peer role on this project."""
 
 
 SandboxEnvironmentTemplateEgressControlConfigDnsPeeringConfigOrDict = Union[
@@ -17586,10 +17616,6 @@ class SandboxEnvironmentTemplateEgressControlConfig(_common.BaseModel):
     internet_access: Optional[bool] = Field(
         default=None, description="""Optional. Whether to allow internet access."""
     )
-    customer_vpc_network: Optional[str] = Field(
-        default=None,
-        description="""Optional. The customer VPC network that sandbox egress is routed into.""",
-    )
     dns_peering_configs: Optional[
         list[SandboxEnvironmentTemplateEgressControlConfigDnsPeeringConfig]
     ] = Field(
@@ -17598,7 +17624,7 @@ class SandboxEnvironmentTemplateEgressControlConfig(_common.BaseModel):
     )
     network_attachment: Optional[str] = Field(
         default=None,
-        description="""Optional. The name of the customer VPC NetworkAttachment used to draw a PSC interface IP into the customer VPC for sandbox egress.""",
+        description="""Optional. The name of the customer VPC `NetworkAttachment` used to draw a PSC interface IP into the customer VPC for sandbox egress.""",
     )
 
 
@@ -17608,16 +17634,13 @@ class SandboxEnvironmentTemplateEgressControlConfigDict(TypedDict, total=False):
     internet_access: Optional[bool]
     """Optional. Whether to allow internet access."""
 
-    customer_vpc_network: Optional[str]
-    """Optional. The customer VPC network that sandbox egress is routed into."""
-
     dns_peering_configs: Optional[
         list[SandboxEnvironmentTemplateEgressControlConfigDnsPeeringConfigDict]
     ]
     """Optional. DNS peering configurations that allow sandbox egress to resolve customer-internal domains via the customer VPC."""
 
     network_attachment: Optional[str]
-    """Optional. The name of the customer VPC NetworkAttachment used to draw a PSC interface IP into the customer VPC for sandbox egress."""
+    """Optional. The name of the customer VPC `NetworkAttachment` used to draw a PSC interface IP into the customer VPC for sandbox egress."""
 
 
 SandboxEnvironmentTemplateEgressControlConfigOrDict = Union[
@@ -17873,6 +17896,10 @@ class SandboxEnvironmentTemplate(_common.BaseModel):
         default=None,
         description="""Optional. The configuration for private ingress (PSC-E) of this template. When set, the sandbox router is exposed privately via a PSC service attachment so VPC-SC customers can connect from their VPC over a private endpoint instead of the public internet. The resulting service attachment is surfaced on `SandboxEnvironment.connection_info.service_attachment`. Only the PSC-E (service-attachment/ingress) portion of `PrivateServiceConnectConfig` applies here: `enable_private_service_connect` and `project_allowlist` (the consumer projects allowed to connect). The nested `psc_interface_config` (PSC-I / egress) is not used for sandbox ingress; sandbox egress is configured via `egress_control_config` instead.""",
     )
+    use_gke_td: Optional[bool] = Field(
+        default=None,
+        description="""Optional. Immutable. Whether to provision the SandboxEnvironmentTemplate via the GKE TD pool.""",
+    )
 
 
 class SandboxEnvironmentTemplateDict(TypedDict, total=False):
@@ -17909,6 +17936,9 @@ class SandboxEnvironmentTemplateDict(TypedDict, total=False):
     ingress_control_config: Optional[PrivateServiceConnectConfigDict]
     """Optional. The configuration for private ingress (PSC-E) of this template. When set, the sandbox router is exposed privately via a PSC service attachment so VPC-SC customers can connect from their VPC over a private endpoint instead of the public internet. The resulting service attachment is surfaced on `SandboxEnvironment.connection_info.service_attachment`. Only the PSC-E (service-attachment/ingress) portion of `PrivateServiceConnectConfig` applies here: `enable_private_service_connect` and `project_allowlist` (the consumer projects allowed to connect). The nested `psc_interface_config` (PSC-I / egress) is not used for sandbox ingress; sandbox egress is configured via `egress_control_config` instead."""
 
+    use_gke_td: Optional[bool]
+    """Optional. Immutable. Whether to provision the SandboxEnvironmentTemplate via the GKE TD pool."""
+
 
 SandboxEnvironmentTemplateOrDict = Union[
     SandboxEnvironmentTemplate, SandboxEnvironmentTemplateDict
@@ -18340,6 +18370,10 @@ class SandboxEnvironmentSnapshot(_common.BaseModel):
         default=None,
         description="""Output only. The timestamp when this SandboxEnvironment was most recently updated.""",
     )
+    use_gke_td: Optional[bool] = Field(
+        default=None,
+        description="""Output only. Whether the source SandboxEnvironment uses the GKE TD pool.""",
+    )
 
 
 class SandboxEnvironmentSnapshotDict(TypedDict, total=False):
@@ -18379,6 +18413,9 @@ class SandboxEnvironmentSnapshotDict(TypedDict, total=False):
     update_time: Optional[datetime.datetime]
     """Output only. The timestamp when this SandboxEnvironment was most recently updated."""
 
+    use_gke_td: Optional[bool]
+    """Output only. Whether the source SandboxEnvironment uses the GKE TD pool."""
+
 
 SandboxEnvironmentSnapshotOrDict = Union[
     SandboxEnvironmentSnapshot, SandboxEnvironmentSnapshotDict
@@ -20965,6 +21002,10 @@ class SchemaPromptSpecAppBuilderData(_common.BaseModel):
             description="""Linked resources attached to the application by the user.""",
         )
     )
+    deployed_regions: Optional[list[str]] = Field(
+        default=None,
+        description="""Optional. The Cloud Run regions in which the application is currently deployed. Used to rediscover and redeploy the app in the regions it already runs in, which may differ from the prompt's location.""",
+    )
 
 
 class SchemaPromptSpecAppBuilderDataDict(TypedDict, total=False):
@@ -20979,6 +21020,9 @@ class SchemaPromptSpecAppBuilderDataDict(TypedDict, total=False):
     linked_resources: Optional[list[SchemaPromptSpecAppBuilderDataLinkedResourceDict]]
     """Linked resources attached to the application by the user."""
 
+    deployed_regions: Optional[list[str]]
+    """Optional. The Cloud Run regions in which the application is currently deployed. Used to rediscover and redeploy the app in the regions it already runs in, which may differ from the prompt's location."""
+
 
 SchemaPromptSpecAppBuilderDataOrDict = Union[
     SchemaPromptSpecAppBuilderData, SchemaPromptSpecAppBuilderDataDict
@@ -28666,11 +28710,11 @@ AudioTranscriptionWordInfoOrDict = Union[
 
 
 class AudioTranscription(_common.BaseModel):
-    """The transcription of an audio part. For multi-speaker audio, each speaker segment is a separate Part with its own AudioTranscription carrying the speaker_label."""
+    """The transcription of an audio part. For multi-speaker audio, each speaker segment is a separate `Part` with its own `AudioTranscription` carrying the `speaker_label`."""
 
     speaker_label: Optional[str] = Field(
         default=None,
-        description="""Optional. A label identifying the speaker of this audio segment (e.g. "spk_1", "spk_2"). Present when diarization is set.""",
+        description="""Optional. A label identifying the speaker of this audio segment (e.g. `spk_1`, `spk_2`). Present when `diarization` is set.""",
     )
     text: Optional[str] = Field(
         default=None,
@@ -28678,21 +28722,21 @@ class AudioTranscription(_common.BaseModel):
     )
     words: Optional[list[AudioTranscriptionWordInfo]] = Field(
         default=None,
-        description="""Optional. Detailed word-level transcriptions and timing details. Present when word_timestamp is set.""",
+        description="""Optional. Detailed word-level transcriptions and timing details. Present when `word_timestamp` is set.""",
     )
 
 
 class AudioTranscriptionDict(TypedDict, total=False):
-    """The transcription of an audio part. For multi-speaker audio, each speaker segment is a separate Part with its own AudioTranscription carrying the speaker_label."""
+    """The transcription of an audio part. For multi-speaker audio, each speaker segment is a separate `Part` with its own `AudioTranscription` carrying the `speaker_label`."""
 
     speaker_label: Optional[str]
-    """Optional. A label identifying the speaker of this audio segment (e.g. "spk_1", "spk_2"). Present when diarization is set."""
+    """Optional. A label identifying the speaker of this audio segment (e.g. `spk_1`, `spk_2`). Present when `diarization` is set."""
 
     text: Optional[str]
     """Required. The transcription text of this audio segment."""
 
     words: Optional[list[AudioTranscriptionWordInfoDict]]
-    """Optional. Detailed word-level transcriptions and timing details. Present when word_timestamp is set."""
+    """Optional. Detailed word-level transcriptions and timing details. Present when `word_timestamp` is set."""
 
 
 AudioTranscriptionOrDict = Union[AudioTranscription, AudioTranscriptionDict]

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0}/agentplatform/version.py

@@ -14,4 +14,4 @@
 #
 """Version file for agentplatform."""
 
-__version__ = "2.1.0"
+__version__ = "2.2.0"

{google_cloud_agentplatform-2.1.0 → google_cloud_agentplatform-2.2.0/google_cloud_agentplatform.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-agentplatform
-Version: 2.1.0
+Version: 2.2.0
 Summary: Google Cloud Agent Platform Client Library
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0