From 5beb9f19316a6b4e1b781b111878631dab5d9e03 Mon Sep 17 00:00:00 2001 From: Amer Elsheikh Date: Mon, 12 Jan 2026 04:58:08 -0800 Subject: [PATCH] internal test change. fix: test fix PiperOrigin-RevId: 855176412 --- .../genai/replays/test_create_agent_engine.py | 25 +++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/unit/vertexai/genai/replays/test_create_agent_engine.py b/tests/unit/vertexai/genai/replays/test_create_agent_engine.py index e68373ac5f..477c903d07 100644 --- a/tests/unit/vertexai/genai/replays/test_create_agent_engine.py +++ b/tests/unit/vertexai/genai/replays/test_create_agent_engine.py @@ -16,6 +16,7 @@ import os import re +import sys from tests.unit.vertexai.genai.replays import pytest_helper from vertexai._genai import types @@ -124,6 +125,30 @@ def test_create_with_source_packages( mock_agent_engine_create_path_exists, ): """Tests creating an agent engine with source packages.""" + 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 with ( mock_agent_engine_create_base64_encoded_tarball, mock_agent_engine_create_path_exists,