google-cloud-agentplatform 2.1.3__tar.gz → 2.3.0__tar.gz

{google_cloud_agentplatform-2.1.3/google_cloud_agentplatform.egg-info → google_cloud_agentplatform-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-agentplatform
-Version: 2.1.3
+Version: 2.3.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.3 → google_cloud_agentplatform-2.3.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.3 → google_cloud_agentplatform-2.3.0}/agentplatform/_genai/_evals_common.py

@@ -37,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
@@ -775,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.
@@ -1689,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)

{google_cloud_agentplatform-2.1.3 → google_cloud_agentplatform-2.3.0}/agentplatform/_genai/client.py

@@ -62,6 +62,10 @@ if TYPE_CHECKING:
     from agentplatform._genai import (
         memory_banks as memory_banks_module,
     )
+    from agentplatform._genai import (
+        serving_profiles as serving_profiles_module,
+    )
+
 
 _GENAI_MODULES_TELEMETRY_HEADER = "vertex-genai-modules"
 
@@ -112,6 +116,7 @@ class AsyncClient:
         self._sessions: Optional[ModuleType] = None
         self._sandboxes: Optional[ModuleType] = None
         self._memory_banks: Optional[ModuleType] = None
+        self._serving_profiles: Optional[ModuleType] = None
 
     @property
     @_common.experimental_warning(
@@ -287,6 +292,17 @@ class AsyncClient:
             self._memory_banks = importlib.import_module(".memory_banks", __package__)
         return self._memory_banks.AsyncMemoryBanks(self._api_client)  # type: ignore[no-any-return]
 
+    @property
+    def serving_profiles(
+        self,
+    ) -> "serving_profiles_module.AsyncServingProfiles":
+        if self._serving_profiles is None:
+            self._serving_profiles = importlib.import_module(
+                ".serving_profiles",
+                __package__,
+            )
+        return self._serving_profiles.AsyncServingProfiles(self._api_client)  # type: ignore[no-any-return]
+
     async def aclose(self) -> None:
         """Closes the async client explicitly.
 
@@ -399,6 +415,7 @@ class Client:
         self._sessions: Optional[ModuleType] = None
         self._sandboxes: Optional[ModuleType] = None
         self._memory_banks: Optional[ModuleType] = None
+        self._serving_profiles: Optional[ModuleType] = None
 
     @property
     def evals(self) -> "evals_module.Evals":
@@ -598,3 +615,12 @@ class Client:
         if self._memory_banks is None:
             self._memory_banks = importlib.import_module(".memory_banks", __package__)
         return self._memory_banks.MemoryBanks(self._api_client)  # type: ignore[no-any-return]
+
+    @property
+    def serving_profiles(self) -> "serving_profiles_module.ServingProfiles":
+        if self._serving_profiles is None:
+            self._serving_profiles = importlib.import_module(
+                ".serving_profiles",
+                __package__,
+            )
+        return self._serving_profiles.ServingProfiles(self._api_client)  # type: ignore[no-any-return]

{google_cloud_agentplatform-2.1.3 → google_cloud_agentplatform-2.3.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.3 → google_cloud_agentplatform-2.3.0}/agentplatform/_genai/memory_banks.py

@@ -504,6 +504,13 @@ def _ReasoningEngine_from_vertex(
     if getv(from_object, ["trafficConfig"]) is not None:
         setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"]))
 
+    if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None:
+        setv(
+            to_object,
+            ["revision_garbage_collection_strategy"],
+            getv(from_object, ["revisionGarbageCollectionStrategy"]),
+        )
+
     return to_object
 
 

{google_cloud_agentplatform-2.1.3 → google_cloud_agentplatform-2.3.0}/agentplatform/_genai/runtimes.py

@@ -484,6 +484,13 @@ def _ReasoningEngine_from_vertex(
     if getv(from_object, ["trafficConfig"]) is not None:
         setv(to_object, ["traffic_config"], getv(from_object, ["trafficConfig"]))
 
+    if getv(from_object, ["revisionGarbageCollectionStrategy"]) is not None:
+        setv(
+            to_object,
+            ["revision_garbage_collection_strategy"],
+            getv(from_object, ["revisionGarbageCollectionStrategy"]),
+        )
+
     return to_object