Skip to content

Commit 914c63c

Browse files
docs: A comment for field partner_model_tuning_spec in message .google.cloud.aiplatform.v1beta1.TuningJob is changed (#4538)
* chore: Update gapic-generator-python to v1.19.1 PiperOrigin-RevId: 684571179 Source-Link: googleapis/googleapis@fbdc238 Source-Link: googleapis/googleapis-gen@3a2cdcf Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiM2EyY2RjZmI4MGMyZDBmNWVjMGNjNjYzYzJiYWIwYTk0ODYyMjlkMCJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * docs: A comment for field `partner_model_tuning_spec` in message `.google.cloud.aiplatform.v1beta1.TuningJob` is changed PiperOrigin-RevId: 684636851 Source-Link: googleapis/googleapis@de509e3 Source-Link: googleapis/googleapis-gen@5ec9807 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNWVjOTgwN2ZhMjI1ZGI5MWM2YzZkMzM1MzM4YTVmMGZiMzZlMzRiOSJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
1 parent 4268f28 commit 914c63c

File tree

568 files changed

+487318
-393455
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

568 files changed

+487318
-393455
lines changed

google/cloud/aiplatform_v1/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,7 @@
103103
from .types.content import GroundingSupport
104104
from .types.content import LogprobsResult
105105
from .types.content import Part
106+
from .types.content import RetrievalMetadata
106107
from .types.content import SafetyRating
107108
from .types.content import SafetySetting
108109
from .types.content import SearchEntryPoint
@@ -863,6 +864,7 @@
863864
from .types.tensorboard_service import WriteTensorboardRunDataRequest
864865
from .types.tensorboard_service import WriteTensorboardRunDataResponse
865866
from .types.tensorboard_time_series import TensorboardTimeSeries
867+
from .types.tool import DynamicRetrievalConfig
866868
from .types.tool import FunctionCall
867869
from .types.tool import FunctionCallingConfig
868870
from .types.tool import FunctionDeclaration
@@ -1162,6 +1164,7 @@
11621164
"DirectRawPredictResponse",
11631165
"DiskSpec",
11641166
"DoubleArray",
1167+
"DynamicRetrievalConfig",
11651168
"EncryptionSpec",
11661169
"Endpoint",
11671170
"EndpointServiceClient",
@@ -1596,6 +1599,7 @@
15961599
"ResumeModelDeploymentMonitoringJobRequest",
15971600
"ResumeScheduleRequest",
15981601
"Retrieval",
1602+
"RetrievalMetadata",
15991603
"RougeInput",
16001604
"RougeInstance",
16011605
"RougeMetricValue",

google/cloud/aiplatform_v1/services/dataset_service/async_client.py

+12-50
Original file line numberDiff line numberDiff line change
@@ -2663,11 +2663,7 @@ async def list_operations(
26632663

26642664
# Wrap the RPC method; this adds retry and timeout information,
26652665
# and friendly error handling.
2666-
rpc = gapic_v1.method_async.wrap_method(
2667-
self._client._transport.list_operations,
2668-
default_timeout=None,
2669-
client_info=DEFAULT_CLIENT_INFO,
2670-
)
2666+
rpc = self.transport._wrapped_methods[self._client._transport.list_operations]
26712667

26722668
# Certain fields should be provided within the metadata header;
26732669
# add these here.
@@ -2720,11 +2716,7 @@ async def get_operation(
27202716

27212717
# Wrap the RPC method; this adds retry and timeout information,
27222718
# and friendly error handling.
2723-
rpc = gapic_v1.method_async.wrap_method(
2724-
self._client._transport.get_operation,
2725-
default_timeout=None,
2726-
client_info=DEFAULT_CLIENT_INFO,
2727-
)
2719+
rpc = self.transport._wrapped_methods[self._client._transport.get_operation]
27282720

27292721
# Certain fields should be provided within the metadata header;
27302722
# add these here.
@@ -2781,11 +2773,7 @@ async def delete_operation(
27812773

27822774
# Wrap the RPC method; this adds retry and timeout information,
27832775
# and friendly error handling.
2784-
rpc = gapic_v1.method_async.wrap_method(
2785-
self._client._transport.delete_operation,
2786-
default_timeout=None,
2787-
client_info=DEFAULT_CLIENT_INFO,
2788-
)
2776+
rpc = self.transport._wrapped_methods[self._client._transport.delete_operation]
27892777

27902778
# Certain fields should be provided within the metadata header;
27912779
# add these here.
@@ -2838,11 +2826,7 @@ async def cancel_operation(
28382826

28392827
# Wrap the RPC method; this adds retry and timeout information,
28402828
# and friendly error handling.
2841-
rpc = gapic_v1.method_async.wrap_method(
2842-
self._client._transport.cancel_operation,
2843-
default_timeout=None,
2844-
client_info=DEFAULT_CLIENT_INFO,
2845-
)
2829+
rpc = self.transport._wrapped_methods[self._client._transport.cancel_operation]
28462830

28472831
# Certain fields should be provided within the metadata header;
28482832
# add these here.
@@ -2898,11 +2882,7 @@ async def wait_operation(
28982882

28992883
# Wrap the RPC method; this adds retry and timeout information,
29002884
# and friendly error handling.
2901-
rpc = gapic_v1.method_async.wrap_method(
2902-
self._client._transport.wait_operation,
2903-
default_timeout=None,
2904-
client_info=DEFAULT_CLIENT_INFO,
2905-
)
2885+
rpc = self.transport._wrapped_methods[self._client._transport.wait_operation]
29062886

29072887
# Certain fields should be provided within the metadata header;
29082888
# add these here.
@@ -3021,11 +3001,7 @@ async def set_iam_policy(
30213001

30223002
# Wrap the RPC method; this adds retry and timeout information,
30233003
# and friendly error handling.
3024-
rpc = gapic_v1.method_async.wrap_method(
3025-
self._client._transport.set_iam_policy,
3026-
default_timeout=None,
3027-
client_info=DEFAULT_CLIENT_INFO,
3028-
)
3004+
rpc = self.transport._wrapped_methods[self._client._transport.set_iam_policy]
30293005

30303006
# Certain fields should be provided within the metadata header;
30313007
# add these here.
@@ -3145,11 +3121,7 @@ async def get_iam_policy(
31453121

31463122
# Wrap the RPC method; this adds retry and timeout information,
31473123
# and friendly error handling.
3148-
rpc = gapic_v1.method_async.wrap_method(
3149-
self._client._transport.get_iam_policy,
3150-
default_timeout=None,
3151-
client_info=DEFAULT_CLIENT_INFO,
3152-
)
3124+
rpc = self.transport._wrapped_methods[self._client._transport.get_iam_policy]
31533125

31543126
# Certain fields should be provided within the metadata header;
31553127
# add these here.
@@ -3207,11 +3179,9 @@ async def test_iam_permissions(
32073179

32083180
# Wrap the RPC method; this adds retry and timeout information,
32093181
# and friendly error handling.
3210-
rpc = gapic_v1.method_async.wrap_method(
3211-
self._client._transport.test_iam_permissions,
3212-
default_timeout=None,
3213-
client_info=DEFAULT_CLIENT_INFO,
3214-
)
3182+
rpc = self.transport._wrapped_methods[
3183+
self._client._transport.test_iam_permissions
3184+
]
32153185

32163186
# Certain fields should be provided within the metadata header;
32173187
# add these here.
@@ -3264,11 +3234,7 @@ async def get_location(
32643234

32653235
# Wrap the RPC method; this adds retry and timeout information,
32663236
# and friendly error handling.
3267-
rpc = gapic_v1.method_async.wrap_method(
3268-
self._client._transport.get_location,
3269-
default_timeout=None,
3270-
client_info=DEFAULT_CLIENT_INFO,
3271-
)
3237+
rpc = self.transport._wrapped_methods[self._client._transport.get_location]
32723238

32733239
# Certain fields should be provided within the metadata header;
32743240
# add these here.
@@ -3321,11 +3287,7 @@ async def list_locations(
33213287

33223288
# Wrap the RPC method; this adds retry and timeout information,
33233289
# and friendly error handling.
3324-
rpc = gapic_v1.method_async.wrap_method(
3325-
self._client._transport.list_locations,
3326-
default_timeout=None,
3327-
client_info=DEFAULT_CLIENT_INFO,
3328-
)
3290+
rpc = self.transport._wrapped_methods[self._client._transport.list_locations]
33293291

33303292
# Certain fields should be provided within the metadata header;
33313293
# add these here.

google/cloud/aiplatform_v1/services/dataset_service/client.py

+10-50
Original file line numberDiff line numberDiff line change
@@ -3150,11 +3150,7 @@ def list_operations(
31503150

31513151
# Wrap the RPC method; this adds retry and timeout information,
31523152
# and friendly error handling.
3153-
rpc = gapic_v1.method.wrap_method(
3154-
self._transport.list_operations,
3155-
default_timeout=None,
3156-
client_info=DEFAULT_CLIENT_INFO,
3157-
)
3153+
rpc = self._transport._wrapped_methods[self._transport.list_operations]
31583154

31593155
# Certain fields should be provided within the metadata header;
31603156
# add these here.
@@ -3207,11 +3203,7 @@ def get_operation(
32073203

32083204
# Wrap the RPC method; this adds retry and timeout information,
32093205
# and friendly error handling.
3210-
rpc = gapic_v1.method.wrap_method(
3211-
self._transport.get_operation,
3212-
default_timeout=None,
3213-
client_info=DEFAULT_CLIENT_INFO,
3214-
)
3206+
rpc = self._transport._wrapped_methods[self._transport.get_operation]
32153207

32163208
# Certain fields should be provided within the metadata header;
32173209
# add these here.
@@ -3268,11 +3260,7 @@ def delete_operation(
32683260

32693261
# Wrap the RPC method; this adds retry and timeout information,
32703262
# and friendly error handling.
3271-
rpc = gapic_v1.method.wrap_method(
3272-
self._transport.delete_operation,
3273-
default_timeout=None,
3274-
client_info=DEFAULT_CLIENT_INFO,
3275-
)
3263+
rpc = self._transport._wrapped_methods[self._transport.delete_operation]
32763264

32773265
# Certain fields should be provided within the metadata header;
32783266
# add these here.
@@ -3325,11 +3313,7 @@ def cancel_operation(
33253313

33263314
# Wrap the RPC method; this adds retry and timeout information,
33273315
# and friendly error handling.
3328-
rpc = gapic_v1.method.wrap_method(
3329-
self._transport.cancel_operation,
3330-
default_timeout=None,
3331-
client_info=DEFAULT_CLIENT_INFO,
3332-
)
3316+
rpc = self._transport._wrapped_methods[self._transport.cancel_operation]
33333317

33343318
# Certain fields should be provided within the metadata header;
33353319
# add these here.
@@ -3385,11 +3369,7 @@ def wait_operation(
33853369

33863370
# Wrap the RPC method; this adds retry and timeout information,
33873371
# and friendly error handling.
3388-
rpc = gapic_v1.method.wrap_method(
3389-
self._transport.wait_operation,
3390-
default_timeout=None,
3391-
client_info=DEFAULT_CLIENT_INFO,
3392-
)
3372+
rpc = self._transport._wrapped_methods[self._transport.wait_operation]
33933373

33943374
# Certain fields should be provided within the metadata header;
33953375
# add these here.
@@ -3508,11 +3488,7 @@ def set_iam_policy(
35083488

35093489
# Wrap the RPC method; this adds retry and timeout information,
35103490
# and friendly error handling.
3511-
rpc = gapic_v1.method.wrap_method(
3512-
self._transport.set_iam_policy,
3513-
default_timeout=None,
3514-
client_info=DEFAULT_CLIENT_INFO,
3515-
)
3491+
rpc = self._transport._wrapped_methods[self._transport.set_iam_policy]
35163492

35173493
# Certain fields should be provided within the metadata header;
35183494
# add these here.
@@ -3632,11 +3608,7 @@ def get_iam_policy(
36323608

36333609
# Wrap the RPC method; this adds retry and timeout information,
36343610
# and friendly error handling.
3635-
rpc = gapic_v1.method.wrap_method(
3636-
self._transport.get_iam_policy,
3637-
default_timeout=None,
3638-
client_info=DEFAULT_CLIENT_INFO,
3639-
)
3611+
rpc = self._transport._wrapped_methods[self._transport.get_iam_policy]
36403612

36413613
# Certain fields should be provided within the metadata header;
36423614
# add these here.
@@ -3694,11 +3666,7 @@ def test_iam_permissions(
36943666

36953667
# Wrap the RPC method; this adds retry and timeout information,
36963668
# and friendly error handling.
3697-
rpc = gapic_v1.method.wrap_method(
3698-
self._transport.test_iam_permissions,
3699-
default_timeout=None,
3700-
client_info=DEFAULT_CLIENT_INFO,
3701-
)
3669+
rpc = self._transport._wrapped_methods[self._transport.test_iam_permissions]
37023670

37033671
# Certain fields should be provided within the metadata header;
37043672
# add these here.
@@ -3751,11 +3719,7 @@ def get_location(
37513719

37523720
# Wrap the RPC method; this adds retry and timeout information,
37533721
# and friendly error handling.
3754-
rpc = gapic_v1.method.wrap_method(
3755-
self._transport.get_location,
3756-
default_timeout=None,
3757-
client_info=DEFAULT_CLIENT_INFO,
3758-
)
3722+
rpc = self._transport._wrapped_methods[self._transport.get_location]
37593723

37603724
# Certain fields should be provided within the metadata header;
37613725
# add these here.
@@ -3808,11 +3772,7 @@ def list_locations(
38083772

38093773
# Wrap the RPC method; this adds retry and timeout information,
38103774
# and friendly error handling.
3811-
rpc = gapic_v1.method.wrap_method(
3812-
self._transport.list_locations,
3813-
default_timeout=None,
3814-
client_info=DEFAULT_CLIENT_INFO,
3815-
)
3775+
rpc = self._transport._wrapped_methods[self._transport.list_locations]
38163776

38173777
# Certain fields should be provided within the metadata header;
38183778
# add these here.
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
2+
transport inheritance structure
3+
_______________________________
4+
5+
`DatasetServiceTransport` is the ABC for all transports.
6+
- public child `DatasetServiceGrpcTransport` for sync gRPC transport (defined in `grpc.py`).
7+
- public child `DatasetServiceGrpcAsyncIOTransport` for async gRPC transport (defined in `grpc_asyncio.py`).
8+
- private child `_BaseDatasetServiceRestTransport` for base REST transport with inner classes `_BaseMETHOD` (defined in `rest_base.py`).
9+
- public child `DatasetServiceRestTransport` for sync REST transport with inner classes `METHOD` derived from the parent's corresponding `_BaseMETHOD` classes (defined in `rest.py`).

google/cloud/aiplatform_v1/services/dataset_service/transports/base.py

+50
Original file line numberDiff line numberDiff line change
@@ -234,6 +234,56 @@ def _prep_wrapped_messages(self, client_info):
234234
default_timeout=None,
235235
client_info=client_info,
236236
),
237+
self.get_location: gapic_v1.method.wrap_method(
238+
self.get_location,
239+
default_timeout=None,
240+
client_info=client_info,
241+
),
242+
self.list_locations: gapic_v1.method.wrap_method(
243+
self.list_locations,
244+
default_timeout=None,
245+
client_info=client_info,
246+
),
247+
self.get_iam_policy: gapic_v1.method.wrap_method(
248+
self.get_iam_policy,
249+
default_timeout=None,
250+
client_info=client_info,
251+
),
252+
self.set_iam_policy: gapic_v1.method.wrap_method(
253+
self.set_iam_policy,
254+
default_timeout=None,
255+
client_info=client_info,
256+
),
257+
self.test_iam_permissions: gapic_v1.method.wrap_method(
258+
self.test_iam_permissions,
259+
default_timeout=None,
260+
client_info=client_info,
261+
),
262+
self.cancel_operation: gapic_v1.method.wrap_method(
263+
self.cancel_operation,
264+
default_timeout=None,
265+
client_info=client_info,
266+
),
267+
self.delete_operation: gapic_v1.method.wrap_method(
268+
self.delete_operation,
269+
default_timeout=None,
270+
client_info=client_info,
271+
),
272+
self.get_operation: gapic_v1.method.wrap_method(
273+
self.get_operation,
274+
default_timeout=None,
275+
client_info=client_info,
276+
),
277+
self.list_operations: gapic_v1.method.wrap_method(
278+
self.list_operations,
279+
default_timeout=None,
280+
client_info=client_info,
281+
),
282+
self.wait_operation: gapic_v1.method.wrap_method(
283+
self.wait_operation,
284+
default_timeout=None,
285+
client_info=client_info,
286+
),
237287
}
238288

239289
def close(self):

google/cloud/aiplatform_v1/services/dataset_service/transports/grpc.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -830,7 +830,7 @@ def wait_operation(
830830
# the request.
831831
# gRPC handles serialization and deserialization, so we just need
832832
# to pass in the functions for each.
833-
if "delete_operation" not in self._stubs:
833+
if "wait_operation" not in self._stubs:
834834
self._stubs["wait_operation"] = self.grpc_channel.unary_unary(
835835
"/google.longrunning.Operations/WaitOperation",
836836
request_serializer=operations_pb2.WaitOperationRequest.SerializeToString,

0 commit comments

Comments
 (0)