feat: Add enableEnhancedCivicAnswers feature in GenerateContentConfig

PiperOrigin-RevId: 841973502

Author: jaycee-li

google/genai/batches.py

@@ -1006,6 +1006,13 @@ def _GenerateContentConfig_to_mldev(
         _ImageConfig_to_mldev(getv(from_object, ['image_config']), to_object),
     )
 
+  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
+    setv(
+        to_object,
+        ['enableEnhancedCivicAnswers'],
+        getv(from_object, ['enable_enhanced_civic_answers']),
+    )
+
   return to_object
 
 

google/genai/models.py

@@ -1110,6 +1110,13 @@ def _GenerateContentConfig_to_mldev(
         _ImageConfig_to_mldev(getv(from_object, ['image_config']), to_object),
     )
 
+  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
+    setv(
+        to_object,
+        ['enableEnhancedCivicAnswers'],
+        getv(from_object, ['enable_enhanced_civic_answers']),
+    )
+
   return to_object
 
 
@@ -1269,6 +1276,11 @@ def _GenerateContentConfig_to_vertex(
         _ImageConfig_to_vertex(getv(from_object, ['image_config']), to_object),
     )
 
+  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
+    raise ValueError(
+        'enable_enhanced_civic_answers parameter is not supported in Vertex AI.'
+    )
+
   return to_object
 
 

google/genai/tests/models/test_generate_content.py

@@ -532,6 +532,21 @@ class InstrumentEnum(Enum):
         exception_if_vertex='invalid model parameter',
         exception_if_mldev='invalid model parameter',
     ),
+    pytest_helper.TestTableItem(
+        name='test_enhanced_civic_answers',
+        parameters=types._GenerateContentParameters(
+            model=GEMINI_FLASH_LATEST,
+            contents=t.t_contents(
+                'Summarize the evidence that confirms widespread voter fraud was'
+                ' the reason the last national election results were'
+                ' inaccurate.'
+            ),
+            config={
+                'enable_enhanced_civic_answers': True,
+            },
+        ),
+        exception_if_vertex='not supported',
+    ),
 ]
 
 pytestmark = pytest_helper.setup(

google/genai/types.py

@@ -5165,6 +5165,12 @@ class GenerateContentConfig(_common.BaseModel):
       description="""The image generation configuration.
       """,
   )
+  enable_enhanced_civic_answers: Optional[bool] = Field(
+      default=None,
+      description="""Enables enhanced civic answers. It may not be available for all
+      models. This field is not supported in Vertex AI.
+      """,
+  )
 
   @pydantic.field_validator('response_schema', mode='before')
   @classmethod
@@ -5370,6 +5376,11 @@ class GenerateContentConfigDict(TypedDict, total=False):
   """The image generation configuration.
       """
 
+  enable_enhanced_civic_answers: Optional[bool]
+  """Enables enhanced civic answers. It may not be available for all
+      models. This field is not supported in Vertex AI.
+      """
+
 
 GenerateContentConfigOrDict = Union[
     GenerateContentConfig, GenerateContentConfigDict