feat: Vertex AI Agent Engine GA SDK

PiperOrigin-RevId: 731121127

Author: yeesian

google/cloud/aiplatform/compat/services/__init__.py

@@ -167,6 +167,12 @@
 from google.cloud.aiplatform_v1.services.prediction_service import (
     async_client as prediction_service_async_client_v1,
 )
+from google.cloud.aiplatform_v1.services.reasoning_engine_service import (
+    client as reasoning_engine_service_client_v1,
+)
+from google.cloud.aiplatform_v1.services.reasoning_engine_execution_service import (
+    client as reasoning_engine_execution_service_client_v1,
+)
 from google.cloud.aiplatform_v1.services.schedule_service import (
     client as schedule_service_client_v1,
 )
@@ -209,6 +215,8 @@
     pipeline_service_client_v1,
     prediction_service_client_v1,
     prediction_service_async_client_v1,
+    reasoning_engine_execution_service_client_v1,
+    reasoning_engine_service_client_v1,
     schedule_service_client_v1,
     specialist_pool_service_client_v1,
     tensorboard_service_client_v1,

google/cloud/aiplatform/utils/__init__.py

@@ -86,6 +86,8 @@
     pipeline_service_client_v1,
     prediction_service_client_v1,
     prediction_service_async_client_v1,
+    reasoning_engine_service_client_v1,
+    reasoning_engine_execution_service_client_v1,
     schedule_service_client_v1,
     tensorboard_service_client_v1,
     vizier_service_client_v1,
@@ -971,6 +973,28 @@ class ReasoningEngineExecutionClientWithOverride(ClientWithOverride):
     )
 
 
+class AgentEngineClientWithOverride(ClientWithOverride):
+    _is_temporary = True
+    _default_version = compat.V1
+    _version_map = (
+        (
+            compat.V1,
+            reasoning_engine_service_client_v1.ReasoningEngineServiceClient,
+        ),
+    )
+
+
+class AgentEngineExecutionClientWithOverride(ClientWithOverride):
+    _is_temporary = True
+    _default_version = compat.V1
+    _version_map = (
+        (
+            compat.V1,
+            reasoning_engine_execution_service_client_v1.ReasoningEngineExecutionServiceClient,
+        ),
+    )
+
+
 class VertexRagDataClientWithOverride(ClientWithOverride):
     _is_temporary = True
     _default_version = compat.DEFAULT_VERSION
@@ -1026,6 +1050,8 @@ class VertexRagClientWithOverride(ClientWithOverride):
     PersistentResourceClientWithOverride,
     ReasoningEngineClientWithOverride,
     ReasoningEngineExecutionClientWithOverride,
+    AgentEngineClientWithOverride,
+    AgentEngineExecutionClientWithOverride,
     ModelMonitoringClientWithOverride,
 )
 

setup.py

@@ -148,6 +148,15 @@
     "typing_extensions",
 ]
 
+agent_engines_extra_require = [
+    "packaging >= 24.0",
+    "cloudpickle >= 3.0, < 4.0",
+    "google-cloud-trace < 2",
+    "google-cloud-logging < 4",
+    "pydantic >= 2.10, < 3",
+    "typing_extensions",
+]
+
 evaluation_extra_require = [
     "pandas >= 1.0.0",
     "tqdm>=4.23.0",
@@ -290,6 +299,7 @@
         "ray": ray_extra_require,
         "ray_testing": ray_testing_extra_require,
         "reasoningengine": reasoning_engine_extra_require,
+        "agent_engines": agent_engines_extra_require,
         "evaluation": evaluation_extra_require,
         "langchain": langchain_extra_require,
         "langchain_testing": langchain_testing_extra_require,