From 0fc74dea6d9e8826f29b7b9c1a10f348be911f6a Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Tue, 21 Oct 2025 18:42:02 -0700 Subject: [PATCH] fix: GenAI SDK client - Fix log showing how to get an Agent Engine. Positional arguments are not allowed. PiperOrigin-RevId: 822361058 --- tests/unit/vertexai/genai/test_agent_engines.py | 4 ++-- vertexai/_genai/agent_engines.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unit/vertexai/genai/test_agent_engines.py b/tests/unit/vertexai/genai/test_agent_engines.py index 5da202dfb2..0fbdb6b53f 100644 --- a/tests/unit/vertexai/genai/test_agent_engines.py +++ b/tests/unit/vertexai/genai/test_agent_engines.py @@ -1263,7 +1263,7 @@ def test_create_agent_engine(self, mock_await_operation, mock_prepare, caplog): assert "View progress and logs at" in caplog.text assert "Agent Engine created. To use it in another session:" in caplog.text assert ( - f"agent_engine=client.agent_engines.get(" + f"agent_engine=client.agent_engines.get(name=" f"'{_TEST_AGENT_ENGINE_RESOURCE_NAME}')" in caplog.text ) @@ -1687,7 +1687,7 @@ def test_update_agent_engine_requirements( assert "Agent Engine updated. To use it in another session:" in caplog.text assert ( f"agent_engine=client.agent_engines.get(" - f"'{_TEST_AGENT_ENGINE_RESOURCE_NAME}')" in caplog.text + f"name='{_TEST_AGENT_ENGINE_RESOURCE_NAME}')" in caplog.text ) @mock.patch.object(_agent_engines_utils, "_prepare") diff --git a/vertexai/_genai/agent_engines.py b/vertexai/_genai/agent_engines.py index 3612c12d70..f5eecc853a 100644 --- a/vertexai/_genai/agent_engines.py +++ b/vertexai/_genai/agent_engines.py @@ -895,7 +895,7 @@ def create( if agent_engine.api_resource: logger.info("Agent Engine created. To use it in another session:") logger.info( - f"agent_engine=client.agent_engines.get('{agent_engine.api_resource.name}')" + f"agent_engine=client.agent_engines.get(name='{agent_engine.api_resource.name}')" ) elif operation.error: raise RuntimeError(f"Failed to create Agent Engine: {operation.error}") @@ -1313,7 +1313,7 @@ def update( if agent_engine.api_resource: logger.info("Agent Engine updated. To use it in another session:") logger.info( - f"agent_engine=client.agent_engines.get('{agent_engine.api_resource.name}')" + f"agent_engine=client.agent_engines.get(name='{agent_engine.api_resource.name}')" ) elif operation.error: raise RuntimeError(f"Failed to update Agent Engine: {operation.error}")