google-cloud-agentplatform 2.8.0__tar.gz → 2.9.0__tar.gz

{google_cloud_agentplatform-2.8.0/google_cloud_agentplatform.egg-info → google_cloud_agentplatform-2.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-agentplatform
-Version: 2.8.0
+Version: 2.9.0
 Summary: Google Cloud Agent Platform Client Library
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -61,7 +61,7 @@ Provides-Extra: evaluation
 Requires-Dist: pandas>=1.0.0; extra == "evaluation"
 Requires-Dist: tqdm>=4.23.0; extra == "evaluation"
 Requires-Dist: pyyaml; extra == "evaluation"
-Requires-Dist: litellm<1.107.0,>=1.106.5; python_version < "3.15" and extra == "evaluation"
+Requires-Dist: litellm!=1.107.0,<1.110.0,>=1.106.5; python_version < "3.15" and extra == "evaluation"
 Requires-Dist: google-cloud-storage<4.0.0,>=1.32.0; extra == "evaluation"
 Provides-Extra: bigquery
 Requires-Dist: google-cloud-bigquery<4.0.0,>=1.15.0; extra == "bigquery"

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/agentplatform/_genai/_runtimes_utils.py

@@ -1322,6 +1322,8 @@ def _upload_extra_packages(
     tar_fileobj = io.BytesIO()
     with tarfile.open(fileobj=tar_fileobj, mode="w|gz") as tar:
         for file in extra_packages:
+            # Note: tar.add(file) preserves each path relative to the current
+            # working directory in the archive (extracted into /code).
             tar.add(file)
     tar_fileobj.seek(0)
     blob = gcs_bucket.blob(f"{gcs_dir_name}/{_EXTRA_PACKAGES_FILE}")
@@ -1360,78 +1362,14 @@ def _validate_packages_or_raise(
     build_options: Optional[Dict[str, Sequence[str]]] = None,
 ) -> Sequence[str]:
     """Tries to validates the packages."""
+    del build_options
     packages = packages or []
-    if build_options and _INSTALLATION_SUBDIR in build_options:
-        _validate_installation_scripts_or_raise(
-            script_paths=build_options[_INSTALLATION_SUBDIR],
-            packages=packages,
-        )
     for package in packages:
         if not os.path.exists(package):
             raise FileNotFoundError(f"Package specified but not found: {package=}")
     return packages
 
 
-def _validate_installation_scripts_or_raise(
-    *,
-    script_paths: Sequence[str],
-    packages: Sequence[str],
-) -> None:
-    """Validates the installation scripts' path explicitly provided by the user.
-
-    Args:
-        script_paths (Sequence[str]):
-            Required. The paths to the installation scripts.
-        packages (Sequence[str]):
-            Required. The user-provided packages.
-
-    Raises:
-        ValueError: If a user-defined script is not under the expected
-            subdirectory, or not in `packages`, or if a package is
-            in the installation scripts subdirectory, but is not specified as an
-            installation script.
-    """
-    for script_path in script_paths:
-        if not script_path.startswith(_INSTALLATION_SUBDIR):
-            logger.warning(
-                f"User-defined installation script '{script_path}' is not in "
-                f"the expected '{_INSTALLATION_SUBDIR}' subdirectory. "
-                f"Ensure it is placed in '{_INSTALLATION_SUBDIR}' within your "
-                f"'extra_packages' or 'source_packages'."
-            )
-            raise ValueError(
-                f"Required installation script '{script_path}' "
-                f"is not under '{_INSTALLATION_SUBDIR}'"
-            )
-
-        if script_path not in packages:
-            logger.warning(
-                f"User-defined installation script '{script_path}' is not in "
-                f"'extra_packages' or 'source_packages'. Ensure it is added to "
-                f"'extra_packages' or 'source_packages'."
-            )
-            raise ValueError(
-                f"User-defined installation script '{script_path}' "
-                f"does not exist in 'extra_packages' or 'source_packages'."
-            )
-
-    for package in packages:
-        if package.startswith(_INSTALLATION_SUBDIR) and package not in script_paths:
-            logger.warning(
-                f"Package '{package}' is in the installation "
-                "scripts subdirectory, but is not specified as an installation "
-                "script in `build_options`. "
-                "Ensure it is added to installation_scripts for "
-                "automatic execution."
-            )
-            raise ValueError(
-                f"Package '{package}' is in the installation "
-                "scripts subdirectory, but is not specified as an installation "
-                "script in `build_options`."
-            )
-    return
-
-
 def _validate_staging_bucket_or_raise(*, staging_bucket: str) -> str:
     """Tries to validate the staging bucket."""
     if not staging_bucket:
@@ -1875,7 +1813,8 @@ def _wrap_a2a_operation(method_name: str, agent_card: str) -> Callable[..., list
                         f"Bearer {self.api_client._api_client._credentials.token}"
                     )
                 },
-                timeout=(
+                timeout=kwargs.pop("timeout", None)
+                or (
                     self.api_client._api_client._http_options.timeout / 1000.0
                     if self.api_client._api_client._http_options.timeout
                     else None

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/agentplatform/_genai/feedback_entries.py

@@ -532,7 +532,7 @@ class FeedbackEntries(_api_module.BaseModule):
         Args:
             name (str): Required. Name of the Feedback Entry to update. Format:
                 `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/feedbackEntries/{feedback_entry_id}`.
-            feedback_type (shared.FeedbackType): Optional. The type of feedback provided.
+            feedback_type (FeedbackType): Optional. The type of feedback provided.
             config (UpdateRuntimeFeedbackEntryConfig): Optional. The configuration for updating the Feedback Entry.
 
         Returns:
@@ -1245,7 +1245,7 @@ class AsyncFeedbackEntries(_api_module.BaseModule):
         Args:
             name (str): Required. Name of the Feedback Entry to update. Format:
                 `projects/{project}/locations/{location}/reasoningEngines/{resource_id}/feedbackEntries/{feedback_entry_id}`.
-            feedback_type (shared.FeedbackType): Optional. The type of feedback provided.
+            feedback_type (FeedbackType): Optional. The type of feedback provided.
             config (UpdateRuntimeFeedbackEntryConfig): Optional. The configuration for updating the Feedback Entry.
 
         Returns:

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/agentplatform/_genai/types/__init__.py

@@ -2050,6 +2050,9 @@ from .common import SandboxEnvironmentTemplateResourceRequirements
 from .common import SandboxEnvironmentTemplateResourceRequirementsDict
 from .common import SandboxEnvironmentTemplateResourceRequirementsOrDict
 from .common import SandboxEnvironmentTemplateState
+from .common import SandboxEnvironmentTemplateWarmPoolConfig
+from .common import SandboxEnvironmentTemplateWarmPoolConfigDict
+from .common import SandboxEnvironmentTemplateWarmPoolConfigOrDict
 from .common import SandboxState
 from .common import SavedQuery
 from .common import SavedQueryDict
@@ -2368,6 +2371,7 @@ from .common import UpdateServingProfileConfigOrDict
 from .common import UpdateSkillConfig
 from .common import UpdateSkillConfigDict
 from .common import UpdateSkillConfigOrDict
+from .common import UpdateStrategy
 from .common import UploadRagFileConfig
 from .common import UploadRagFileConfigDict
 from .common import UploadRagFileConfigOrDict
@@ -3619,6 +3623,9 @@ __all__ = [
     "SandboxEnvironmentTemplatePersistentDiskConfig",
     "SandboxEnvironmentTemplatePersistentDiskConfigDict",
     "SandboxEnvironmentTemplatePersistentDiskConfigOrDict",
+    "SandboxEnvironmentTemplateWarmPoolConfig",
+    "SandboxEnvironmentTemplateWarmPoolConfigDict",
+    "SandboxEnvironmentTemplateWarmPoolConfigOrDict",
     "SandboxEnvironmentTemplate",
     "SandboxEnvironmentTemplateDict",
     "SandboxEnvironmentTemplateOrDict",
@@ -4526,6 +4533,7 @@ __all__ = [
     "DefaultContainerCategory",
     "PscAutomationState",
     "SandboxEnvironmentTemplateState",
+    "UpdateStrategy",
     "PostSnapshotAction",
     "Framework",
     "SkillSource",

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/agentplatform/_genai/types/common.py

@@ -243,7 +243,7 @@ class State(_common.CaseInSensitiveEnum):
 
 
 class MemoryType(_common.CaseInSensitiveEnum):
-    """The type of the memory."""
+    """Represents the type of the memory. If not set, the `NATURAL_LANGUAGE_COLLECTION` type is used. If `STRUCTURED_COLLECTION` or `STRUCTURED_PROFILE` is used, then `structured_data` must be provided."""
 
     MEMORY_TYPE_UNSPECIFIED = "MEMORY_TYPE_UNSPECIFIED"
     """Represents an unspecified memory type. This value should not be used."""
@@ -385,6 +385,17 @@ class SandboxEnvironmentTemplateState(_common.CaseInSensitiveEnum):
     """Sandbox has failed to provision."""
 
 
+class UpdateStrategy(_common.CaseInSensitiveEnum):
+    """The strategy for updating warm pool instances when the template is updated."""
+
+    UPDATE_STRATEGY_UNSPECIFIED = "UPDATE_STRATEGY_UNSPECIFIED"
+    """Unspecified update strategy. Defaults to ON_REPLENISH."""
+    ON_REPLENISH = "ON_REPLENISH"
+    """Existing warm pool pods are left running and new pods use the updated template as the warm pool is replenished."""
+    RECREATE = "RECREATE"
+    """Existing warm pool pods are recreated immediately with the updated template."""
+
+
 class PostSnapshotAction(_common.CaseInSensitiveEnum):
     """Input only. Action to take on the source SandboxEnvironment after the snapshot is taken. This field is only used in CreateSandboxEnvironmentSnapshotRequest and it is not stored in the resource."""
 
@@ -419,7 +430,7 @@ class SkillSource(_common.CaseInSensitiveEnum):
 
 
 class SkillState(_common.CaseInSensitiveEnum):
-    """State of the Skill."""
+    """Output only. The state of the Skill."""
 
     STATE_UNSPECIFIED = "STATE_UNSPECIFIED"
     """The state of the Skill is unspecified."""
@@ -490,14 +501,14 @@ class QuotaState(_common.CaseInSensitiveEnum):
 
 
 class FeedbackType(_common.CaseInSensitiveEnum):
-    """The type of the feedback."""
+    """The coarse-grained type of feedback provided by the user. Must be set to a value other than `FEEDBACK_TYPE_UNSPECIFIED`."""
 
     FEEDBACK_TYPE_UNSPECIFIED = "FEEDBACK_TYPE_UNSPECIFIED"
-    """Default value."""
+    """This is the default value meaning the type has not been set."""
     THUMBS_UP = "THUMBS_UP"
-    """Indicates positive feedback (e.g., a "thumbs up")."""
+    """Indicates positive feedback."""
     THUMBS_DOWN = "THUMBS_DOWN"
-    """Indicates a thumbs down feedback (e.g., a "thumbs down")."""
+    """Indicates negative feedback."""
 
 
 class Encoding(_common.CaseInSensitiveEnum):
@@ -664,13 +675,13 @@ class ServingProfileScope(_common.CaseInSensitiveEnum):
     """The specific API this ServingProfile applies to."""
 
     SERVING_PROFILE_SCOPE_UNSPECIFIED = "SERVING_PROFILE_SCOPE_UNSPECIFIED"
-    """Unspecified scope."""
+    """Default value. This value is unused. When users create a ServingProfile, they must choose a scope."""
     GEMINI_LIVE = "GEMINI_LIVE"
-    """Scope for Gemini Live."""
+    """The scope for Gemini Live."""
     INTERACTIONS_API = "INTERACTIONS_API"
-    """Scope for Interactions API."""
+    """The scope for Interactions API."""
     RESPONSE_API = "RESPONSE_API"
-    """Scope for Response API."""
+    """The scope for Response API."""
 
 
 class EvaluationExperimentMergeStrategy(_common.CaseInSensitiveEnum):
@@ -18532,6 +18543,35 @@ SandboxEnvironmentTemplatePersistentDiskConfigOrDict = Union[
 ]
 
 
+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.""",
+    )
+    update_strategy: Optional[UpdateStrategy] = Field(
+        default=None,
+        description="""Optional. The strategy for updating warm pool instances when the template is updated.""",
+    )
+
+
+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."""
+
+    update_strategy: Optional[UpdateStrategy]
+    """Optional. The strategy for updating warm pool instances when the template is updated."""
+
+
+SandboxEnvironmentTemplateWarmPoolConfigOrDict = Union[
+    SandboxEnvironmentTemplateWarmPoolConfig,
+    SandboxEnvironmentTemplateWarmPoolConfigDict,
+]
+
+
 class SandboxEnvironmentTemplate(_common.BaseModel):
     """A sandbox environment template."""
 
@@ -18587,6 +18627,10 @@ class SandboxEnvironmentTemplate(_common.BaseModel):
             description="""Optional. Configuration for attaching a persistent disk (PD) to each SandboxEnvironment created from this template. When unset (or when `enabled` is `false`), sandboxes created from this template are not backed by a persistent disk and rely on ephemeral storage only. See PersistentDiskConfig for details.""",
         )
     )
+    warm_pool_config: Optional[SandboxEnvironmentTemplateWarmPoolConfig] = Field(
+        default=None,
+        description="""Optional. The configuration for the warm pool of this template.""",
+    )
 
 
 class SandboxEnvironmentTemplateDict(TypedDict, total=False):
@@ -18629,6 +18673,9 @@ class SandboxEnvironmentTemplateDict(TypedDict, total=False):
     persistent_disk_config: Optional[SandboxEnvironmentTemplatePersistentDiskConfigDict]
     """Optional. Configuration for attaching a persistent disk (PD) to each SandboxEnvironment created from this template. When unset (or when `enabled` is `false`), sandboxes created from this template are not backed by a persistent disk and rely on ephemeral storage only. See PersistentDiskConfig for details."""
 
+    warm_pool_config: Optional[SandboxEnvironmentTemplateWarmPoolConfigDict]
+    """Optional. The configuration for the warm pool of this template."""
+
 
 SandboxEnvironmentTemplateOrDict = Union[
     SandboxEnvironmentTemplate, SandboxEnvironmentTemplateDict
@@ -26080,6 +26127,10 @@ class DeployRequestDeployConfig(_common.BaseModel):
         default=None,
         description="""Optional. System labels for Model Garden deployments. These labels are managed by Google and for tracking purposes only.""",
     )
+    service_account: Optional[str] = Field(
+        default=None,
+        description="""Optional. The service account that the container of the deployed model runs as. Specify the email address of the service account. If not specified, the container runs as a service account that has no access to the resource project. Users deploying the model must have the `iam.serviceAccounts.actAs` permission on this service account.""",
+    )
 
 
 class DeployRequestDeployConfigDict(TypedDict, total=False):
@@ -26094,6 +26145,9 @@ class DeployRequestDeployConfigDict(TypedDict, total=False):
     system_labels: Optional[dict[str, str]]
     """Optional. System labels for Model Garden deployments. These labels are managed by Google and for tracking purposes only."""
 
+    service_account: Optional[str]
+    """Optional. The service account that the container of the deployed model runs as. Specify the email address of the service account. If not specified, the container runs as a service account that has no access to the resource project. Users deploying the model must have the `iam.serviceAccounts.actAs` permission on this service account."""
+
 
 DeployRequestDeployConfigOrDict = Union[
     DeployRequestDeployConfig, DeployRequestDeployConfigDict

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/agentplatform/frameworks/adk.py

@@ -1336,8 +1336,12 @@ class AdkApp:
         """Streams responses asynchronously from the ADK application.
 
         Args:
-            message (str):
-                Required. The message to stream responses for.
+            message (Union[str, Dict[str, Any]]):
+                Required. The message to stream responses for. Either a text
+                string or a dictionary representing a
+                `google.genai.types.Content` object (for example,
+                `content.model_dump(exclude_none=True, mode="json")` for
+                multimodal inputs).
             user_id (str):
                 Required. The ID of the user.
             session_id (str):
@@ -1362,7 +1366,7 @@ class AdkApp:
 
         Raises:
             TypeError: If message is not a string or a dictionary representing
-            a Content object.
+                a Content object.
             ValueError: If both session_id and session_events are specified.
         """
         from agentplatform._genai import _runtimes_utils

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/agentplatform/version.py

@@ -14,4 +14,4 @@
 #
 """Version file for agentplatform."""
 
-__version__ = "2.8.0"
+__version__ = "2.9.0"

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0/google_cloud_agentplatform.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-agentplatform
-Version: 2.8.0
+Version: 2.9.0
 Summary: Google Cloud Agent Platform Client Library
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache-2.0
@@ -61,7 +61,7 @@ Provides-Extra: evaluation
 Requires-Dist: pandas>=1.0.0; extra == "evaluation"
 Requires-Dist: tqdm>=4.23.0; extra == "evaluation"
 Requires-Dist: pyyaml; extra == "evaluation"
-Requires-Dist: litellm<1.107.0,>=1.106.5; python_version < "3.15" and extra == "evaluation"
+Requires-Dist: litellm!=1.107.0,<1.110.0,>=1.106.5; python_version < "3.15" and extra == "evaluation"
 Requires-Dist: google-cloud-storage<4.0.0,>=1.32.0; extra == "evaluation"
 Provides-Extra: bigquery
 Requires-Dist: google-cloud-bigquery<4.0.0,>=1.15.0; extra == "bigquery"

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/google_cloud_agentplatform.egg-info/requires.txt

@@ -46,7 +46,7 @@ pyyaml
 google-cloud-storage<4.0.0,>=1.32.0
 
 [evaluation:python_version < "3.15"]
-litellm<1.107.0,>=1.106.5
+litellm!=1.107.0,<1.110.0,>=1.106.5
 
 [langchain]
 google-cloud-agentplatform[agent_engines]

{google_cloud_agentplatform-2.8.0 → google_cloud_agentplatform-2.9.0}/pyproject.toml

@@ -80,9 +80,9 @@ evaluation = [
     "pandas >= 1.0.0",
     "tqdm >= 4.23.0",
     "pyyaml",
-    # Lower bound >=1.106.5 includes security advisory patches; litellm
-    # requires Python <3.15.
-    "litellm >= 1.106.5, < 1.107.0; python_version < '3.15'",
+    # Lower bound >=1.106.5 includes security advisory patches; 1.107.0 was
+    # compromised; litellm requires Python <3.15.
+    "litellm >= 1.106.5, != 1.107.0, < 1.110.0; python_version < '3.15'",
     "google-cloud-storage >= 1.32.0, < 4.0.0",
 ]
 bigquery = [