diff --git a/google/genai/_live_converters.py b/google/genai/_live_converters.py index 54489055d..a8f25d7f5 100644 --- a/google/genai/_live_converters.py +++ b/google/genai/_live_converters.py @@ -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 @@ -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.' ) diff --git a/google/genai/_tokens_converters.py b/google/genai/_tokens_converters.py index a22306cfd..0a8ea2f37 100644 --- a/google/genai/_tokens_converters.py +++ b/google/genai/_tokens_converters.py @@ -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 diff --git a/google/genai/tests/live/test_live.py b/google/genai/tests/live/test_live.py index 9298f2553..d029442ee 100644 --- a/google/genai/tests/live/test_live.py +++ b/google/genai/tests/live/test_live.py @@ -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', }, @@ -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', ) diff --git a/google/genai/types.py b/google/genai/types.py index 9f846a74b..cc426d3d7 100644 --- a/google/genai/types.py +++ b/google/genai/types.py @@ -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( @@ -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] @@ -19823,8 +19823,8 @@ class StreamTranslationConfigDict(TypedDict, total=False): language codes (e.g. "en", "es", "fr").""" -StreamTranslationConfigOrDict = Union[ - StreamTranslationConfig, StreamTranslationConfigDict +StreamingTranslationConfigOrDict = Union[ + StreamingTranslationConfig, StreamingTranslationConfigDict ] @@ -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.""" ) @@ -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]