fix: fix AFC logging

PiperOrigin-RevId: 808221859

Author: qiaodev

google/genai/models.py

@@ -6565,10 +6565,6 @@ def generate_content(
       response = self._generate_content(
           model=model, contents=contents, config=parsed_config
       )
-      logger.info(f'AFC remote call {i} is done.')
-      remaining_remote_calls_afc -= 1
-      if remaining_remote_calls_afc == 0:
-        logger.info('Reached max remote calls for automatic function calling.')
 
       function_map = _extra_utils.get_function_map(parsed_config)
       if not function_map:
@@ -6586,6 +6582,11 @@ def generate_content(
       )
       if not func_response_parts:
         break
+      logger.info(f'AFC remote call {i} is done.')
+      remaining_remote_calls_afc -= 1
+      if remaining_remote_calls_afc == 0:
+        logger.info('Reached max remote calls for automatic function calling.')
+
       func_call_content = response.candidates[0].content
       func_response_content = types.Content(
           role='user',
@@ -6703,10 +6704,6 @@ def generate_content_stream(
       response = self._generate_content_stream(
           model=model, contents=contents, config=parsed_config
       )
-      logger.info(f'AFC remote call {i} is done.')
-      remaining_remote_calls_afc -= 1
-      if remaining_remote_calls_afc == 0:
-        logger.info('Reached max remote calls for automatic function calling.')
 
       function_map = _extra_utils.get_function_map(parsed_config)
 
@@ -6756,6 +6753,10 @@ def generate_content_stream(
         break
       if not func_response_parts:
         break
+      logger.info(f'AFC remote call {i} is done.')
+      remaining_remote_calls_afc -= 1
+      if remaining_remote_calls_afc == 0:
+        logger.info('Reached max remote calls for automatic function calling.')
 
       # Append function response parts to contents for the next request.
       if chunk is not None and chunk.candidates is not None:

google/genai/tests/models/test_generate_content_tools.py

@@ -1411,7 +1411,7 @@ def test_afc_logs_to_logger_instance(client, caplog):
           'tools': [divide_integers],
           'automatic_function_calling': {
               'disable': False,
-              'maximum_remote_calls': 2,
+              'maximum_remote_calls': 1,
               'ignore_call_history': True,
           },
       },
@@ -1420,9 +1420,8 @@ def test_afc_logs_to_logger_instance(client, caplog):
     assert log.levelname == 'INFO'
     assert log.name == 'google_genai.models'
 
-  assert 'AFC is enabled with max remote calls: 2' in caplog.text
+  assert 'AFC is enabled with max remote calls: 1' in caplog.text
   assert 'remote call 1 is done' in caplog.text
-  assert 'remote call 2 is done' in caplog.text
   assert 'Reached max remote calls' in caplog.text