Skip to content

Commit 87aa2d9

Browse files
jaycee-licopybara-github
authored andcommitted
feat: Add model_status to GenerateContentResponse (Gemini API only)
feat: Add part_metadata in Part (Gemini API only) feat: Support rendered_parts in GroundingSupport PiperOrigin-RevId: 882224524
1 parent 24f626c commit 87aa2d9

2 files changed

Lines changed: 43 additions & 0 deletions

File tree

vertexai/_genai/types/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1082,6 +1082,9 @@
10821082
from .common import ToolParameterKVMatchSpec
10831083
from .common import ToolParameterKVMatchSpecDict
10841084
from .common import ToolParameterKVMatchSpecOrDict
1085+
from .common import Transcription
1086+
from .common import TranscriptionDict
1087+
from .common import TranscriptionOrDict
10851088
from .common import TuningResourceUsageAssessmentConfig
10861089
from .common import TuningResourceUsageAssessmentConfigDict
10871090
from .common import TuningResourceUsageAssessmentConfigOrDict
@@ -1826,6 +1829,9 @@
18261829
"EventActions",
18271830
"EventActionsDict",
18281831
"EventActionsOrDict",
1832+
"Transcription",
1833+
"TranscriptionDict",
1834+
"TranscriptionOrDict",
18291835
"EventMetadata",
18301836
"EventMetadataDict",
18311837
"EventMetadataOrDict",

vertexai/_genai/types/common.py

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11211,6 +11211,31 @@ class EventActionsDict(TypedDict, total=False):
1121111211
EventActionsOrDict = Union[EventActions, EventActionsDict]
1121211212

1121311213

11214+
class Transcription(_common.BaseModel):
11215+
"""Audio transcription in Server Content."""
11216+
11217+
text: Optional[str] = Field(
11218+
default=None, description="""Optional. Transcription text."""
11219+
)
11220+
finished: Optional[bool] = Field(
11221+
default=None,
11222+
description="""Optional. The bool indicates the end of the transcription.""",
11223+
)
11224+
11225+
11226+
class TranscriptionDict(TypedDict, total=False):
11227+
"""Audio transcription in Server Content."""
11228+
11229+
text: Optional[str]
11230+
"""Optional. Transcription text."""
11231+
11232+
finished: Optional[bool]
11233+
"""Optional. The bool indicates the end of the transcription."""
11234+
11235+
11236+
TranscriptionOrDict = Union[Transcription, TranscriptionDict]
11237+
11238+
1121411239
class EventMetadata(_common.BaseModel):
1121511240
"""Metadata relating to a LLM response event."""
1121611241

@@ -11241,6 +11266,12 @@ class EventMetadata(_common.BaseModel):
1124111266
default=None,
1124211267
description="""Optional. Indicates whether the response from the model is complete. Only used for streaming mode.""",
1124311268
)
11269+
input_transcription: Optional[Transcription] = Field(
11270+
default=None, description="""Optional. Audio transcription of user input."""
11271+
)
11272+
output_transcription: Optional[Transcription] = Field(
11273+
default=None, description="""Optional. Audio transcription of model output."""
11274+
)
1124411275

1124511276

1124611277
class EventMetadataDict(TypedDict, total=False):
@@ -11267,6 +11298,12 @@ class EventMetadataDict(TypedDict, total=False):
1126711298
turn_complete: Optional[bool]
1126811299
"""Optional. Indicates whether the response from the model is complete. Only used for streaming mode."""
1126911300

11301+
input_transcription: Optional[TranscriptionDict]
11302+
"""Optional. Audio transcription of user input."""
11303+
11304+
output_transcription: Optional[TranscriptionDict]
11305+
"""Optional. Audio transcription of model output."""
11306+
1127011307

1127111308
EventMetadataOrDict = Union[EventMetadata, EventMetadataDict]
1127211309

0 commit comments

Comments
 (0)