Skip to content

Commit 6dfbad7

Browse files
gcf-owl-bot[bot]copybara-github
gcf-owl-bot[bot]
authored and
copybara-github
committed
Copybara import of the project:
-- 709b546 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add grpc_ports to UploadModel ModelContainerSpec feat: add DirectPredict to PredictionService feat: add DirectRawPredict to PredictionService feat: add StreamingPredict to PredictionService. feat: add StreamingRawPredict to PredictionService. PiperOrigin-RevId: 583502805 Source-Link: googleapis/googleapis@31c70c9 Source-Link: googleapis/googleapis-gen@a041c3d Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYTA0MWMzZDgwNjhjYjk5YTFkY2Y4MzEyNGQwY2Q2M2ViYTkxYTRjNyJ9 -- adaa9f5 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 3285b1b by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: chore: Update gapic-generator-python to v1.12.0 PiperOrigin-RevId: 586356061 Source-Link: googleapis/googleapis@72a1f55 Source-Link: googleapis/googleapis-gen@558a04b Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNTU4YTA0YmNkMWNjMDU3NmU4ZmFjMTA4OWU0OGU0OGIyN2FjMTYxYiJ9 -- ca369f6 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 3f862d6 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add auto_populated_fields to google.api.MethodSettings PiperOrigin-RevId: 586359350 Source-Link: googleapis/googleapis@fce431c Source-Link: googleapis/googleapis-gen@df4b8f4 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZGY0YjhmNDEyZGZhOWZjODAyMDU3NTA1M2M3YmM1ZWFmM2UyY2QwYiJ9 -- 0ed2597 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- af4c4d4 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add grpc_ports to UploadModel ModelContainerSpec feat: add DirectPredict to PredictionService feat: add DirectRawPredict to PredictionService feat: add StreamingPredict to PredictionService. feat: add StreamingRawPredict to PredictionService. PiperOrigin-RevId: 586417186 Source-Link: googleapis/googleapis@362f61e Source-Link: googleapis/googleapis-gen@a81e534 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYTgxZTUzNDNjZTZjYTllZTRhNTQ1NTQ5ZGMzMmVkYzIzNTA1MzBhYSJ9 -- 5831667 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- f938450 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: added Generator API docs: updated doc for speech mode PiperOrigin-RevId: 586469693 Source-Link: googleapis/googleapis@e8148d6 Source-Link: googleapis/googleapis-gen@85136bd Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiODUxMzZiZDA0MzgzZWQ3MTcyYmIxOGI3YjhkMjIwZGQ3ZmY2YjNhMCJ9 -- 6888c6c by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md COPYBARA_INTEGRATE_REVIEW=#2993 from googleapis:owl-bot-copy 55d5110 PiperOrigin-RevId: 587056958
1 parent ab31abd commit 6dfbad7

File tree

129 files changed

+7630
-2058
lines changed

Some content is hidden

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

129 files changed

+7630
-2058
lines changed

‎google/cloud/aiplatform_v1/__init__.py

+12
Original file line numberDiff line numberDiff line change
@@ -531,13 +531,19 @@
531531
from .types.pipeline_state import PipelineState
532532
from .types.prediction_service import CountTokensRequest
533533
from .types.prediction_service import CountTokensResponse
534+
from .types.prediction_service import DirectPredictRequest
535+
from .types.prediction_service import DirectPredictResponse
536+
from .types.prediction_service import DirectRawPredictRequest
537+
from .types.prediction_service import DirectRawPredictResponse
534538
from .types.prediction_service import ExplainRequest
535539
from .types.prediction_service import ExplainResponse
536540
from .types.prediction_service import PredictRequest
537541
from .types.prediction_service import PredictResponse
538542
from .types.prediction_service import RawPredictRequest
539543
from .types.prediction_service import StreamingPredictRequest
540544
from .types.prediction_service import StreamingPredictResponse
545+
from .types.prediction_service import StreamingRawPredictRequest
546+
from .types.prediction_service import StreamingRawPredictResponse
541547
from .types.publisher_model import PublisherModel
542548
from .types.saved_query import SavedQuery
543549
from .types.schedule import Schedule
@@ -854,6 +860,10 @@
854860
"DeployedModel",
855861
"DeployedModelRef",
856862
"DestinationFeatureSetting",
863+
"DirectPredictRequest",
864+
"DirectPredictResponse",
865+
"DirectRawPredictRequest",
866+
"DirectRawPredictResponse",
857867
"DiskSpec",
858868
"DoubleArray",
859869
"EncryptionSpec",
@@ -1193,6 +1203,8 @@
11931203
"StratifiedSplit",
11941204
"StreamingPredictRequest",
11951205
"StreamingPredictResponse",
1206+
"StreamingRawPredictRequest",
1207+
"StreamingRawPredictResponse",
11961208
"StreamingReadFeatureValuesRequest",
11971209
"StringArray",
11981210
"Study",

‎google/cloud/aiplatform_v1/gapic_metadata.json

+40
Original file line numberDiff line numberDiff line change
@@ -2158,6 +2158,16 @@
21582158
"grpc": {
21592159
"libraryClient": "PredictionServiceClient",
21602160
"rpcs": {
2161+
"DirectPredict": {
2162+
"methods": [
2163+
"direct_predict"
2164+
]
2165+
},
2166+
"DirectRawPredict": {
2167+
"methods": [
2168+
"direct_raw_predict"
2169+
]
2170+
},
21612171
"Explain": {
21622172
"methods": [
21632173
"explain"
@@ -2177,12 +2187,32 @@
21772187
"methods": [
21782188
"server_streaming_predict"
21792189
]
2190+
},
2191+
"StreamingPredict": {
2192+
"methods": [
2193+
"streaming_predict"
2194+
]
2195+
},
2196+
"StreamingRawPredict": {
2197+
"methods": [
2198+
"streaming_raw_predict"
2199+
]
21802200
}
21812201
}
21822202
},
21832203
"grpc-async": {
21842204
"libraryClient": "PredictionServiceAsyncClient",
21852205
"rpcs": {
2206+
"DirectPredict": {
2207+
"methods": [
2208+
"direct_predict"
2209+
]
2210+
},
2211+
"DirectRawPredict": {
2212+
"methods": [
2213+
"direct_raw_predict"
2214+
]
2215+
},
21862216
"Explain": {
21872217
"methods": [
21882218
"explain"
@@ -2202,6 +2232,16 @@
22022232
"methods": [
22032233
"server_streaming_predict"
22042234
]
2235+
},
2236+
"StreamingPredict": {
2237+
"methods": [
2238+
"streaming_predict"
2239+
]
2240+
},
2241+
"StreamingRawPredict": {
2242+
"methods": [
2243+
"streaming_raw_predict"
2244+
]
22052245
}
22062246
}
22072247
}

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

+41-41
Large diffs are not rendered by default.

‎google/cloud/aiplatform_v1/services/endpoint_service/async_client.py

+31-31
Original file line numberDiff line numberDiff line change
@@ -33,14 +33,14 @@
3333
from google.api_core.client_options import ClientOptions
3434
from google.api_core import exceptions as core_exceptions
3535
from google.api_core import gapic_v1
36-
from google.api_core import retry as retries
36+
from google.api_core import retry_async as retries
3737
from google.auth import credentials as ga_credentials # type: ignore
3838
from google.oauth2 import service_account # type: ignore
3939

4040
try:
41-
OptionalRetry = Union[retries.Retry, gapic_v1.method._MethodDefault]
41+
OptionalRetry = Union[retries.AsyncRetry, gapic_v1.method._MethodDefault]
4242
except AttributeError: # pragma: NO COVER
43-
OptionalRetry = Union[retries.Retry, object] # type: ignore
43+
OptionalRetry = Union[retries.AsyncRetry, object] # type: ignore
4444

4545
from google.api_core import operation as gac_operation # type: ignore
4646
from google.api_core import operation_async # type: ignore
@@ -323,7 +323,7 @@ async def sample_create_endpoint():
323323
This corresponds to the ``endpoint_id`` field
324324
on the ``request`` instance; if ``request`` is provided, this
325325
should not be set.
326-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
326+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
327327
should be retried.
328328
timeout (float): The timeout for this request.
329329
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -439,7 +439,7 @@ async def sample_get_endpoint():
439439
This corresponds to the ``name`` field
440440
on the ``request`` instance; if ``request`` is provided, this
441441
should not be set.
442-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
442+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
443443
should be retried.
444444
timeout (float): The timeout for this request.
445445
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -544,7 +544,7 @@ async def sample_list_endpoints():
544544
This corresponds to the ``parent`` field
545545
on the ``request`` instance; if ``request`` is provided, this
546546
should not be set.
547-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
547+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
548548
should be retried.
549549
timeout (float): The timeout for this request.
550550
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -669,7 +669,7 @@ async def sample_update_endpoint():
669669
This corresponds to the ``update_mask`` field
670670
on the ``request`` instance; if ``request`` is provided, this
671671
should not be set.
672-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
672+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
673673
should be retried.
674674
timeout (float): The timeout for this request.
675675
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -781,7 +781,7 @@ async def sample_delete_endpoint():
781781
This corresponds to the ``name`` field
782782
on the ``request`` instance; if ``request`` is provided, this
783783
should not be set.
784-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
784+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
785785
should be retried.
786786
timeout (float): The timeout for this request.
787787
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -945,7 +945,7 @@ async def sample_deploy_model():
945945
This corresponds to the ``traffic_split`` field
946946
on the ``request`` instance; if ``request`` is provided, this
947947
should not be set.
948-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
948+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
949949
should be retried.
950950
timeout (float): The timeout for this request.
951951
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1093,7 +1093,7 @@ async def sample_undeploy_model():
10931093
This corresponds to the ``traffic_split`` field
10941094
on the ``request`` instance; if ``request`` is provided, this
10951095
should not be set.
1096-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
1096+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
10971097
should be retried.
10981098
timeout (float): The timeout for this request.
10991099
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1253,7 +1253,7 @@ async def sample_mutate_deployed_model():
12531253
This corresponds to the ``update_mask`` field
12541254
on the ``request`` instance; if ``request`` is provided, this
12551255
should not be set.
1256-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
1256+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
12571257
should be retried.
12581258
timeout (float): The timeout for this request.
12591259
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1335,7 +1335,7 @@ async def list_operations(
13351335
request (:class:`~.operations_pb2.ListOperationsRequest`):
13361336
The request object. Request message for
13371337
`ListOperations` method.
1338-
retry (google.api_core.retry.Retry): Designation of what errors,
1338+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
13391339
if any, should be retried.
13401340
timeout (float): The timeout for this request.
13411341
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1352,7 +1352,7 @@ async def list_operations(
13521352

13531353
# Wrap the RPC method; this adds retry and timeout information,
13541354
# and friendly error handling.
1355-
rpc = gapic_v1.method.wrap_method(
1355+
rpc = gapic_v1.method_async.wrap_method(
13561356
self._client._transport.list_operations,
13571357
default_timeout=None,
13581358
client_info=DEFAULT_CLIENT_INFO,
@@ -1389,7 +1389,7 @@ async def get_operation(
13891389
request (:class:`~.operations_pb2.GetOperationRequest`):
13901390
The request object. Request message for
13911391
`GetOperation` method.
1392-
retry (google.api_core.retry.Retry): Designation of what errors,
1392+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
13931393
if any, should be retried.
13941394
timeout (float): The timeout for this request.
13951395
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1406,7 +1406,7 @@ async def get_operation(
14061406

14071407
# Wrap the RPC method; this adds retry and timeout information,
14081408
# and friendly error handling.
1409-
rpc = gapic_v1.method.wrap_method(
1409+
rpc = gapic_v1.method_async.wrap_method(
14101410
self._client._transport.get_operation,
14111411
default_timeout=None,
14121412
client_info=DEFAULT_CLIENT_INFO,
@@ -1448,7 +1448,7 @@ async def delete_operation(
14481448
request (:class:`~.operations_pb2.DeleteOperationRequest`):
14491449
The request object. Request message for
14501450
`DeleteOperation` method.
1451-
retry (google.api_core.retry.Retry): Designation of what errors,
1451+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
14521452
if any, should be retried.
14531453
timeout (float): The timeout for this request.
14541454
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1464,7 +1464,7 @@ async def delete_operation(
14641464

14651465
# Wrap the RPC method; this adds retry and timeout information,
14661466
# and friendly error handling.
1467-
rpc = gapic_v1.method.wrap_method(
1467+
rpc = gapic_v1.method_async.wrap_method(
14681468
self._client._transport.delete_operation,
14691469
default_timeout=None,
14701470
client_info=DEFAULT_CLIENT_INFO,
@@ -1502,7 +1502,7 @@ async def cancel_operation(
15021502
request (:class:`~.operations_pb2.CancelOperationRequest`):
15031503
The request object. Request message for
15041504
`CancelOperation` method.
1505-
retry (google.api_core.retry.Retry): Designation of what errors,
1505+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
15061506
if any, should be retried.
15071507
timeout (float): The timeout for this request.
15081508
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1518,7 +1518,7 @@ async def cancel_operation(
15181518

15191519
# Wrap the RPC method; this adds retry and timeout information,
15201520
# and friendly error handling.
1521-
rpc = gapic_v1.method.wrap_method(
1521+
rpc = gapic_v1.method_async.wrap_method(
15221522
self._client._transport.cancel_operation,
15231523
default_timeout=None,
15241524
client_info=DEFAULT_CLIENT_INFO,
@@ -1558,7 +1558,7 @@ async def wait_operation(
15581558
request (:class:`~.operations_pb2.WaitOperationRequest`):
15591559
The request object. Request message for
15601560
`WaitOperation` method.
1561-
retry (google.api_core.retry.Retry): Designation of what errors,
1561+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
15621562
if any, should be retried.
15631563
timeout (float): The timeout for this request.
15641564
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1575,7 +1575,7 @@ async def wait_operation(
15751575

15761576
# Wrap the RPC method; this adds retry and timeout information,
15771577
# and friendly error handling.
1578-
rpc = gapic_v1.method.wrap_method(
1578+
rpc = gapic_v1.method_async.wrap_method(
15791579
self._client._transport.wait_operation,
15801580
default_timeout=None,
15811581
client_info=DEFAULT_CLIENT_INFO,
@@ -1614,7 +1614,7 @@ async def set_iam_policy(
16141614
request (:class:`~.iam_policy_pb2.SetIamPolicyRequest`):
16151615
The request object. Request message for `SetIamPolicy`
16161616
method.
1617-
retry (google.api_core.retry.Retry): Designation of what errors, if any,
1617+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if any,
16181618
should be retried.
16191619
timeout (float): The timeout for this request.
16201620
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1695,7 +1695,7 @@ async def set_iam_policy(
16951695

16961696
# Wrap the RPC method; this adds retry and timeout information,
16971697
# and friendly error handling.
1698-
rpc = gapic_v1.method.wrap_method(
1698+
rpc = gapic_v1.method_async.wrap_method(
16991699
self._client._transport.set_iam_policy,
17001700
default_timeout=None,
17011701
client_info=DEFAULT_CLIENT_INFO,
@@ -1735,7 +1735,7 @@ async def get_iam_policy(
17351735
request (:class:`~.iam_policy_pb2.GetIamPolicyRequest`):
17361736
The request object. Request message for `GetIamPolicy`
17371737
method.
1738-
retry (google.api_core.retry.Retry): Designation of what errors, if
1738+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors, if
17391739
any, should be retried.
17401740
timeout (float): The timeout for this request.
17411741
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1816,7 +1816,7 @@ async def get_iam_policy(
18161816

18171817
# Wrap the RPC method; this adds retry and timeout information,
18181818
# and friendly error handling.
1819-
rpc = gapic_v1.method.wrap_method(
1819+
rpc = gapic_v1.method_async.wrap_method(
18201820
self._client._transport.get_iam_policy,
18211821
default_timeout=None,
18221822
client_info=DEFAULT_CLIENT_INFO,
@@ -1857,7 +1857,7 @@ async def test_iam_permissions(
18571857
request (:class:`~.iam_policy_pb2.TestIamPermissionsRequest`):
18581858
The request object. Request message for
18591859
`TestIamPermissions` method.
1860-
retry (google.api_core.retry.Retry): Designation of what errors,
1860+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
18611861
if any, should be retried.
18621862
timeout (float): The timeout for this request.
18631863
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1875,7 +1875,7 @@ async def test_iam_permissions(
18751875

18761876
# Wrap the RPC method; this adds retry and timeout information,
18771877
# and friendly error handling.
1878-
rpc = gapic_v1.method.wrap_method(
1878+
rpc = gapic_v1.method_async.wrap_method(
18791879
self._client._transport.test_iam_permissions,
18801880
default_timeout=None,
18811881
client_info=DEFAULT_CLIENT_INFO,
@@ -1912,7 +1912,7 @@ async def get_location(
19121912
request (:class:`~.location_pb2.GetLocationRequest`):
19131913
The request object. Request message for
19141914
`GetLocation` method.
1915-
retry (google.api_core.retry.Retry): Designation of what errors,
1915+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
19161916
if any, should be retried.
19171917
timeout (float): The timeout for this request.
19181918
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1929,7 +1929,7 @@ async def get_location(
19291929

19301930
# Wrap the RPC method; this adds retry and timeout information,
19311931
# and friendly error handling.
1932-
rpc = gapic_v1.method.wrap_method(
1932+
rpc = gapic_v1.method_async.wrap_method(
19331933
self._client._transport.get_location,
19341934
default_timeout=None,
19351935
client_info=DEFAULT_CLIENT_INFO,
@@ -1966,7 +1966,7 @@ async def list_locations(
19661966
request (:class:`~.location_pb2.ListLocationsRequest`):
19671967
The request object. Request message for
19681968
`ListLocations` method.
1969-
retry (google.api_core.retry.Retry): Designation of what errors,
1969+
retry (google.api_core.retry_async.AsyncRetry): Designation of what errors,
19701970
if any, should be retried.
19711971
timeout (float): The timeout for this request.
19721972
metadata (Sequence[Tuple[str, str]]): Strings which should be
@@ -1983,7 +1983,7 @@ async def list_locations(
19831983

19841984
# Wrap the RPC method; this adds retry and timeout information,
19851985
# and friendly error handling.
1986-
rpc = gapic_v1.method.wrap_method(
1986+
rpc = gapic_v1.method_async.wrap_method(
19871987
self._client._transport.list_locations,
19881988
default_timeout=None,
19891989
client_info=DEFAULT_CLIENT_INFO,

0 commit comments

Comments
 (0)