feat: Add Create Skill method for Vertex AI Skill Registry SDK

PiperOrigin-RevId: 910161153

Author: vertex-sdk-bot

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

@@ -0,0 +1,78 @@
+# 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.
+#
+"""Tests the skills.create() method against the Vertex AI endpoint using replays."""
+
+import io
+import os
+import tempfile
+import zipfile
+
+from tests.unit.vertexai.genai.replays import pytest_helper
+from vertexai._genai import types
+
+# MANDATORY: Initialize the replay test framework for this module
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+)
+
+
+def test_create_skill(client):
+    """Tests the creation of a skill using `client.skills.create()`."""
+    # Target the autopush sandbox endpoint for the Skill Registry API
+    client._api_client._http_options.base_url = (
+        "https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
+    )
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        # Create a dummy skill structure (SKILL.md is required by the spec)
+        with open(os.path.join(tmpdir, "SKILL.md"), "w") as f:
+            f.write("# My Replay Skill\nThis is a test skill for replay tests.")
+
+        skill = client.skills.create(
+            display_name="My Replay Skill",
+            description="My Replay Skill Description",
+            config=types.CreateSkillConfig(local_path=tmpdir, wait_for_completion=True),
+        )
+
+        assert skill.name is not None
+        assert skill.display_name == "My Replay Skill"
+        assert skill.description == "My Replay Skill Description"
+
+
+def test_create_skill_with_prezipped_bytes(client):
+    """Tests the creation of a skill with pre-zipped bytes."""
+    # Target the autopush sandbox endpoint for the Skill Registry API
+    client._api_client._http_options.base_url = (
+        "https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
+    )
+
+    zip_buffer = io.BytesIO()
+    zinfo = zipfile.ZipInfo("SKILL.md", date_time=(1980, 1, 1, 0, 0, 0))
+    with zipfile.ZipFile(zip_buffer, "w") as zip_file:
+        zip_file.writestr(zinfo, "# My Zipped Replay Skill\nThis is a test.")
+    zipped_bytes = zip_buffer.getvalue()
+
+    skill = client.skills.create(
+        display_name="My Zipped Replay Skill",
+        description="My Zipped Replay Skill Description",
+        config=types.CreateSkillConfig(
+            zipped_filesystem=zipped_bytes, wait_for_completion=True
+        ),
+    )
+
+    assert skill.name is not None
+    assert skill.display_name == "My Zipped Replay Skill"
+    assert skill.description == "My Zipped Replay Skill Description"

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

@@ -0,0 +1,33 @@
+"""Tests the skills.get() method against the autopush endpoint."""
+
+from google.api_core import exceptions
+from tests.unit.vertexai.genai.replays import pytest_helper
+import pytest
+
+PROJECT_ID = "demo-project"
+REGION = "us-central1"
+SKILL_ID = "7184367305562783744"
+# target the autopush sandbox endpoint for the Skill Registry API
+ENDPOINT = f"{REGION}-autopush-aiplatform.sandbox.googleapis.com"
+
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+)
+
+
+def test_get_skill(client):  # client fixture is injected by pytest_helper.setup
+    """Tests the skills.get() method against the autopush endpoint."""
+
+    client._api_client._http_options.base_url = (
+        "https://us-central1-autopush-aiplatform.sandbox.googleapis.com"
+    )
+    skill_name = f"projects/{PROJECT_ID}/locations/{REGION}/skills/{SKILL_ID}"
+
+    try:
+        skill = client.skills.get(name=skill_name)
+        assert skill.name == skill_name
+
+    except exceptions.GoogleAPIError as e:
+        pytest.fail(f"Error calling client.skills.get(): {e}")

tests/unit/vertexai/genai/test_genai_skills.py

@@ -0,0 +1,44 @@
+# //third_party/py/google/cloud/aiplatform/tests/unit/vertexai/genai/test_genai_skills.py
+import json
+from unittest import mock
+from vertexai import _genai as genai
+from vertexai._genai import client as vertexai_client
+from google.genai import types as genai_types
+import pytest
+
+
+@pytest.fixture
+def skills_client():
+    creds = mock.MagicMock()
+    creds.token = "test_token"
+    client = vertexai_client.Client(
+        project="test-project", location="test-location", credentials=creds
+    )
+    return client.skills
+
+
+class TestGenaiSkills:
+    mock_get_skill_response = {
+        "name": "projects/test-project/locations/test-location/skills/test-skill",
+        "displayName": "My Test Skill",
+    }
+
+    def test_get_skill(self, skills_client):
+        """Tests the get_skill method."""
+        with mock.patch.object(skills_client._api_client, "request") as request_mock:
+            request_mock.return_value = genai_types.HttpResponse(
+                body=json.dumps(self.mock_get_skill_response)
+            )
+            skill_name = (
+                "projects/test-project/locations/test-location/skills/test-skill"
+            )
+            skill = skills_client.get(name=skill_name)
+            request_mock.assert_called_with(
+                "get",
+                skill_name,
+                {"_url": {"name": skill_name}},
+                None,
+            )
+            assert isinstance(skill, genai.types.Skill)
+            assert skill.name == skill_name
+            assert skill.display_name == "My Test Skill"

vertexai/_genai/_skills_utils.py

@@ -0,0 +1,116 @@
+# 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.
+#
+"""Utility functions for Skills."""
+
+import asyncio
+import base64
+import io
+import os
+import time
+from typing import Any, Awaitable, Callable
+import zipfile
+
+
+def zip_directory(directory_path: str) -> bytes:
+    """Zips a directory into memory and returns the bytes.
+
+    Args:
+        directory_path (str): Required. The local path to the directory.
+
+    Returns:
+        bytes: The zipped directory content.
+    """
+    if not os.path.isdir(directory_path):
+        raise ValueError(f"Path is not a directory: {directory_path}")
+
+    zip_buffer = io.BytesIO()
+    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
+        for root, _, files in os.walk(directory_path):
+            for file in files:
+                file_path = os.path.join(root, file)
+                arcname = os.path.relpath(file_path, directory_path)
+
+                # Read actual file data
+                with open(file_path, "rb") as f:
+                    file_data = f.read()
+
+                # Use deterministic ZipInfo (mtime: 1980-01-01 00:00:00)
+                zinfo = zipfile.ZipInfo(arcname, date_time=(1980, 1, 1, 0, 0, 0))
+                zinfo.compress_type = zipfile.ZIP_DEFLATED
+                zinfo.external_attr = 0o644 << 16  # Constant file permissions
+
+                zip_file.writestr(zinfo, file_data)
+    return zip_buffer.getvalue()
+
+
+def get_zipped_filesystem_payload(directory_path: str) -> str:
+    """Zips a directory and base64-encodes the result to a UTF-8 string.
+
+    Args:
+        directory_path (str): Required. The local path to the directory.
+
+    Returns:
+        str: The base64-encoded zipped directory.
+    """
+    zip_bytes = zip_directory(directory_path)
+    return base64.b64encode(zip_bytes).decode("utf-8")
+
+
+def await_operation(
+    *,
+    operation_name: str,
+    get_operation_fn: Callable[..., Any],
+    poll_interval_seconds: float = 10.0,
+) -> Any:
+    """Waits for a long running operation to complete.
+
+    Args:
+        operation_name (str): Required. The name of the operation.
+        get_operation_fn (Callable): Required. Function to get the operation
+          status.
+        poll_interval_seconds (float): The interval between polls in seconds.
+
+    Returns:
+        Any: The completed operation.
+    """
+    operation = get_operation_fn(operation_name=operation_name)
+    while not operation.done:
+        time.sleep(poll_interval_seconds)
+        operation = get_operation_fn(operation_name=operation.name)
+    return operation
+
+
+async def await_operation_async(
+    *,
+    operation_name: str,
+    get_operation_fn: Callable[..., Awaitable[Any]],
+    poll_interval_seconds: float = 10.0,
+) -> Any:
+    """Waits for a long running operation to complete asynchronously.
+
+    Args:
+        operation_name (str): Required. The name of the operation.
+        get_operation_fn (Callable): Required. Async function to get the operation
+          status.
+        poll_interval_seconds (float): The interval between polls in seconds.
+
+    Returns:
+        Any: The completed operation.
+    """
+    operation = await get_operation_fn(operation_name=operation_name)
+    while not operation.done:
+        await asyncio.sleep(poll_interval_seconds)
+        operation = await get_operation_fn(operation_name=operation.name)
+    return operation

vertexai/_genai/client.py

@@ -35,6 +35,7 @@
         prompt_optimizer as prompt_optimizer_module,
     )
     from vertexai._genai import prompts as prompts_module
+    from vertexai._genai import skills as skills_module
     from vertexai._genai import live as live_module
 
 
@@ -52,6 +53,7 @@ def __init__(self, api_client: genai_client.BaseApiClient):  # type: ignore[name
         self._prompt_optimizer: Optional[ModuleType] = None
         self._prompts: Optional[ModuleType] = None
         self._datasets: Optional[ModuleType] = None
+        self._skills: Optional[ModuleType] = None
 
     @property
     @_common.experimental_warning(
@@ -124,6 +126,15 @@ def datasets(self) -> "datasets_module.AsyncDatasets":
             )
         return self._datasets.AsyncDatasets(self._api_client)  # type: ignore[no-any-return]
 
+    @property
+    def skills(self) -> "skills_module.AsyncSkills":
+        if self._skills is None:
+            self._skills = importlib.import_module(
+                ".skills",
+                __package__,
+            )
+        return self._skills.AsyncSkills(self._api_client)  # type: ignore[no-any-return]
+
     async def aclose(self) -> None:
         """Closes the async client explicitly.
 
@@ -239,6 +250,7 @@ def __init__(
         self._agent_engines: Optional[ModuleType] = None
         self._prompts: Optional[ModuleType] = None
         self._datasets: Optional[ModuleType] = None
+        self._skills: Optional[ModuleType] = None
 
     @property
     def evals(self) -> "evals_module.Evals":
@@ -335,3 +347,12 @@ def datasets(self) -> "datasets_module.Datasets":
                 __package__,
             )
         return self._datasets.Datasets(self._api_client)  # type: ignore[no-any-return]
+
+    @property
+    def skills(self) -> "skills_module.Skills":
+        if self._skills is None:
+            self._skills = importlib.import_module(
+                ".skills",
+                __package__,
+            )
+        return self._skills.Skills(self._api_client)  # type: ignore[no-any-return]