feat: Vertex AI Extension SDK Public Preview

PiperOrigin-RevId: 621918294

Author: yeesian

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

@@ -24,6 +24,12 @@
 from google.cloud.aiplatform_v1beta1.services.endpoint_service import (
     client as endpoint_service_client_v1beta1,
 )
+from google.cloud.aiplatform_v1beta1.services.extension_execution_service import (
+    client as extension_execution_service_client_v1beta1,
+)
+from google.cloud.aiplatform_v1beta1.services.extension_registry_service import (
+    client as extension_registry_service_client_v1beta1,
+)
 from google.cloud.aiplatform_v1beta1.services.feature_online_store_service import (
     client as feature_online_store_service_client_v1beta1,
 )

google/cloud/aiplatform/utils/__init__.py

@@ -39,6 +39,8 @@
     dataset_service_client_v1beta1,
     deployment_resource_pool_service_client_v1beta1,
     endpoint_service_client_v1beta1,
+    extension_execution_service_client_v1beta1,
+    extension_registry_service_client_v1beta1,
     feature_online_store_admin_service_client_v1beta1,
     feature_online_store_service_client_v1beta1,
     featurestore_online_serving_service_client_v1beta1,
@@ -555,6 +557,28 @@ class EndpointClientWithOverride(ClientWithOverride):
     )
 
 
+class ExtensionExecutionClientWithOverride(ClientWithOverride):
+    _is_temporary = True
+    _default_version = compat.V1BETA1
+    _version_map = (
+        (
+            compat.V1BETA1,
+            extension_execution_service_client_v1beta1.ExtensionExecutionServiceClient,
+        ),
+    )
+
+
+class ExtensionRegistryClientWithOverride(ClientWithOverride):
+    _is_temporary = True
+    _default_version = compat.V1BETA1
+    _version_map = (
+        (
+            compat.V1BETA1,
+            extension_registry_service_client_v1beta1.ExtensionRegistryServiceClient,
+        ),
+    )
+
+
 class IndexClientWithOverride(ClientWithOverride):
     _is_temporary = True
     _default_version = compat.DEFAULT_VERSION