Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions google/genai/_live_converters.py
Original file line number Diff line number Diff line change
Expand Up @@ -973,11 +973,11 @@ def _LiveConnectConfig_to_mldev(
],
)

if getv(from_object, ['stream_translation_config']) is not None:
if getv(from_object, ['streaming_translation_config']) is not None:
setv(
parent_object,
['setup', 'generationConfig', 'streamTranslationConfig'],
getv(from_object, ['stream_translation_config']),
['setup', 'generationConfig', 'streamingTranslationConfig'],
getv(from_object, ['streaming_translation_config']),
)

return to_object
Expand Down Expand Up @@ -1160,9 +1160,9 @@ def _LiveConnectConfig_to_vertex(
[item for item in getv(from_object, ['safety_settings'])],
)

if getv(from_object, ['stream_translation_config']) is not None:
if getv(from_object, ['streaming_translation_config']) is not None:
raise ValueError(
'stream_translation_config parameter is only supported in Gemini'
'streaming_translation_config parameter is only supported in Gemini'
' Developer API mode, not in Gemini Enterprise Agent Platform mode.'
)

Expand Down
6 changes: 3 additions & 3 deletions google/genai/_tokens_converters.py
Original file line number Diff line number Diff line change
Expand Up @@ -474,11 +474,11 @@ def _LiveConnectConfig_to_mldev(
],
)

if getv(from_object, ['stream_translation_config']) is not None:
if getv(from_object, ['streaming_translation_config']) is not None:
setv(
parent_object,
['setup', 'generationConfig', 'streamTranslationConfig'],
getv(from_object, ['stream_translation_config']),
['setup', 'generationConfig', 'streamingTranslationConfig'],
getv(from_object, ['streaming_translation_config']),
)

return to_object
Expand Down
28 changes: 14 additions & 14 deletions google/genai/tests/live/test_live.py
Original file line number Diff line number Diff line change
Expand Up @@ -1612,12 +1612,12 @@ async def test_bidi_setup_to_api_with_transparent_session_resumption(vertexai):

@pytest.mark.parametrize('vertexai', [True, False])
@pytest.mark.asyncio
async def test_bidi_setup_to_api_with_stream_translation_config(vertexai):
async def test_bidi_setup_to_api_with_streaming_translation_config(vertexai):
api_client = mock_api_client(vertexai=vertexai)

# Test 1: Config defined using dict representation.
config_dict = {
'stream_translation_config': {
'streaming_translation_config': {
'echo_target_language': True,
'target_language_code': 'es',
},
Expand All @@ -1628,23 +1628,23 @@ async def test_bidi_setup_to_api_with_stream_translation_config(vertexai):
api_client=api_client, model='test_model', config=config_dict
)

expected_result = {
'setup': {
'model': 'models/test_model',
'generationConfig': {
'streamingTranslationConfig': {
'echo_target_language': True,
'target_language_code': 'es',
},
},
}
}
if not vertexai:
expected_result = {
'setup': {
'model': 'models/test_model',
'generationConfig': {
'streamTranslationConfig': {
'echo_target_language': True,
'target_language_code': 'es',
},
},
}
}
assert result == expected_result

# Test 2: Config defined using types.LiveConnectConfig.
config = types.LiveConnectConfig(
stream_translation_config=types.StreamTranslationConfig(
streaming_translation_config=types.StreamingTranslationConfig(
echo_target_language=True,
target_language_code='es',
)
Expand Down
16 changes: 8 additions & 8 deletions google/genai/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -19794,7 +19794,7 @@ class RealtimeInputConfigDict(TypedDict, total=False):
RealtimeInputConfigOrDict = Union[RealtimeInputConfig, RealtimeInputConfigDict]


class StreamTranslationConfig(_common.BaseModel):
class StreamingTranslationConfig(_common.BaseModel):
"""Config for stream translation."""

echo_target_language: Optional[bool] = Field(
Expand All @@ -19810,7 +19810,7 @@ class StreamTranslationConfig(_common.BaseModel):
)


class StreamTranslationConfigDict(TypedDict, total=False):
class StreamingTranslationConfigDict(TypedDict, total=False):
"""Config for stream translation."""

echo_target_language: Optional[bool]
Expand All @@ -19823,8 +19823,8 @@ class StreamTranslationConfigDict(TypedDict, total=False):
language codes (e.g. "en", "es", "fr")."""


StreamTranslationConfigOrDict = Union[
StreamTranslationConfig, StreamTranslationConfigDict
StreamingTranslationConfigOrDict = Union[
StreamingTranslationConfig, StreamingTranslationConfigDict
]


Expand Down Expand Up @@ -19966,8 +19966,8 @@ class LiveConnectConfig(_common.BaseModel):
response.
""",
)
stream_translation_config: Optional[StreamTranslationConfig] = Field(
default=None, description="""Config for stream translation."""
streaming_translation_config: Optional[StreamingTranslationConfig] = Field(
default=None, description="""Config for streaming translation."""
)


Expand Down Expand Up @@ -20087,8 +20087,8 @@ class LiveConnectConfigDict(TypedDict, total=False):
response.
"""

stream_translation_config: Optional[StreamTranslationConfigDict]
"""Config for stream translation."""
streaming_translation_config: Optional[StreamingTranslationConfigDict]
"""Config for streaming translation."""


LiveConnectConfigOrDict = Union[LiveConnectConfig, LiveConnectConfigDict]
Expand Down
Loading