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

google/genai/tests/models/test_generate_content_cached_content.py

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

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

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

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

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