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]>
googleapis · May 24, 2024 · 5436d88 · 5436d88
1 parent 0e8b8cd
commit 5436d88
Show file tree

Hide file tree

Showing 27 changed files with 17,930 additions and 0 deletions.
diff --git a/docs/aiplatform_v1beta1/gen_ai_tuning_service.rst b/docs/aiplatform_v1beta1/gen_ai_tuning_service.rst
@@ -0,0 +1,10 @@
+GenAiTuningService
+------------------------------------
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_tuning_service
+    :members:
+    :inherited-members:
+
+.. automodule:: google.cloud.aiplatform_v1beta1.services.gen_ai_tuning_service.pagers
+    :members:
+    :inherited-members:
diff --git a/docs/aiplatform_v1beta1/services_.rst b/docs/aiplatform_v1beta1/services_.rst
@@ -14,6 +14,7 @@ Services for Google Cloud Aiplatform v1beta1 API
     feature_registry_service
     featurestore_online_serving_service
     featurestore_service
+    gen_ai_tuning_service
     index_endpoint_service
     index_service
     job_service

diff --git a/google/cloud/aiplatform_v1beta1/__init__.py b/google/cloud/aiplatform_v1beta1/__init__.py
@@ -52,6 +52,8 @@
 )
 from .services.featurestore_service import FeaturestoreServiceClient
 from .services.featurestore_service import FeaturestoreServiceAsyncClient
+from .services.gen_ai_tuning_service import GenAiTuningServiceClient
+from .services.gen_ai_tuning_service import GenAiTuningServiceAsyncClient
 from .services.index_endpoint_service import IndexEndpointServiceClient
 from .services.index_endpoint_service import IndexEndpointServiceAsyncClient
 from .services.index_service import IndexServiceClient
@@ -454,6 +456,11 @@
 from .types.featurestore_service import UpdateFeatureRequest
 from .types.featurestore_service import UpdateFeaturestoreOperationMetadata
 from .types.featurestore_service import UpdateFeaturestoreRequest
+from .types.genai_tuning_service import CancelTuningJobRequest
+from .types.genai_tuning_service import CreateTuningJobRequest
+from .types.genai_tuning_service import GetTuningJobRequest
+from .types.genai_tuning_service import ListTuningJobsRequest
+from .types.genai_tuning_service import ListTuningJobsResponse
 from .types.hyperparameter_tuning_job import HyperparameterTuningJob
 from .types.index import Index
 from .types.index import IndexDatapoint
@@ -939,6 +946,13 @@
 from .types.training_pipeline import StratifiedSplit
 from .types.training_pipeline import TimestampSplit
 from .types.training_pipeline import TrainingPipeline
+from .types.tuning_job import SupervisedHyperParameters
+from .types.tuning_job import SupervisedTuningDatasetDistribution
+from .types.tuning_job import SupervisedTuningDataStats
+from .types.tuning_job import SupervisedTuningSpec
+from .types.tuning_job import TunedModel
+from .types.tuning_job import TuningDataStats
+from .types.tuning_job import TuningJob
 from .types.types import BoolArray
 from .types.types import DoubleArray
 from .types.types import Int64Array
@@ -1006,6 +1020,7 @@
     "FeatureRegistryServiceAsyncClient",
     "FeaturestoreOnlineServingServiceAsyncClient",
     "FeaturestoreServiceAsyncClient",
+    "GenAiTuningServiceAsyncClient",
     "IndexEndpointServiceAsyncClient",
     "IndexServiceAsyncClient",
     "JobServiceAsyncClient",
@@ -1091,6 +1106,7 @@
     "CancelNasJobRequest",
     "CancelPipelineJobRequest",
     "CancelTrainingPipelineRequest",
+    "CancelTuningJobRequest",
     "Candidate",
     "CheckTrialEarlyStoppingStateMetatdata",
     "CheckTrialEarlyStoppingStateRequest",
@@ -1175,6 +1191,7 @@
     "CreateTensorboardTimeSeriesRequest",
     "CreateTrainingPipelineRequest",
     "CreateTrialRequest",
+    "CreateTuningJobRequest",
     "CsvDestination",
     "CsvSource",
     "CustomJob",
@@ -1349,6 +1366,7 @@
     "FunctionResponse",
     "GcsDestination",
     "GcsSource",
+    "GenAiTuningServiceClient",
     "GenerateContentRequest",
     "GenerateContentResponse",
     "GenerationConfig",
@@ -1404,6 +1422,7 @@
     "GetTensorboardTimeSeriesRequest",
     "GetTrainingPipelineRequest",
     "GetTrialRequest",
+    "GetTuningJobRequest",
     "GoogleDriveSource",
     "GoogleSearchRetrieval",
     "GroundednessInput",
@@ -1552,6 +1571,8 @@
     "ListTrainingPipelinesResponse",
     "ListTrialsRequest",
     "ListTrialsResponse",
+    "ListTuningJobsRequest",
+    "ListTuningJobsResponse",
     "LlmUtilityServiceClient",
     "LookupStudyRequest",
     "MachineSpec",
@@ -1817,6 +1838,10 @@
     "SummarizationVerbosityInstance",
     "SummarizationVerbosityResult",
     "SummarizationVerbositySpec",
+    "SupervisedHyperParameters",
+    "SupervisedTuningDataStats",
+    "SupervisedTuningDatasetDistribution",
+    "SupervisedTuningSpec",
     "SyncFeatureViewRequest",
     "SyncFeatureViewResponse",
     "TFRecordDestination",
@@ -1861,6 +1886,9 @@
     "TrainingPipeline",
     "Trial",
     "TrialContext",
+    "TunedModel",
+    "TuningDataStats",
+    "TuningJob",
     "Type",
     "UndeployIndexOperationMetadata",
     "UndeployIndexRequest",

diff --git a/google/cloud/aiplatform_v1beta1/gapic_metadata.json b/google/cloud/aiplatform_v1beta1/gapic_metadata.json
@@ -1564,6 +1564,85 @@
         }
       }
     },
+    "GenAiTuningService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "GenAiTuningServiceClient",
+          "rpcs": {
+            "CancelTuningJob": {
+              "methods": [
+                "cancel_tuning_job"
+              ]
+            },
+            "CreateTuningJob": {
+              "methods": [
+                "create_tuning_job"
+              ]
+            },
+            "GetTuningJob": {
+              "methods": [
+                "get_tuning_job"
+              ]
+            },
+            "ListTuningJobs": {
+              "methods": [
+                "list_tuning_jobs"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "GenAiTuningServiceAsyncClient",
+          "rpcs": {
+            "CancelTuningJob": {
+              "methods": [
+                "cancel_tuning_job"
+              ]
+            },
+            "CreateTuningJob": {
+              "methods": [
+                "create_tuning_job"
+              ]
+            },
+            "GetTuningJob": {
+              "methods": [
+                "get_tuning_job"
+              ]
+            },
+            "ListTuningJobs": {
+              "methods": [
+                "list_tuning_jobs"
+              ]
+            }
+          }
+        },
+        "rest": {
+          "libraryClient": "GenAiTuningServiceClient",
+          "rpcs": {
+            "CancelTuningJob": {
+              "methods": [
+                "cancel_tuning_job"
+              ]
+            },
+            "CreateTuningJob": {
+              "methods": [
+                "create_tuning_job"
+              ]
+            },
+            "GetTuningJob": {
+              "methods": [
+                "get_tuning_job"
+              ]
+            },
+            "ListTuningJobs": {
+              "methods": [
+                "list_tuning_jobs"
+              ]
+            }
+          }
+        }
+      }
+    },
     "IndexEndpointService": {
       "clients": {
         "grpc": {

diff --git a/google/cloud/aiplatform_v1beta1/services/gen_ai_tuning_service/__init__.py b/google/cloud/aiplatform_v1beta1/services/gen_ai_tuning_service/__init__.py
@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+from .client import GenAiTuningServiceClient
+from .async_client import GenAiTuningServiceAsyncClient
+
+__all__ = (
+    "GenAiTuningServiceClient",
+    "GenAiTuningServiceAsyncClient",
+)