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

google/genai/tests/shared/batches/test_create_get_cancel.py

@@ -1,56 +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 batches.create(), batches.get(), batches.cancel()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-def create_get_cancel(client, parameters):
-  batch_job = client.batches.create(
-      model=parameters.model,
-      src=parameters.src,
-      config=parameters.config,
-  )
-  batch_job = client.batches.get(name=batch_job.name)
-  client.batches.cancel(name=batch_job.name)
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name="test_create_get_cancel_mldev",
-        parameters=genai_types._CreateBatchJobParameters(
-            model="gemini-2.5-flash",
-            src=[{
-                "contents": [{
-                    "parts": [{"text": "Why is the sky blue?"}],
-                    "role": "user",
-                }]
-            }],
-        ),
-        exception_if_vertex="not supported in Vertex AI.",
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method="create_get_cancel",
-    test_table=test_table,
-)
-
-pytest_plugins = ("pytest_asyncio",)

google/genai/tests/shared/batches/test_list.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.
-#
-
-"""Tests for batches.list()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_list_batch_jobs',
-        parameters=genai_types._ListBatchJobsParameters(
-            config=genai_types.ListBatchJobsConfig(
-                page_size=2,
-            ),
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='batches.list',
-    test_table=test_table,
-)
-
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/shared/caches/__init__.py

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

google/genai/tests/shared/caches/test_create_get_delete.py

@@ -1,67 +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 caches.create(), caches.get(), and caches.delete()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-def create_get_delete(client, parameters):
-  if client.vertexai:
-    cache = client.caches.create(
-        model=parameters.model,
-        config=parameters.config,
-    )
-  else:
-    # For MLDev, we must create the file before creating the cache.
-    file = client.files.upload(file="tests/data/a-man-and-a-dog.png")
-    config = genai_types.CreateCachedContentConfig(
-        contents=[
-            genai_types.Part.from_uri(
-                file_uri=file.uri,
-                mime_type=file.mime_type,
-            )
-        ]
-        * 5
-    )
-    cache = client.caches.create(model=parameters.model, config=config)
-  got_cache = client.caches.get(name=cache.name)
-  client.caches.delete(name=got_cache.name)
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name="test_create_get_delete",
-        parameters=genai_types._CreateCachedContentParameters(
-            model="gemini-2.5-flash",
-            config=genai_types.CreateCachedContentConfig(
-                contents=[genai_types.Part.from_uri(
-                    file_uri="gs://cloud-samples-data/generative-ai/image/a-man-and-a-dog.png",
-                    mime_type="image/png",
-                )] * 5
-            ),
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method="create_get_delete",
-    test_table=test_table,
-)
-
-pytest_plugins = ("pytest_asyncio",)

google/genai/tests/shared/caches/test_create_update_get.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 for caches.create(), caches.update(), and caches.get()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-def create_update_get(client, parameters):
-  if client.vertexai:
-    cache = client.caches.create(
-        model=parameters.model,
-        config=parameters.config,
-    )
-  else:
-    # For MLDev, we must create the file before creating the cache.
-    file = client.files.upload(file="tests/data/a-man-and-a-dog.png")
-    config = genai_types.CreateCachedContentConfig(
-        contents=[
-            genai_types.Part.from_uri(
-                file_uri=file.uri,
-                mime_type=file.mime_type,
-            )
-        ]
-        * 5
-    )
-    cache = client.caches.create(model=parameters.model, config=config)
-
-  updated_cache = client.caches.update(
-      name=cache.name,
-      config=genai_types.UpdateCachedContentConfig(ttl="7200s"),
-  )
-  _ = client.caches.get(name=updated_cache.name)
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name="test_create_update_get",
-        parameters=genai_types._CreateCachedContentParameters(
-            model="gemini-2.5-flash",
-            config=genai_types.CreateCachedContentConfig(
-                contents=[genai_types.Part.from_uri(
-                    file_uri="gs://cloud-samples-data/generative-ai/image/a-man-and-a-dog.png",
-                    mime_type="image/png",
-                )] * 5
-            ),
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method="create_update_get",
-    test_table=test_table,
-)
-
-pytest_plugins = ("pytest_asyncio",)

google/genai/tests/shared/caches/test_list.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.
-#
-
-"""Tests for caches.list()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_list_cached_contents',
-        parameters=genai_types._ListCachedContentsParameters(
-            config=genai_types.ListCachedContentsConfig(
-                page_size=2,
-            ),
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='caches.list',
-    test_table=test_table,
-)
-
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/shared/chats/__init__.py

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

google/genai/tests/shared/chats/test_send_message.py

@@ -1,48 +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.
-#
-
-"""Table tests for chats.send_message()."""
-
-from ... import pytest_helper
-from pydantic import BaseModel
-
-
-class _SendMessageParameters(BaseModel):
-  model: str
-  message: str
-
-
-def send_message(client, parameters):
-  chat = client.chats.create(model=parameters.model)
-  chat.send_message(parameters.message)
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name="test_send_message",
-        parameters=_SendMessageParameters(
-            model="gemini-2.5-flash",
-            message="Hello",
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method="send_message",
-    test_table=test_table,
-)
-
-pytest_plugins = ("pytest_asyncio",)

google/genai/tests/shared/chats/test_send_message_stream.py

@@ -1,50 +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.
-#
-
-"""Table tests for chats.send_message_stream()."""
-
-from ... import pytest_helper
-from pydantic import BaseModel
-
-
-class _SendMessageStreamParameters(BaseModel):
-    model: str
-    message: str
-
-
-def send_message_stream(client, parameters):
-  chat = client.chats.create(model=parameters.model)
-  response = chat.send_message_stream(parameters.message)
-  for _ in response:
-    pass
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name="test_send_message_stream",
-        parameters=_SendMessageStreamParameters(
-            model="gemini-2.5-flash",
-            message="Tell a joke.",
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method="send_message_stream",
-    test_table=test_table,
-)
-
-pytest_plugins = ("pytest_asyncio",)

google/genai/tests/shared/files/__init__.py

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

google/genai/tests/shared/files/test_list.py

@@ -1,41 +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 files.list()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_list_files',
-        parameters=genai_types._ListFilesParameters(
-            config=genai_types.ListFilesConfig(
-                page_size=2,
-            ),
-        ),
-        exception_if_vertex='only supported in the Gemini Developer client',
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='files.list',
-    test_table=test_table,
-)
-
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/shared/files/test_upload_get_delete.py

@@ -1,54 +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 files.upload(), files.get(), and files.delete()."""
-
-from pydantic import BaseModel
-from ... import pytest_helper
-
-
-class _UploadGetDeleteParameters(BaseModel):
-  file_path: str
-
-
-def upload_get_delete(client, parameters):
-  file = None
-  try:
-    file = client.files.upload(file=parameters.file_path)
-    assert file.name
-    got_file = client.files.get(name=file.name)
-    assert got_file.name == file.name
-  finally:
-    if file and file.name:
-      client.files.delete(name=file.name)
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name="test_upload_get_delete_image",
-        parameters=_UploadGetDeleteParameters(
-            file_path="tests/data/google.png",
-        ),
-        exception_if_vertex='only supported in the Gemini Developer client',
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method="upload_get_delete",
-    test_table=test_table,
-)
-
-pytest_plugins = ("pytest_asyncio",)

google/genai/tests/shared/models/__init__.py

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

google/genai/tests/shared/models/test_compute_tokens.py

@@ -1,41 +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 models.compute_tokens()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_compute_tokens',
-        parameters=genai_types._ComputeTokensParameters(
-            model='gemini-2.5-flash',
-            contents='The quick brown fox jumps over the lazy dog.',
-        ),
-        exception_if_mldev='only supported in the Vertex AI',
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.compute_tokens',
-    test_table=test_table,
-)
-
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/shared/models/test_count_tokens.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.
-#
-
-
-"""Tests for models.count_tokens()."""
-
-from .... import types as genai_types
-from ... import pytest_helper
-
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_count_tokens',
-        parameters=genai_types._CountTokensParameters(
-            model='gemini-2.5-flash',
-            contents='The quick brown fox jumps over the lazy dog.',
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.count_tokens',
-    test_table=test_table,
-)
-
-pytest_plugins = ('pytest_asyncio',)

google/genai/tests/shared/models/test_edit_image.py

@@ -1,67 +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 edit_image."""
-
-import os
-
-from .... import types
-from ... import pytest_helper
-
-CAPABILITY_MODEL_NAME = 'imagen-3.0-capability-001'
-
-IMAGE_FILE_PATH = os.path.abspath(
-    os.path.join(os.path.dirname(__file__), '../../data/google.png')
-)
-
-MASK_FILE_PATH = os.path.abspath(
-    os.path.join(os.path.dirname(__file__), '../../data/checkerboard.png')
-)
-
-raw_ref_image = types.RawReferenceImage(
-    reference_id=1,
-    reference_image=types.Image.from_file(location=IMAGE_FILE_PATH),
-)
-
-mask_ref_image = types.MaskReferenceImage(
-    reference_id=2,
-    config=types.MaskReferenceConfig(
-        mask_mode='MASK_MODE_BACKGROUND',
-        mask_dilation=0.06,
-    ),
-)
-
-test_table: list[pytest_helper.TestTableItem] = [
-    pytest_helper.TestTableItem(
-        name='test_edit_mask_inpaint_insert',
-        exception_if_mldev='only supported in the Vertex AI client',
-        parameters=types._EditImageParameters(
-            model=CAPABILITY_MODEL_NAME,
-            prompt='Sunlight and clear weather',
-            reference_images=[raw_ref_image, mask_ref_image],
-            config=types.EditImageConfig(
-                edit_mode=types.EditMode.EDIT_MODE_INPAINT_INSERTION,
-            ),
-        ),
-    ),
-]
-
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-    test_method='models.edit_image',
-    test_table=test_table,
-)