chore: add httpOptions to LiveConnectConfig

PiperOrigin-RevId: 738166931

Author: happy-qiao

google/genai/live.py

@@ -849,6 +849,10 @@ async def connect(
         async for message in session.receive():
           print(message)
     """
+    # TODO(b/404946806): Remove this check once httpOptions is supported.
+    if config and 'http_options' in config:
+      raise ValueError('http_options is not supported for live connect.')
+
     base_url = self._api_client._websocket_base_url()
     transformed_model = t.t_model(self._api_client, model)
     # Ensure the config is a LiveConnectConfig.

google/genai/tests/live/test_live.py

@@ -899,9 +899,9 @@ def test_parse_client_message_function_response(
       api_client=mock_api_client(vertexai=vertexai), websocket=mock_websocket
   )
   input = types.FunctionResponse(
-    id='test_id',
-    name='test_name',
-    response={'result': 'test_response'},
+      id='test_id',
+      name='test_name',
+      response={'result': 'test_response'},
   )
   result = session._parse_client_message(input)
   assert 'tool_response' in result
@@ -928,11 +928,11 @@ def test_parse_client_message_tool_response_dict_with_only_response(
       api_client=mock_api_client(vertexai=vertexai), websocket=mock_websocket
   )
   input = {
-    'id': 'test_id',
-    'name': 'test_name',
-    'response': {
-        'result': 'test_response',
-    }
+      'id': 'test_id',
+      'name': 'test_name',
+      'response': {
+          'result': 'test_response',
+      },
   }
   result = session._parse_client_message(input)
   assert 'tool_response' in result
@@ -985,3 +985,19 @@ def test_parse_client_message_realtime_tool_response(
           ],
       }
   }
+
+
+@pytest.mark.asyncio
+async def test_connect_with_http_options_throws_error(
+    mock_api_client, mock_websocket
+):
+  with pytest.raises(ValueError, match='http_options is not supported'):
+    await live.AsyncLive(mock_api_client()).connect(
+        model='models/gemini-pro',
+        config=types.LiveConnectConfig(http_options={'proxy': 'test-proxy'}),
+    )
+  with pytest.raises(ValueError, match='http_options is not supported'):
+    await live.AsyncLive(mock_api_client()).connect(
+        model='models/gemini-pro',
+        config=types.LiveConnectConfig({'http_options': {'proxy': 'test-proxy'}}),
+    )

google/genai/types.py

@@ -9129,6 +9129,9 @@ class LiveClientMessageDict(TypedDict, total=False):
 class LiveConnectConfig(_common.BaseModel):
   """Session config for the API connection."""
 
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   generation_config: Optional[GenerationConfig] = Field(
       default=None,
       description="""The generation configuration for the session.""",
@@ -9163,6 +9166,9 @@ class LiveConnectConfig(_common.BaseModel):
 class LiveConnectConfigDict(TypedDict, total=False):
   """Session config for the API connection."""
 
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
   generation_config: Optional[GenerationConfigDict]
   """The generation configuration for the session."""