Skip to content

Commit 50fbdee

Browse files
yeesiancopybara-github
authored andcommitted
feat: ExampleStore SDK in public preview
PiperOrigin-RevId: 742235855
1 parent a7453da commit 50fbdee

File tree

7 files changed

+1197
-9
lines changed

7 files changed

+1197
-9
lines changed

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

+4
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,9 @@
2424
from google.cloud.aiplatform_v1beta1.services.endpoint_service import (
2525
client as endpoint_service_client_v1beta1,
2626
)
27+
from google.cloud.aiplatform_v1beta1.services.example_store_service import (
28+
client as example_store_service_client_v1beta1,
29+
)
2730
from google.cloud.aiplatform_v1beta1.services.extension_execution_service import (
2831
client as extension_execution_service_client_v1beta1,
2932
)
@@ -231,6 +234,7 @@
231234
dataset_service_client_v1beta1,
232235
deployment_resource_pool_service_client_v1beta1,
233236
endpoint_service_client_v1beta1,
237+
example_store_service_client_v1beta1,
234238
feature_online_store_service_client_v1beta1,
235239
feature_online_store_admin_service_client_v1beta1,
236240
feature_registry_service_client_v1beta1,

google/cloud/aiplatform/utils/__init__.py

+12
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
dataset_service_client_v1beta1,
4040
deployment_resource_pool_service_client_v1beta1,
4141
endpoint_service_client_v1beta1,
42+
example_store_service_client_v1beta1,
4243
extension_execution_service_client_v1beta1,
4344
extension_registry_service_client_v1beta1,
4445
feature_online_store_admin_service_client_v1beta1,
@@ -951,6 +952,17 @@ class PersistentResourceClientWithOverride(ClientWithOverride):
951952
)
952953

953954

955+
class ExampleStoreClientWithOverride(ClientWithOverride):
956+
_is_temporary = True
957+
_default_version = compat.V1BETA1
958+
_version_map = (
959+
(
960+
compat.V1BETA1,
961+
example_store_service_client_v1beta1.ExampleStoreServiceClient,
962+
),
963+
)
964+
965+
954966
class ReasoningEngineClientWithOverride(ClientWithOverride):
955967
_is_temporary = True
956968
_default_version = compat.V1BETA1

0 commit comments

Comments
 (0)