Skip to content

Commit 6ae7da7

Browse files
1 parent 64b08bc commit 6ae7da7

File tree

2,790 files changed

+95
-2545548
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,790 files changed

+95
-2545548
lines changed

google/cloud/aiplatform_v1/types/tool.py

Lines changed: 79 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -663,6 +663,8 @@ class RagRetrievalConfig(proto.Message):
663663
Optional. The number of contexts to retrieve.
664664
filter (google.cloud.aiplatform_v1.types.RagRetrievalConfig.Filter):
665665
Optional. Config for filters.
666+
ranking (google.cloud.aiplatform_v1.types.RagRetrievalConfig.Ranking):
667+
Optional. Config for ranking and reranking.
666668
"""
667669

668670
class Filter(proto.Message):
@@ -705,6 +707,78 @@ class Filter(proto.Message):
705707
number=2,
706708
)
707709

710+
class Ranking(proto.Message):
711+
r"""Config for ranking and reranking.
712+
713+
This message has `oneof`_ fields (mutually exclusive fields).
714+
For each oneof, at most one member field can be set at the same time.
715+
Setting any member of the oneof automatically clears all other
716+
members.
717+
718+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
719+
720+
Attributes:
721+
rank_service (google.cloud.aiplatform_v1.types.RagRetrievalConfig.Ranking.RankService):
722+
Optional. Config for Rank Service.
723+
724+
This field is a member of `oneof`_ ``ranking_config``.
725+
llm_ranker (google.cloud.aiplatform_v1.types.RagRetrievalConfig.Ranking.LlmRanker):
726+
Optional. Config for LlmRanker.
727+
728+
This field is a member of `oneof`_ ``ranking_config``.
729+
"""
730+
731+
class RankService(proto.Message):
732+
r"""Config for Rank Service.
733+
734+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
735+
736+
Attributes:
737+
model_name (str):
738+
Optional. The model name of the rank service. Format:
739+
``semantic-ranker-512@latest``
740+
741+
This field is a member of `oneof`_ ``_model_name``.
742+
"""
743+
744+
model_name: str = proto.Field(
745+
proto.STRING,
746+
number=1,
747+
optional=True,
748+
)
749+
750+
class LlmRanker(proto.Message):
751+
r"""Config for LlmRanker.
752+
753+
.. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
754+
755+
Attributes:
756+
model_name (str):
757+
Optional. The model name used for ranking. Format:
758+
``gemini-1.5-pro``
759+
760+
This field is a member of `oneof`_ ``_model_name``.
761+
"""
762+
763+
model_name: str = proto.Field(
764+
proto.STRING,
765+
number=1,
766+
optional=True,
767+
)
768+
769+
rank_service: "RagRetrievalConfig.Ranking.RankService" = proto.Field(
770+
proto.MESSAGE,
771+
number=1,
772+
oneof="ranking_config",
773+
message="RagRetrievalConfig.Ranking.RankService",
774+
)
775+
llm_ranker: "RagRetrievalConfig.Ranking.LlmRanker" = proto.Field(
776+
proto.MESSAGE,
777+
number=3,
778+
oneof="ranking_config",
779+
message="RagRetrievalConfig.Ranking.LlmRanker",
780+
)
781+
708782
top_k: int = proto.Field(
709783
proto.INT32,
710784
number=1,
@@ -714,6 +788,11 @@ class Filter(proto.Message):
714788
number=3,
715789
message=Filter,
716790
)
791+
ranking: Ranking = proto.Field(
792+
proto.MESSAGE,
793+
number=4,
794+
message=Ranking,
795+
)
717796

718797

719798
__all__ = tuple(sorted(__protobuf__.manifest))

owl-bot-staging/v1/.coveragerc

Lines changed: 0 additions & 13 deletions
This file was deleted.

owl-bot-staging/v1/.flake8

Lines changed: 0 additions & 33 deletions
This file was deleted.

owl-bot-staging/v1/MANIFEST.in

Lines changed: 0 additions & 2 deletions
This file was deleted.

owl-bot-staging/v1/README.rst

Lines changed: 0 additions & 143 deletions
This file was deleted.

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

Lines changed: 0 additions & 3 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

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

Lines changed: 0 additions & 6 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

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

Lines changed: 0 additions & 6 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

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

Lines changed: 0 additions & 6 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

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

Lines changed: 0 additions & 10 deletions
This file was deleted.

0 commit comments

Comments
 (0)