# -*- coding: utf-8 -*- # Copyright 2025 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 __future__ import annotations from typing import MutableMapping, MutableSequence import proto # type: ignore from cloudsdk.google.protobuf import timestamp_pb2 # type: ignore from googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types import api_auth __protobuf__ = proto.module( package='google.cloud.aiplatform.v1', manifest={ 'AvroSource', 'CsvSource', 'GcsSource', 'GcsDestination', 'BigQuerySource', 'BigQueryDestination', 'CsvDestination', 'TFRecordDestination', 'ContainerRegistryDestination', 'GoogleDriveSource', 'DirectUploadSource', 'SlackSource', 'JiraSource', 'SharePointSources', }, ) class AvroSource(proto.Message): r"""The storage details for Avro input content. Attributes: gcs_source (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.GcsSource): Required. Google Cloud Storage location. """ gcs_source: 'GcsSource' = proto.Field( proto.MESSAGE, number=1, message='GcsSource', ) class CsvSource(proto.Message): r"""The storage details for CSV input content. Attributes: gcs_source (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.GcsSource): Required. Google Cloud Storage location. """ gcs_source: 'GcsSource' = proto.Field( proto.MESSAGE, number=1, message='GcsSource', ) class GcsSource(proto.Message): r"""The Google Cloud Storage location for the input content. Attributes: uris (MutableSequence[str]): Required. Google Cloud Storage URI(-s) to the input file(s). May contain wildcards. For more information on wildcards, see https://cloud.google.com/storage/docs/wildcards. """ uris: MutableSequence[str] = proto.RepeatedField( proto.STRING, number=1, ) class GcsDestination(proto.Message): r"""The Google Cloud Storage location where the output is to be written to. Attributes: output_uri_prefix (str): Required. Google Cloud Storage URI to output directory. If the uri doesn't end with '/', a '/' will be automatically appended. The directory is created if it doesn't exist. """ output_uri_prefix: str = proto.Field( proto.STRING, number=1, ) class BigQuerySource(proto.Message): r"""The BigQuery location for the input content. Attributes: input_uri (str): Required. BigQuery URI to a table, up to 2000 characters long. Accepted forms: - BigQuery path. For example: ``bq://projectId.bqDatasetId.bqTableId``. """ input_uri: str = proto.Field( proto.STRING, number=1, ) class BigQueryDestination(proto.Message): r"""The BigQuery location for the output content. Attributes: output_uri (str): Required. BigQuery URI to a project or table, up to 2000 characters long. When only the project is specified, the Dataset and Table is created. When the full table reference is specified, the Dataset must exist and table must not exist. Accepted forms: - BigQuery path. For example: ``bq://projectId`` or ``bq://projectId.bqDatasetId`` or ``bq://projectId.bqDatasetId.bqTableId``. """ output_uri: str = proto.Field( proto.STRING, number=1, ) class CsvDestination(proto.Message): r"""The storage details for CSV output content. Attributes: gcs_destination (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.GcsDestination): Required. Google Cloud Storage location. """ gcs_destination: 'GcsDestination' = proto.Field( proto.MESSAGE, number=1, message='GcsDestination', ) class TFRecordDestination(proto.Message): r"""The storage details for TFRecord output content. Attributes: gcs_destination (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.GcsDestination): Required. Google Cloud Storage location. """ gcs_destination: 'GcsDestination' = proto.Field( proto.MESSAGE, number=1, message='GcsDestination', ) class ContainerRegistryDestination(proto.Message): r"""The Container Registry location for the container image. Attributes: output_uri (str): Required. Container Registry URI of a container image. Only Google Container Registry and Artifact Registry are supported now. Accepted forms: - Google Container Registry path. For example: ``gcr.io/projectId/imageName:tag``. - Artifact Registry path. For example: ``us-central1-docker.pkg.dev/projectId/repoName/imageName:tag``. If a tag is not specified, "latest" will be used as the default tag. """ output_uri: str = proto.Field( proto.STRING, number=1, ) class GoogleDriveSource(proto.Message): r"""The Google Drive location for the input content. Attributes: resource_ids (MutableSequence[googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.GoogleDriveSource.ResourceId]): Required. Google Drive resource IDs. """ class ResourceId(proto.Message): r"""The type and ID of the Google Drive resource. Attributes: resource_type (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.GoogleDriveSource.ResourceId.ResourceType): Required. The type of the Google Drive resource. resource_id (str): Required. The ID of the Google Drive resource. """ class ResourceType(proto.Enum): r"""The type of the Google Drive resource. Values: RESOURCE_TYPE_UNSPECIFIED (0): Unspecified resource type. RESOURCE_TYPE_FILE (1): File resource type. RESOURCE_TYPE_FOLDER (2): Folder resource type. """ RESOURCE_TYPE_UNSPECIFIED = 0 RESOURCE_TYPE_FILE = 1 RESOURCE_TYPE_FOLDER = 2 resource_type: 'GoogleDriveSource.ResourceId.ResourceType' = proto.Field( proto.ENUM, number=1, enum='GoogleDriveSource.ResourceId.ResourceType', ) resource_id: str = proto.Field( proto.STRING, number=2, ) resource_ids: MutableSequence[ResourceId] = proto.RepeatedField( proto.MESSAGE, number=1, message=ResourceId, ) class DirectUploadSource(proto.Message): r"""The input content is encapsulated and uploaded in the request. """ class SlackSource(proto.Message): r"""The Slack source for the ImportRagFilesRequest. Attributes: channels (MutableSequence[googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.SlackSource.SlackChannels]): Required. The Slack channels. """ class SlackChannels(proto.Message): r"""SlackChannels contains the Slack channels and corresponding access token. Attributes: channels (MutableSequence[googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.SlackSource.SlackChannels.SlackChannel]): Required. The Slack channel IDs. api_key_config (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.ApiAuth.ApiKeyConfig): Required. The SecretManager secret version resource name (e.g. projects/{project}/secrets/{secret}/versions/{version}) storing the Slack channel access token that has access to the slack channel IDs. See: https://api.slack.com/tutorials/tracks/getting-a-token. """ class SlackChannel(proto.Message): r"""SlackChannel contains the Slack channel ID and the time range to import. Attributes: channel_id (str): Required. The Slack channel ID. start_time (google.protobuf.timestamp_pb2.Timestamp): Optional. The starting timestamp for messages to import. end_time (google.protobuf.timestamp_pb2.Timestamp): Optional. The ending timestamp for messages to import. """ channel_id: str = proto.Field( proto.STRING, number=1, ) start_time: timestamp_pb2.Timestamp = proto.Field( proto.MESSAGE, number=2, message=timestamp_pb2.Timestamp, ) end_time: timestamp_pb2.Timestamp = proto.Field( proto.MESSAGE, number=3, message=timestamp_pb2.Timestamp, ) channels: MutableSequence['SlackSource.SlackChannels.SlackChannel'] = proto.RepeatedField( proto.MESSAGE, number=1, message='SlackSource.SlackChannels.SlackChannel', ) api_key_config: api_auth.ApiAuth.ApiKeyConfig = proto.Field( proto.MESSAGE, number=3, message=api_auth.ApiAuth.ApiKeyConfig, ) channels: MutableSequence[SlackChannels] = proto.RepeatedField( proto.MESSAGE, number=1, message=SlackChannels, ) class JiraSource(proto.Message): r"""The Jira source for the ImportRagFilesRequest. Attributes: jira_queries (MutableSequence[googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.JiraSource.JiraQueries]): Required. The Jira queries. """ class JiraQueries(proto.Message): r"""JiraQueries contains the Jira queries and corresponding authentication. Attributes: projects (MutableSequence[str]): A list of Jira projects to import in their entirety. custom_queries (MutableSequence[str]): A list of custom Jira queries to import. For information about JQL (Jira Query Language), see https://support.atlassian.com/jira-service-management-cloud/docs/use-advanced-search-with-jira-query-language-jql/ email (str): Required. The Jira email address. server_uri (str): Required. The Jira server URI. api_key_config (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.ApiAuth.ApiKeyConfig): Required. The SecretManager secret version resource name (e.g. projects/{project}/secrets/{secret}/versions/{version}) storing the Jira API key. See `Manage API tokens for your Atlassian account `__. """ projects: MutableSequence[str] = proto.RepeatedField( proto.STRING, number=3, ) custom_queries: MutableSequence[str] = proto.RepeatedField( proto.STRING, number=4, ) email: str = proto.Field( proto.STRING, number=5, ) server_uri: str = proto.Field( proto.STRING, number=6, ) api_key_config: api_auth.ApiAuth.ApiKeyConfig = proto.Field( proto.MESSAGE, number=7, message=api_auth.ApiAuth.ApiKeyConfig, ) jira_queries: MutableSequence[JiraQueries] = proto.RepeatedField( proto.MESSAGE, number=1, message=JiraQueries, ) class SharePointSources(proto.Message): r"""The SharePointSources to pass to ImportRagFiles. Attributes: share_point_sources (MutableSequence[googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.SharePointSources.SharePointSource]): The SharePoint sources. """ class SharePointSource(proto.Message): r"""An individual SharePointSource. 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: sharepoint_folder_path (str): The path of the SharePoint folder to download from. This field is a member of `oneof`_ ``folder_source``. sharepoint_folder_id (str): The ID of the SharePoint folder to download from. This field is a member of `oneof`_ ``folder_source``. drive_name (str): The name of the drive to download from. This field is a member of `oneof`_ ``drive_source``. drive_id (str): The ID of the drive to download from. This field is a member of `oneof`_ ``drive_source``. client_id (str): The Application ID for the app registered in Microsoft Azure Portal. The application must also be configured with MS Graph permissions "Files.ReadAll", "Sites.ReadAll" and BrowserSiteLists.Read.All. client_secret (googlecloudsdk.generated_clients.gapic_clients.aiplatform_v1.types.ApiAuth.ApiKeyConfig): The application secret for the app registered in Azure. tenant_id (str): Unique identifier of the Azure Active Directory Instance. sharepoint_site_name (str): The name of the SharePoint site to download from. This can be the site name or the site id. file_id (str): Output only. The SharePoint file id. Output only. """ sharepoint_folder_path: str = proto.Field( proto.STRING, number=5, oneof='folder_source', ) sharepoint_folder_id: str = proto.Field( proto.STRING, number=6, oneof='folder_source', ) drive_name: str = proto.Field( proto.STRING, number=7, oneof='drive_source', ) drive_id: str = proto.Field( proto.STRING, number=8, oneof='drive_source', ) client_id: str = proto.Field( proto.STRING, number=1, ) client_secret: api_auth.ApiAuth.ApiKeyConfig = proto.Field( proto.MESSAGE, number=2, message=api_auth.ApiAuth.ApiKeyConfig, ) tenant_id: str = proto.Field( proto.STRING, number=3, ) sharepoint_site_name: str = proto.Field( proto.STRING, number=4, ) file_id: str = proto.Field( proto.STRING, number=9, ) share_point_sources: MutableSequence[SharePointSource] = proto.RepeatedField( proto.MESSAGE, number=1, message=SharePointSource, ) __all__ = tuple(sorted(__protobuf__.manifest))