From 2c844a82b0c17feaffe7cc193f9b86093a9b3dfc Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Tue, 12 May 2026 11:57:11 -0700 Subject: [PATCH] chore: Update Skill Registry tests to target prod endpoints PiperOrigin-RevId: 914404430 --- tests/unit/vertexai/genai/replays/test_skills_delete.py | 6 +----- tests/unit/vertexai/genai/replays/test_skills_get.py | 9 ++------- tests/unit/vertexai/genai/replays/test_skills_list.py | 6 +----- tests/unit/vertexai/genai/replays/test_skills_update.py | 8 -------- 4 files changed, 4 insertions(+), 25 deletions(-) diff --git a/tests/unit/vertexai/genai/replays/test_skills_delete.py b/tests/unit/vertexai/genai/replays/test_skills_delete.py index 92bb1a8228..298ef3bfed 100644 --- a/tests/unit/vertexai/genai/replays/test_skills_delete.py +++ b/tests/unit/vertexai/genai/replays/test_skills_delete.py @@ -1,4 +1,4 @@ -"""Tests the skills.delete() method against the autopush endpoint.""" +"""Tests the skills.delete() method against the prod endpoint.""" from tests.unit.vertexai.genai.replays import pytest_helper from vertexai._genai import types @@ -12,10 +12,6 @@ def test_delete_skill(client, tmp_path): - # 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" - ) # 1. Create a fresh unique skill first with open(tmp_path / "SKILL.md", "w") as f: diff --git a/tests/unit/vertexai/genai/replays/test_skills_get.py b/tests/unit/vertexai/genai/replays/test_skills_get.py index 824c0921f1..24ff223caa 100644 --- a/tests/unit/vertexai/genai/replays/test_skills_get.py +++ b/tests/unit/vertexai/genai/replays/test_skills_get.py @@ -1,4 +1,4 @@ -"""Tests the skills.get() method against the autopush endpoint.""" +"""Tests the skills.get() method against the prod endpoint.""" from google.api_core import exceptions from tests.unit.vertexai.genai.replays import pytest_helper @@ -7,8 +7,6 @@ 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( @@ -18,11 +16,8 @@ def test_get_skill(client): # client fixture is injected by pytest_helper.setup - """Tests the skills.get() method against the autopush endpoint.""" + """Tests the skills.get() method against the prod 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: diff --git a/tests/unit/vertexai/genai/replays/test_skills_list.py b/tests/unit/vertexai/genai/replays/test_skills_list.py index 173095cc75..4a412783a9 100644 --- a/tests/unit/vertexai/genai/replays/test_skills_list.py +++ b/tests/unit/vertexai/genai/replays/test_skills_list.py @@ -1,4 +1,4 @@ -"""Tests the skills.list() method against the autopush endpoint.""" +"""Tests the skills.list() method against the prod endpoint.""" from tests.unit.vertexai.genai.replays import pytest_helper from vertexai._genai import types @@ -10,10 +10,6 @@ def test_list_skills(client): - # 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" - ) skills = client.skills.list() for skill in skills: diff --git a/tests/unit/vertexai/genai/replays/test_skills_update.py b/tests/unit/vertexai/genai/replays/test_skills_update.py index 6d72897989..ef7affd03c 100644 --- a/tests/unit/vertexai/genai/replays/test_skills_update.py +++ b/tests/unit/vertexai/genai/replays/test_skills_update.py @@ -28,10 +28,6 @@ def test_update_skill(client, tmp_path): - # 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" - ) # 1. Create a fresh unique skill first with open(tmp_path / "SKILL.md", "w") as f: @@ -61,10 +57,6 @@ def test_update_skill(client, tmp_path): def test_update_skill_with_zipped_bytes(client, tmp_path): - # 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" - ) # 1. Create a fresh unique skill first with open(tmp_path / "SKILL.md", "w") as f: