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

google/genai/tests/files/test_upload.py

@@ -0,0 +1,255 @@
+# 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.
+#
+
+
+"""Test files upload method."""
+
+
+import io
+import pathlib
+import pytest
+from ... import types
+from ... import errors
+from .. import pytest_helper
+
+# Upload method is not pydantic.
+test_table: list[pytest_helper.TestTableItem] = []
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method='files.upload',
+    test_table=test_table,
+)
+
+
+def test_image_png_upload(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(file='tests/data/google.png')
+    assert file.name.startswith('files/')
+
+def test_image_png_upload_with_path(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    p = pathlib.Path('tests/data/google.png')
+    file = client.files.upload(
+        file=p,
+        config=types.UploadFileConfig(display_name='test_image_png_path'),
+    )
+    assert file.name.startswith('files/')
+
+def test_image_png_upload_with_bytesio(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    with open('tests/data/google.png', 'rb') as f:
+      with io.BytesIO(f.read()) as buffer:
+        file = client.files.upload(
+            file=buffer,
+            config=types.UploadFileConfig(mime_type='image/png'),
+        )
+
+        assert file.name.startswith('files/')
+
+def test_image_png_upload_with_fd(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    with open('tests/data/google.png', 'rb') as f:
+      file = client.files.upload(
+          file=f,
+          config=types.UploadFileConfig(mime_type='image/png'),
+      )
+
+    assert file.name.startswith('files/')
+
+def test_image_png_upload_with_config(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/google.png',
+        config=types.UploadFileConfig(display_name='test_image_png'),
+    )
+    assert file.name.startswith('files/')
+
+
+def test_image_png_upload_with_config_dict(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/google.png', config={'display_name': 'test_image_png'}
+    )
+    assert file.name.startswith('files/')
+
+
+def test_image_jpg_upload(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(file='tests/data/google.jpg')
+    assert file.name.startswith('files/')
+
+
+def test_image_jpg_upload_with_config(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/google.jpg',
+        config=types.UploadFileConfig(display_name='test_image_jpg'),
+    )
+    assert file.name.startswith('files/')
+
+
+def test_image_jpg_upload_with_config_dict(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/google.jpg', config={'display_name': 'test_image_jpg'}
+    )
+    assert file.name.startswith('files/')
+
+
+def test_application_pdf_file_upload(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(file='tests/data/story.pdf')
+    assert file.name.startswith('files/')
+
+
+def test_application_pdf_upload_with_config(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/story.pdf',
+        config=types.UploadFileConfig(display_name='test_application_pdf'),
+    )
+    assert file.name.startswith('files/')
+
+
+def test_application_pdf_upload_with_config_dict(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/story.pdf',
+        config={'display_name': 'test_application_pdf'},
+    )
+    assert file.name.startswith('files/')
+
+
+def test_video_mp4_file_upload(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(file='tests/data/animal.mp4')
+    assert file.name.startswith('files/')
+
+
+def test_video_mp4_upload_with_config(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/animal.mp4',
+        config=types.UploadFileConfig(display_name='test_video_mp4'),
+    )
+    assert file.name.startswith('files/')
+
+
+def test_video_mp4_upload_with_config_dict(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/animal.mp4', config={'display_name': 'test_video_mp4'}
+    )
+    assert file.name.startswith('files/')
+
+
+def test_audio_m4a_file_upload(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/pixel.m4a',
+        config=types.UploadFileConfig(mime_type='audio/mp4'),
+    )
+    assert file.name.startswith('files/')
+
+
+def test_audio_m4a_upload_with_config(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/pixel.m4a',
+        config=types.UploadFileConfig(
+            display_name='test_audio_m4a', mime_type='audio/mp4'
+        ),
+    )
+    assert file.name.startswith('files/')
+
+
+def test_audio_m4a_upload_with_config_dict(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(
+        file='tests/data/pixel.m4a',
+        config={'display_name': 'test_audio_m4a', 'mime_type': 'audio/mp4'},
+    )
+    assert file.name.startswith('files/')
+
+
+def test_bad_mime_type(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    with pytest.raises(errors.APIError, match="Unsupported MIME"):
+      file = client.files.upload(
+          file=io.BytesIO(b'test'),
+          config={'mime_type': 'bad/mime_type'},
+      )
+
+
+@pytest.mark.asyncio
+async def test_image_upload_async(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = await client.aio.files.upload(file='tests/data/google.png')
+    assert file.name.startswith('files/')
+
+
+@pytest.mark.asyncio
+async def test_image_upload_with_config_async(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = await client.aio.files.upload(
+        file='tests/data/google.png',
+        config=types.UploadFileConfig(display_name='test_image'),
+    )
+    assert file.name.startswith('files/')
+
+
+@pytest.mark.asyncio
+async def test_image_upload_with_config_dict_async(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = await client.aio.files.upload(
+        file='tests/data/google.png',
+        config={
+            'display_name': 'test_image',
+            'http_options': {'timeout': '8000'},
+        },
+    )
+    assert file.name.startswith('files/')
+
+
+@pytest.mark.asyncio
+async def test_image_upload_with_bytesio_async(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    with open('tests/data/google.png', 'rb') as f:
+      buffer = io.BytesIO(f.read())
+    file = await client.aio.files.upload(
+        file=buffer,
+        config=types.UploadFileConfig(
+            mime_type='image/png'),
+    )
+    assert file.name.startswith('files/')
+
+
+@pytest.mark.asyncio
+async def test_unknown_path_upload_async(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    try:
+      await client.aio.files.upload(file='unknown_path')
+    except FileNotFoundError as e:
+      assert 'is not a valid file path' in str(e)
+
+@pytest.mark.asyncio
+async def test_bad_mime_type_async(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    with pytest.raises(errors.APIError, match="Unsupported MIME"):
+      file = await client.aio.files.upload(
+          file=io.BytesIO(b'test'),
+          config={'mime_type': 'bad/mime_type'},
+      )

google/genai/tests/imports/test_no_optional_imports.py

@@ -0,0 +1,28 @@
+# 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 os
+import pytest
+
+IS_NOT_GITHUB_ACTIONS = os.getenv('GITHUB_ACTIONS') != 'true'
+
+
+@pytest.mark.skipif(IS_NOT_GITHUB_ACTIONS,
+                    reason='This test is only run on GitHub Actions.')
+def test_library_can_be_imported_without_optional_dependencies():
+  """Tests that the library can be imported without optional dependencies.
+  """
+  from google import genai
+  from google.genai import types

google/genai/tests/interactions/test_integration.py

@@ -0,0 +1,80 @@
+# 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 unittest import mock
+import pytest
+from ... import client as client_lib
+
+pytest_plugins = ("pytest_asyncio",)
+
+
+def test_client_future_warning():
+  with mock.patch.object(
+      client_lib, "_interactions_experimental_warned", new=False
+  ):
+    client = client_lib.Client(
+        api_key="placeholder",
+        http_options={
+            "api_version": "v1alpha",
+        }
+    )
+    with pytest.warns(
+        UserWarning, match="Interactions.*experimental"
+    ):
+      _ = client.interactions
+
+
+def test_client_timeout():
+  with mock.patch.object(
+      client_lib, "GeminiNextGenAPIClient", spec_set=True
+  ) as mock_nextgen_client:
+
+    client = client_lib.Client(
+        api_key="placeholder",
+        http_options={"api_version": "v1alpha", "timeout": 5000},
+    )
+
+    _ = client.interactions
+
+    mock_nextgen_client.assert_called_once_with(
+        base_url=mock.ANY,
+        api_key="placeholder",
+        default_headers=mock.ANY,
+        http_client=mock.ANY,
+        timeout=5.0,
+        max_retries=mock.ANY,
+    )
+
+
+@pytest.mark.asyncio
+async def test_async_client_timeout():
+  with mock.patch.object(
+      client_lib, "AsyncGeminiNextGenAPIClient", spec_set=True
+  ) as mock_nextgen_client:
+
+    client = client_lib.Client(
+        api_key="placeholder",
+        http_options={"api_version": "v1alpha", "timeout": 5000},
+    )
+
+    _ = client.aio.interactions
+
+    mock_nextgen_client.assert_called_once_with(
+        base_url=mock.ANY,
+        api_key="placeholder",
+        default_headers=mock.ANY,
+        http_client=mock.ANY,
+        timeout=5.0,
+        max_retries=mock.ANY,
+    )

google/genai/tests/live/__init__.py

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