feat: Add translation_config SDK support for GEAP.

PiperOrigin-RevId: 971644297

Author: google-genai-bot

google/genai/_live_converters.py

@@ -1154,9 +1154,10 @@ def _LiveConnectConfig_to_vertex(
     )
 
   if getv(from_object, ['translation_config']) is not None:
-    raise ValueError(
-        'translation_config parameter is only supported in Gemini Developer API'
-        ' mode, not in Gemini Enterprise Agent Platform mode.'
+    setv(
+        parent_object,
+        ['setup', 'generationConfig', 'translationConfig'],
+        getv(from_object, ['translation_config']),
     )
 
   return to_object

google/genai/tests/live/test_live.py

@@ -1624,12 +1624,26 @@ async def test_bidi_setup_to_api_with_translation_config(vertexai):
       },
   }
 
-  with pytest_helper.exception_if_vertex(api_client, ValueError):
-    result = await get_connect_message(
-        api_client=api_client, model='test_model', config=config_dict
-    )
+  result = await get_connect_message(
+      api_client=api_client, model='test_model', config=config_dict
+  )
 
-  if not vertexai:
+  if vertexai:
+    expected_result = {
+        'setup': {
+            'model': (
+                'projects/test_project/locations/us-central1/publishers/google/models/test_model'
+            ),
+            'generationConfig': {
+                'responseModalities': ['AUDIO'],
+                'translationConfig': {
+                    'echo_target_language': True,
+                    'target_language_code': 'es',
+                },
+            },
+        }
+    }
+  else:
     expected_result = {
         'setup': {
             'model': 'models/test_model',
@@ -1641,7 +1655,7 @@ async def test_bidi_setup_to_api_with_translation_config(vertexai):
             },
         }
     }
-    assert result == expected_result
+  assert result == expected_result
 
   # Test 2: Config defined using types.LiveConnectConfig.
   config = types.LiveConnectConfig(
@@ -1651,13 +1665,11 @@ async def test_bidi_setup_to_api_with_translation_config(vertexai):
       )
   )
 
-  with pytest_helper.exception_if_vertex(api_client, ValueError):
-    result = await get_connect_message(
-        api_client=api_client, model='test_model', config=config
-    )
+  result = await get_connect_message(
+      api_client=api_client, model='test_model', config=config
+  )
 
-  if not vertexai:
-    assert result == expected_result
+  assert result == expected_result
 
 
 @pytest.mark.parametrize('vertexai', [True, False])
@@ -2303,4 +2315,3 @@ async def test_bidi_setup_replicated_voice_config_with_consent(vertexai):
       replicated_sig['voice_consent_signature'].get('signature')
       == 'test_sig_abc123'
   )
-

google/genai/types.py

@@ -11395,22 +11395,22 @@ class TranslationConfig(_common.BaseModel):
 
   echo_target_language: Optional[bool] = Field(
       default=None,
-      description="""Optional. If true, the model will generate audio when the target language is spoken, essentially it will parrot the input. If false, we will not produce audio for the target language.""",
+      description="""If true, the model will generate audio when the target language is spoken, essentially it will parrot the input. If false, we will not produce audio for the target language.""",
   )
   target_language_code: Optional[str] = Field(
       default=None,
-      description="""Required. The target language for translation. Supported values are BCP-47 language codes (e.g. "en", "es", "fr").""",
+      description="""The target language for translation. Supported values are BCP-47 language codes (e.g. "en", "es", "fr").""",
   )
 
 
 class TranslationConfigDict(TypedDict, total=False):
   """Config for stream translation."""
 
   echo_target_language: Optional[bool]
-  """Optional. If true, the model will generate audio when the target language is spoken, essentially it will parrot the input. If false, we will not produce audio for the target language."""
+  """If true, the model will generate audio when the target language is spoken, essentially it will parrot the input. If false, we will not produce audio for the target language."""
 
   target_language_code: Optional[str]
-  """Required. The target language for translation. Supported values are BCP-47 language codes (e.g. "en", "es", "fr")."""
+  """The target language for translation. Supported values are BCP-47 language codes (e.g. "en", "es", "fr")."""
 
 
 TranslationConfigOrDict = Union[TranslationConfig, TranslationConfigDict]