Skip to content

Commit 64386e9

Browse files
gcf-owl-bot[bot]copybara-github
gcf-owl-bot[bot]
authored and
copybara-github
committed
Copybara import of the project:
-- 913f9d6 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add UpdateRagEngineConfig rpc feat: add RagEngineConfig to specify RAG project-level config feat: add rag_managed_db_config to RagEngineConfig for specifying Basic or Enterprise RagManagedDb tiers PiperOrigin-RevId: 746505624 Source-Link: googleapis/googleapis@8c1862f Source-Link: googleapis/googleapis-gen@3b98fff Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiM2I5OGZmZmUzYzFlZWJlNzhmYTA3OTNkZThhOTNhN2M2NjZmYjBiNCJ9 -- 4951ad3 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=#5165 from googleapis:owl-bot-copy 4951ad3 PiperOrigin-RevId: 746601899
1 parent a6a1ec5 commit 64386e9

File tree

123 files changed

+10850
-1671
lines changed

Some content is hidden

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

123 files changed

+10850
-1671
lines changed

‎google/cloud/aiplatform_v1beta1/__init__.py‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -513,6 +513,8 @@
513513
from .types.feature_online_store_admin_service import UpdateFeatureViewOperationMetadata
514514
from .types.feature_online_store_admin_service import UpdateFeatureViewRequest
515515
from .types.feature_online_store_service import FeatureViewDataKey
516+
from .types.feature_online_store_service import FeatureViewDirectWriteRequest
517+
from .types.feature_online_store_service import FeatureViewDirectWriteResponse
516518
from .types.feature_online_store_service import FetchFeatureValuesRequest
517519
from .types.feature_online_store_service import FetchFeatureValuesResponse
518520
from .types.feature_online_store_service import NearestNeighborQuery
@@ -1186,10 +1188,12 @@
11861188
from .types.vertex_rag_data import RagChunk
11871189
from .types.vertex_rag_data import RagCorpus
11881190
from .types.vertex_rag_data import RagEmbeddingModelConfig
1191+
from .types.vertex_rag_data import RagEngineConfig
11891192
from .types.vertex_rag_data import RagFile
11901193
from .types.vertex_rag_data import RagFileChunkingConfig
11911194
from .types.vertex_rag_data import RagFileParsingConfig
11921195
from .types.vertex_rag_data import RagFileTransformationConfig
1196+
from .types.vertex_rag_data import RagManagedDbConfig
11931197
from .types.vertex_rag_data import RagVectorDbConfig
11941198
from .types.vertex_rag_data import UploadRagFileConfig
11951199
from .types.vertex_rag_data import VertexAiSearchConfig
@@ -1198,6 +1202,7 @@
11981202
from .types.vertex_rag_data_service import DeleteRagCorpusRequest
11991203
from .types.vertex_rag_data_service import DeleteRagFileRequest
12001204
from .types.vertex_rag_data_service import GetRagCorpusRequest
1205+
from .types.vertex_rag_data_service import GetRagEngineConfigRequest
12011206
from .types.vertex_rag_data_service import GetRagFileRequest
12021207
from .types.vertex_rag_data_service import ImportRagFilesOperationMetadata
12031208
from .types.vertex_rag_data_service import ImportRagFilesRequest
@@ -1208,6 +1213,8 @@
12081213
from .types.vertex_rag_data_service import ListRagFilesResponse
12091214
from .types.vertex_rag_data_service import UpdateRagCorpusOperationMetadata
12101215
from .types.vertex_rag_data_service import UpdateRagCorpusRequest
1216+
from .types.vertex_rag_data_service import UpdateRagEngineConfigOperationMetadata
1217+
from .types.vertex_rag_data_service import UpdateRagEngineConfigRequest
12111218
from .types.vertex_rag_data_service import UploadRagFileRequest
12121219
from .types.vertex_rag_data_service import UploadRagFileResponse
12131220
from .types.vertex_rag_service import AugmentPromptRequest
@@ -1656,6 +1663,8 @@
16561663
"FeatureView",
16571664
"FeatureViewDataFormat",
16581665
"FeatureViewDataKey",
1666+
"FeatureViewDirectWriteRequest",
1667+
"FeatureViewDirectWriteResponse",
16591668
"FeatureViewSync",
16601669
"Featurestore",
16611670
"FeaturestoreMonitoringConfig",
@@ -1739,6 +1748,7 @@
17391748
"GetPipelineJobRequest",
17401749
"GetPublisherModelRequest",
17411750
"GetRagCorpusRequest",
1751+
"GetRagEngineConfigRequest",
17421752
"GetRagFileRequest",
17431753
"GetReasoningEngineRequest",
17441754
"GetScheduleRequest",
@@ -2097,10 +2107,12 @@
20972107
"RagContexts",
20982108
"RagCorpus",
20992109
"RagEmbeddingModelConfig",
2110+
"RagEngineConfig",
21002111
"RagFile",
21012112
"RagFileChunkingConfig",
21022113
"RagFileParsingConfig",
21032114
"RagFileTransformationConfig",
2115+
"RagManagedDbConfig",
21042116
"RagQuery",
21052117
"RagRetrievalConfig",
21062118
"RagVectorDbConfig",
@@ -2392,6 +2404,8 @@
23922404
"UpdatePersistentResourceRequest",
23932405
"UpdateRagCorpusOperationMetadata",
23942406
"UpdateRagCorpusRequest",
2407+
"UpdateRagEngineConfigOperationMetadata",
2408+
"UpdateRagEngineConfigRequest",
23952409
"UpdateReasoningEngineOperationMetadata",
23962410
"UpdateReasoningEngineRequest",
23972411
"UpdateScheduleRequest",

‎google/cloud/aiplatform_v1beta1/gapic_metadata.json‎

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1197,6 +1197,11 @@
11971197
"grpc": {
11981198
"libraryClient": "FeatureOnlineStoreServiceClient",
11991199
"rpcs": {
1200+
"FeatureViewDirectWrite": {
1201+
"methods": [
1202+
"feature_view_direct_write"
1203+
]
1204+
},
12001205
"FetchFeatureValues": {
12011206
"methods": [
12021207
"fetch_feature_values"
@@ -1217,6 +1222,11 @@
12171222
"grpc-async": {
12181223
"libraryClient": "FeatureOnlineStoreServiceAsyncClient",
12191224
"rpcs": {
1225+
"FeatureViewDirectWrite": {
1226+
"methods": [
1227+
"feature_view_direct_write"
1228+
]
1229+
},
12201230
"FetchFeatureValues": {
12211231
"methods": [
12221232
"fetch_feature_values"
@@ -1237,6 +1247,11 @@
12371247
"rest": {
12381248
"libraryClient": "FeatureOnlineStoreServiceClient",
12391249
"rpcs": {
1250+
"FeatureViewDirectWrite": {
1251+
"methods": [
1252+
"feature_view_direct_write"
1253+
]
1254+
},
12401255
"FetchFeatureValues": {
12411256
"methods": [
12421257
"fetch_feature_values"
@@ -5956,6 +5971,11 @@
59565971
"get_rag_corpus"
59575972
]
59585973
},
5974+
"GetRagEngineConfig": {
5975+
"methods": [
5976+
"get_rag_engine_config"
5977+
]
5978+
},
59595979
"GetRagFile": {
59605980
"methods": [
59615981
"get_rag_file"
@@ -5981,6 +6001,11 @@
59816001
"update_rag_corpus"
59826002
]
59836003
},
6004+
"UpdateRagEngineConfig": {
6005+
"methods": [
6006+
"update_rag_engine_config"
6007+
]
6008+
},
59846009
"UploadRagFile": {
59856010
"methods": [
59866011
"upload_rag_file"
@@ -6011,6 +6036,11 @@
60116036
"get_rag_corpus"
60126037
]
60136038
},
6039+
"GetRagEngineConfig": {
6040+
"methods": [
6041+
"get_rag_engine_config"
6042+
]
6043+
},
60146044
"GetRagFile": {
60156045
"methods": [
60166046
"get_rag_file"
@@ -6036,6 +6066,11 @@
60366066
"update_rag_corpus"
60376067
]
60386068
},
6069+
"UpdateRagEngineConfig": {
6070+
"methods": [
6071+
"update_rag_engine_config"
6072+
]
6073+
},
60396074
"UploadRagFile": {
60406075
"methods": [
60416076
"upload_rag_file"
@@ -6066,6 +6101,11 @@
60666101
"get_rag_corpus"
60676102
]
60686103
},
6104+
"GetRagEngineConfig": {
6105+
"methods": [
6106+
"get_rag_engine_config"
6107+
]
6108+
},
60696109
"GetRagFile": {
60706110
"methods": [
60716111
"get_rag_file"
@@ -6091,6 +6131,11 @@
60916131
"update_rag_corpus"
60926132
]
60936133
},
6134+
"UpdateRagEngineConfig": {
6135+
"methods": [
6136+
"update_rag_engine_config"
6137+
]
6138+
},
60946139
"UploadRagFile": {
60956140
"methods": [
60966141
"upload_rag_file"

‎google/cloud/aiplatform_v1beta1/services/dataset_service/transports/rest.py‎

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1788,6 +1788,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
17881788
"method": "post",
17891789
"uri": "/ui/{name=projects/*/locations/*/schedules/*/operations/*}:cancel",
17901790
},
1791+
{
1792+
"method": "post",
1793+
"uri": "/ui/{name=projects/*/locations/*/ragEngineConfig/operations/*}:cancel",
1794+
},
17911795
{
17921796
"method": "post",
17931797
"uri": "/ui/{name=projects/*/locations/*/specialistPools/*/operations/*}:cancel",
@@ -1948,6 +1952,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
19481952
"method": "post",
19491953
"uri": "/v1beta1/{name=projects/*/locations/*/persistentResources/*/operations/*}:cancel",
19501954
},
1955+
{
1956+
"method": "post",
1957+
"uri": "/v1beta1/{name=projects/*/locations/*/ragEngineConfig/operations/*}:cancel",
1958+
},
19511959
{
19521960
"method": "post",
19531961
"uri": "/v1beta1/{name=projects/*/locations/*/ragCorpora/*/operations/*}:cancel",
@@ -2162,6 +2170,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
21622170
"method": "delete",
21632171
"uri": "/ui/{name=projects/*/locations/*/specialistPools/*/operations/*}",
21642172
},
2173+
{
2174+
"method": "delete",
2175+
"uri": "/ui/{name=projects/*/locations/*/ragEngineConfig/operations/*}",
2176+
},
21652177
{
21662178
"method": "delete",
21672179
"uri": "/ui/{name=projects/*/locations/*/tensorboards/*/operations/*}",
@@ -2342,6 +2354,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
23422354
"method": "delete",
23432355
"uri": "/v1beta1/{name=projects/*/locations/*/persistentResources/*/operations/*}",
23442356
},
2357+
{
2358+
"method": "delete",
2359+
"uri": "/v1beta1/{name=projects/*/locations/*/ragEngineConfig/operations/*}",
2360+
},
23452361
{
23462362
"method": "delete",
23472363
"uri": "/v1beta1/{name=projects/*/locations/*/ragCorpora/*/operations/*}",
@@ -2580,6 +2596,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
25802596
"method": "get",
25812597
"uri": "/ui/{name=projects/*/locations/*/pipelineJobs/*/operations/*}",
25822598
},
2599+
{
2600+
"method": "get",
2601+
"uri": "/ui/{name=projects/*/locations/*/ragEngineConfig/operations/*}",
2602+
},
25832603
{
25842604
"method": "get",
25852605
"uri": "/ui/{name=projects/*/locations/*/schedules/*/operations/*}",
@@ -2768,6 +2788,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
27682788
"method": "get",
27692789
"uri": "/v1beta1/{name=projects/*/locations/*/persistentResources/*/operations/*}",
27702790
},
2791+
{
2792+
"method": "get",
2793+
"uri": "/v1beta1/{name=projects/*/locations/*/ragEngineConfig/operations/*}",
2794+
},
27712795
{
27722796
"method": "get",
27732797
"uri": "/v1beta1/{name=projects/*/locations/*/ragCorpora/*/operations/*}",
@@ -3002,6 +3026,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
30023026
"method": "get",
30033027
"uri": "/ui/{name=projects/*/locations/*/pipelineJobs/*}/operations",
30043028
},
3029+
{
3030+
"method": "get",
3031+
"uri": "/ui/{name=projects/*/locations/*/ragEngineConfig}/operations",
3032+
},
30053033
{
30063034
"method": "get",
30073035
"uri": "/ui/{name=projects/*/locations/*/schedules/*}/operations",
@@ -3190,6 +3218,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
31903218
"method": "get",
31913219
"uri": "/v1beta1/{name=projects/*/locations/*/persistentResources/*}/operations",
31923220
},
3221+
{
3222+
"method": "get",
3223+
"uri": "/v1beta1/{name=projects/*/locations/*/ragEngineConfig}/operations",
3224+
},
31933225
{
31943226
"method": "get",
31953227
"uri": "/v1beta1/{name=projects/*/locations/*/ragCorpora/*}/operations",
@@ -3432,6 +3464,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
34323464
"method": "post",
34333465
"uri": "/ui/{name=projects/*/locations/*/specialistPools/*/operations/*}:wait",
34343466
},
3467+
{
3468+
"method": "post",
3469+
"uri": "/ui/{name=projects/*/locations/*/ragEngineConfig/operations/*}:wait",
3470+
},
34353471
{
34363472
"method": "post",
34373473
"uri": "/ui/{name=projects/*/locations/*/tensorboards/*/operations/*}:wait",
@@ -3612,6 +3648,10 @@ def operations_client(self) -> operations_v1.AbstractOperationsClient:
36123648
"method": "post",
36133649
"uri": "/v1beta1/{name=projects/*/locations/*/persistentResources/*/operations/*}:wait",
36143650
},
3651+
{
3652+
"method": "post",
3653+
"uri": "/v1beta1/{name=projects/*/locations/*/ragEngineConfig/operations/*}:wait",
3654+
},
36153655
{
36163656
"method": "post",
36173657
"uri": "/v1beta1/{name=projects/*/locations/*/ragCorpora/*/operations/*}:wait",

0 commit comments

Comments
 (0)