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

google/genai/tests/models/test_generate_content_cached_content.py

@@ -1,132 +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.
-#
-
-
-from ... import _transformers as t
-from ... import types
-from .. import pytest_helper
-
-VERTEX_CACHED_CONTENT_NAME = '8163416782997028864'
-MLDEV_CACHED_CONTENT_NAME = 'idv6zldt3p2f1lbm31vn41tx2ld42qngpcnka2s3'
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_cached_content_wrong_name',
-        exception_if_vertex='INVALID_ARGUMENT',
-        exception_if_mldev='INVALID_ARGUMENT',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is in these docs?'),
-            config={
-                'cached_content': 'batchPredictionJobs/123',
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_cached_content_partial_vertex_resource_name_0',
-        exception_if_mldev='INVALID_ARGUMENT',
-        skip_in_api_mode=(
-            'CachedContent API has expiration and permission issues'
-        ),
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is in these docs?'),
-            config={
-                'cached_content': f'locations/us-central1/cachedContents/{VERTEX_CACHED_CONTENT_NAME}',
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_cached_content_partial_vertex_resource_name_1',
-        exception_if_mldev='PERMISSION_DENIED',
-        skip_in_api_mode=(
-            'CachedContent API has expiration and permission issues'
-        ),
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is in these docs?'),
-            config={
-                'cached_content': (
-                    f'cachedContents/{VERTEX_CACHED_CONTENT_NAME}'
-                ),
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_cached_content_partial_vertex_resource_name_2',
-        exception_if_mldev='PERMISSION_DENIED',
-        skip_in_api_mode=(
-            'CachedContent API has expiration and permission issues'
-        ),
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is in these docs?'),
-            config={
-                'cached_content': VERTEX_CACHED_CONTENT_NAME,
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_cached_content_partial_mldev_resource_name_1',
-        exception_if_vertex='NOT_FOUND',
-        skip_in_api_mode=(
-            'CachedContent API has expiration and permission issues'
-        ),
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is in these docs?'),
-            config={
-                'cached_content': f'{MLDEV_CACHED_CONTENT_NAME}',
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_cached_content_for_vertex',
-        exception_if_mldev='INVALID_ARGUMENT',
-        skip_in_api_mode=(
-            'CachedContent API has expiration and permission issues'
-        ),
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is in these docs?'),
-            config={
-                'cached_content': f'projects/964831358985/locations/us-central1/cachedContents/{VERTEX_CACHED_CONTENT_NAME}',
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_cached_content_for_mldev',
-        exception_if_vertex='NOT_FOUND',
-        skip_in_api_mode=(
-            'CachedContent API has expiration and permission issues'
-        ),
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('Tell me a story in 300 words.'),
-            config={
-                'cached_content': f'cachedContents/{MLDEV_CACHED_CONTENT_NAME}',
-            },
-        ),
-    ),
-]
-
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.generate_content',
-    test_table=test_table,
-)
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/models/test_generate_content_config_zero_value.py

@@ -1,103 +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.
-#
-
-
-from pydantic import BaseModel, ValidationError
-import pytest
-from ... import _transformers as t
-from ... import errors
-from ... import types
-from .. import pytest_helper
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_candidate_count_zero',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is your name?'),
-            config={
-                'candidate_count': 0,
-            },
-        ),
-        exception_if_mldev='400'
-    ),
-    pytest_helper.TestTableItem(
-        name='test_max_output_tokens_zero',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is your name?'),
-            config={
-                'max_output_tokens': 0,
-            },
-        ),
-        exception_if_vertex='400',
-        exception_if_mldev='400',
-    ),
-    pytest_helper.TestTableItem(
-        name='test_logprobs_zero',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is your name?'),
-            config={
-                'logprobs': 0,
-            },
-        ),
-        exception_if_mldev='response_logprobs is true',
-        exception_if_vertex='setting response_logprobs to be true',
-        skip_in_api_mode='it will encounter 400 for api mode',
-    ),
-    pytest_helper.TestTableItem(
-        name='test_logprobs_zero_with_response_logprobs_true',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is your name?'),
-            config={
-                'response_logprobs': True,
-                'logprobs': 0,
-            },
-        ),
-        exception_if_mldev=(
-            'Logprobs is not enabled for models/gemini-2.5-flash'
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_presence_penalty_zero',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is your name?'),
-            config={
-                'presence_penalty': 0,
-            },
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_frequency_penalty_zero',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash',
-            contents=t.t_contents('What is your name?'),
-            config={
-                'frequency_penalty': 0,
-            },
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.generate_content',
-    test_table=test_table,
-)
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/models/test_generate_content_from_apikey.py

@@ -1,44 +0,0 @@
-import pytest
-from .. import pytest_helper
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.generate_content',
-)
-
-
-def test_simple_request(client):
-  # TODO(b/388917450): Add Vertex AI in Express mode test suite
-  client._api_client.project = None
-  client._api_client.location = None
-
-  # To record a replay file, replace with api key (from Vertex AI Express).
-  # API mode will not work if the API key is a ML Dev API key.
-  # After recording, change the string back to 'key'.
-  client._api_client._http_options.headers['x-goog-api-key'] = 'key'
-  if not client._api_client.vertexai:
-    return
-  response = client.models.generate_content(
-      model='gemini-2.0-flash-001', contents='Tell me a joke.'
-  )
-  assert response.text
-
-
-def test_simple_request_stream(client):
-  # TODO(b/388917450): Add Vertex AI in Express mode test suite
-  client._api_client.project = None
-  client._api_client.location = None
-
-  # To record a replay file, replace with api key (from Vertex AI Express).
-  # API mode will not work if the API key is a ML Dev API key.
-  # After recording, change the string back to 'key'.
-  client._api_client._http_options.headers['x-goog-api-key'] = 'key'
-  if not client._api_client.vertexai:
-    return
-
-  response = client.models.generate_content_stream(
-      model='gemini-2.0-flash-001', contents='Tell me a joke.'
-  )
-
-  assert any(chunk.text for chunk in response)

google/genai/tests/models/test_generate_content_http_options.py

@@ -1,40 +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.
-#
-
-
-from ... import _transformers as t
-from ... import types
-from .. import pytest_helper
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_generate_content_with_http_options',
-        parameters=types._GenerateContentParameters(
-            model='models/gemini-2.5-flash',
-            contents=t.t_contents('how are you doing?'),
-        ),
-        exception_if_vertex='404',
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.generate_content',
-    test_table=test_table,
-    http_options={
-        'api_version': 'v1alpha',
-    },
-)

google/genai/tests/models/test_generate_content_image_generation.py

@@ -1,143 +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.
-#
-
-import pydantic
-import pytest
-
-from ... import _api_client
-from ... import _transformers as t
-from ... import errors
-from ... import types
-from .. import pytest_helper
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_image_generation_config',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash-image',
-            contents=t.t_contents('A photorealistic red apple on a table.'),
-            config=types.GenerateContentConfig(
-                response_modalities=['IMAGE'],
-                image_config=types.ImageConfig(
-                    aspect_ratio='16:9',
-                )
-            ),
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_image_generation_config_output_mime_type',
-        exception_if_mldev='not supported in Gemini API',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash-image',
-            contents=t.t_contents('A photorealistic red apple on a table.'),
-            config=types.GenerateContentConfig(
-                response_modalities=['IMAGE'],
-                image_config=types.ImageConfig(
-                    aspect_ratio='16:9',
-                    output_mime_type='image/jpeg',
-                )
-            ),
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_image_generation_config_output_compression_quality',
-        exception_if_mldev='not supported in Gemini API',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash-image',
-            contents=t.t_contents('A photorealistic red apple on a table.'),
-            config=types.GenerateContentConfig(
-                response_modalities=['IMAGE'],
-                image_config=types.ImageConfig(
-                    aspect_ratio='16:9',
-                    output_compression_quality=80,
-                )
-            ),
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_image_generation_filtered',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash-image',
-            contents=t.t_contents('Make a zombie anime style'),
-            config=types.GenerateContentConfig(
-                response_modalities=['IMAGE'],
-                image_config=types.ImageConfig(
-                    aspect_ratio='16:9',
-                )
-            ),
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_image_generation_no_image',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash-image',
-            contents=t.t_contents('What is your name?'),
-            config=types.GenerateContentConfig(
-                response_modalities=['IMAGE'],
-                image_config=types.ImageConfig(
-                    aspect_ratio='16:9',
-                )
-            ),
-        ),
-    ),
-    pytest_helper.TestTableItem(
-        name='test_image_generation_config_validation_none',
-        parameters=types._GenerateContentParameters(
-            model='gemini-2.5-flash-image',
-            contents=t.t_contents('A photorealistic red apple on a table.'),
-            config=types.GenerateContentConfig(
-                response_modalities=['IMAGE'],
-                image_config=None
-            ),
-        ),
-    ),
-]
-
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.generate_content',
-    test_table=test_table,
-)
-
-
-def test_image_generation_wrong_config(client):
-  with pytest.raises(pydantic.ValidationError):
-    client.models.generate_content(
-        model='gemini-2.5-flash-image',
-        contents=t.t_contents('A photorealistic red apple on a table.'),
-        config=types.GenerateContentConfig(
-            response_modalities=['IMAGE'],
-            image_config=types.GenerateImagesConfig(
-                aspect_ratio='16:9',
-                number_of_images=1,
-            )
-        ),
-    )
-
-
-def test_image_generation_validation_model_dump(client):
-  config = types.GenerateContentConfig(
-      response_modalities=['IMAGE'],
-  )
-
-  class Foo(pydantic.BaseModel):
-    value: types.GenerateContentConfig
-
-  f = Foo(value=config)
-  in_memory = f.model_dump(mode='json')
-  Foo.model_validate(in_memory)