feat: Add model_status to GenerateContentResponse (Gemini API only)

google/genai/_live_converters.py

@@ -1461,6 +1461,9 @@ def _Part_to_mldev(
   if getv(from_object, ['tool_response']) is not None:
     setv(to_object, ['toolResponse'], getv(from_object, ['tool_response']))
 
+  if getv(from_object, ['part_metadata']) is not None:
+    setv(to_object, ['partMetadata'], getv(from_object, ['part_metadata']))
+
   return to_object
 
 
@@ -1522,6 +1525,9 @@ def _Part_to_vertex(
   if getv(from_object, ['tool_response']) is not None:
     raise ValueError('tool_response parameter is not supported in Vertex AI.')
 
+  if getv(from_object, ['part_metadata']) is not None:
+    raise ValueError('part_metadata parameter is not supported in Vertex AI.')
+
   return to_object
 
 

google/genai/_tokens_converters.py

@@ -521,6 +521,9 @@ def _Part_to_mldev(
   if getv(from_object, ['tool_response']) is not None:
     setv(to_object, ['toolResponse'], getv(from_object, ['tool_response']))
 
+  if getv(from_object, ['part_metadata']) is not None:
+    setv(to_object, ['partMetadata'], getv(from_object, ['part_metadata']))
+
   return to_object
 
 

google/genai/batches.py

@@ -1087,6 +1087,9 @@ def _GenerateContentResponse_from_mldev(
   if getv(from_object, ['usageMetadata']) is not None:
     setv(to_object, ['usage_metadata'], getv(from_object, ['usageMetadata']))
 
+  if getv(from_object, ['modelStatus']) is not None:
+    setv(to_object, ['model_status'], getv(from_object, ['modelStatus']))
+
   return to_object
 
 
@@ -1460,6 +1463,9 @@ def _Part_to_mldev(
   if getv(from_object, ['tool_response']) is not None:
     setv(to_object, ['toolResponse'], getv(from_object, ['tool_response']))
 
+  if getv(from_object, ['part_metadata']) is not None:
+    setv(to_object, ['partMetadata'], getv(from_object, ['part_metadata']))
+
   return to_object
 
 

google/genai/caches.py

@@ -703,6 +703,9 @@ def _Part_to_mldev(
   if getv(from_object, ['tool_response']) is not None:
     setv(to_object, ['toolResponse'], getv(from_object, ['tool_response']))
 
+  if getv(from_object, ['part_metadata']) is not None:
+    setv(to_object, ['partMetadata'], getv(from_object, ['part_metadata']))
+
   return to_object
 
 
@@ -764,6 +767,9 @@ def _Part_to_vertex(
   if getv(from_object, ['tool_response']) is not None:
     raise ValueError('tool_response parameter is not supported in Vertex AI.')
 
+  if getv(from_object, ['part_metadata']) is not None:
+    raise ValueError('part_metadata parameter is not supported in Vertex AI.')
+
   return to_object
 
 

google/genai/models.py

@@ -1606,6 +1606,9 @@ def _GenerateContentResponse_from_mldev(
   if getv(from_object, ['usageMetadata']) is not None:
     setv(to_object, ['usage_metadata'], getv(from_object, ['usageMetadata']))
 
+  if getv(from_object, ['modelStatus']) is not None:
+    setv(to_object, ['model_status'], getv(from_object, ['modelStatus']))
+
   return to_object
 
 
@@ -3375,6 +3378,9 @@ def _Part_to_mldev(
   if getv(from_object, ['tool_response']) is not None:
     setv(to_object, ['toolResponse'], getv(from_object, ['tool_response']))
 
+  if getv(from_object, ['part_metadata']) is not None:
+    setv(to_object, ['partMetadata'], getv(from_object, ['part_metadata']))
+
   return to_object
 
 
@@ -3437,6 +3443,9 @@ def _Part_to_vertex(
   if getv(from_object, ['tool_response']) is not None:
     raise ValueError('tool_response parameter is not supported in Vertex AI.')
 
+  if getv(from_object, ['part_metadata']) is not None:
+    raise ValueError('part_metadata parameter is not supported in Vertex AI.')
+
   return to_object
 
 

google/genai/tests/models/test_generate_content_tools.py

@@ -607,7 +607,8 @@ def divide_floats(a: float, b: float) -> float:
         name='test_include_server_side_tool_invocations',
         parameters=types._GenerateContentParameters(
             model='gemini-3.1-pro-preview',
-            contents=t.t_contents('Why is the sky blue?'),
+            contents=t.t_contents(
+                'Use Google Search to tell me about the 1970 world cup match'),
             config=types.GenerateContentConfig(
                 tools=[
                     types.Tool(

google/genai/tunings.py

@@ -1314,6 +1314,20 @@ def _TuningJob_from_vertex(
   if getv(from_object, ['veoTuningSpec']) is not None:
     setv(to_object, ['veo_tuning_spec'], getv(from_object, ['veoTuningSpec']))
 
+  if getv(from_object, ['distillationSamplingSpec']) is not None:
+    setv(
+        to_object,
+        ['distillation_sampling_spec'],
+        getv(from_object, ['distillationSamplingSpec']),
+    )
+
+  if getv(from_object, ['tuningJobMetadata']) is not None:
+    setv(
+        to_object,
+        ['tuning_job_metadata'],
+        getv(from_object, ['tuningJobMetadata']),
+    )
+
   return to_object