Skip to content

Commit 183c038

Browse files
gcf-owl-bot[bot]d-goog
gcf-owl-bot[bot]
and
d-goog
authored
fix: annotate PipelineJob and PipelineTaskRerunConfig fields as optional (#5698)
* feat: add TunedModelRef and RebaseTunedModel Api for Vertex GenAiTuningService docs: limit comment `SupervisedTuningSpec` for 1p tuning PiperOrigin-RevId: 676969679 Source-Link: googleapis/googleapis@2a98f79 Source-Link: googleapis/googleapis-gen@eb6e7f5 Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6ImViNmU3ZjU5NWE4NmIzODVmMWY0NTkzZTQzY2JlMTAxNWMyOWVmNzkifQ== * feat: add TunedModelRef and RebaseTunedModel Api for Vertex GenAiTuningService docs: limit comment `SupervisedTuningSpec` for 1p tuning PiperOrigin-RevId: 676970579 Source-Link: googleapis/googleapis@30009c9 Source-Link: googleapis/googleapis-gen@6d9cc8d Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6IjZkOWNjOGQ2MGFmZjY1MWE2YzQ4ZmIwYTk1MWRkOGE4M2UwNmEyNjcifQ== * fix: annotate PipelineJob and PipelineTaskRerunConfig fields as optional PiperOrigin-RevId: 679707420 Source-Link: googleapis/googleapis@d6f1848 Source-Link: googleapis/googleapis-gen@f6c16c2 Copy-Tag: eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWFpcGxhdGZvcm0vLk93bEJvdC55YW1sIiwiaCI6ImY2YzE2YzIwMDBiMGRmMTU4MWRhZjIzZTY1NWYzZmE0ZTMzZGZmOTkifQ== * 🦉 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> Co-authored-by: Daniel Bankhead <danielbankhead@google.com>
1 parent fd737ec commit 183c038

22 files changed

+8186
-39
lines changed

‎packages/google-cloud-aiplatform/README.md‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ Samples are in the [`samples/`](https://github.com/googleapis/google-cloud-node/
189189
| Gen_ai_tuning_service.create_tuning_job | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.create_tuning_job.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.create_tuning_job.js,packages/google-cloud-aiplatform/samples/README.md) |
190190
| Gen_ai_tuning_service.get_tuning_job | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.get_tuning_job.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.get_tuning_job.js,packages/google-cloud-aiplatform/samples/README.md) |
191191
| Gen_ai_tuning_service.list_tuning_jobs | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.list_tuning_jobs.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.list_tuning_jobs.js,packages/google-cloud-aiplatform/samples/README.md) |
192+
| Gen_ai_tuning_service.rebase_tuned_model | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.rebase_tuned_model.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/gen_ai_tuning_service.rebase_tuned_model.js,packages/google-cloud-aiplatform/samples/README.md) |
192193
| Index_endpoint_service.create_index_endpoint | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/index_endpoint_service.create_index_endpoint.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/index_endpoint_service.create_index_endpoint.js,packages/google-cloud-aiplatform/samples/README.md) |
193194
| Index_endpoint_service.delete_index_endpoint | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/index_endpoint_service.delete_index_endpoint.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/index_endpoint_service.delete_index_endpoint.js,packages/google-cloud-aiplatform/samples/README.md) |
194195
| Index_endpoint_service.deploy_index | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1/index_endpoint_service.deploy_index.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1/index_endpoint_service.deploy_index.js,packages/google-cloud-aiplatform/samples/README.md) |
@@ -499,6 +500,7 @@ Samples are in the [`samples/`](https://github.com/googleapis/google-cloud-node/
499500
| Gen_ai_tuning_service.create_tuning_job | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.create_tuning_job.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.create_tuning_job.js,packages/google-cloud-aiplatform/samples/README.md) |
500501
| Gen_ai_tuning_service.get_tuning_job | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.get_tuning_job.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.get_tuning_job.js,packages/google-cloud-aiplatform/samples/README.md) |
501502
| Gen_ai_tuning_service.list_tuning_jobs | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.list_tuning_jobs.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.list_tuning_jobs.js,packages/google-cloud-aiplatform/samples/README.md) |
503+
| Gen_ai_tuning_service.rebase_tuned_model | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.rebase_tuned_model.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/gen_ai_tuning_service.rebase_tuned_model.js,packages/google-cloud-aiplatform/samples/README.md) |
502504
| Index_endpoint_service.create_index_endpoint | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/index_endpoint_service.create_index_endpoint.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/index_endpoint_service.create_index_endpoint.js,packages/google-cloud-aiplatform/samples/README.md) |
503505
| Index_endpoint_service.delete_index_endpoint | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/index_endpoint_service.delete_index_endpoint.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/index_endpoint_service.delete_index_endpoint.js,packages/google-cloud-aiplatform/samples/README.md) |
504506
| Index_endpoint_service.deploy_index | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-aiplatform/samples/generated/v1beta1/index_endpoint_service.deploy_index.js) | [![Open in Cloud Shell][shell_img]](https://console.cloud.google.com/cloudshell/open?git_repo=https://github.com/googleapis/google-cloud-node&page=editor&open_in_editor=packages/google-cloud-aiplatform/samples/generated/v1beta1/index_endpoint_service.deploy_index.js,packages/google-cloud-aiplatform/samples/README.md) |

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/genai_tuning_service.proto‎

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,10 @@ import "google/api/annotations.proto";
2020
import "google/api/client.proto";
2121
import "google/api/field_behavior.proto";
2222
import "google/api/resource.proto";
23+
import "google/cloud/aiplatform/v1/io.proto";
24+
import "google/cloud/aiplatform/v1/operation.proto";
2325
import "google/cloud/aiplatform/v1/tuning_job.proto";
26+
import "google/longrunning/operations.proto";
2427
import "google/protobuf/empty.proto";
2528

2629
option csharp_namespace = "Google.Cloud.AIPlatform.V1";
@@ -82,6 +85,22 @@ service GenAiTuningService {
8285
};
8386
option (google.api.method_signature) = "name";
8487
}
88+
89+
// Rebase a TunedModel.
90+
// Creates a LongRunningOperation that takes a legacy Tuned GenAI model
91+
// Reference and creates a TuningJob based on newly available model.
92+
rpc RebaseTunedModel(RebaseTunedModelRequest)
93+
returns (google.longrunning.Operation) {
94+
option (google.api.http) = {
95+
post: "/v1/{parent=projects/*/locations/*}/tuningJobs:rebaseTunedModel"
96+
body: "*"
97+
};
98+
option (google.api.method_signature) = "parent,tuned_model_ref";
99+
option (google.longrunning.operation_info) = {
100+
response_type: "TuningJob"
101+
metadata_type: "RebaseTunedModelOperationMetadata"
102+
};
103+
}
85104
}
86105

87106
// Request message for
@@ -162,3 +181,39 @@ message CancelTuningJobRequest {
162181
}
163182
];
164183
}
184+
185+
// Request message for
186+
// [GenAiTuningService.RebaseTunedModel][google.cloud.aiplatform.v1.GenAiTuningService.RebaseTunedModel].
187+
message RebaseTunedModelRequest {
188+
// Required. The resource name of the Location into which to rebase the Model.
189+
// Format: `projects/{project}/locations/{location}`
190+
string parent = 1 [
191+
(google.api.field_behavior) = REQUIRED,
192+
(google.api.resource_reference) = {
193+
type: "locations.googleapis.com/Location"
194+
}
195+
];
196+
197+
// Required. TunedModel reference to retrieve the legacy model information.
198+
TunedModelRef tuned_model_ref = 2 [(google.api.field_behavior) = REQUIRED];
199+
200+
// Optional. The TuningJob to be updated. Users can use this TuningJob field
201+
// to overwrite tuning configs.
202+
TuningJob tuning_job = 3 [(google.api.field_behavior) = OPTIONAL];
203+
204+
// Optional. The Google Cloud Storage location to write the artifacts.
205+
GcsDestination artifact_destination = 4
206+
[(google.api.field_behavior) = OPTIONAL];
207+
208+
// Optional. By default, bison to gemini migration will always create new
209+
// model/endpoint, but for gemini-1.0 to gemini-1.5 migration, we default
210+
// deploy to the same endpoint. See details in this Section.
211+
bool deploy_to_same_endpoint = 5 [(google.api.field_behavior) = OPTIONAL];
212+
}
213+
214+
// Runtime operation information for
215+
// [GenAiTuningService.RebaseTunedModel][google.cloud.aiplatform.v1.GenAiTuningService.RebaseTunedModel].
216+
message RebaseTunedModelOperationMetadata {
217+
// The common part of the operation generic information.
218+
GenericOperationMetadata generic_metadata = 1;
219+
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1/tuning_job.proto‎

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ message SupervisedHyperParameters {
285285
AdapterSize adapter_size = 3 [(google.api.field_behavior) = OPTIONAL];
286286
}
287287

288-
// Tuning Spec for Supervised Tuning.
288+
// Tuning Spec for Supervised Tuning for first party models.
289289
message SupervisedTuningSpec {
290290
// Required. Cloud Storage path to file containing training dataset for
291291
// tuning. The dataset must be formatted as a JSONL file.
@@ -299,3 +299,26 @@ message SupervisedTuningSpec {
299299
SupervisedHyperParameters hyper_parameters = 3
300300
[(google.api.field_behavior) = OPTIONAL];
301301
}
302+
303+
// TunedModel Reference for legacy model migration.
304+
message TunedModelRef {
305+
// The Tuned Model Reference for the model.
306+
oneof tuned_model_ref {
307+
// Support migration from model registry.
308+
string tuned_model = 1 [(google.api.resource_reference) = {
309+
type: "aiplatform.googleapis.com/Model"
310+
}];
311+
312+
// Support migration from tuning job list page, from gemini-1.0-pro-002
313+
// to 1.5 and above.
314+
string tuning_job = 2 [(google.api.resource_reference) = {
315+
type: "aiplatform.googleapis.com/TuningJob"
316+
}];
317+
318+
// Support migration from tuning job list page, from bison model to gemini
319+
// model.
320+
string pipeline_job = 3 [(google.api.resource_reference) = {
321+
type: "aiplatform.googleapis.com/PipelineJob"
322+
}];
323+
}
324+
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1beta1/genai_tuning_service.proto‎

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,10 @@ import "google/api/annotations.proto";
2020
import "google/api/client.proto";
2121
import "google/api/field_behavior.proto";
2222
import "google/api/resource.proto";
23+
import "google/cloud/aiplatform/v1beta1/io.proto";
24+
import "google/cloud/aiplatform/v1beta1/operation.proto";
2325
import "google/cloud/aiplatform/v1beta1/tuning_job.proto";
26+
import "google/longrunning/operations.proto";
2427
import "google/protobuf/empty.proto";
2528

2629
option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
@@ -82,6 +85,22 @@ service GenAiTuningService {
8285
};
8386
option (google.api.method_signature) = "name";
8487
}
88+
89+
// Rebase a TunedModel.
90+
// Creates a LongRunningOperation that takes a legacy Tuned GenAI model
91+
// Reference and creates a TuningJob based on newly available model.
92+
rpc RebaseTunedModel(RebaseTunedModelRequest)
93+
returns (google.longrunning.Operation) {
94+
option (google.api.http) = {
95+
post: "/v1beta1/{parent=projects/*/locations/*}/tuningJobs:rebaseTunedModel"
96+
body: "*"
97+
};
98+
option (google.api.method_signature) = "parent,tuned_model_ref";
99+
option (google.longrunning.operation_info) = {
100+
response_type: "TuningJob"
101+
metadata_type: "RebaseTunedModelOperationMetadata"
102+
};
103+
}
85104
}
86105

87106
// Request message for
@@ -162,3 +181,39 @@ message CancelTuningJobRequest {
162181
}
163182
];
164183
}
184+
185+
// Request message for
186+
// [GenAiTuningService.RebaseTunedModel][google.cloud.aiplatform.v1beta1.GenAiTuningService.RebaseTunedModel].
187+
message RebaseTunedModelRequest {
188+
// Required. The resource name of the Location into which to rebase the Model.
189+
// Format: `projects/{project}/locations/{location}`
190+
string parent = 1 [
191+
(google.api.field_behavior) = REQUIRED,
192+
(google.api.resource_reference) = {
193+
type: "locations.googleapis.com/Location"
194+
}
195+
];
196+
197+
// Required. TunedModel reference to retrieve the legacy model information.
198+
TunedModelRef tuned_model_ref = 2 [(google.api.field_behavior) = REQUIRED];
199+
200+
// Optional. The TuningJob to be updated. Users can use this TuningJob field
201+
// to overwrite tuning configs.
202+
TuningJob tuning_job = 3 [(google.api.field_behavior) = OPTIONAL];
203+
204+
// Optional. The Google Cloud Storage location to write the artifacts.
205+
GcsDestination artifact_destination = 4
206+
[(google.api.field_behavior) = OPTIONAL];
207+
208+
// Optional. By default, bison to gemini migration will always create new
209+
// model/endpoint, but for gemini-1.0 to gemini-1.5 migration, we default
210+
// deploy to the same endpoint. See details in this Section.
211+
bool deploy_to_same_endpoint = 5 [(google.api.field_behavior) = OPTIONAL];
212+
}
213+
214+
// Runtime operation information for
215+
// [GenAiTuningService.RebaseTunedModel][google.cloud.aiplatform.v1beta1.GenAiTuningService.RebaseTunedModel].
216+
message RebaseTunedModelOperationMetadata {
217+
// The common part of the operation generic information.
218+
GenericOperationMetadata generic_metadata = 1;
219+
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1beta1/pipeline_job.proto‎

Lines changed: 21 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -224,12 +224,11 @@ message PipelineJob {
224224
// Output only. Reserved for future use.
225225
bool satisfies_pzi = 28 [(google.api.field_behavior) = OUTPUT_ONLY];
226226

227-
// Output only. The original pipeline job id if this pipeline job is a rerun
228-
// of a previous pipeline job.
229-
int64 original_pipeline_job_id = 29
230-
[(google.api.field_behavior) = OUTPUT_ONLY];
227+
// Optional. The original pipeline job id if this pipeline job is a rerun of a
228+
// previous pipeline job.
229+
int64 original_pipeline_job_id = 29 [(google.api.field_behavior) = OPTIONAL];
231230

232-
// Output only. The rerun configs for each task in the pipeline job.
231+
// Optional. The rerun configs for each task in the pipeline job.
233232
// By default, the rerun will:
234233
// 1. Use the same input artifacts as the original run.
235234
// 2. Use the same input parameters as the original run.
@@ -238,7 +237,7 @@ message PipelineJob {
238237
// By providing this field, users can override the default behavior and
239238
// specify the rerun config for each task.
240239
repeated PipelineTaskRerunConfig pipeline_task_rerun_configs = 30
241-
[(google.api.field_behavior) = OUTPUT_ONLY];
240+
[(google.api.field_behavior) = OPTIONAL];
242241
}
243242

244243
// Pipeline template metadata if
@@ -459,35 +458,34 @@ message PipelineTaskExecutorDetail {
459458
message PipelineTaskRerunConfig {
460459
// A list of artifact metadata.
461460
message ArtifactList {
462-
// Output only. A list of artifact metadata.
461+
// Optional. A list of artifact metadata.
463462
repeated RuntimeArtifact artifacts = 1
464-
[(google.api.field_behavior) = OUTPUT_ONLY];
463+
[(google.api.field_behavior) = OPTIONAL];
465464
}
466465

467466
// Runtime inputs data of the task.
468467
message Inputs {
469-
// Output only. Input artifacts.
468+
// Optional. Input artifacts.
470469
map<string, ArtifactList> artifacts = 1
471-
[(google.api.field_behavior) = OUTPUT_ONLY];
470+
[(google.api.field_behavior) = OPTIONAL];
472471

473-
// Output only. Input parameters.
472+
// Optional. Input parameters.
474473
map<string, google.protobuf.Value> parameter_values = 2
475-
[(google.api.field_behavior) = OUTPUT_ONLY];
474+
[(google.api.field_behavior) = OPTIONAL];
476475
}
477476

478-
// Output only. The system generated ID of the task. Retrieved from original
479-
// run.
480-
int64 task_id = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
477+
// Optional. The system generated ID of the task. Retrieved from original run.
478+
int64 task_id = 1 [(google.api.field_behavior) = OPTIONAL];
481479

482-
// Output only. The name of the task.
483-
string task_name = 2 [(google.api.field_behavior) = OUTPUT_ONLY];
480+
// Optional. The name of the task.
481+
string task_name = 2 [(google.api.field_behavior) = OPTIONAL];
484482

485-
// Output only. The runtime input of the task overridden by the user.
486-
Inputs inputs = 3 [(google.api.field_behavior) = OUTPUT_ONLY];
483+
// Optional. The runtime input of the task overridden by the user.
484+
Inputs inputs = 3 [(google.api.field_behavior) = OPTIONAL];
487485

488-
// Output only. Whether to skip this task. Default value is False.
489-
bool skip_task = 4 [(google.api.field_behavior) = OUTPUT_ONLY];
486+
// Optional. Whether to skip this task. Default value is False.
487+
bool skip_task = 4 [(google.api.field_behavior) = OPTIONAL];
490488

491-
// Output only. Whether to skip downstream tasks. Default value is False.
492-
bool skip_downstream_tasks = 5 [(google.api.field_behavior) = OUTPUT_ONLY];
489+
// Optional. Whether to skip downstream tasks. Default value is False.
490+
bool skip_downstream_tasks = 5 [(google.api.field_behavior) = OPTIONAL];
493491
}

‎packages/google-cloud-aiplatform/protos/google/cloud/aiplatform/v1beta1/tuning_job.proto‎

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,7 @@ message SupervisedHyperParameters {
384384
AdapterSize adapter_size = 3 [(google.api.field_behavior) = OPTIONAL];
385385
}
386386

387-
// Tuning Spec for Supervised Tuning.
387+
// Tuning Spec for Supervised Tuning for first party models.
388388
message SupervisedTuningSpec {
389389
// Required. Cloud Storage path to file containing training dataset for
390390
// tuning. The dataset must be formatted as a JSONL file.
@@ -450,3 +450,26 @@ message DistillationHyperParameters {
450450
SupervisedHyperParameters.AdapterSize adapter_size = 3
451451
[(google.api.field_behavior) = OPTIONAL];
452452
}
453+
454+
// TunedModel Reference for legacy model migration.
455+
message TunedModelRef {
456+
// The Tuned Model Reference for the model.
457+
oneof tuned_model_ref {
458+
// Support migration from model registry.
459+
string tuned_model = 1 [(google.api.resource_reference) = {
460+
type: "aiplatform.googleapis.com/Model"
461+
}];
462+
463+
// Support migration from tuning job list page, from gemini-1.0-pro-002
464+
// to 1.5 and above.
465+
string tuning_job = 2 [(google.api.resource_reference) = {
466+
type: "aiplatform.googleapis.com/TuningJob"
467+
}];
468+
469+
// Support migration from tuning job list page, from bison model to gemini
470+
// model.
471+
string pipeline_job = 3 [(google.api.resource_reference) = {
472+
type: "aiplatform.googleapis.com/PipelineJob"
473+
}];
474+
}
475+
}

0 commit comments

Comments
 (0)