feat: Add support for snapshots for Sandboxes in Vertex AI GenAI SDK.

PiperOrigin-RevId: 897294488

Author: vertex-sdk-bot

tests/unit/vertexai/genai/replays/test_ae_sandbox_snapshots_create.py

@@ -0,0 +1,40 @@
+"""Tests for Sandbox Snapshot create operation."""
+
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# pylint: disable=protected-access,bad-continuation,missing-function-docstring,bad-indentation
+
+from tests.unit.vertexai.genai.replays import pytest_helper
+from vertexai._genai import types
+
+
+def test_create_sandbox_snapshot(client):
+    snapshot = client.agent_engines.sandboxes.snapshots.create(
+        source_sandbox_environment_name="projects/802583348448/locations/us-central1/reasoningEngines/6130241318758121472/sandboxEnvironments/525190525100228608",
+        config={
+            "display_name": "test_snapshot",
+            "ttl": "3600s",
+            "owner": "test_owner",
+        },
+    )
+
+    assert isinstance(snapshot, types.AgentEngineSandboxSnapshotOperation)
+
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method="agent_engines.sandboxes.snapshots.create",
+)

tests/unit/vertexai/genai/replays/test_ae_sandbox_snapshots_delete.py

@@ -0,0 +1,35 @@
+"""Tests for Sandbox Snapshot delete operation."""
+
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# pylint: disable=protected-access,bad-continuation,missing-function-docstring,bad-indentation
+
+from tests.unit.vertexai.genai.replays import pytest_helper
+from vertexai._genai import types
+
+
+def test_delete_sandbox_snapshot(client):
+    result = client.agent_engines.sandboxes.snapshots.delete(
+        name="projects/802583348448/locations/us-central1/reasoningEngines/6130241318758121472/sandboxEnvironmentSnapshots/421086565159141376",
+    )
+
+    assert isinstance(result, types.DeleteSandboxEnvironmentSnapshotOperation)
+
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method="agent_engines.sandboxes.snapshots.delete",
+)

tests/unit/vertexai/genai/replays/test_ae_sandbox_snapshots_get.py

@@ -0,0 +1,37 @@
+"""Tests for Sandbox Snapshot get operation."""
+
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# pylint: disable=protected-access,bad-continuation,missing-function-docstring,bad-indentation
+
+from tests.unit.vertexai.genai.replays import pytest_helper
+from vertexai._genai import types
+
+
+def test_get_sandbox_snapshot(client):
+    snapshot_name = "projects/802583348448/locations/us-central1/reasoningEngines/6130241318758121472/sandboxEnvironmentSnapshots/421086565159141376"
+    fetched_snapshot = client.agent_engines.sandboxes.snapshots.get(
+        name=snapshot_name,
+    )
+
+    assert isinstance(fetched_snapshot, types.SandboxEnvironmentSnapshot)
+    assert fetched_snapshot.name == snapshot_name
+
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method="agent_engines.sandboxes.snapshots.get",
+)

tests/unit/vertexai/genai/replays/test_ae_sandbox_snapshots_list.py

@@ -0,0 +1,41 @@
+"""Tests for Sandbox Snapshot list operation."""
+
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# pylint: disable=protected-access,bad-continuation,missing-function-docstring,bad-indentation
+
+from tests.unit.vertexai.genai.replays import pytest_helper
+from vertexai._genai import types
+
+
+def test_list_sandbox_snapshots(client):
+    snapshots_list_operation = client.agent_engines.sandboxes.snapshots.list(
+        name="projects/802583348448/locations/us-central1/reasoningEngines/6130241318758121472",
+    )
+
+    assert isinstance(
+        snapshots_list_operation.sandbox_environment_snapshots[0],
+        types.SandboxEnvironmentSnapshot,
+    )
+    assert isinstance(
+        snapshots_list_operation, types.ListSandboxEnvironmentSnapshotsResponse
+    )
+
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method="agent_engines.sandboxes.snapshots.list",
+)