@@ -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
16291708class RagEngineConfig (proto .Message ):
0 commit comments