feat: add LiveServerContent.waiting_for_input flag.

PiperOrigin-RevId: 803169639

Author: MarkDaoust

google/genai/_live_converters.py

@@ -2655,6 +2655,11 @@ def _LiveServerContent_from_mldev(
         ),
     )
 
+  if getv(from_object, ['waitingForInput']) is not None:
+    setv(
+        to_object, ['waiting_for_input'], getv(from_object, ['waitingForInput'])
+    )
+
   return to_object
 
 
@@ -3331,6 +3336,11 @@ def _LiveServerContent_from_vertex(
         ),
     )
 
+  if getv(from_object, ['waitingForInput']) is not None:
+    setv(
+        to_object, ['waiting_for_input'], getv(from_object, ['waitingForInput'])
+    )
+
   return to_object
 
 

google/genai/tests/live/test_live_response.py

@@ -97,7 +97,8 @@ async def test_receive_server_content(mock_websocket, vertexai):
                       "title": "Search results",
                   }
               }]
-          }
+          },
+          "waitingForInput": True,
       }
   })
   mock_websocket.recv.return_value = raw_response_json
@@ -118,6 +119,9 @@ async def test_receive_server_content(mock_websocket, vertexai):
   assert result.server_content.grounding_metadata.web_search_queries == ["test query"]
   assert result.server_content.grounding_metadata.grounding_chunks[0].web.domain == "google.com"
   assert result.server_content.grounding_metadata.grounding_chunks[0].web.title == "Search results"
+
+  assert result.server_content.waiting_for_input is True
+
   # Verify usageMetadata was parsed
   assert isinstance(result.usage_metadata, types.UsageMetadata)
   assert result.usage_metadata.prompt_token_count == 15

google/genai/types.py

@@ -13103,6 +13103,12 @@ class LiveServerContent(_common.BaseModel):
       default=None,
       description="""Metadata related to url context retrieval tool.""",
   )
+  waiting_for_input: Optional[bool] = Field(
+      default=None,
+      description="""If true, indicates that the model is not generating content because
+      it is waiting for more input from the user, e.g. because it expects the
+      user to continue talking.""",
+  )
 
 
 class LiveServerContentDict(TypedDict, total=False):
@@ -13148,6 +13154,11 @@ class LiveServerContentDict(TypedDict, total=False):
   url_context_metadata: Optional[UrlContextMetadataDict]
   """Metadata related to url context retrieval tool."""
 
+  waiting_for_input: Optional[bool]
+  """If true, indicates that the model is not generating content because
+      it is waiting for more input from the user, e.g. because it expects the
+      user to continue talking."""
+
 
 LiveServerContentOrDict = Union[LiveServerContent, LiveServerContentDict]