Copybara import of the project:

-- 94bb896 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: Add service_account to Reasoning Engine public protos PiperOrigin-RevId: 785977703 Source-Link: googleapis/googleapis@a0a150a Source-Link: googleapis/googleapis-gen@1a2482c Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWEyNDgyYzMxOTBjMTUzMGMxOTkxNjNhNmY4MjI0NjU5NzJhOWNkOSJ9 feat: Add service_account to Reasoning Engine public protos PiperOrigin-RevId: 785969680 Source-Link: googleapis/googleapis@2e4cb44 Source-Link: googleapis/googleapis-gen@cdae6f6 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiY2RhZTZmNjRmMzY0ZjRkY2NhMjk5NWIzNmE2NjdiMWNiMTg4MzVkNSJ9 fix!: remove gemini_template_config and request_column_name fields from DatasetService.AssessData and DatasetService.AssembleData PiperOrigin-RevId: 784583197 Source-Link: googleapis/googleapis@fc59d68 Source-Link: googleapis/googleapis-gen@d334e99 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZDMzNGU5OWFkZDg1YmM3YmQ5NDg1NDc2MTcwNWQzY2RiMzRiMzEyYSJ9 -- 6bcc7cd 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=#5572 from googleapis:owl-bot-copy b9e825a PiperOrigin-RevId: 786831098

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1/types/reasoning_engine.py

@@ -36,7 +36,20 @@
 class ReasoningEngineSpec(proto.Message):
  r"""ReasoningEngine configurations
 
+ .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
  Attributes:
+ service_account (str):
+ Optional. The service account that the
+ Reasoning Engine artifact runs as. It should
+ have "roles/storage.objectViewer" for reading
+ the user project's Cloud Storage and
+ "roles/aiplatform.user" for using Vertex
+ extensions. If not specified, the Vertex AI
+ Reasoning Engine Service Agent in the project
+ will be used.
+
+ This field is a member of `oneof`_ ``_service_account``.
  package_spec (google.cloud.aiplatform_v1.types.ReasoningEngineSpec.PackageSpec):
  Optional. User provided package spec of the ReasoningEngine.
  Ignored when users directly specify a deployment image
@@ -121,6 +134,11 @@ class DeploymentSpec(proto.Message):
  message=env_var.SecretEnvVar,
  )
 
+ service_account: str = proto.Field(
+ proto.STRING,
+ number=1,
+ optional=True,
+ )
  package_spec: PackageSpec = proto.Field(
  proto.MESSAGE,
  number=2,

google/cloud/aiplatform_v1beta1/services/dataset_service/async_client.py

@@ -2796,12 +2796,8 @@ async def sample_assess_data():
  tuning_validation_assessment_config.model_name = "model_name_value"
  tuning_validation_assessment_config.dataset_usage = "SFT_VALIDATION"
 
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssessDataRequest(
  tuning_validation_assessment_config=tuning_validation_assessment_config,
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 
@@ -2903,11 +2899,7 @@ async def sample_assemble_data():
  client = aiplatform_v1beta1.DatasetServiceAsyncClient()
 
  # Initialize request argument(s)
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssembleDataRequest(
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 

google/cloud/aiplatform_v1beta1/services/dataset_service/client.py

@@ -3367,12 +3367,8 @@ def sample_assess_data():
  tuning_validation_assessment_config.model_name = "model_name_value"
  tuning_validation_assessment_config.dataset_usage = "SFT_VALIDATION"
 
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssessDataRequest(
  tuning_validation_assessment_config=tuning_validation_assessment_config,
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 
@@ -3472,11 +3468,7 @@ def sample_assemble_data():
  client = aiplatform_v1beta1.DatasetServiceClient()
 
  # Initialize request argument(s)
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssembleDataRequest(
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 

google/cloud/aiplatform_v1beta1/types/dataset_service.py

@@ -1164,18 +1164,6 @@ class AssessDataRequest(proto.Message):
  prediction resource usage assessment.
 
  This field is a member of `oneof`_ ``assessment_config``.
- gemini_template_config (google.cloud.aiplatform_v1beta1.types.GeminiTemplateConfig):
- Optional. Config for assembling templates
- with a Gemini API structure to assess assembled
- data.
-
- This field is a member of `oneof`_ ``read_config``.
- request_column_name (str):
- Optional. The column name in the underlying
- table that contains already fully assembled
- requests.
-
- This field is a member of `oneof`_ ``read_config``.
  name (str):
  Required. The name of the Dataset resource. Used only for
  MULTIMODAL datasets. Format:
@@ -1291,17 +1279,6 @@ class BatchPredictionResourceUsageAssessmentConfig(proto.Message):
  oneof="assessment_config",
  message=BatchPredictionResourceUsageAssessmentConfig,
  )
- gemini_template_config: "GeminiTemplateConfig" = proto.Field(
- proto.MESSAGE,
- number=4,
- oneof="read_config",
- message="GeminiTemplateConfig",
- )
- request_column_name: str = proto.Field(
- proto.STRING,
- number=5,
- oneof="read_config",
- )
  name: str = proto.Field(
  proto.STRING,
  number=1,
@@ -1629,26 +1606,7 @@ class AssembleDataRequest(proto.Message):
  [DatasetService.AssembleData][google.cloud.aiplatform.v1beta1.DatasetService.AssembleData].
  Used only for MULTIMODAL datasets.
 
- This message has `oneof`_ fields (mutually exclusive fields).
- For each oneof, at most one member field can be set at the same time.
- Setting any member of the oneof automatically clears all other
- members.
-
- .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
-
  Attributes:
- gemini_template_config (google.cloud.aiplatform_v1beta1.types.GeminiTemplateConfig):
- Optional. Config for assembling templates
- with a Gemini API structure.
-
- This field is a member of `oneof`_ ``read_config``.
- request_column_name (str):
- Optional. The column name in the underlying
- table that contains already fully assembled
- requests. If this field is set, the original
- request will be copied to the output table.
-
- This field is a member of `oneof`_ ``read_config``.
  name (str):
  Required. The name of the Dataset resource (used only for
  MULTIMODAL datasets). Format:
@@ -1657,17 +1615,6 @@ class AssembleDataRequest(proto.Message):
  Optional. The read config for the dataset.
  """
 
- gemini_template_config: "GeminiTemplateConfig" = proto.Field(
- proto.MESSAGE,
- number=2,
- oneof="read_config",
- message="GeminiTemplateConfig",
- )
- request_column_name: str = proto.Field(
- proto.STRING,
- number=5,
- oneof="read_config",
- )
  name: str = proto.Field(
  proto.STRING,
  number=1,

google/cloud/aiplatform_v1beta1/types/reasoning_engine.py

@@ -37,7 +37,20 @@
 class ReasoningEngineSpec(proto.Message):
  r"""ReasoningEngine configurations
 
+ .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
+
  Attributes:
+ service_account (str):
+ Optional. The service account that the
+ Reasoning Engine artifact runs as. It should
+ have "roles/storage.objectViewer" for reading
+ the user project's Cloud Storage and
+ "roles/aiplatform.user" for using Vertex
+ extensions. If not specified, the Vertex AI
+ Reasoning Engine Service Agent in the project
+ will be used.
+
+ This field is a member of `oneof`_ ``_service_account``.
  package_spec (google.cloud.aiplatform_v1beta1.types.ReasoningEngineSpec.PackageSpec):
  Optional. User provided package spec of the ReasoningEngine.
  Ignored when users directly specify a deployment image
@@ -123,6 +136,11 @@ class DeploymentSpec(proto.Message):
  message=env_var.SecretEnvVar,
  )
 
+ service_account: str = proto.Field(
+ proto.STRING,
+ number=1,
+ optional=True,
+ )
  package_spec: PackageSpec = proto.Field(
  proto.MESSAGE,
  number=2,

samples/generated_samples/aiplatform_v1beta1_generated_dataset_service_assemble_data_async.py

@@ -39,11 +39,7 @@ async def sample_assemble_data():
  client = aiplatform_v1beta1.DatasetServiceAsyncClient()
 
  # Initialize request argument(s)
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssembleDataRequest(
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 

samples/generated_samples/aiplatform_v1beta1_generated_dataset_service_assemble_data_sync.py

@@ -39,11 +39,7 @@ def sample_assemble_data():
  client = aiplatform_v1beta1.DatasetServiceClient()
 
  # Initialize request argument(s)
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssembleDataRequest(
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 

samples/generated_samples/aiplatform_v1beta1_generated_dataset_service_assess_data_async.py

@@ -43,12 +43,8 @@ async def sample_assess_data():
  tuning_validation_assessment_config.model_name = "model_name_value"
  tuning_validation_assessment_config.dataset_usage = "SFT_VALIDATION"
 
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssessDataRequest(
  tuning_validation_assessment_config=tuning_validation_assessment_config,
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 

samples/generated_samples/aiplatform_v1beta1_generated_dataset_service_assess_data_sync.py

@@ -43,12 +43,8 @@ def sample_assess_data():
  tuning_validation_assessment_config.model_name = "model_name_value"
  tuning_validation_assessment_config.dataset_usage = "SFT_VALIDATION"
 
- gemini_template_config = aiplatform_v1beta1.GeminiTemplateConfig()
- gemini_template_config.gemini_example.contents.parts.text = "text_value"
-
  request = aiplatform_v1beta1.AssessDataRequest(
  tuning_validation_assessment_config=tuning_validation_assessment_config,
- gemini_template_config=gemini_template_config,
  name="name_value",
  )
 

samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json

@@ -8,7 +8,7 @@
  ],
  "language": "PYTHON",
  "name": "google-cloud-aiplatform",
- "version": "1.105.0"
+ "version": "0.1.0"
  },
  "snippets": [
  {