google-genai 1.56.0__py3-none-any.whl → 1.58.0__py3-none-any.whl

google/genai/tests/tokens/test_create.py

@@ -0,0 +1,154 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+
+"""Tests for tokens.create()."""
+
+import pytest
+
+from ... import types
+from .. import pytest_helper
+
+
+_MODEL = 'models/gemini-live-2.5-flash-preview'
+
+
+# All tests will be run for both Vertex and MLDev.
+test_table: list[pytest_helper.TestTableItem] = [
+    pytest_helper.TestTableItem(
+        name='test_create_no_lock',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={'http_options': {'api_version': 'v1alpha'}},
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_create_global_lock',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={
+                'http_options': {'api_version': 'v1alpha'},
+                'uses': 2,
+                'live_connect_constraints': {
+                    'model': _MODEL,
+                    'config': {
+                        'response_modalities': ['TEXT'],
+                        'temperature': 0.7,
+                    },
+                },
+            },
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_create_lock_non_null_fields',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={
+                'http_options': {'api_version': 'v1alpha'},
+                'uses': 2,
+                'live_connect_constraints': {
+                    'model': _MODEL,
+                    'config': {
+                        'response_modalities': ['TEXT'],
+                        'temperature': 0.7,
+                    },
+                },
+                'lock_additional_fields': [],
+            },
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_create_lock_unset_fields_as_default',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={
+                'http_options': {'api_version': 'v1alpha'},
+                'uses': 2,
+                'live_connect_constraints': {
+                    'model': _MODEL,
+                    'config': {
+                        'response_modalities': ['TEXT'],
+                        'temperature': 0.7,
+                    },
+                },
+                'lock_additional_fields': ['output_audio_transcription'],
+            },
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_create_lock_additional_fields',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={
+                'http_options': {'api_version': 'v1alpha'},
+                'uses': 2,
+                'live_connect_constraints': {
+                    'model': _MODEL,
+                    'config': {
+                        'response_modalities': ['TEXT'],
+                        'temperature': 0.7,
+                    },
+                },
+                'lock_additional_fields': ['top_k'],
+            },
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_create_lock_with_no_params',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={
+                'http_options': {'api_version': 'v1alpha'},
+                'lock_additional_fields': ['output_audio_transcription'],
+            },
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_create_lock_with_empty_params',
+        parameters=types.CreateAuthTokenParameters(
+            # remove v1alpha after v1beta is available.
+            config={
+                'http_options': {'api_version': 'v1alpha'},
+                'lock_additional_fields': ['output_audio_transcription'],
+                'live_connect_constraints': {},
+            },
+        ),
+        exception_if_vertex='only supported in the Gemini Developer client',
+    ),
+]
+
+pytestmark = [
+    pytest_helper.setup(
+        file=__file__,
+        globals_for_file=globals(),
+        test_method='auth_tokens.create',
+        test_table=test_table,
+    ),
+]
+
+
+@pytest.mark.asyncio
+async def test_async_create_no_lock(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    config = types.CreateAuthTokenConfig(
+        http_options=types.HttpOptions(api_version='v1alpha')
+    )
+    token = await client.aio.auth_tokens.create(config=config)

google/genai/tests/transformers/__init__.py

@@ -0,0 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+
+"""Tests for the Google GenAI SDK's _transformers module."""

google/genai/tests/transformers/test_blobs.py

@@ -0,0 +1,84 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+
+"""Tests t_bytes methods in the _transformers module."""
+
+import os
+
+import PIL.Image
+import pytest
+
+from ... import _transformers as t
+from ... import types
+
+
+def test_blob_dict():
+  blob = t.t_blob({
+      'data': bytes([0, 0, 0, 0, 0, 0]),
+      'mime_type': 'audio/pcm',
+  })
+  assert blob.data == bytes([0, 0, 0, 0, 0, 0])
+  assert blob.mime_type == 'audio/pcm'
+
+
+def test_blob():
+  blob = t.t_blob(
+      types.Blob(data=bytes([0, 0, 0, 0, 0, 0]), mime_type='audio/pcm')
+  )
+  assert blob.data == bytes([0, 0, 0, 0, 0, 0])
+  assert blob.mime_type == 'audio/pcm'
+
+
+def test_image(image_jpeg):
+  blob = t.t_blob(image_jpeg)
+  assert blob.data[6:10] == b'JFIF'
+  assert blob.mime_type == 'image/jpeg'
+
+  round_trip_image = blob.as_image()._pil_image
+  assert round_trip_image is not None
+  assert round_trip_image.size == image_jpeg.size
+  assert round_trip_image.mode == image_jpeg.mode
+  assert round_trip_image.format == image_jpeg.format
+
+
+def test_not_image():
+  blob = types.Blob(data=bytes([0, 0, 0, 0, 0, 0]), mime_type='audio/pcm')
+  assert blob.as_image() is None
+
+
+def test_part_image(image_jpeg):
+  part = t.t_part(image_jpeg)
+  assert part.inline_data.data[6:10] == b'JFIF'
+  assert part.inline_data.mime_type == 'image/jpeg'
+
+  round_trip_image = part.as_image()._pil_image
+  assert round_trip_image is not None
+  assert round_trip_image.size == image_jpeg.size
+  assert round_trip_image.mode == image_jpeg.mode
+  assert round_trip_image.format == image_jpeg.format
+
+
+def test_part_not_image():
+  part = t.t_part('hello world')
+  assert part.as_image() is None
+
+
+def test_pil_to_blob_with_memory_pil_image():
+  img = PIL.Image.new('RGB', (1, 1), color='red')
+  blob = t.pil_to_blob(img)
+  assert blob.mime_type == 'image/png'
+  assert blob.data and len(blob.data) == 69
+  assert blob.data[0:4] == b'\x89PNG'

google/genai/tests/transformers/test_bytes.py

@@ -0,0 +1,15 @@
+"""Tests t_bytes methods in the _transformers module."""
+
+import base64
+
+from ... import _base_transformers as t
+
+_RAW_BYTES = (
+    b'\xfb\xf6\x9bq\xd7\x9f\x82\x18\xa3\x92Y\xa7\xa2\x9a\xab\xb2\xdb\xaf\xc3\x1c\xb3\x00\x10\x83\x10Q\x87'
+    b' \x92\x8b0\xd3\x8fA\x14\x93QU\x97a\x9d5\xdb~9\xeb\xbf='
+)
+
+
+def test_t_bytes():
+  assert t.t_bytes(_RAW_BYTES) == base64.b64encode(_RAW_BYTES).decode('ascii')
+  assert t.t_bytes('string') == 'string'

google/genai/tests/transformers/test_duck_type.py

@@ -0,0 +1,96 @@
+import unittest
+
+import pydantic
+
+from ... import _transformers
+
+
+class TestIsDuckTypeOf(unittest.TestCase):
+
+  class FakePydanticModel(pydantic.BaseModel):
+    field1: str
+    field2: int
+    field3: str
+    field4: str
+    field5: str
+
+  class FakePydanticModelWithLessFields(pydantic.BaseModel):
+    field1: str
+    field2: int
+    field3: str
+    field4: str
+
+  def test_is_duck_type_of_true_for_pydantic_object(self):
+    obj = self.FakePydanticModel(
+        field1="a",
+        field2=1,
+        field3="b",
+        field4="c",
+        field5="d",
+    )
+    self.assertTrue(_transformers._is_duck_type_of(obj, self.FakePydanticModel))
+
+  def test_is_duck_type_of_true_for_duck_typed_object(self):
+    class DuckTypedObject:
+
+      def __init__(self):
+        self.field1 = "a"
+        self.field2 = 1
+        self.field3 = "b"
+        self.field4 = "c"
+        self.field5 = "d"
+
+    obj = DuckTypedObject()
+    self.assertTrue(_transformers._is_duck_type_of(obj, self.FakePydanticModel))
+
+  def test_is_duck_type_of_false_for_different_many_fields(self):
+    class DifferentFieldsObject:
+
+      def __init__(self):
+        self.fielda = "a"
+
+    obj = DifferentFieldsObject()
+    self.assertFalse(
+        _transformers._is_duck_type_of(obj, self.FakePydanticModel)
+    )
+
+  def test_is_duck_type_of_false_for_missing_fields(self):
+
+    obj = self.FakePydanticModelWithLessFields(
+        field1="a", field2=1, field3="b", field4="c"
+    )
+    self.assertFalse(_transformers._is_duck_type_of(obj, self.FakePydanticModel))
+
+  def test_is_duck_type_of_false_for_dict(self):
+    obj = {"field1": "a", "field2": 1}
+    self.assertFalse(
+        _transformers._is_duck_type_of(obj, self.FakePydanticModel)
+    )
+
+  def test_is_duck_type_of_false_for_non_pydantic_class(self):
+    class NonPydanticModel:
+      pass
+
+    class SomeObject:
+      pass
+
+    obj = SomeObject()
+    self.assertFalse(_transformers._is_duck_type_of(obj, NonPydanticModel))
+
+  def test_is_duck_type_of_true_with_extra_fields(self):
+    class ExtraFieldsObject:
+
+      def __init__(self):
+        self.field1 = "a"
+        self.field2 = 1
+        self.field3 = "extra"
+        self.field4 = "extra"
+        self.field5 = "extra"
+        self.field6 = "extra"
+
+    obj = ExtraFieldsObject()
+    self.assertTrue(_transformers._is_duck_type_of(obj, self.FakePydanticModel))
+
+
+if __name__ == "__main__":
+  unittest.main()

google/genai/tests/transformers/test_function_responses.py

@@ -0,0 +1,72 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+
+"""Tests for live.py."""
+from ... import types
+from ... import _transformers as t
+
+def test_function_response_dict():
+  input = {
+      'name': 'get_current_weather',
+      'response': {'temperature': 14.5, 'unit': 'C'},
+      'id': 'some-id',
+  }
+
+  function_responses = t.t_function_responses(input)
+
+  assert len(function_responses) == 1
+  assert function_responses[0].name == 'get_current_weather'
+  assert function_responses[0].response['temperature'] == 14.5
+  assert function_responses[0].response['unit'] == 'C'
+
+
+def test_send_function_response():
+  input = types.FunctionResponse(
+      name='get_current_weather',
+      response={'temperature': 14.5, 'unit': 'C'},
+      id='some-id',
+  )
+  
+  function_responses = t.t_function_responses(input)
+
+  assert len(function_responses) == 1
+  assert function_responses[0].name == 'get_current_weather'
+  assert function_responses[0].response['temperature'] == 14.5
+  assert function_responses[0].response['unit'] == 'C'
+
+
+def test_send_function_response_list():
+
+  input1 = {
+      'name': 'get_current_weather',
+      'response': {'temperature': 14.5, 'unit': 'C'},
+      'id': '1',
+  }
+  input2 = {
+      'name': 'get_current_weather',
+      'response': {'temperature': 99.9, 'unit': 'C'},
+      'id': '2',
+  }
+
+  function_responses = t.t_function_responses([input1, input2])
+
+  assert len(function_responses) == 2
+  assert function_responses[0].name == 'get_current_weather'
+  assert function_responses[0].response['temperature'] == 14.5
+  assert function_responses[0].response['unit'] == 'C'
+  assert function_responses[1].name == 'get_current_weather'
+  assert function_responses[1].response['temperature'] == 99.9
+  assert function_responses[1].response['unit'] == 'C'