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

google/genai/tests/models/test_generate_content_mcp.py

@@ -0,0 +1,343 @@
+# 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 typing import Any
+import pytest
+from ... import _transformers as t
+from ... import errors
+from ... import types
+from .. import pytest_helper
+
+try:
+  from mcp import types as mcp_types
+  from mcp import ClientSession as McpClientSession
+except ImportError as e:
+  import sys
+
+  if sys.version_info < (3, 10):
+    raise ImportError(
+        'MCP Tool requires Python 3.10 or above. Please upgrade your Python'
+        ' version.'
+    ) from e
+  else:
+    raise e
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method='models.generate_content',
+)
+
+
+# Cannot be included in test_table because MCP integration is only supported
+# in the Python SDK.
+@pytest.mark.asyncio
+async def test_mcp_tools_async(client):
+  response = await client.aio.models.generate_content(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is the weather in Boston?'),
+      config={
+          'tools': [
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              )
+          ],
+      },
+  )
+  assert response.function_calls == [
+      types.FunctionCall(
+          name='get_weather',
+          args={'location': 'Boston'},
+      )
+  ]
+
+
+@pytest.mark.asyncio
+async def test_mcp_tools_with_custom_headers_async(client):
+  config = {
+      'http_options': {
+          'headers': {
+              'x-goog-api-client': 'google-genai-sdk/1.0.0 gl-python/1.0.0',
+          },
+      },
+      'tools': [
+          mcp_types.Tool(
+              name='get_weather',
+              description='Get the weather in a city.',
+              inputSchema={
+                  'type': 'object',
+                  'properties': {'location': {'type': 'string'}},
+              },
+          )
+      ],
+  }
+  response = await client.aio.models.generate_content(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is the weather in Boston?'),
+      config=config,
+  )
+  assert response.function_calls == [
+      types.FunctionCall(
+          name='get_weather',
+          args={'location': 'Boston'},
+      )
+  ]
+  # Assert config is not modified.
+  assert config['http_options']['headers'] == {
+      'x-goog-api-client': 'google-genai-sdk/1.0.0 gl-python/1.0.0'
+  }
+
+
+@pytest.mark.asyncio
+async def test_mcp_tools_subsequent_calls_async(client):
+  class MockMcpClientSession(McpClientSession):
+
+    def __init__(self):
+      self._read_stream = None
+      self._write_stream = None
+
+    async def list_tools(self):
+      return mcp_types.ListToolsResult(
+          tools=[
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+              mcp_types.Tool(
+                  name='add_numbers',
+                  description='Add two numbers together.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {
+                          'a': {'type': 'number'},
+                          'b': {'type': 'number'},
+                      },
+                  },
+              ),
+          ]
+      )
+
+    async def call_tool(
+        self,
+        name: str,
+        arguments: dict[str, Any],
+    ):
+      if name == 'get_weather':
+        return mcp_types.CallToolResult(
+            content=[mcp_types.TextContent(type='text', text='Sunny')]
+        )
+      else:
+        return mcp_types.CallToolResult(
+            content=[mcp_types.TextContent(type='text', text='100')]
+        )
+
+  config = {
+      'tools': [MockMcpClientSession()],
+  }
+
+  response = await client.aio.models.generate_content(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is the weather in Boston?'),
+      config=config,
+  )
+  assert 'sunny' in response.text.lower()
+
+  response_2 = await client.aio.models.generate_content(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is 50 + 50?'),
+      config=config,
+  )
+  assert '100' in response_2.text
+
+
+@pytest.mark.asyncio
+async def test_mcp_tools_duplicate_tool_name_raises_error(client):
+  class MockMcpClientSession(McpClientSession):
+
+    def __init__(self):
+      self._read_stream = None
+      self._write_stream = None
+
+    async def list_tools(self):
+      return mcp_types.ListToolsResult(
+          tools=[
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Different tool to get the weather.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+          ]
+      )
+
+  with pytest.raises(ValueError):
+    await client.aio.models.generate_content(
+        model='gemini-2.5-flash',
+        contents=t.t_contents('What is the weather in Boston?'),
+        config={
+            'tools': [MockMcpClientSession()],
+        },
+    )
+
+
+def test_mcp_tools_synchronous_call(client):
+  response = client.models.generate_content(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is the weather in Boston?'),
+      config={
+          'tools': [
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              )
+          ]
+      },
+  )
+  assert response.function_calls == [
+      types.FunctionCall(
+          name='get_weather',
+          args={'location': 'Boston'},
+      )
+  ]
+
+
+def test_mcp_session_synchronous_call_raises_error(client):
+  class MockMcpClientSession(McpClientSession):
+
+    def __init__(self):
+      self._read_stream = None
+      self._write_stream = None
+
+    async def list_tools(self):
+      return mcp_types.ListToolsResult(
+          tools=[
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Different tool to get the weather.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+          ]
+      )
+
+  with pytest.raises(errors.UnsupportedFunctionError):
+    client.models.generate_content(
+        model='gemini-2.5-flash',
+        contents=t.t_contents('What is the weather in Boston?'),
+        config={
+            'tools': [MockMcpClientSession()],
+        },
+    )
+
+
+def test_mcp_tools_synchronous_stream_call(client):
+  response = client.models.generate_content_stream(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is the weather in Boston?'),
+      config={
+          'tools': [
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              )
+          ]
+      },
+  )
+  for chunk in response:
+    assert chunk.function_calls == [
+        types.FunctionCall(
+            name='get_weather',
+            args={'location': 'Boston'},
+        )
+    ]
+
+
+def test_mcp_session_synchronous_stream_call_raises_error(client):
+  class MockMcpClientSession(McpClientSession):
+
+    def __init__(self):
+      self._read_stream = None
+      self._write_stream = None
+
+    async def list_tools(self):
+      return mcp_types.ListToolsResult(
+          tools=[
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Get the weather in a city.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+              mcp_types.Tool(
+                  name='get_weather',
+                  description='Different tool to get the weather.',
+                  inputSchema={
+                      'type': 'object',
+                      'properties': {'location': {'type': 'string'}},
+                  },
+              ),
+          ]
+      )
+
+  response = client.models.generate_content_stream(
+      model='gemini-2.5-flash',
+      contents=t.t_contents('What is the weather in Boston?'),
+      config={
+          'tools': [MockMcpClientSession()],
+      },
+  )
+
+  with pytest.raises(errors.UnsupportedFunctionError):
+    for chunk in response:
+      pass

google/genai/tests/models/test_generate_content_media_resolution.py

@@ -0,0 +1,97 @@
+# 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 types
+from .. import pytest_helper
+
+test_table: list[pytest_helper.TestTableItem] = [
+    pytest_helper.TestTableItem(
+        name='test_video_audio_uri_with_media_resolution',
+        parameters=types._GenerateContentParameters(
+            model='gemini-2.5-flash',
+            contents=[
+                types.Content(
+                    role='user',
+                    parts=[
+                        types.Part.from_text(
+                            text=(
+                                'Is the audio related to the video? '
+                                'If so, how? '
+                                'What are the common themes? '
+                                'What are the different emphases?'
+                            )
+                        )
+                    ],
+                ),
+                types.Content(
+                    role='user',
+                    parts=[
+                        types.Part.from_uri(
+                            file_uri='gs://cloud-samples-data/generative-ai/video/pixel8.mp4',
+                            mime_type='video/mp4',
+                        )
+                    ],
+                ),
+                types.Content(
+                    role='user',
+                    parts=[
+                        types.Part.from_uri(
+                            file_uri='gs://cloud-samples-data/generative-ai/audio/pixel.mp3',
+                            mime_type='audio/mpeg',
+                        )
+                    ],
+                ),
+            ],
+            config={
+                'system_instruction': types.Content(
+                    role='user',
+                    parts=[
+                        types.Part.from_text(
+                            text=(
+                                'you are a helpful assistant for people with '
+                                'visual and hearing disabilities.'
+                            )
+                        )
+                    ],
+                ),
+                'media_resolution': 'MEDIA_RESOLUTION_LOW',
+            },
+        ),
+        exception_if_mldev='400',
+    )
+]
+
+
+def test_low_media_resolution(client):
+  with pytest_helper.exception_if_vertex(client, ValueError):
+    file = client.files.upload(file='tests/data/google.png')
+    response = client.models.generate_content(
+        model='gemini-2.5-flash',
+        contents=[file, 'Describe the image.'],
+        config=types.GenerateContentConfig(
+            media_resolution='MEDIA_RESOLUTION_LOW',
+            http_options= types.HttpOptions(api_version='v1alpha', base_url='https://generativelanguage.googleapis.com')
+            ),
+        )
+    assert response.text
+
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method='models.generate_content',
+    test_table=test_table,
+)

google/genai/tests/models/test_generate_content_model.py

@@ -0,0 +1,139 @@
+# 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 pytest
+
+from ... import _transformers as t
+from ... import errors
+from ... import types
+from .. import pytest_helper
+
+tuned_model_endpoint = (
+    'projects/801452371447/locations/us-central1/endpoints/4095574160837705728'
+)
+
+
+test_table: list[pytest_helper.TestTableItem] = [
+    pytest_helper.TestTableItem(
+        name='test_tuned_model',
+        parameters=types._GenerateContentParameters(
+            model=tuned_model_endpoint,
+            contents=t.t_contents('Tell me a story in 300 words.'),
+        ),
+        exception_if_mldev='404',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_start_with_publishers',
+        parameters=types._GenerateContentParameters(
+            model='publishers/google/models/gemini-2.5-flash',
+            contents=t.t_contents('Tell me a story in 50 words.'),
+        ),
+        exception_if_mldev='404',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_start_with_models',
+        parameters=types._GenerateContentParameters(
+            model='models/gemini-2.5-flash',
+            contents=t.t_contents('Tell me a story in 50 words.'),
+        ),
+        exception_if_vertex='404',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_publisher_model',
+        parameters=types._GenerateContentParameters(
+            model='google/gemini-2.5-flash',
+            contents=t.t_contents('Tell me a story in 50 words.'),
+        ),
+        exception_if_mldev='404',
+    ),
+    pytest_helper.TestTableItem(
+        name='test_empty_model',
+        parameters=types._GenerateContentParameters(
+            model='',
+            contents=t.t_contents('Tell me a story in 50 words.'),
+        ),
+        exception_if_mldev='model',
+        exception_if_vertex='model',
+    ),
+]
+
+pytestmark = pytest_helper.setup(
+    file=__file__,
+    globals_for_file=globals(),
+    test_method='models.generate_content',
+    test_table=test_table,
+)
+pytest_plugins = ('pytest_asyncio',)
+
+
+def test_tuned_model_stream(client):
+  # Vertex AI endpoints is not supported in MLDev.
+  with pytest_helper.exception_if_mldev(client, errors.ClientError):
+    chunks = 0
+    for chunk in client.models.generate_content_stream(
+        model=tuned_model_endpoint,
+        contents='Tell me a story in 300 words.',
+    ):
+      chunks += 1
+      assert chunk.text is not None or chunk.candidates[0].finish_reason
+    assert chunks >= 2
+
+
+def test_start_with_models_stream(client):
+  # vertex ai require publishers/ prefix for gemini
+  with pytest_helper.exception_if_vertex(client, errors.ClientError):
+    chunks = 0
+    for chunk in client.models.generate_content_stream(
+        model='models/gemini-2.5-flash',
+        contents='Tell me a story in 50 words.',
+    ):
+      chunks += 1
+      assert chunk.text is not None or chunk.candidates[0].finish_reason
+    assert chunks >= 2
+
+
+def test_models_stream_with_non_empty_last_chunk(client):
+  chunks = list(
+      client.models.generate_content_stream(
+          model='gemini-2.5-flash',
+          contents='Tell me a story in 300 words.',
+      )
+  )
+  assert chunks[-1].text
+
+
+@pytest.mark.asyncio
+async def test_start_with_models_stream_async(client):
+  # vertex ai require publishers/ prefix for gemini
+  with pytest_helper.exception_if_vertex(client, errors.ClientError):
+    chunks = 0
+    async for chunk in await client.aio.models.generate_content_stream(
+        model='models/gemini-2.5-flash',
+        contents='Tell me a story in 300 words.',
+    ):
+      chunks += 1
+      assert chunk.text is not None or chunk.candidates[0].finish_reason
+    assert chunks > 2
+
+
+@pytest.mark.asyncio
+async def test_start_with_models_async(client):
+  # vertex ai require publishers/ prefix for gemini
+  with pytest_helper.exception_if_vertex(client, errors.ClientError):
+    await client.aio.models.generate_content(
+        model='models/gemini-2.5-flash',
+        contents='Tell me a story in 50 words.',
+    )