Skip to content

Commit d1735ab

Browse files
speedstorm1copybara-github
authored andcommitted
Copybara import of the project:
-- 3e05303 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 -- 75aa48e by Ayush Agrawal <ayushagra@google.com>: Revert generated metadata files to resolve conflicts COPYBARA_INTEGRATE_REVIEW=#6212 from googleapis:owl-bot-copy 75aa48e PiperOrigin-RevId: 858867075
1 parent 8f5bfc5 commit d1735ab

2 files changed

Lines changed: 90 additions & 7 deletions

File tree

google/cloud/aiplatform_v1beta1/types/vertex_rag_data.py

Lines changed: 86 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1546,20 +1546,32 @@ class RagManagedDbConfig(proto.Message):
15461546
15471547
This field is a member of `oneof`_ ``tier``.
15481548
scaled (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Scaled):
1549-
Sets the RagManagedDb to the Scaled tier.
1550-
This is the default tier if not explicitly
1551-
chosen.
1549+
Deprecated: Use ``mode`` instead to set the tier under
1550+
Spanner. Sets the RagManagedDb to the Scaled tier.
15521551
15531552
This field is a member of `oneof`_ ``tier``.
15541553
basic (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Basic):
1555-
Sets the RagManagedDb to the Basic tier.
1554+
Deprecated: Use ``mode`` instead to set the tier under
1555+
Spanner. Sets the RagManagedDb to the Basic tier.
15561556
15571557
This field is a member of `oneof`_ ``tier``.
15581558
unprovisioned (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Unprovisioned):
1559-
Sets the RagManagedDb to the Unprovisioned
1560-
tier.
1559+
Deprecated: Use ``mode`` instead to set the tier under
1560+
Spanner. Sets the RagManagedDb to the Unprovisioned tier.
15611561
15621562
This field is a member of `oneof`_ ``tier``.
1563+
serverless (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Serverless):
1564+
Sets the backend to be the serverless mode
1565+
offered by RAG Engine.
1566+
1567+
This field is a member of `oneof`_ ``mode``.
1568+
spanner (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Spanner):
1569+
Sets the RAG Engine backend to be
1570+
RagManagedDb, built on top of Spanner.
1571+
NOTE: This is the default mode (w/ Basic Tier)
1572+
if not explicitly chosen.
1573+
1574+
This field is a member of `oneof`_ ``mode``.
15631575
"""
15641576

15651577
class Enterprise(proto.Message):
@@ -1585,7 +1597,8 @@ class Basic(proto.Message):
15851597
- Latency insensitive workload.
15861598
- Only using RAG Engine with external vector DBs.
15871599
1588-
NOTE: This is the default tier if not explicitly chosen.
1600+
NOTE: This is the default tier under Spanner mode if not explicitly
1601+
chosen.
15891602
15901603
"""
15911604

@@ -1600,6 +1613,60 @@ class Unprovisioned(proto.Message):
16001613
16011614
"""
16021615

1616+
class Spanner(proto.Message):
1617+
r"""Message to configure the Spanner database used by
1618+
RagManagedDb.
1619+
1620+
This message has `oneof`_ fields (mutually exclusive fields).
1621+
For each oneof, at most one member field can be set at the same time.
1622+
Setting any member of the oneof automatically clears all other
1623+
members.
1624+
1625+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
1626+
1627+
Attributes:
1628+
scaled (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Scaled):
1629+
Sets the RagManagedDb to the Scaled tier.
1630+
1631+
This field is a member of `oneof`_ ``tier``.
1632+
basic (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Basic):
1633+
Sets the RagManagedDb to the Basic tier. This
1634+
is the default tier for Spanner mode if not
1635+
explicitly chosen.
1636+
1637+
This field is a member of `oneof`_ ``tier``.
1638+
unprovisioned (google.cloud.aiplatform_v1beta1.types.RagManagedDbConfig.Unprovisioned):
1639+
Sets the RagManagedDb to the Unprovisioned
1640+
tier.
1641+
1642+
This field is a member of `oneof`_ ``tier``.
1643+
"""
1644+
1645+
scaled: "RagManagedDbConfig.Scaled" = proto.Field(
1646+
proto.MESSAGE,
1647+
number=1,
1648+
oneof="tier",
1649+
message="RagManagedDbConfig.Scaled",
1650+
)
1651+
basic: "RagManagedDbConfig.Basic" = proto.Field(
1652+
proto.MESSAGE,
1653+
number=2,
1654+
oneof="tier",
1655+
message="RagManagedDbConfig.Basic",
1656+
)
1657+
unprovisioned: "RagManagedDbConfig.Unprovisioned" = proto.Field(
1658+
proto.MESSAGE,
1659+
number=3,
1660+
oneof="tier",
1661+
message="RagManagedDbConfig.Unprovisioned",
1662+
)
1663+
1664+
class Serverless(proto.Message):
1665+
r"""Message to configure the serverless mode offered by RAG
1666+
Engine.
1667+
1668+
"""
1669+
16031670
enterprise: Enterprise = proto.Field(
16041671
proto.MESSAGE,
16051672
number=1,
@@ -1624,6 +1691,18 @@ class Unprovisioned(proto.Message):
16241691
oneof="tier",
16251692
message=Unprovisioned,
16261693
)
1694+
serverless: Serverless = proto.Field(
1695+
proto.MESSAGE,
1696+
number=5,
1697+
oneof="mode",
1698+
message=Serverless,
1699+
)
1700+
spanner: Spanner = proto.Field(
1701+
proto.MESSAGE,
1702+
number=6,
1703+
oneof="mode",
1704+
message=Spanner,
1705+
)
16271706

16281707

16291708
class RagEngineConfig(proto.Message):

tests/unit/gapic/aiplatform_v1beta1/test_vertex_rag_data_service.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10756,6 +10756,8 @@ def test_update_rag_engine_config_rest_call_success(request_type):
1075610756
"scaled": {},
1075710757
"basic": {},
1075810758
"unprovisioned": {},
10759+
"serverless": {},
10760+
"spanner": {"scaled": {}, "basic": {}, "unprovisioned": {}},
1075910761
},
1076010762
}
1076110763
# The version of a generated dependency at test runtime may differ from the version used during generation.
@@ -13773,6 +13775,8 @@ async def test_update_rag_engine_config_rest_asyncio_call_success(request_type):
1377313775
"scaled": {},
1377413776
"basic": {},
1377513777
"unprovisioned": {},
13778+
"serverless": {},
13779+
"spanner": {"scaled": {}, "basic": {}, "unprovisioned": {}},
1377613780
},
1377713781
}
1377813782
# The version of a generated dependency at test runtime may differ from the version used during generation.

0 commit comments

Comments
 (0)