Skip to content

Commit b2bac31

Browse files
1 parent f72e523 commit b2bac31

File tree

2,863 files changed

+42
-2620212
lines changed

Some content is hidden

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

2,863 files changed

+42
-2620212
lines changed

google/cloud/aiplatform_v1/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -979,6 +979,7 @@
979979
from .types.vertex_rag_data import RagFileTransformationConfig
980980
from .types.vertex_rag_data import RagVectorDbConfig
981981
from .types.vertex_rag_data import UploadRagFileConfig
982+
from .types.vertex_rag_data import VertexAiSearchConfig
982983
from .types.vertex_rag_data_service import CreateRagCorpusOperationMetadata
983984
from .types.vertex_rag_data_service import CreateRagCorpusRequest
984985
from .types.vertex_rag_data_service import DeleteRagCorpusRequest
@@ -1993,6 +1994,7 @@
19931994
"UserActionReference",
19941995
"Value",
19951996
"VertexAISearch",
1997+
"VertexAiSearchConfig",
19961998
"VertexRagDataServiceClient",
19971999
"VertexRagServiceClient",
19982000
"VertexRagStore",

google/cloud/aiplatform_v1/types/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -1108,6 +1108,7 @@
11081108
RagFileTransformationConfig,
11091109
RagVectorDbConfig,
11101110
UploadRagFileConfig,
1111+
VertexAiSearchConfig,
11111112
)
11121113
from .vertex_rag_data_service import (
11131114
CreateRagCorpusOperationMetadata,
@@ -2030,6 +2031,7 @@
20302031
"RagFileTransformationConfig",
20312032
"RagVectorDbConfig",
20322033
"UploadRagFileConfig",
2034+
"VertexAiSearchConfig",
20332035
"CreateRagCorpusOperationMetadata",
20342036
"CreateRagCorpusRequest",
20352037
"DeleteRagCorpusRequest",

google/cloud/aiplatform_v1/types/vertex_rag_data.py

+34
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
"RagEmbeddingModelConfig",
3131
"RagVectorDbConfig",
3232
"FileStatus",
33+
"VertexAiSearchConfig",
3334
"CorpusStatus",
3435
"RagCorpus",
3536
"RagFile",
@@ -238,6 +239,24 @@ class State(proto.Enum):
238239
)
239240

240241

242+
class VertexAiSearchConfig(proto.Message):
243+
r"""Config for the Vertex AI Search.
244+
245+
Attributes:
246+
serving_config (str):
247+
Vertex AI Search Serving Config resource full name. For
248+
example,
249+
``projects/{project}/locations/{location}/collections/{collection}/engines/{engine}/servingConfigs/{serving_config}``
250+
or
251+
``projects/{project}/locations/{location}/collections/{collection}/dataStores/{data_store}/servingConfigs/{serving_config}``.
252+
"""
253+
254+
serving_config: str = proto.Field(
255+
proto.STRING,
256+
number=1,
257+
)
258+
259+
241260
class CorpusStatus(proto.Message):
242261
r"""RagCorpus status.
243262
@@ -284,6 +303,10 @@ class RagCorpus(proto.Message):
284303
r"""A RagCorpus is a RagFile container and a project can have
285304
multiple RagCorpora.
286305
306+
This message has `oneof`_ fields (mutually exclusive fields).
307+
For each oneof, at most one member field can be set at the same time.
308+
Setting any member of the oneof automatically clears all other
309+
members.
287310
288311
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
289312
@@ -309,6 +332,11 @@ class RagCorpus(proto.Message):
309332
Optional. Immutable. The config for the
310333
Vector DBs.
311334
335+
This field is a member of `oneof`_ ``backend_config``.
336+
vertex_ai_search_config (google.cloud.aiplatform_v1.types.VertexAiSearchConfig):
337+
Optional. Immutable. The config for the
338+
Vertex AI Search.
339+
312340
This field is a member of `oneof`_ ``backend_config``.
313341
"""
314342

@@ -345,6 +373,12 @@ class RagCorpus(proto.Message):
345373
oneof="backend_config",
346374
message="RagVectorDbConfig",
347375
)
376+
vertex_ai_search_config: "VertexAiSearchConfig" = proto.Field(
377+
proto.MESSAGE,
378+
number=10,
379+
oneof="backend_config",
380+
message="VertexAiSearchConfig",
381+
)
348382

349383

350384
class RagFile(proto.Message):

owl-bot-staging/v1/.coveragerc

-13
This file was deleted.

owl-bot-staging/v1/.flake8

-33
This file was deleted.

owl-bot-staging/v1/MANIFEST.in

-2
This file was deleted.

owl-bot-staging/v1/README.rst

-143
This file was deleted.

owl-bot-staging/v1/docs/_static/custom.css

-3
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/dataset_service.rst

-10
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/deployment_resource_pool_service.rst

-10
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/endpoint_service.rst

-10
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/evaluation_service.rst

-6
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/feature_online_store_admin_service.rst

-10
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/feature_online_store_service.rst

-6
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/feature_registry_service.rst

-10
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/featurestore_online_serving_service.rst

-6
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/featurestore_service.rst

-10
This file was deleted.

owl-bot-staging/v1/docs/aiplatform_v1/gen_ai_cache_service.rst

-10
This file was deleted.

0 commit comments

Comments
 (0)