Skip to content

Commit 5436d88

Browse files
feat: Added the new GenerationConfig.response_schema field (#3772)
* feat: Added the v1beta1 version of the GenAI Tuning Service PiperOrigin-RevId: 632610890 Source-Link: googleapis/googleapis@e5eef0f Source-Link: googleapis/googleapis-gen@39bd6f8 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMzliZDZmOGRiYzNmZDY0OTgyYjU1OWE0MzkyNmU2NzEwY2M3YjMyOCJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * feat: Added the new `GenerationConfig.response_schema` field PiperOrigin-RevId: 636671748 Source-Link: googleapis/googleapis@9b92004 Source-Link: googleapis/googleapis-gen@63661cc Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNjM2NjFjY2Q1ZGY3ZjNmODM3NDQwZDM2NmI5MjY2YjFhZjI2YzNlNCJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * Update snippet_metadata_google.cloud.aiplatform.v1.json * Update snippet_metadata_google.cloud.aiplatform.v1beta1.json --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com> Co-authored-by: Amy Wu <[email protected]>
1 parent 0e8b8cd commit 5436d88

27 files changed

+17930
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
GenAiTuningService
2+
------------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_tuning_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_tuning_service.pagers
9+
:members:
10+
:inherited-members:

docs/aiplatform_v1beta1/services_.rst

+1
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ Services for Google Cloud Aiplatform v1beta1 API
1414
feature_registry_service
1515
featurestore_online_serving_service
1616
featurestore_service
17+
gen_ai_tuning_service
1718
index_endpoint_service
1819
index_service
1920
job_service

google/cloud/aiplatform_v1beta1/__init__.py

+28
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@
5252
)
5353
from .services.featurestore_service import FeaturestoreServiceClient
5454
from .services.featurestore_service import FeaturestoreServiceAsyncClient
55+
from .services.gen_ai_tuning_service import GenAiTuningServiceClient
56+
from .services.gen_ai_tuning_service import GenAiTuningServiceAsyncClient
5557
from .services.index_endpoint_service import IndexEndpointServiceClient
5658
from .services.index_endpoint_service import IndexEndpointServiceAsyncClient
5759
from .services.index_service import IndexServiceClient
@@ -454,6 +456,11 @@
454456
from .types.featurestore_service import UpdateFeatureRequest
455457
from .types.featurestore_service import UpdateFeaturestoreOperationMetadata
456458
from .types.featurestore_service import UpdateFeaturestoreRequest
459+
from .types.genai_tuning_service import CancelTuningJobRequest
460+
from .types.genai_tuning_service import CreateTuningJobRequest
461+
from .types.genai_tuning_service import GetTuningJobRequest
462+
from .types.genai_tuning_service import ListTuningJobsRequest
463+
from .types.genai_tuning_service import ListTuningJobsResponse
457464
from .types.hyperparameter_tuning_job import HyperparameterTuningJob
458465
from .types.index import Index
459466
from .types.index import IndexDatapoint
@@ -939,6 +946,13 @@
939946
from .types.training_pipeline import StratifiedSplit
940947
from .types.training_pipeline import TimestampSplit
941948
from .types.training_pipeline import TrainingPipeline
949+
from .types.tuning_job import SupervisedHyperParameters
950+
from .types.tuning_job import SupervisedTuningDatasetDistribution
951+
from .types.tuning_job import SupervisedTuningDataStats
952+
from .types.tuning_job import SupervisedTuningSpec
953+
from .types.tuning_job import TunedModel
954+
from .types.tuning_job import TuningDataStats
955+
from .types.tuning_job import TuningJob
942956
from .types.types import BoolArray
943957
from .types.types import DoubleArray
944958
from .types.types import Int64Array
@@ -1006,6 +1020,7 @@
10061020
"FeatureRegistryServiceAsyncClient",
10071021
"FeaturestoreOnlineServingServiceAsyncClient",
10081022
"FeaturestoreServiceAsyncClient",
1023+
"GenAiTuningServiceAsyncClient",
10091024
"IndexEndpointServiceAsyncClient",
10101025
"IndexServiceAsyncClient",
10111026
"JobServiceAsyncClient",
@@ -1091,6 +1106,7 @@
10911106
"CancelNasJobRequest",
10921107
"CancelPipelineJobRequest",
10931108
"CancelTrainingPipelineRequest",
1109+
"CancelTuningJobRequest",
10941110
"Candidate",
10951111
"CheckTrialEarlyStoppingStateMetatdata",
10961112
"CheckTrialEarlyStoppingStateRequest",
@@ -1175,6 +1191,7 @@
11751191
"CreateTensorboardTimeSeriesRequest",
11761192
"CreateTrainingPipelineRequest",
11771193
"CreateTrialRequest",
1194+
"CreateTuningJobRequest",
11781195
"CsvDestination",
11791196
"CsvSource",
11801197
"CustomJob",
@@ -1349,6 +1366,7 @@
13491366
"FunctionResponse",
13501367
"GcsDestination",
13511368
"GcsSource",
1369+
"GenAiTuningServiceClient",
13521370
"GenerateContentRequest",
13531371
"GenerateContentResponse",
13541372
"GenerationConfig",
@@ -1404,6 +1422,7 @@
14041422
"GetTensorboardTimeSeriesRequest",
14051423
"GetTrainingPipelineRequest",
14061424
"GetTrialRequest",
1425+
"GetTuningJobRequest",
14071426
"GoogleDriveSource",
14081427
"GoogleSearchRetrieval",
14091428
"GroundednessInput",
@@ -1552,6 +1571,8 @@
15521571
"ListTrainingPipelinesResponse",
15531572
"ListTrialsRequest",
15541573
"ListTrialsResponse",
1574+
"ListTuningJobsRequest",
1575+
"ListTuningJobsResponse",
15551576
"LlmUtilityServiceClient",
15561577
"LookupStudyRequest",
15571578
"MachineSpec",
@@ -1817,6 +1838,10 @@
18171838
"SummarizationVerbosityInstance",
18181839
"SummarizationVerbosityResult",
18191840
"SummarizationVerbositySpec",
1841+
"SupervisedHyperParameters",
1842+
"SupervisedTuningDataStats",
1843+
"SupervisedTuningDatasetDistribution",
1844+
"SupervisedTuningSpec",
18201845
"SyncFeatureViewRequest",
18211846
"SyncFeatureViewResponse",
18221847
"TFRecordDestination",
@@ -1861,6 +1886,9 @@
18611886
"TrainingPipeline",
18621887
"Trial",
18631888
"TrialContext",
1889+
"TunedModel",
1890+
"TuningDataStats",
1891+
"TuningJob",
18641892
"Type",
18651893
"UndeployIndexOperationMetadata",
18661894
"UndeployIndexRequest",

google/cloud/aiplatform_v1beta1/gapic_metadata.json

+79
Original file line numberDiff line numberDiff line change
@@ -1564,6 +1564,85 @@
15641564
}
15651565
}
15661566
},
1567+
"GenAiTuningService": {
1568+
"clients": {
1569+
"grpc": {
1570+
"libraryClient": "GenAiTuningServiceClient",
1571+
"rpcs": {
1572+
"CancelTuningJob": {
1573+
"methods": [
1574+
"cancel_tuning_job"
1575+
]
1576+
},
1577+
"CreateTuningJob": {
1578+
"methods": [
1579+
"create_tuning_job"
1580+
]
1581+
},
1582+
"GetTuningJob": {
1583+
"methods": [
1584+
"get_tuning_job"
1585+
]
1586+
},
1587+
"ListTuningJobs": {
1588+
"methods": [
1589+
"list_tuning_jobs"
1590+
]
1591+
}
1592+
}
1593+
},
1594+
"grpc-async": {
1595+
"libraryClient": "GenAiTuningServiceAsyncClient",
1596+
"rpcs": {
1597+
"CancelTuningJob": {
1598+
"methods": [
1599+
"cancel_tuning_job"
1600+
]
1601+
},
1602+
"CreateTuningJob": {
1603+
"methods": [
1604+
"create_tuning_job"
1605+
]
1606+
},
1607+
"GetTuningJob": {
1608+
"methods": [
1609+
"get_tuning_job"
1610+
]
1611+
},
1612+
"ListTuningJobs": {
1613+
"methods": [
1614+
"list_tuning_jobs"
1615+
]
1616+
}
1617+
}
1618+
},
1619+
"rest": {
1620+
"libraryClient": "GenAiTuningServiceClient",
1621+
"rpcs": {
1622+
"CancelTuningJob": {
1623+
"methods": [
1624+
"cancel_tuning_job"
1625+
]
1626+
},
1627+
"CreateTuningJob": {
1628+
"methods": [
1629+
"create_tuning_job"
1630+
]
1631+
},
1632+
"GetTuningJob": {
1633+
"methods": [
1634+
"get_tuning_job"
1635+
]
1636+
},
1637+
"ListTuningJobs": {
1638+
"methods": [
1639+
"list_tuning_jobs"
1640+
]
1641+
}
1642+
}
1643+
}
1644+
}
1645+
},
15671646
"IndexEndpointService": {
15681647
"clients": {
15691648
"grpc": {
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 GenAiTuningServiceClient
17+
from .async_client import GenAiTuningServiceAsyncClient
18+
19+
__all__ = (
20+
"GenAiTuningServiceClient",
21+
"GenAiTuningServiceAsyncClient",
22+
)

0 commit comments

Comments
 (0)