test: for non-stream generate_content, automatic function calling is only available via Chat.send_message. Direct call to Model.generate_content does not support automatic function calling anymore.

PiperOrigin-RevId: 929461973

Author: yyyu-google

google/genai/tests/afc/test_generate_content_stream_afc.py

@@ -21,6 +21,14 @@
 from ... import types
 
 
+pytestmark = [
+    pytest.mark.skipif(
+        "config.getoption('--private')",
+        reason="AFC logic in private SDK is re-written",
+    ),
+]
+
+
 TEST_NO_AFC_PART = types.Part(
     text=(
         'Okay, here is the weather in San Francisco'

google/genai/tests/afc/test_should_disable_afc.py

@@ -17,9 +17,17 @@
 """Tests for should_disable_afc."""
 
 import pytest
+from .. import pytest_helper
 from ... import types
 from ..._extra_utils import should_disable_afc
 
+pytestmark = [
+    pytest.mark.skipif(
+        "config.getoption('--private')",
+        reason="AFC re-written for private SDK",
+    ),
+]
+
 
 def test_config_is_none():
   assert should_disable_afc(None) is False

google/genai/tests/chats/test_get_history.py

@@ -562,6 +562,10 @@ def test_chat_stream_with_empty_content(
   assert not chat.get_history(curated=True)
 
 
+@pytest.mark.skipif(
+    'config.getoption("--private")',
+    reason='AFC logic in private is re-written',
+)
 def test_chat_with_afc_history(mock_generate_content_afc_history):
   models_module = models.Models(mock_api_client)
   chats_module = chats.Chats(modules=models_module)

google/genai/tests/chats/test_send_message.py

@@ -40,10 +40,16 @@
     raise e
 
 
-pytestmark = pytest_helper.setup(
-    file=__file__,
-    globals_for_file=globals(),
-)
+pytestmark = [
+    pytest_helper.setup(
+        file=__file__,
+        globals_for_file=globals(),
+    ),
+    pytest.mark.skipif(
+        "config.getoption('--private')",
+        reason="AFC re-written for private SDK",
+    ),
+]
 pytest_plugins = ('pytest_asyncio',)
 
 

google/genai/tests/chats/test_validate_response.py

@@ -14,8 +14,10 @@
 #
 
 
+from ... import chats
 from ... import types
-from ...chats import _validate_response
+
+_validate_response = chats._validate_response
 
 
 def test_validate_response_default_response():

google/genai/tests/models/test_function_call_streaming.py

@@ -148,6 +148,11 @@
     test_method='models.generate_content_stream',
 )
 
+
+@pytest.mark.skipif(
+    'config.getoption("--private")',
+    reason='in private it was not able to find the replay file',
+)
 def test_streaming_with_python_native_no_afc_config(client):
   """Tests streaming function calls with native python AFC without disabling AFC."""
   if not client.vertexai:
@@ -173,6 +178,10 @@ def test_streaming_with_python_native_no_afc_config(client):
   assert 'not compatible with automatic function calling (AFC)' in str(e.value)
 
 
+@pytest.mark.skipif(
+    'config.getoption("--private")',
+    reason='in private it was not able to find the replay file',
+)
 def test_streaming_with_python_afc_disabled_false(client):
   """Tests streaming function calls with native python AFC without disabling AFC."""
   if not client.vertexai:

google/genai/tests/models/test_generate_content.py

@@ -2159,6 +2159,10 @@ class Foo(BaseModel):
     )
 
 
+@pytest.mark.skipif(
+    'config.getoption("--private")',
+    reason='AFC removed from private models.py',
+)
 def test_function(client):
   def get_weather(city: str) -> str:
     """Returns the weather in a city."""

google/genai/tests/models/test_generate_content_mcp.py

@@ -105,6 +105,10 @@ async def test_mcp_tools_with_custom_headers_async(client):
   }
 
 
+@pytest.mark.skipif(
+    'config.getoption("--private")',
+    reason='AFC by default is disabled in private models.py',
+)
 @pytest.mark.asyncio
 async def test_mcp_tools_subsequent_calls_async(client):
   class MockMcpClientSession(McpClientSession):