Skip to content

Commit 2df4992

Browse files
1 parent fda3bd1 commit 2df4992

3,073 files changed

Lines changed: 85 additions & 2933348 deletions

File tree

Some content is hidden

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

google/cloud/aiplatform_v1/types/content.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1977,6 +1977,11 @@ class GroundingMetadata(proto.Message):
19771977
following-up web searches.
19781978
19791979
This field is a member of `oneof`_ ``_search_entry_point``.
1980+
retrieval_queries (MutableSequence[str]):
1981+
Optional. The queries that were executed by
1982+
the retrieval tools. This field is populated
1983+
only when the grounding source is a retrieval
1984+
tool, such as Vertex AI Search.
19801985
grounding_chunks (MutableSequence[google.cloud.aiplatform_v1.types.GroundingChunk]):
19811986
List of supporting references retrieved from
19821987
specified grounding source.
@@ -2035,6 +2040,10 @@ class SourceFlaggingUri(proto.Message):
20352040
optional=True,
20362041
message="SearchEntryPoint",
20372042
)
2043+
retrieval_queries: MutableSequence[str] = proto.RepeatedField(
2044+
proto.STRING,
2045+
number=3,
2046+
)
20382047
grounding_chunks: MutableSequence["GroundingChunk"] = proto.RepeatedField(
20392048
proto.MESSAGE,
20402049
number=5,

google/cloud/aiplatform_v1/types/reasoning_engine.py

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,35 @@ class ReasoningEngineSpec(proto.Message):
8585
develop the agent. Currently supported values:
8686
"google-adk", "langchain", "langgraph", "ag2",
8787
"llama-index", "custom".
88+
identity_type (google.cloud.aiplatform_v1.types.ReasoningEngineSpec.IdentityType):
89+
Optional. The identity type to use for the Reasoning Engine.
90+
If not specified, the ``service_account`` field will be used
91+
if set, otherwise the default Vertex AI Reasoning Engine
92+
Service Agent in the project will be used.
8893
"""
8994

95+
class IdentityType(proto.Enum):
96+
r"""The identity type to use for the Reasoning Engine.
97+
98+
Values:
99+
IDENTITY_TYPE_UNSPECIFIED (0):
100+
Default value. Use a custom service account if the
101+
``service_account`` field is set, otherwise use the default
102+
Vertex AI Reasoning Engine Service Agent in the project.
103+
Same behavior as SERVICE_ACCOUNT.
104+
SERVICE_ACCOUNT (2):
105+
Use a custom service account if the ``service_account``
106+
field is set, otherwise use the default Vertex AI Reasoning
107+
Engine Service Agent in the project.
108+
AGENT_IDENTITY (3):
109+
Use Agent Identity. The ``service_account`` field must not
110+
be set.
111+
"""
112+
113+
IDENTITY_TYPE_UNSPECIFIED = 0
114+
SERVICE_ACCOUNT = 2
115+
AGENT_IDENTITY = 3
116+
90117
class PackageSpec(proto.Message):
91118
r"""User-provided package specification, containing pickled
92119
object and package requirements.
@@ -453,6 +480,11 @@ class ContainerSpec(proto.Message):
453480
proto.STRING,
454481
number=5,
455482
)
483+
identity_type: IdentityType = proto.Field(
484+
proto.ENUM,
485+
number=12,
486+
enum=IdentityType,
487+
)
456488

457489

458490
class ReasoningEngine(proto.Message):

google/cloud/aiplatform_v1beta1/types/reasoning_engine.py

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,35 @@ class ReasoningEngineSpec(proto.Message):
8787
develop the agent. Currently supported values:
8888
"google-adk", "langchain", "langgraph", "ag2",
8989
"llama-index", "custom".
90+
identity_type (google.cloud.aiplatform_v1beta1.types.ReasoningEngineSpec.IdentityType):
91+
Optional. The identity type to use for the Reasoning Engine.
92+
If not specified, the ``service_account`` field will be used
93+
if set, otherwise the default Vertex AI Reasoning Engine
94+
Service Agent in the project will be used.
9095
"""
9196

97+
class IdentityType(proto.Enum):
98+
r"""The identity type to use for the Reasoning Engine.
99+
100+
Values:
101+
IDENTITY_TYPE_UNSPECIFIED (0):
102+
Default value. Use a custom service account if the
103+
``service_account`` field is set, otherwise use the default
104+
Vertex AI Reasoning Engine Service Agent in the project.
105+
Same behavior as SERVICE_ACCOUNT.
106+
SERVICE_ACCOUNT (2):
107+
Use a custom service account if the ``service_account``
108+
field is set, otherwise use the default Vertex AI Reasoning
109+
Engine Service Agent in the project.
110+
AGENT_IDENTITY (3):
111+
Use Agent Identity. The ``service_account`` field must not
112+
be set.
113+
"""
114+
115+
IDENTITY_TYPE_UNSPECIFIED = 0
116+
SERVICE_ACCOUNT = 2
117+
AGENT_IDENTITY = 3
118+
92119
class PackageSpec(proto.Message):
93120
r"""User-provided package specification, containing pickled
94121
object and package requirements.
@@ -455,6 +482,11 @@ class ContainerSpec(proto.Message):
455482
proto.STRING,
456483
number=5,
457484
)
485+
identity_type: IdentityType = proto.Field(
486+
proto.ENUM,
487+
number=12,
488+
enum=IdentityType,
489+
)
458490

459491

460492
class ReasoningEngine(proto.Message):

owl-bot-staging/v1/.coveragerc

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

owl-bot-staging/v1/.flake8

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

owl-bot-staging/v1/LICENSE

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

owl-bot-staging/v1/MANIFEST.in

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

0 commit comments

Comments
 (0)