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

google/genai/tests/tokens/test_create.py

@@ -1,154 +0,0 @@
-# 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

@@ -1,17 +0,0 @@
-# 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

@@ -1,71 +0,0 @@
-# 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

google/genai/tests/transformers/test_bytes.py

@@ -1,15 +0,0 @@
-"""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

@@ -1,96 +0,0 @@
-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

@@ -1,72 +0,0 @@
-# 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'