diff --git a/google/genai/_gaos/google_genai.py b/google/genai/_gaos/google_genai.py index 7802554f4..302a54802 100644 --- a/google/genai/_gaos/google_genai.py +++ b/google/genai/_gaos/google_genai.py @@ -296,7 +296,7 @@ def get( api_version: Optional[str] = None, include_input: Any = None, last_event_id: Any = None, - stream: Any = False, + stream: Any = None, extra_headers: Optional[Mapping[str, str]] = None, extra_query: Optional[Mapping[str, Any]] = None, timeout: Optional[Union[float, httpx.Timeout]] = None, @@ -304,7 +304,7 @@ def get( interactions.Interaction, eventstreaming.Stream[interactions.InteractionSSEEvent], ]: - stream_bool = bool(_optional_bool(stream, default=False)) + stream_bool = _optional_bool(stream) response = wrap_sdk_call( super().get, id=id, @@ -448,7 +448,7 @@ async def get( api_version: Optional[str] = None, include_input: Any = None, last_event_id: Any = None, - stream: Any = False, + stream: Any = None, extra_headers: Optional[Mapping[str, str]] = None, extra_query: Optional[Mapping[str, Any]] = None, timeout: Optional[Union[float, httpx.Timeout]] = None, @@ -456,7 +456,7 @@ async def get( interactions.Interaction, eventstreaming.AsyncStream[interactions.InteractionSSEEvent], ]: - stream_bool = bool(_optional_bool(stream, default=False)) + stream_bool = _optional_bool(stream) response = await async_wrap_sdk_call( super().get, id=id, diff --git a/google/genai/tests/interactions/test_paths.py b/google/genai/tests/interactions/test_paths.py index 118a8e62d..194640b04 100644 --- a/google/genai/tests/interactions/test_paths.py +++ b/google/genai/tests/interactions/test_paths.py @@ -49,8 +49,22 @@ def test_interactions_paths(mock_auth_default, client): client.interactions.get(id=interaction_id) mock_send.assert_called_once() request = mock_send.call_args[0][0] + assert str(request.url) == f'{expected_base_url}/interactions/{interaction_id}' + + mock_send.reset_mock() + mock_send.return_value = Response(200, request=Request('GET', ''), headers={'content-type': 'application/json'}, content='{"status": "completed"}') + client.interactions.get(id=interaction_id, stream=False) + mock_send.assert_called_once() + request = mock_send.call_args[0][0] assert str(request.url) == f'{expected_base_url}/interactions/{interaction_id}?stream=false' + mock_send.reset_mock() + mock_send.return_value = Response(200, request=Request('GET', ''), headers={'content-type': 'application/json'}, content='{"status": "completed"}') + client.interactions.get(id=interaction_id, stream=True) + mock_send.assert_called_once() + request = mock_send.call_args[0][0] + assert str(request.url) == f'{expected_base_url}/interactions/{interaction_id}?stream=true' + mock_send.reset_mock() mock_send.return_value = Response(200, request=Request('POST', ''), headers={'content-type': 'application/json'}, content='{"status": "completed"}') client.interactions.cancel(id=interaction_id) @@ -90,8 +104,22 @@ async def test_async_interactions_paths(mock_auth_default, client): await client.aio.interactions.get(id=interaction_id) mock_send.assert_called_once() request = mock_send.call_args[0][0] + assert str(request.url) == f'{expected_base_url}/interactions/{interaction_id}' + + mock_send.reset_mock() + mock_send.return_value = Response(200, request=Request('GET', ''), headers={'content-type': 'application/json'}, content='{"status": "completed"}') + await client.aio.interactions.get(id=interaction_id, stream=False) + mock_send.assert_called_once() + request = mock_send.call_args[0][0] assert str(request.url) == f'{expected_base_url}/interactions/{interaction_id}?stream=false' + mock_send.reset_mock() + mock_send.return_value = Response(200, request=Request('GET', ''), headers={'content-type': 'application/json'}, content='{"status": "completed"}') + await client.aio.interactions.get(id=interaction_id, stream=True) + mock_send.assert_called_once() + request = mock_send.call_args[0][0] + assert str(request.url) == f'{expected_base_url}/interactions/{interaction_id}?stream=true' + mock_send.reset_mock() mock_send.return_value = Response(200, request=Request('POST', ''), headers={'content-type': 'application/json'}, content='{"status": "completed"}') await client.aio.interactions.cancel(id=interaction_id)