feat: Support enableWidget feature in GoogleMaps

PiperOrigin-RevId: 817775114

Author: jaycee-li

google/genai/_live_converters.py

@@ -237,6 +237,9 @@ def _GoogleMaps_to_mldev(
   if getv(from_object, ['auth_config']) is not None:
     raise ValueError('auth_config parameter is not supported in Gemini API.')
 
+  if getv(from_object, ['enable_widget']) is not None:
+    setv(to_object, ['enableWidget'], getv(from_object, ['enable_widget']))
+
   return to_object
 
 

google/genai/_tokens_converters.py

@@ -156,6 +156,9 @@ def _GoogleMaps_to_mldev(
   if getv(from_object, ['auth_config']) is not None:
     raise ValueError('auth_config parameter is not supported in Gemini API.')
 
+  if getv(from_object, ['enable_widget']) is not None:
+    setv(to_object, ['enableWidget'], getv(from_object, ['enable_widget']))
+
   return to_object
 
 

google/genai/batches.py

@@ -1024,6 +1024,9 @@ def _GoogleMaps_to_mldev(
   if getv(from_object, ['auth_config']) is not None:
     raise ValueError('auth_config parameter is not supported in Gemini API.')
 
+  if getv(from_object, ['enable_widget']) is not None:
+    setv(to_object, ['enableWidget'], getv(from_object, ['enable_widget']))
+
   return to_object
 
 

google/genai/caches.py

@@ -370,6 +370,9 @@ def _GoogleMaps_to_mldev(
   if getv(from_object, ['auth_config']) is not None:
     raise ValueError('auth_config parameter is not supported in Gemini API.')
 
+  if getv(from_object, ['enable_widget']) is not None:
+    setv(to_object, ['enableWidget'], getv(from_object, ['enable_widget']))
+
   return to_object
 
 

google/genai/models.py

@@ -2441,6 +2441,9 @@ def _GoogleMaps_to_mldev(
   if getv(from_object, ['auth_config']) is not None:
     raise ValueError('auth_config parameter is not supported in Gemini API.')
 
+  if getv(from_object, ['enable_widget']) is not None:
+    setv(to_object, ['enableWidget'], getv(from_object, ['enable_widget']))
+
   return to_object
 
 

google/genai/tests/models/test_generate_content_tools.py

@@ -511,6 +511,20 @@ def divide_floats(a: float, b: float) -> float:
             },
         ),
     ),
+    pytest_helper.TestTableItem(
+        name='test_google_maps_with_enable_widget',
+        parameters=types._GenerateContentParameters(
+            model='gemini-2.5-flash',
+            contents=t.t_contents('What is the nearest airport to Seattle?'),
+            config={
+                'tools': [
+                    {'google_maps': {'enable_widget': True}}
+                ]
+            },
+        ),
+        # TODO(b/450916996): Remove this once the feature is launched in Gemini.
+        exception_if_mldev='400',
+    ),
 ]
 
 

google/genai/types.py

@@ -2823,6 +2823,10 @@ class GoogleMaps(_common.BaseModel):
       default=None,
       description="""Optional. Auth config for the Google Maps tool.""",
   )
+  enable_widget: Optional[bool] = Field(
+      default=None,
+      description="""Optional. If true, include the widget context token in the response.""",
+  )
 
 
 class GoogleMapsDict(TypedDict, total=False):
@@ -2831,6 +2835,9 @@ class GoogleMapsDict(TypedDict, total=False):
   auth_config: Optional[AuthConfigDict]
   """Optional. Auth config for the Google Maps tool."""
 
+  enable_widget: Optional[bool]
+  """Optional. If true, include the widget context token in the response."""
+
 
 GoogleMapsOrDict = Union[GoogleMaps, GoogleMapsDict]
 
@@ -4850,6 +4857,12 @@ class GroundingChunkMapsPlaceAnswerSourcesReviewSnippet(_common.BaseModel):
       default=None,
       description="""A reference representing this place review which may be used to look up this place review again.""",
   )
+  review_id: Optional[str] = Field(
+      default=None, description="""Id of the review referencing the place."""
+  )
+  title: Optional[str] = Field(
+      default=None, description="""Title of the review."""
+  )
 
 
 class GroundingChunkMapsPlaceAnswerSourcesReviewSnippetDict(
@@ -4874,6 +4887,12 @@ class GroundingChunkMapsPlaceAnswerSourcesReviewSnippetDict(
   review: Optional[str]
   """A reference representing this place review which may be used to look up this place review again."""
 
+  review_id: Optional[str]
+  """Id of the review referencing the place."""
+
+  title: Optional[str]
+  """Title of the review."""
+
 
 GroundingChunkMapsPlaceAnswerSourcesReviewSnippetOrDict = Union[
     GroundingChunkMapsPlaceAnswerSourcesReviewSnippet,
@@ -5234,6 +5253,39 @@ class SearchEntryPointDict(TypedDict, total=False):
 SearchEntryPointOrDict = Union[SearchEntryPoint, SearchEntryPointDict]
 
 
+class GroundingMetadataSourceFlaggingUri(_common.BaseModel):
+  """Source content flagging uri for a place or review.
+
+  This is currently populated only for Google Maps grounding.
+  """
+
+  flag_content_uri: Optional[str] = Field(
+      default=None,
+      description="""A link where users can flag a problem with the source (place or review).""",
+  )
+  source_id: Optional[str] = Field(
+      default=None, description="""Id of the place or review."""
+  )
+
+
+class GroundingMetadataSourceFlaggingUriDict(TypedDict, total=False):
+  """Source content flagging uri for a place or review.
+
+  This is currently populated only for Google Maps grounding.
+  """
+
+  flag_content_uri: Optional[str]
+  """A link where users can flag a problem with the source (place or review)."""
+
+  source_id: Optional[str]
+  """Id of the place or review."""
+
+
+GroundingMetadataSourceFlaggingUriOrDict = Union[
+    GroundingMetadataSourceFlaggingUri, GroundingMetadataSourceFlaggingUriDict
+]
+
+
 class GroundingMetadata(_common.BaseModel):
   """Metadata returned to client when grounding is enabled."""
 
@@ -5259,6 +5311,12 @@ class GroundingMetadata(_common.BaseModel):
       default=None,
       description="""Optional. Google search entry for the following-up web searches.""",
   )
+  source_flagging_uris: Optional[list[GroundingMetadataSourceFlaggingUri]] = (
+      Field(
+          default=None,
+          description="""Optional. Output only. List of source flagging uris. This is currently populated only for Google Maps grounding.""",
+      )
+  )
   web_search_queries: Optional[list[str]] = Field(
       default=None,
       description="""Optional. Web search queries for the following-up web search.""",
@@ -5286,6 +5344,9 @@ class GroundingMetadataDict(TypedDict, total=False):
   search_entry_point: Optional[SearchEntryPointDict]
   """Optional. Google search entry for the following-up web searches."""
 
+  source_flagging_uris: Optional[list[GroundingMetadataSourceFlaggingUriDict]]
+  """Optional. Output only. List of source flagging uris. This is currently populated only for Google Maps grounding."""
+
   web_search_queries: Optional[list[str]]
   """Optional. Web search queries for the following-up web search."""