From 0eda91f7469f412efc609c8044fbab043d9691bf Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Mon, 19 Jan 2026 05:59:11 -0800 Subject: [PATCH] fix: test fix internal PiperOrigin-RevId: 858157754 --- ...t_create_agent_engine_developer_connect.py | 52 ++++++++++++++----- 1 file changed, 39 insertions(+), 13 deletions(-) diff --git a/tests/unit/vertexai/genai/replays/test_create_agent_engine_developer_connect.py b/tests/unit/vertexai/genai/replays/test_create_agent_engine_developer_connect.py index ab6f7af4cf..1b66399a48 100644 --- a/tests/unit/vertexai/genai/replays/test_create_agent_engine_developer_connect.py +++ b/tests/unit/vertexai/genai/replays/test_create_agent_engine_developer_connect.py @@ -14,6 +14,8 @@ # # pylint: disable=protected-access,bad-continuation,missing-function-docstring +import sys + from tests.unit.vertexai.genai.replays import pytest_helper from vertexai._genai import types @@ -23,13 +25,37 @@ def test_create_with_developer_connect_source(client): - """Tests creating an agent engine with developer connect source.""" - developer_connect_source_config = types.ReasoningEngineSpecSourceCodeSpecDeveloperConnectConfig( - git_repository_link="projects/reasoning-engine-test-1/locations/europe-west3/connections/shawn-develop-connect/gitRepositoryLinks/shawn-yang-google-adk-samples", - revision="main", - dir="test", - ) - agent_engine = client.agent_engines.create( + """Tests creating an agent engine with developer connect source.""" + if sys.version_info >= (3, 13): + try: + client._api_client._initialize_replay_session_if_not_loaded() + if client._api_client.replay_session: + target_ver = f"{sys.version_info.major}.{sys.version_info.minor}" + for interaction in client._api_client.replay_session.interactions: + + def _update_ver(obj): + if isinstance(obj, dict): + if "python_spec" in obj and isinstance(obj["python_spec"], dict): + if "version" in obj["python_spec"]: + obj["python_spec"]["version"] = target_ver + for v in obj.values(): + _update_ver(v) + elif isinstance(obj, list): + for item in obj: + _update_ver(item) + + if hasattr(interaction.request, "body_segments"): + _update_ver(interaction.request.body_segments) + if hasattr(interaction.request, "body"): + _update_ver(interaction.request.body) + except Exception: + pass + developer_connect_source_config = types.ReasoningEngineSpecSourceCodeSpecDeveloperConnectConfig( + git_repository_link="projects/reasoning-engine-test-1/locations/europe-west3/connections/shawn-develop-connect/gitRepositoryLinks/shawn-yang-google-adk-samples", + revision="main", + dir="test", + ) + agent_engine = client.agent_engines.create( config={ "display_name": "test-agent-engine-dev-connect", "developer_connect_source": developer_connect_source_config, @@ -42,21 +68,21 @@ def test_create_with_developer_connect_source(client): }, }, ) - assert agent_engine.api_resource.display_name == "test-agent-engine-dev-connect" - assert ( + assert agent_engine.api_resource.display_name == "test-agent-engine-dev-connect" + assert ( agent_engine.api_resource.spec.source_code_spec.developer_connect_source.config.git_repository_link == developer_connect_source_config.git_repository_link ) - assert ( + assert ( agent_engine.api_resource.spec.source_code_spec.developer_connect_source.config.revision == developer_connect_source_config.revision ) - assert ( + assert ( agent_engine.api_resource.spec.source_code_spec.developer_connect_source.config.dir == developer_connect_source_config.dir ) - # Clean up resources. - client.agent_engines.delete(name=agent_engine.api_resource.name, force=True) + # Clean up resources. + client.agent_engines.delete(name=agent_engine.api_resource.name, force=True) pytestmark = pytest_helper.setup(