Skip to content

Commit a56e4dd

Browse files
gcf-owl-bot[bot]copybara-github
gcf-owl-bot[bot]
authored and
copybara-github
committed
Copybara import of the project:
-- dcb9e67 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add Vertex RAG service proto to v1 PiperOrigin-RevId: 698990917 Source-Link: googleapis/googleapis@6f7652b Source-Link: googleapis/googleapis-gen@b15d3ee Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYjE1ZDNlZThlNWRhN2UwNDBmNGMxNjA1NGFiZjNmOWFiMjMzMmY5ZiJ9 -- 70a666a 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=#4699 from googleapis:owl-bot-copy 15a7986 PiperOrigin-RevId: 699227215
1 parent 4ce1d96 commit a56e4dd

File tree

140 files changed

+61954
-48
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

140 files changed

+61954
-48
lines changed

‎docs/aiplatform_v1/services_.rst

+2
Original file line numberDiff line numberDiff line change
@@ -29,4 +29,6 @@ Services for Google Cloud Aiplatform v1 API
2929
schedule_service
3030
specialist_pool_service
3131
tensorboard_service
32+
vertex_rag_data_service
33+
vertex_rag_service
3234
vizier_service

‎docs/aiplatform_v1/vertex_rag_data_service.rst

+10
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
VertexRagDataService
2+
--------------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1.services.vertex_rag_data_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.aiplatform_v1.services.vertex_rag_data_service.pagers
9+
:members:
10+
:inherited-members:

‎docs/aiplatform_v1/vertex_rag_service.rst

+6
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
VertexRagService
2+
----------------------------------
3+
4+
.. automodule:: google.cloud.aiplatform_v1.services.vertex_rag_service
5+
:members:
6+
:inherited-members:

‎google/cloud/aiplatform_v1/__init__.py

+98
Original file line numberDiff line numberDiff line change
@@ -82,12 +82,17 @@
8282
from .services.specialist_pool_service import SpecialistPoolServiceAsyncClient
8383
from .services.tensorboard_service import TensorboardServiceClient
8484
from .services.tensorboard_service import TensorboardServiceAsyncClient
85+
from .services.vertex_rag_data_service import VertexRagDataServiceClient
86+
from .services.vertex_rag_data_service import VertexRagDataServiceAsyncClient
87+
from .services.vertex_rag_service import VertexRagServiceClient
88+
from .services.vertex_rag_service import VertexRagServiceAsyncClient
8589
from .services.vizier_service import VizierServiceClient
8690
from .services.vizier_service import VizierServiceAsyncClient
8791

8892
from .types.accelerator_type import AcceleratorType
8993
from .types.annotation import Annotation
9094
from .types.annotation_spec import AnnotationSpec
95+
from .types.api_auth import ApiAuth
9196
from .types.artifact import Artifact
9297
from .types.batch_prediction_job import BatchPredictionJob
9398
from .types.completion_stats import CompletionStats
@@ -494,8 +499,13 @@
494499
from .types.io import ContainerRegistryDestination
495500
from .types.io import CsvDestination
496501
from .types.io import CsvSource
502+
from .types.io import DirectUploadSource
497503
from .types.io import GcsDestination
498504
from .types.io import GcsSource
505+
from .types.io import GoogleDriveSource
506+
from .types.io import JiraSource
507+
from .types.io import SharePointSources
508+
from .types.io import SlackSource
499509
from .types.io import TFRecordDestination
500510
from .types.job_service import CancelBatchPredictionJobRequest
501511
from .types.job_service import CancelCustomJobRequest
@@ -885,10 +895,12 @@
885895
from .types.tool import FunctionDeclaration
886896
from .types.tool import FunctionResponse
887897
from .types.tool import GoogleSearchRetrieval
898+
from .types.tool import RagRetrievalConfig
888899
from .types.tool import Retrieval
889900
from .types.tool import Tool
890901
from .types.tool import ToolConfig
891902
from .types.tool import VertexAISearch
903+
from .types.tool import VertexRagStore
892904
from .types.training_pipeline import FilterSplit
893905
from .types.training_pipeline import FractionSplit
894906
from .types.training_pipeline import InputDataConfig
@@ -912,6 +924,43 @@
912924
from .types.unmanaged_container_model import UnmanagedContainerModel
913925
from .types.user_action_reference import UserActionReference
914926
from .types.value import Value
927+
from .types.vertex_rag_data import CorpusStatus
928+
from .types.vertex_rag_data import FileStatus
929+
from .types.vertex_rag_data import ImportRagFilesConfig
930+
from .types.vertex_rag_data import RagCorpus
931+
from .types.vertex_rag_data import RagEmbeddingModelConfig
932+
from .types.vertex_rag_data import RagFile
933+
from .types.vertex_rag_data import RagFileChunkingConfig
934+
from .types.vertex_rag_data import RagFileTransformationConfig
935+
from .types.vertex_rag_data import RagVectorDbConfig
936+
from .types.vertex_rag_data import UploadRagFileConfig
937+
from .types.vertex_rag_data_service import CreateRagCorpusOperationMetadata
938+
from .types.vertex_rag_data_service import CreateRagCorpusRequest
939+
from .types.vertex_rag_data_service import DeleteRagCorpusRequest
940+
from .types.vertex_rag_data_service import DeleteRagFileRequest
941+
from .types.vertex_rag_data_service import GetRagCorpusRequest
942+
from .types.vertex_rag_data_service import GetRagFileRequest
943+
from .types.vertex_rag_data_service import ImportRagFilesOperationMetadata
944+
from .types.vertex_rag_data_service import ImportRagFilesRequest
945+
from .types.vertex_rag_data_service import ImportRagFilesResponse
946+
from .types.vertex_rag_data_service import ListRagCorporaRequest
947+
from .types.vertex_rag_data_service import ListRagCorporaResponse
948+
from .types.vertex_rag_data_service import ListRagFilesRequest
949+
from .types.vertex_rag_data_service import ListRagFilesResponse
950+
from .types.vertex_rag_data_service import UpdateRagCorpusOperationMetadata
951+
from .types.vertex_rag_data_service import UpdateRagCorpusRequest
952+
from .types.vertex_rag_data_service import UploadRagFileRequest
953+
from .types.vertex_rag_data_service import UploadRagFileResponse
954+
from .types.vertex_rag_service import AugmentPromptRequest
955+
from .types.vertex_rag_service import AugmentPromptResponse
956+
from .types.vertex_rag_service import Claim
957+
from .types.vertex_rag_service import CorroborateContentRequest
958+
from .types.vertex_rag_service import CorroborateContentResponse
959+
from .types.vertex_rag_service import Fact
960+
from .types.vertex_rag_service import RagContexts
961+
from .types.vertex_rag_service import RagQuery
962+
from .types.vertex_rag_service import RetrieveContextsRequest
963+
from .types.vertex_rag_service import RetrieveContextsResponse
915964
from .types.vizier_service import AddTrialMeasurementRequest
916965
from .types.vizier_service import CheckTrialEarlyStoppingStateMetatdata
917966
from .types.vizier_service import CheckTrialEarlyStoppingStateRequest
@@ -962,6 +1011,8 @@
9621011
"ScheduleServiceAsyncClient",
9631012
"SpecialistPoolServiceAsyncClient",
9641013
"TensorboardServiceAsyncClient",
1014+
"VertexRagDataServiceAsyncClient",
1015+
"VertexRagServiceAsyncClient",
9651016
"VizierServiceAsyncClient",
9661017
"AcceleratorType",
9671018
"ActiveLearningConfig",
@@ -974,10 +1025,13 @@
9741025
"AddTrialMeasurementRequest",
9751026
"Annotation",
9761027
"AnnotationSpec",
1028+
"ApiAuth",
9771029
"Artifact",
9781030
"AssignNotebookRuntimeOperationMetadata",
9791031
"AssignNotebookRuntimeRequest",
9801032
"Attribution",
1033+
"AugmentPromptRequest",
1034+
"AugmentPromptResponse",
9811035
"AutomaticResources",
9821036
"AutoscalingMetricSpec",
9831037
"AvroSource",
@@ -1031,6 +1085,7 @@
10311085
"CheckTrialEarlyStoppingStateResponse",
10321086
"Citation",
10331087
"CitationMetadata",
1088+
"Claim",
10341089
"ClientConnectionConfig",
10351090
"CoherenceInput",
10361091
"CoherenceInstance",
@@ -1051,6 +1106,9 @@
10511106
"CopyModelOperationMetadata",
10521107
"CopyModelRequest",
10531108
"CopyModelResponse",
1109+
"CorpusStatus",
1110+
"CorroborateContentRequest",
1111+
"CorroborateContentResponse",
10541112
"CountTokensRequest",
10551113
"CountTokensResponse",
10561114
"CreateArtifactRequest",
@@ -1096,6 +1154,8 @@
10961154
"CreatePersistentResourceOperationMetadata",
10971155
"CreatePersistentResourceRequest",
10981156
"CreatePipelineJobRequest",
1157+
"CreateRagCorpusOperationMetadata",
1158+
"CreateRagCorpusRequest",
10991159
"CreateRegistryFeatureOperationMetadata",
11001160
"CreateScheduleRequest",
11011161
"CreateSpecialistPoolOperationMetadata",
@@ -1154,6 +1214,8 @@
11541214
"DeleteOperationMetadata",
11551215
"DeletePersistentResourceRequest",
11561216
"DeletePipelineJobRequest",
1217+
"DeleteRagCorpusRequest",
1218+
"DeleteRagFileRequest",
11571219
"DeleteSavedQueryRequest",
11581220
"DeleteScheduleRequest",
11591221
"DeleteSpecialistPoolRequest",
@@ -1182,6 +1244,7 @@
11821244
"DirectPredictResponse",
11831245
"DirectRawPredictRequest",
11841246
"DirectRawPredictResponse",
1247+
"DirectUploadSource",
11851248
"DiskSpec",
11861249
"DoubleArray",
11871250
"DynamicRetrievalConfig",
@@ -1229,6 +1292,7 @@
12291292
"ExportModelResponse",
12301293
"ExportTensorboardTimeSeriesDataRequest",
12311294
"ExportTensorboardTimeSeriesDataResponse",
1295+
"Fact",
12321296
"FasterDeploymentConfig",
12331297
"Feature",
12341298
"FeatureGroup",
@@ -1253,6 +1317,7 @@
12531317
"FetchFeatureValuesRequest",
12541318
"FetchFeatureValuesResponse",
12551319
"FileData",
1320+
"FileStatus",
12561321
"FilterSplit",
12571322
"FindNeighborsRequest",
12581323
"FindNeighborsResponse",
@@ -1312,6 +1377,8 @@
13121377
"GetPersistentResourceRequest",
13131378
"GetPipelineJobRequest",
13141379
"GetPublisherModelRequest",
1380+
"GetRagCorpusRequest",
1381+
"GetRagFileRequest",
13151382
"GetScheduleRequest",
13161383
"GetSpecialistPoolRequest",
13171384
"GetStudyRequest",
@@ -1322,6 +1389,7 @@
13221389
"GetTrainingPipelineRequest",
13231390
"GetTrialRequest",
13241391
"GetTuningJobRequest",
1392+
"GoogleDriveSource",
13251393
"GoogleSearchRetrieval",
13261394
"GroundednessInput",
13271395
"GroundednessInstance",
@@ -1341,6 +1409,10 @@
13411409
"ImportFeatureValuesRequest",
13421410
"ImportFeatureValuesResponse",
13431411
"ImportModelEvaluationRequest",
1412+
"ImportRagFilesConfig",
1413+
"ImportRagFilesOperationMetadata",
1414+
"ImportRagFilesRequest",
1415+
"ImportRagFilesResponse",
13441416
"Index",
13451417
"IndexDatapoint",
13461418
"IndexEndpoint",
@@ -1351,6 +1423,7 @@
13511423
"InputDataConfig",
13521424
"Int64Array",
13531425
"IntegratedGradientsAttribution",
1426+
"JiraSource",
13541427
"JobServiceClient",
13551428
"JobState",
13561429
"LargeModelReference",
@@ -1429,6 +1502,10 @@
14291502
"ListPersistentResourcesResponse",
14301503
"ListPipelineJobsRequest",
14311504
"ListPipelineJobsResponse",
1505+
"ListRagCorporaRequest",
1506+
"ListRagCorporaResponse",
1507+
"ListRagFilesRequest",
1508+
"ListRagFilesResponse",
14321509
"ListSavedQueriesRequest",
14331510
"ListSavedQueriesResponse",
14341511
"ListSchedulesRequest",
@@ -1590,6 +1667,15 @@
15901667
"QuestionAnsweringRelevanceInstance",
15911668
"QuestionAnsweringRelevanceResult",
15921669
"QuestionAnsweringRelevanceSpec",
1670+
"RagContexts",
1671+
"RagCorpus",
1672+
"RagEmbeddingModelConfig",
1673+
"RagFile",
1674+
"RagFileChunkingConfig",
1675+
"RagFileTransformationConfig",
1676+
"RagQuery",
1677+
"RagRetrievalConfig",
1678+
"RagVectorDbConfig",
15931679
"RawPredictRequest",
15941680
"RayLogsSpec",
15951681
"RayMetricSpec",
@@ -1625,6 +1711,8 @@
16251711
"ResumeScheduleRequest",
16261712
"Retrieval",
16271713
"RetrievalMetadata",
1714+
"RetrieveContextsRequest",
1715+
"RetrieveContextsResponse",
16281716
"RougeInput",
16291717
"RougeInstance",
16301718
"RougeMetricValue",
@@ -1658,7 +1746,9 @@
16581746
"SearchNearestEntitiesResponse",
16591747
"Segment",
16601748
"ServiceAccountSpec",
1749+
"SharePointSources",
16611750
"ShieldedVmConfig",
1751+
"SlackSource",
16621752
"SmoothGradConfig",
16631753
"SpecialistPool",
16641754
"SpecialistPoolServiceClient",
@@ -1793,6 +1883,8 @@
17931883
"UpdateNotebookRuntimeTemplateRequest",
17941884
"UpdatePersistentResourceOperationMetadata",
17951885
"UpdatePersistentResourceRequest",
1886+
"UpdateRagCorpusOperationMetadata",
1887+
"UpdateRagCorpusRequest",
17961888
"UpdateScheduleRequest",
17971889
"UpdateSpecialistPoolOperationMetadata",
17981890
"UpdateSpecialistPoolRequest",
@@ -1807,11 +1899,17 @@
18071899
"UploadModelOperationMetadata",
18081900
"UploadModelRequest",
18091901
"UploadModelResponse",
1902+
"UploadRagFileConfig",
1903+
"UploadRagFileRequest",
1904+
"UploadRagFileResponse",
18101905
"UpsertDatapointsRequest",
18111906
"UpsertDatapointsResponse",
18121907
"UserActionReference",
18131908
"Value",
18141909
"VertexAISearch",
1910+
"VertexRagDataServiceClient",
1911+
"VertexRagServiceClient",
1912+
"VertexRagStore",
18151913
"VideoMetadata",
18161914
"VizierServiceClient",
18171915
"WorkerPoolSpec",

0 commit comments

Comments
 (0)