From 49754a671beef1287702d59eec8db316cedb2760 Mon Sep 17 00:00:00 2001 From: Matthew Tang Date: Wed, 4 Mar 2026 11:18:48 -0800 Subject: [PATCH] feat: Add GenAI Client and basic Agent Engine module PiperOrigin-RevId: 878581336 --- vertexai/_genai/a2a_task_events.py | 11 +----- vertexai/_genai/a2a_tasks.py | 19 +--------- vertexai/_genai/agent_engines.py | 37 ++---------------- vertexai/_genai/memories.py | 58 ++++------------------------- vertexai/_genai/memory_revisions.py | 11 +----- vertexai/_genai/sandboxes.py | 28 ++------------ vertexai/_genai/session_events.py | 16 ++------ vertexai/_genai/sessions.py | 33 +++------------- 8 files changed, 32 insertions(+), 181 deletions(-) diff --git a/vertexai/_genai/a2a_task_events.py b/vertexai/_genai/a2a_task_events.py index 1b7c7093cf..8113427370 100644 --- a/vertexai/_genai/a2a_task_events.py +++ b/vertexai/_genai/a2a_task_events.py @@ -50,9 +50,6 @@ def _AppendAgentEngineTaskEventRequestParameters_to_vertex( [item for item in getv(from_object, ["task_events"])], ) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -95,12 +92,8 @@ def _ListAgentEngineTaskEventsRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineTaskEventsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _ListAgentEngineTaskEventsConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object diff --git a/vertexai/_genai/a2a_tasks.py b/vertexai/_genai/a2a_tasks.py index 1f88202f7d..5a83a16aac 100644 --- a/vertexai/_genai/a2a_tasks.py +++ b/vertexai/_genai/a2a_tasks.py @@ -75,13 +75,7 @@ def _CreateAgentEngineTaskRequestParameters_to_vertex( setv(to_object, ["_query", "a2a_task_id"], getv(from_object, ["a2a_task_id"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _CreateAgentEngineTaskConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) + _CreateAgentEngineTaskConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object @@ -94,9 +88,6 @@ def _GetAgentEngineTaskRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -130,13 +121,7 @@ def _ListAgentEngineTasksRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineTasksConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) + _ListAgentEngineTasksConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object diff --git a/vertexai/_genai/agent_engines.py b/vertexai/_genai/agent_engines.py index 819ad034d9..ca6d440da0 100644 --- a/vertexai/_genai/agent_engines.py +++ b/vertexai/_genai/agent_engines.py @@ -119,13 +119,7 @@ def _CreateAgentEngineRequestParameters_to_vertex( ) -> dict[str, Any]: to_object: dict[str, Any] = {} if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _CreateAgentEngineConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) + _CreateAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object @@ -141,9 +135,6 @@ def _DeleteAgentEngineRequestParameters_to_vertex( if getv(from_object, ["force"]) is not None: setv(to_object, ["force"], getv(from_object, ["force"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -157,9 +148,6 @@ def _GetAgentEngineOperationParameters_to_vertex( to_object, ["_url", "operationName"], getv(from_object, ["operation_name"]) ) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -171,9 +159,6 @@ def _GetAgentEngineRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -201,11 +186,7 @@ def _ListAgentEngineRequestParameters_to_vertex( ) -> dict[str, Any]: to_object: dict[str, Any] = {} if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) + _ListAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object @@ -237,11 +218,7 @@ def _QueryAgentEngineRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _QueryAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object), - ) + _QueryAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object @@ -324,13 +301,7 @@ def _UpdateAgentEngineRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _UpdateAgentEngineConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) + _UpdateAgentEngineConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object diff --git a/vertexai/_genai/memories.py b/vertexai/_genai/memories.py index 4b48507e36..e57ef42239 100644 --- a/vertexai/_genai/memories.py +++ b/vertexai/_genai/memories.py @@ -108,13 +108,7 @@ def _CreateAgentEngineMemoryRequestParameters_to_vertex( setv(to_object, ["scope"], getv(from_object, ["scope"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _AgentEngineMemoryConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) + _AgentEngineMemoryConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object @@ -127,9 +121,6 @@ def _DeleteAgentEngineMemoryRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -216,12 +207,8 @@ def _GenerateAgentEngineMemoriesRequestParameters_to_vertex( setv(to_object, ["scope"], getv(from_object, ["scope"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _GenerateAgentEngineMemoriesConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _GenerateAgentEngineMemoriesConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object @@ -237,9 +224,6 @@ def _GetAgentEngineGenerateMemoriesOperationParameters_to_vertex( to_object, ["_url", "operationName"], getv(from_object, ["operation_name"]) ) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -253,9 +237,6 @@ def _GetAgentEngineMemoryOperationParameters_to_vertex( to_object, ["_url", "operationName"], getv(from_object, ["operation_name"]) ) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -267,9 +248,6 @@ def _GetAgentEngineMemoryRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -303,13 +281,7 @@ def _ListAgentEngineMemoryRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineMemoryConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), - ) + _ListAgentEngineMemoryConfig_to_vertex(getv(from_object, ["config"]), to_object) return to_object @@ -335,9 +307,6 @@ def _PurgeAgentEngineMemoriesRequestParameters_to_vertex( if getv(from_object, ["force"]) is not None: setv(to_object, ["force"], getv(from_object, ["force"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -386,12 +355,8 @@ def _RetrieveAgentEngineMemoriesRequestParameters_to_vertex( ) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _RetrieveAgentEngineMemoriesConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _RetrieveAgentEngineMemoriesConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object @@ -408,9 +373,6 @@ def _RollbackAgentEngineMemoryRequestParameters_to_vertex( if getv(from_object, ["target_revision_id"]) is not None: setv(to_object, ["targetRevisionId"], getv(from_object, ["target_revision_id"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -484,12 +446,8 @@ def _UpdateAgentEngineMemoryRequestParameters_to_vertex( setv(to_object, ["scope"], getv(from_object, ["scope"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _UpdateAgentEngineMemoryConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _UpdateAgentEngineMemoryConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object diff --git a/vertexai/_genai/memory_revisions.py b/vertexai/_genai/memory_revisions.py index 765ac32347..c71c5cfcc7 100644 --- a/vertexai/_genai/memory_revisions.py +++ b/vertexai/_genai/memory_revisions.py @@ -43,9 +43,6 @@ def _GetAgentEngineMemoryRevisionRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -76,12 +73,8 @@ def _ListAgentEngineMemoryRevisionsRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineMemoryRevisionsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _ListAgentEngineMemoryRevisionsConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object diff --git a/vertexai/_genai/sandboxes.py b/vertexai/_genai/sandboxes.py index d4f8db5497..6b844848cf 100644 --- a/vertexai/_genai/sandboxes.py +++ b/vertexai/_genai/sandboxes.py @@ -72,12 +72,8 @@ def _CreateAgentEngineSandboxRequestParameters_to_vertex( setv(to_object, ["spec"], getv(from_object, ["spec"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _CreateAgentEngineSandboxConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _CreateAgentEngineSandboxConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object @@ -91,9 +87,6 @@ def _DeleteAgentEngineSandboxRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -108,9 +101,6 @@ def _ExecuteCodeAgentEngineSandboxRequestParameters_to_vertex( if getv(from_object, ["inputs"]) is not None: setv(to_object, ["inputs"], [item for item in getv(from_object, ["inputs"])]) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -124,9 +114,6 @@ def _GetAgentEngineSandboxOperationParameters_to_vertex( to_object, ["_url", "operationName"], getv(from_object, ["operation_name"]) ) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -138,9 +125,6 @@ def _GetAgentEngineSandboxRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -171,12 +155,8 @@ def _ListAgentEngineSandboxesRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineSandboxesConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _ListAgentEngineSandboxesConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object diff --git a/vertexai/_genai/session_events.py b/vertexai/_genai/session_events.py index 2f7a820c29..7890ce8e78 100644 --- a/vertexai/_genai/session_events.py +++ b/vertexai/_genai/session_events.py @@ -78,12 +78,8 @@ def _AppendAgentEngineSessionEventRequestParameters_to_vertex( setv(to_object, ["timestamp"], getv(from_object, ["timestamp"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _AppendAgentEngineSessionEventConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _AppendAgentEngineSessionEventConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object @@ -116,12 +112,8 @@ def _ListAgentEngineSessionEventsRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineSessionEventsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _ListAgentEngineSessionEventsConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object diff --git a/vertexai/_genai/sessions.py b/vertexai/_genai/sessions.py index 43c5bf64b5..994fdf9587 100644 --- a/vertexai/_genai/sessions.py +++ b/vertexai/_genai/sessions.py @@ -79,12 +79,8 @@ def _CreateAgentEngineSessionRequestParameters_to_vertex( setv(to_object, ["userId"], getv(from_object, ["user_id"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _CreateAgentEngineSessionConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _CreateAgentEngineSessionConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object @@ -98,9 +94,6 @@ def _DeleteAgentEngineSessionRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -114,9 +107,6 @@ def _GetAgentEngineSessionOperationParameters_to_vertex( to_object, ["_url", "operationName"], getv(from_object, ["operation_name"]) ) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -128,9 +118,6 @@ def _GetAgentEngineSessionRequestParameters_to_vertex( if getv(from_object, ["name"]) is not None: setv(to_object, ["_url", "name"], getv(from_object, ["name"])) - if getv(from_object, ["config"]) is not None: - setv(to_object, ["config"], getv(from_object, ["config"])) - return to_object @@ -161,12 +148,8 @@ def _ListAgentEngineSessionsRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _ListAgentEngineSessionsConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _ListAgentEngineSessionsConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object @@ -213,12 +196,8 @@ def _UpdateAgentEngineSessionRequestParameters_to_vertex( setv(to_object, ["_url", "name"], getv(from_object, ["name"])) if getv(from_object, ["config"]) is not None: - setv( - to_object, - ["config"], - _UpdateAgentEngineSessionConfig_to_vertex( - getv(from_object, ["config"]), to_object - ), + _UpdateAgentEngineSessionConfig_to_vertex( + getv(from_object, ["config"]), to_object ) return to_object