From 4c04fd0855845140022afcb5a7ac44dcedebf51b Mon Sep 17 00:00:00 2001 From: Sara Robinson Date: Wed, 22 Jul 2026 06:55:18 -0700 Subject: [PATCH] feat: Add bidiGenerateContentSetup, expireTime, fieldMask, newSessionExpireTime, and uses to AuthToken. Add audioTranscriptionConfig to GenerationConfig. Add buildSpec to ReasoningEngineSpec. Add stepCount to ReinforcementTuningHyperParameters. Add enableDataRetention and enableZeroDataRetention to ToolParallelAiSearch. PiperOrigin-RevId: 952092382 --- agentplatform/_genai/types/__init__.py | 6 ++ agentplatform/_genai/types/common.py | 79 ++++++++++++++++++-------- 2 files changed, 60 insertions(+), 25 deletions(-) diff --git a/agentplatform/_genai/types/__init__.py b/agentplatform/_genai/types/__init__.py index e7612f28d1..160fea5088 100644 --- a/agentplatform/_genai/types/__init__.py +++ b/agentplatform/_genai/types/__init__.py @@ -1419,6 +1419,9 @@ from .common import ReasoningEngineRuntimeRevisionDict from .common import ReasoningEngineRuntimeRevisionOrDict from .common import ReasoningEngineSpec +from .common import ReasoningEngineSpecBuildSpec +from .common import ReasoningEngineSpecBuildSpecDict +from .common import ReasoningEngineSpecBuildSpecOrDict from .common import ReasoningEngineSpecContainerSpec from .common import ReasoningEngineSpecContainerSpecDict from .common import ReasoningEngineSpecContainerSpecOrDict @@ -2551,6 +2554,9 @@ "ReasoningEngineSpecContainerSpec", "ReasoningEngineSpecContainerSpecDict", "ReasoningEngineSpecContainerSpecOrDict", + "ReasoningEngineSpecBuildSpec", + "ReasoningEngineSpecBuildSpecDict", + "ReasoningEngineSpecBuildSpecOrDict", "ReasoningEngineSpec", "ReasoningEngineSpecDict", "ReasoningEngineSpecOrDict", diff --git a/agentplatform/_genai/types/common.py b/agentplatform/_genai/types/common.py index cd6fb70b6c..7bbd9c5852 100644 --- a/agentplatform/_genai/types/common.py +++ b/agentplatform/_genai/types/common.py @@ -8676,6 +8676,27 @@ class ReasoningEngineSpecContainerSpecDict(TypedDict, total=False): ] +class ReasoningEngineSpecBuildSpec(_common.BaseModel): + """Specification for building container image.""" + + worker_pool: Optional[str] = Field( + default=None, + description="""Optional. Identifier. The resource name of the Cloud Build WorkerPool to use for the build. Format: `projects/{project}/locations/{location}/workerPools/{worker_pool}`""", + ) + + +class ReasoningEngineSpecBuildSpecDict(TypedDict, total=False): + """Specification for building container image.""" + + worker_pool: Optional[str] + """Optional. Identifier. The resource name of the Cloud Build WorkerPool to use for the build. Format: `projects/{project}/locations/{location}/workerPools/{worker_pool}`""" + + +ReasoningEngineSpecBuildSpecOrDict = Union[ + ReasoningEngineSpecBuildSpec, ReasoningEngineSpecBuildSpecDict +] + + class ReasoningEngineSpec(_common.BaseModel): """The specification of an agent engine.""" @@ -8719,6 +8740,10 @@ class ReasoningEngineSpec(_common.BaseModel): default=None, description="""Deploy from a container image with a defined entrypoint and commands.""", ) + build_spec: Optional[ReasoningEngineSpecBuildSpec] = Field( + default=None, + description="""Optional. Configuration for building container image.""", + ) class ReasoningEngineSpecDict(TypedDict, total=False): @@ -8754,6 +8779,9 @@ class ReasoningEngineSpecDict(TypedDict, total=False): container_spec: Optional[ReasoningEngineSpecContainerSpecDict] """Deploy from a container image with a defined entrypoint and commands.""" + build_spec: Optional[ReasoningEngineSpecBuildSpecDict] + """Optional. Configuration for building container image.""" + ReasoningEngineSpecOrDict = Union[ReasoningEngineSpec, ReasoningEngineSpecDict] @@ -24142,45 +24170,46 @@ class FeedbackEntry(_common.BaseModel): create_time: Optional[datetime.datetime] = Field( default=None, - description="""Output only. Timestamp when the feedback entry was created.""", + description="""Output only. The time at which the entry was created.""", ) custom_metadata: Optional[dict[str, str]] = Field( default=None, - description="""Optional. Additional key-value metadata associated with the feedback. Allows the collect data for which there is no dedicated field in the resource, ex. version, LLM temperature etc.""", + description="""Optional. Additional key-value metadata associated with the feedback.""", ) event_id: Optional[str] = Field( default=None, - description="""Required. The ID of the event to which the feedback relates to.""", + description="""Required. The ID of the event within the session that the feedback relates to.""", ) feedback_labels: Optional[list[str]] = Field( - default=None, - description="""Optional. Specific labels for feedback (non-factual, offensive, etc.).""", + default=None, description="""feedbackLabels""" ) feedback_text: Optional[str] = Field( default=None, description="""Optional. Qualitative free-form comments provided by the user.""", ) feedback_type: Optional[FeedbackType] = Field( - default=None, description="""Required. The type of feedback provided.""" + default=None, + description="""Required. The coarse-grained type of feedback provided by the user. Must be set to a value other than `FEEDBACK_TYPE_UNSPECIFIED`.""", ) name: Optional[str] = Field( default=None, - description="""Identifier. The resource name of the feedback entry. Format: 'projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}'.""", + description="""Identifier. The resource name. Assigned by the server on create. Format: `projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}`""", ) session_id: Optional[str] = Field( default=None, - description="""Required. The ID of the session to which the feedback relates to.""", + description="""Required. The ID of the session that the feedback relates to.""", ) source: Optional[str] = Field( default=None, - description="""Optional. Originating UI surface (e.g. 'ADK Web UI').""", + description="""Optional. The surface that the feedback originated from.""", ) update_time: Optional[datetime.datetime] = Field( default=None, - description="""Output only. Timestamp when the feedback entry was last updated.""", + description="""Output only. The time at which the entry was most recently updated.""", ) user_id: Optional[str] = Field( - default=None, description="""Optional. User provided identifier.""" + default=None, + description="""Optional. A caller-supplied identifier for the user who provided the feedback. The semantics of this field (for example whether it is an opaque token, a hashed value, or a user-visible identifier) are determined by the calling application.""", ) @@ -24188,37 +24217,37 @@ class FeedbackEntryDict(TypedDict, total=False): """A Feedback Entry.""" create_time: Optional[datetime.datetime] - """Output only. Timestamp when the feedback entry was created.""" + """Output only. The time at which the entry was created.""" custom_metadata: Optional[dict[str, str]] - """Optional. Additional key-value metadata associated with the feedback. Allows the collect data for which there is no dedicated field in the resource, ex. version, LLM temperature etc.""" + """Optional. Additional key-value metadata associated with the feedback.""" event_id: Optional[str] - """Required. The ID of the event to which the feedback relates to.""" + """Required. The ID of the event within the session that the feedback relates to.""" feedback_labels: Optional[list[str]] - """Optional. Specific labels for feedback (non-factual, offensive, etc.).""" + """feedbackLabels""" feedback_text: Optional[str] """Optional. Qualitative free-form comments provided by the user.""" feedback_type: Optional[FeedbackType] - """Required. The type of feedback provided.""" + """Required. The coarse-grained type of feedback provided by the user. Must be set to a value other than `FEEDBACK_TYPE_UNSPECIFIED`.""" name: Optional[str] - """Identifier. The resource name of the feedback entry. Format: 'projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}'.""" + """Identifier. The resource name. Assigned by the server on create. Format: `projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}`""" session_id: Optional[str] - """Required. The ID of the session to which the feedback relates to.""" + """Required. The ID of the session that the feedback relates to.""" source: Optional[str] - """Optional. Originating UI surface (e.g. 'ADK Web UI').""" + """Optional. The surface that the feedback originated from.""" update_time: Optional[datetime.datetime] - """Output only. Timestamp when the feedback entry was last updated.""" + """Output only. The time at which the entry was most recently updated.""" user_id: Optional[str] - """Optional. User provided identifier.""" + """Optional. A caller-supplied identifier for the user who provided the feedback. The semantics of this field (for example whether it is an opaque token, a hashed value, or a user-visible identifier) are determined by the calling application.""" FeedbackEntryOrDict = Union[FeedbackEntry, FeedbackEntryDict] @@ -24725,11 +24754,11 @@ class FeedbackContext(_common.BaseModel): context_events: Optional[list[SessionEvent]] = Field( default=None, - description="""Optional. Events from the conversation relevant to the parent feedback entry.""", + description="""Optional. The session events from the originating session.""", ) name: Optional[str] = Field( default=None, - description="""Identifier. The resource name of the feedback context. Format: 'projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}/feedbackContext'.""", + description="""Identifier. The resource name. Assigned by the server on create. Format: `projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}/feedbackContext`""", ) @@ -24737,10 +24766,10 @@ class FeedbackContextDict(TypedDict, total=False): """A Feedback Context.""" context_events: Optional[list[SessionEventDict]] - """Optional. Events from the conversation relevant to the parent feedback entry.""" + """Optional. The session events from the originating session.""" name: Optional[str] - """Identifier. The resource name of the feedback context. Format: 'projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}/feedbackContext'.""" + """Identifier. The resource name. Assigned by the server on create. Format: `projects/{project}/locations/{location}/reasoningEngines/{reasoning_engine}/feedbackEntries/{feedback_entry}/feedbackContext`""" FeedbackContextOrDict = Union[FeedbackContext, FeedbackContextDict]