Skip to content

Commit 2148b18

Browse files
gcf-owl-bot[bot]copybara-github
gcf-owl-bot[bot]
authored and
copybara-github
committed
Copybara import of the project:
-- e43d467 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add retrieval_config to ToolConfig v1beta1 PiperOrigin-RevId: 715248038 Source-Link: googleapis/googleapis@79d2e13 Source-Link: googleapis/googleapis-gen@d82f0eb Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZDgyZjBlYjYzZGRjNzgzNjYzZmM3ZTUxOTNiYWIwMzk2Yzg0ZjFkMSJ9 feat: add retrieval_config to ToolConfig v1 PiperOrigin-RevId: 715245748 Source-Link: googleapis/googleapis@870b703 Source-Link: googleapis/googleapis-gen@0273c42 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMDI3M2M0MmQwN2Y5ZmVjMDliYzZiNDBmOWI3Yjk2NzRkNjUxZGQzYiJ9 feat: add Context Cache to v1 PiperOrigin-RevId: 714842212 Source-Link: googleapis/googleapis@51b75ed Source-Link: googleapis/googleapis-gen@2689f70 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMjY4OWY3MGQyYmRhMThmMjI2YTdlZGU2NDgzZGUxMTk2N2U4MTE1NCJ9 feat: Promote network_configs field to v2 API feat: Make v6e available for GA feat: Add UNKNOWN to TPU node state, This state will be used to prevent a node from being marked as READY during diagnose after it has failed repair feat: Make V5Litepod accelerator config available for v2 feat: Make v5p available for GA docs: A comment for field `network_config` in message `.google.cloud.tpu.v2.Node` is changed PiperOrigin-RevId: 714369166 Source-Link: googleapis/googleapis@276b372 Source-Link: googleapis/googleapis-gen@5b65431 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNWI2NTQzMTUyN2FkN2JhZDk0MGRkZWMwZDBhYjllZTJlZjEyY2U5MiJ9 -- d618c3d 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=#4843 from googleapis:owl-bot-copy d618c3d PiperOrigin-RevId: 715801249
1 parent 30a9403 commit 2148b18

File tree

46 files changed

+26638
-79
lines changed

Some content is hidden

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

46 files changed

+26638
-79
lines changed

‎docs/aiplatform_v1/gen_ai_cache_service.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
GenAiCacheService
2+
-----------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1.services.gen_ai_cache_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.aiplatform_v1.services.gen_ai_cache_service.pagers
9+
:members:
10+
:inherited-members:

‎docs/aiplatform_v1/services_.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ Services for Google Cloud Aiplatform v1 API
1212
feature_registry_service
1313
featurestore_online_serving_service
1414
featurestore_service
15+
gen_ai_cache_service
1516
gen_ai_tuning_service
1617
index_endpoint_service
1718
index_service

‎google/cloud/aiplatform_v1/__init__.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,8 @@
4848
)
4949
from .services.featurestore_service import FeaturestoreServiceClient
5050
from .services.featurestore_service import FeaturestoreServiceAsyncClient
51+
from .services.gen_ai_cache_service import GenAiCacheServiceClient
52+
from .services.gen_ai_cache_service import GenAiCacheServiceAsyncClient
5153
from .services.gen_ai_tuning_service import GenAiTuningServiceClient
5254
from .services.gen_ai_tuning_service import GenAiTuningServiceAsyncClient
5355
from .services.index_endpoint_service import IndexEndpointServiceClient
@@ -95,6 +97,7 @@
9597
from .types.api_auth import ApiAuth
9698
from .types.artifact import Artifact
9799
from .types.batch_prediction_job import BatchPredictionJob
100+
from .types.cached_content import CachedContent
98101
from .types.completion_stats import CompletionStats
99102
from .types.content import Blob
100103
from .types.content import Candidate
@@ -449,6 +452,12 @@
449452
from .types.featurestore_service import UpdateFeatureRequest
450453
from .types.featurestore_service import UpdateFeaturestoreOperationMetadata
451454
from .types.featurestore_service import UpdateFeaturestoreRequest
455+
from .types.gen_ai_cache_service import CreateCachedContentRequest
456+
from .types.gen_ai_cache_service import DeleteCachedContentRequest
457+
from .types.gen_ai_cache_service import GetCachedContentRequest
458+
from .types.gen_ai_cache_service import ListCachedContentsRequest
459+
from .types.gen_ai_cache_service import ListCachedContentsResponse
460+
from .types.gen_ai_cache_service import UpdateCachedContentRequest
452461
from .types.genai_tuning_service import CancelTuningJobRequest
453462
from .types.genai_tuning_service import CreateTuningJobRequest
454463
from .types.genai_tuning_service import GetTuningJobRequest
@@ -897,6 +906,7 @@
897906
from .types.tool import GoogleSearchRetrieval
898907
from .types.tool import RagRetrievalConfig
899908
from .types.tool import Retrieval
909+
from .types.tool import RetrievalConfig
900910
from .types.tool import Tool
901911
from .types.tool import ToolConfig
902912
from .types.tool import VertexAISearch
@@ -994,6 +1004,7 @@
9941004
"FeatureRegistryServiceAsyncClient",
9951005
"FeaturestoreOnlineServingServiceAsyncClient",
9961006
"FeaturestoreServiceAsyncClient",
1007+
"GenAiCacheServiceAsyncClient",
9971008
"GenAiTuningServiceAsyncClient",
9981009
"IndexEndpointServiceAsyncClient",
9991010
"IndexServiceAsyncClient",
@@ -1071,6 +1082,7 @@
10711082
"Blob",
10721083
"BlurBaselineConfig",
10731084
"BoolArray",
1085+
"CachedContent",
10741086
"CancelBatchPredictionJobRequest",
10751087
"CancelCustomJobRequest",
10761088
"CancelDataLabelingJobRequest",
@@ -1113,6 +1125,7 @@
11131125
"CountTokensResponse",
11141126
"CreateArtifactRequest",
11151127
"CreateBatchPredictionJobRequest",
1128+
"CreateCachedContentRequest",
11161129
"CreateContextRequest",
11171130
"CreateCustomJobRequest",
11181131
"CreateDataLabelingJobRequest",
@@ -1182,6 +1195,7 @@
11821195
"DedicatedResources",
11831196
"DeleteArtifactRequest",
11841197
"DeleteBatchPredictionJobRequest",
1198+
"DeleteCachedContentRequest",
11851199
"DeleteContextRequest",
11861200
"DeleteCustomJobRequest",
11871201
"DeleteDataLabelingJobRequest",
@@ -1336,6 +1350,7 @@
13361350
"FunctionResponse",
13371351
"GcsDestination",
13381352
"GcsSource",
1353+
"GenAiCacheServiceClient",
13391354
"GenAiTuningServiceClient",
13401355
"GenerateContentRequest",
13411356
"GenerateContentResponse",
@@ -1345,6 +1360,7 @@
13451360
"GetAnnotationSpecRequest",
13461361
"GetArtifactRequest",
13471362
"GetBatchPredictionJobRequest",
1363+
"GetCachedContentRequest",
13481364
"GetContextRequest",
13491365
"GetCustomJobRequest",
13501366
"GetDataLabelingJobRequest",
@@ -1434,6 +1450,8 @@
14341450
"ListArtifactsResponse",
14351451
"ListBatchPredictionJobsRequest",
14361452
"ListBatchPredictionJobsResponse",
1453+
"ListCachedContentsRequest",
1454+
"ListCachedContentsResponse",
14371455
"ListContextsRequest",
14381456
"ListContextsResponse",
14391457
"ListCustomJobsRequest",
@@ -1710,6 +1728,7 @@
17101728
"ResumeModelDeploymentMonitoringJobRequest",
17111729
"ResumeScheduleRequest",
17121730
"Retrieval",
1731+
"RetrievalConfig",
17131732
"RetrievalMetadata",
17141733
"RetrieveContextsRequest",
17151734
"RetrieveContextsResponse",
@@ -1851,6 +1870,7 @@
18511870
"UndeployModelResponse",
18521871
"UnmanagedContainerModel",
18531872
"UpdateArtifactRequest",
1873+
"UpdateCachedContentRequest",
18541874
"UpdateContextRequest",
18551875
"UpdateDatasetRequest",
18561876
"UpdateDatasetVersionRequest",

‎google/cloud/aiplatform_v1/gapic_metadata.json

Lines changed: 94 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1451,6 +1451,100 @@
14511451
}
14521452
}
14531453
},
1454+
"GenAiCacheService": {
1455+
"clients": {
1456+
"grpc": {
1457+
"libraryClient": "GenAiCacheServiceClient",
1458+
"rpcs": {
1459+
"CreateCachedContent": {
1460+
"methods": [
1461+
"create_cached_content"
1462+
]
1463+
},
1464+
"DeleteCachedContent": {
1465+
"methods": [
1466+
"delete_cached_content"
1467+
]
1468+
},
1469+
"GetCachedContent": {
1470+
"methods": [
1471+
"get_cached_content"
1472+
]
1473+
},
1474+
"ListCachedContents": {
1475+
"methods": [
1476+
"list_cached_contents"
1477+
]
1478+
},
1479+
"UpdateCachedContent": {
1480+
"methods": [
1481+
"update_cached_content"
1482+
]
1483+
}
1484+
}
1485+
},
1486+
"grpc-async": {
1487+
"libraryClient": "GenAiCacheServiceAsyncClient",
1488+
"rpcs": {
1489+
"CreateCachedContent": {
1490+
"methods": [
1491+
"create_cached_content"
1492+
]
1493+
},
1494+
"DeleteCachedContent": {
1495+
"methods": [
1496+
"delete_cached_content"
1497+
]
1498+
},
1499+
"GetCachedContent": {
1500+
"methods": [
1501+
"get_cached_content"
1502+
]
1503+
},
1504+
"ListCachedContents": {
1505+
"methods": [
1506+
"list_cached_contents"
1507+
]
1508+
},
1509+
"UpdateCachedContent": {
1510+
"methods": [
1511+
"update_cached_content"
1512+
]
1513+
}
1514+
}
1515+
},
1516+
"rest": {
1517+
"libraryClient": "GenAiCacheServiceClient",
1518+
"rpcs": {
1519+
"CreateCachedContent": {
1520+
"methods": [
1521+
"create_cached_content"
1522+
]
1523+
},
1524+
"DeleteCachedContent": {
1525+
"methods": [
1526+
"delete_cached_content"
1527+
]
1528+
},
1529+
"GetCachedContent": {
1530+
"methods": [
1531+
"get_cached_content"
1532+
]
1533+
},
1534+
"ListCachedContents": {
1535+
"methods": [
1536+
"list_cached_contents"
1537+
]
1538+
},
1539+
"UpdateCachedContent": {
1540+
"methods": [
1541+
"update_cached_content"
1542+
]
1543+
}
1544+
}
1545+
}
1546+
}
1547+
},
14541548
"GenAiTuningService": {
14551549
"clients": {
14561550
"grpc": {

‎google/cloud/aiplatform_v1/services/gen_ai_cache_service/__init__.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
# -*- coding: utf-8 -*-
2+
# Copyright 2024 Google LLC
3+
#
4+
# Licensed under the Apache License, Version 2.0 (the "License");
5+
# you may not use this file except in compliance with the License.
6+
# You may obtain a copy of the License at
7+
#
8+
# http://www.apache.org/licenses/LICENSE-2.0
9+
#
10+
# Unless required by applicable law or agreed to in writing, software
11+
# distributed under the License is distributed on an "AS IS" BASIS,
12+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
# See the License for the specific language governing permissions and
14+
# limitations under the License.
15+
#
16+
from .client import GenAiCacheServiceClient
17+
from .async_client import GenAiCacheServiceAsyncClient
18+
19+
__all__ = (
20+
"GenAiCacheServiceClient",
21+
"GenAiCacheServiceAsyncClient",
22+
)

0 commit comments

Comments
 (0)