google-genai 1.55.0__py3-none-any.whl → 1.57.0__py3-none-any.whl

google/genai/tests/models/test_function_call_streaming.py

@@ -154,7 +154,7 @@ def test_streaming_with_python_native_no_afc_config(client):
     return
   with pytest.raises(ValueError) as e:
     for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         contents=generate_content_prompt,
         config=types.GenerateContentConfig(
             tools=[
@@ -179,36 +179,18 @@ def test_streaming_with_python_afc_disabled_false(client):
     return
   with pytest.raises(ValueError) as e:
     for chunk in client.models.generate_content_stream(
-      model='gemini-2.5-pro',
-      contents='What is the price of GOOG? And what is the weather in Boston?',
-      config=types.GenerateContentConfig(
-          tools=[
-              test_generate_content_tools.get_weather,
-              test_generate_content_tools.get_stock_price,
-          ],
-          automatic_function_calling=types.AutomaticFunctionCallingConfig(
-              disable=False,
-          ),
-          tool_config=types.ToolConfig(
-              function_calling_config={
-                  'stream_function_call_arguments': True,
-              }
-          ),
-      ),
-    ):
-      pass
-  assert 'not compatible with automatic function calling (AFC)' in str(e.value)
-
-
-def test_streaming_with_json_parameters_without_history(client):
-  """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
-
-  with pytest_helper.exception_if_mldev(client, ValueError):
-    for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
-        contents=generate_content_prompt,
+        model='gemini-3-pro-preview',
+        contents=(
+            'What is the price of GOOG? And what is the weather in Boston?'
+        ),
         config=types.GenerateContentConfig(
-            tools=[{'function_declarations': json_function_declarations}],
+            tools=[
+                test_generate_content_tools.get_weather,
+                test_generate_content_tools.get_stock_price,
+            ],
+            automatic_function_calling=types.AutomaticFunctionCallingConfig(
+                disable=False,
+            ),
             tool_config=types.ToolConfig(
                 function_calling_config={
                     'stream_function_call_arguments': True,
@@ -216,18 +198,17 @@ def test_streaming_with_json_parameters_without_history(client):
             ),
         ),
     ):
-      assert chunk is not None
-      assert chunk.candidates is not None
-      assert chunk.candidates[0].content is not None
-      assert chunk.candidates[0].content.parts is not None
+      pass
+  assert 'not compatible with automatic function calling (AFC)' in str(e.value)
 
 
-def test_streaming_with_json_parameters_with_history(client):
+def test_streaming_with_json_parameters_without_history(client):
   """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
+
   with pytest_helper.exception_if_mldev(client, ValueError):
     for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
-        contents=previous_generate_content_history,
+        model='gemini-3-pro-preview',
+        contents=generate_content_prompt,
         config=types.GenerateContentConfig(
             tools=[{'function_declarations': json_function_declarations}],
             tool_config=types.ToolConfig(
@@ -242,12 +223,13 @@ def test_streaming_with_json_parameters_with_history(client):
       assert chunk.candidates[0].content is not None
       assert chunk.candidates[0].content.parts is not None
 
+
 @pytest.mark.asyncio
 async  def test_streaming_with_json_parameters_async(client):
   """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
   with pytest_helper.exception_if_mldev(client, ValueError):
     async for chunk in await client.aio.models.generate_content_stream(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         contents=generate_content_prompt,
         config=types.GenerateContentConfig(
             tools=[{'function_declarations': json_function_declarations}],
@@ -268,7 +250,7 @@ def test_streaming_with_gemini_parameters_without_history(client):
   """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
   with pytest_helper.exception_if_mldev(client, ValueError):
     for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         contents=generate_content_prompt,
         config=types.GenerateContentConfig(
             tools=[{
@@ -291,7 +273,7 @@ def test_streaming_with_gemini_parameters_with_response(client):
   with pytest_helper.exception_if_mldev(client, ValueError):
     streaming_function_call_content = []
     for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         contents=[
             types.Content(
                 role='user',
@@ -335,7 +317,7 @@ def test_streaming_with_gemini_parameters_with_response(client):
     )
 
     for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         contents=streaming_function_call_content,
         config=types.GenerateContentConfig(
             tools=[{'function_declarations': json_function_declarations}],
@@ -348,41 +330,41 @@ def test_streaming_with_gemini_parameters_with_response(client):
     ):
       pass
 
-def test_streaming_with_gemini_parameters_with_history(client):
-  """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
-  with pytest_helper.exception_if_mldev(client, ValueError):
-    for chunk in client.models.generate_content_stream(
-        model='gemini-2.5-pro',
-        contents=previous_generate_content_history,
-        config=types.GenerateContentConfig(
-            tools=[{
-                'function_declarations': gemini_function_declarations
-            }],
-            tool_config=types.ToolConfig(
-                function_calling_config={
-                    'stream_function_call_arguments': True,
-                }
-            ),
-        ),
-    ):
-      assert chunk is not None
-      assert chunk.candidates is not None
-      assert chunk.candidates[0].content is not None
-      assert chunk.candidates[0].content.parts is not None
-
 def test_chat_streaming_with_json_parameters_with_history(client):
   """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
   with pytest_helper.exception_if_mldev(client, ValueError):
-    messages = [
-        """
-        get the current weather in boston in celsius, the country is U.S., the purpose is to know what to wear today.
-        """,
-        """
-        get the current weather in new brunswick in celsius, the country is U.S., the purpose is to know what to prepare an event today.
-        """,
+    test_parts = [
+        types.Part(
+            text=(
+                'get the current weather in boston in celsius, the'
+                ' country should be US, the purpose is to know'
+                ' what to wear today?'
+            )
+        ),
+        types.Part.from_function_response(
+            name='get_current_weather',
+            response={
+                'temperature': 21,
+                'unit': 'C',
+            },
+        ),
+        types.Part(
+            text=(
+                'get the current weather in new brunswick in celsius, the'
+                ' country should be US, the purpose is to know'
+                ' what to prepare an event today?'
+            )
+        ),
+        types.Part.from_function_response(
+            name='get_current_weather',
+            response={
+                'temperature': 21,
+                'unit': 'C',
+            },
+        ),
     ]
     chat = client.chats.create(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         history=previous_generate_content_history,
         config=types.GenerateContentConfig(
             tools=[{
@@ -395,7 +377,7 @@ def test_chat_streaming_with_json_parameters_with_history(client):
             ),
         ),
     )
-    for message in messages:
+    for message in test_parts:
       result = chat.send_message_stream(message)
       for chunk in result:
         assert chunk is not None
@@ -409,17 +391,39 @@ def test_chat_streaming_with_json_parameters_with_history(client):
 @pytest.mark.asyncio
 async def test_chat_streaming_with_json_parameters_with_history_async(client):
   """Tests streaming function calls with FunctionDeclaration withJSON parameters."""
-  messages = [
-      """
-      get the current weather in boston in celsius, the country is U.S., the purpose is to know what to wear today.
-      """,
-      """
-      get the current weather in new brunswick in celsius, the country is U.S., the purpose is to know what to prepare an event today.
-      """,
+  test_parts = [
+      types.Part(
+          text=(
+              'get the current weather in boston in celsius, the'
+              ' country should be US, the purpose is to know'
+              ' what to wear today?'
+          )
+      ),
+      types.Part.from_function_response(
+          name='get_current_weather',
+          response={
+              'temperature': 21,
+              'unit': 'C',
+          },
+      ),
+      types.Part(
+          text=(
+              'get the current weather in new brunswick in celsius, the'
+              ' country should be US, the purpose is to know'
+              ' what to prepare an event today?'
+          )
+      ),
+      types.Part.from_function_response(
+          name='get_current_weather',
+          response={
+              'temperature': 21,
+              'unit': 'C',
+          },
+      ),
   ]
   with pytest_helper.exception_if_mldev(client, ValueError):
     chat = client.aio.chats.create(
-        model='gemini-2.5-pro',
+        model='gemini-3-pro-preview',
         history=previous_generate_content_history,
         config=types.GenerateContentConfig(
             tools=[{'function_declarations': gemini_function_declarations}],
@@ -430,13 +434,9 @@ async def test_chat_streaming_with_json_parameters_with_history_async(client):
             ),
         ),
     )
-    async for chunk in await chat.send_message_stream(messages[0]):
-      assert chunk is not None
-      assert chunk.candidates is not None
-      assert chunk.candidates[0].content is not None
-      assert chunk.candidates[0].content.parts is not None
-    async for chunk in await chat.send_message_stream(messages[1]):
-      assert chunk is not None
-      assert chunk.candidates is not None
-      assert chunk.candidates[0].content is not None
-      assert chunk.candidates[0].content.parts is not None
+    for message in test_parts:
+      async for chunk in await chat.send_message_stream(message):
+        assert chunk is not None
+        assert chunk.candidates is not None
+        assert chunk.candidates[0].content is not None
+        assert chunk.candidates[0].content.parts is not None

google/genai/tests/models/test_generate_content.py

@@ -374,7 +374,6 @@ test_table: list[pytest_helper.TestTableItem] = [
     ),
     pytest_helper.TestTableItem(
         name='test_speech_with_multi_speaker_voice_config',
-        exception_if_vertex='not supported',
         parameters=types._GenerateContentParameters(
             model='gemini-2.5-flash-preview-tts',
             contents=t.t_contents(
@@ -409,7 +408,7 @@ test_table: list[pytest_helper.TestTableItem] = [
     ),
     pytest_helper.TestTableItem(
         name='test_speech_error_with_speech_config_and_multi_speech_config',
-        exception_if_vertex='not supported',
+        exception_if_vertex='mutually exclusive',
         exception_if_mldev='mutually exclusive',
         parameters=types._GenerateContentParameters(
             model='gemini-2.5-flash-preview-tts',

google/genai/tests/models/test_recontext_image.py

@@ -23,7 +23,7 @@ from .. import pytest_helper
 
 PRODUCT_RECONTEXT_MODEL_LATEST = 'imagen-product-recontext-preview-06-30'
 
-VIRTUAL_TRY_ON_IMAGE_MODEL_LATEST = 'virtual-try-on-preview-08-04'
+VIRTUAL_TRY_ON_IMAGE_MODEL_LATEST = 'virtual-try-on-001'
 
 BACKPACK1 = types.Image(
     gcs_uri='gs://genai-sdk-tests/inputs/images/backpack1.png',

google/genai/tests/pytest_helper.py

@@ -227,3 +227,20 @@ def camel_to_snake(camel_str: str) -> str:
 def get_value_ignore_key_case(obj, key):
   """Returns the value of the key in the object, converting to camelCase or snake_case if necessary."""
   return obj.get(snake_to_camel(key), obj.get(camel_to_snake(key), None))
+
+
+def camel_to_snake_all_keys(data):
+  """Converts all keys in a dictionary or list to snake_case."""
+  if isinstance(data, dict):
+    new_dict = {}
+    for key, value in data.items():
+      if isinstance(key, str):
+        new_key = camel_to_snake(key)
+      else:
+        new_key = key
+      new_dict[new_key] = camel_to_snake_all_keys(value)
+    return new_dict
+  elif isinstance(data, list):
+    return [camel_to_snake_all_keys(item) for item in data]
+  else:
+    return data

google/genai/tunings.py

@@ -674,13 +674,7 @@ def _GenerationConfig_to_vertex(
     setv(to_object, ['seed'], getv(from_object, ['seed']))
 
   if getv(from_object, ['speech_config']) is not None:
-    setv(
-        to_object,
-        ['speechConfig'],
-        _SpeechConfig_to_vertex(
-            getv(from_object, ['speech_config']), to_object, root_object
-        ),
-    )
+    setv(to_object, ['speechConfig'], getv(from_object, ['speech_config']))
 
   if getv(from_object, ['stop_sequences']) is not None:
     setv(to_object, ['stopSequences'], getv(from_object, ['stop_sequences']))
@@ -861,26 +855,6 @@ def _ListTuningJobsResponse_from_vertex(
   return to_object
 
 
-def _SpeechConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-    root_object: Optional[Union[dict[str, Any], object]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['voice_config']) is not None:
-    setv(to_object, ['voiceConfig'], getv(from_object, ['voice_config']))
-
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
-  if getv(from_object, ['multi_speaker_voice_config']) is not None:
-    raise ValueError(
-        'multi_speaker_voice_config parameter is not supported in Vertex AI.'
-    )
-
-  return to_object
-
-
 def _TunedModel_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,