feat: return headers for list method in all modules.

PiperOrigin-RevId: 785947079

Author: wanlin31

google/genai/batches.py

@@ -1622,6 +1622,11 @@ def _ListBatchJobsResponse_from_mldev(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -1784,6 +1789,11 @@ def _ListBatchJobsResponse_from_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -2091,7 +2101,9 @@ def _list(
     return_value = types.ListBatchJobsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 
@@ -2522,7 +2534,9 @@ async def _list(
     return_value = types.ListBatchJobsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 

google/genai/caches.py

@@ -1327,6 +1327,11 @@ def _ListCachedContentsResponse_from_mldev(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -1386,6 +1391,11 @@ def _ListCachedContentsResponse_from_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -1773,7 +1783,9 @@ def _list(
     return_value = types.ListCachedContentsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 
@@ -2165,7 +2177,9 @@ async def _list(
     return_value = types.ListCachedContentsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 

google/genai/files.py

@@ -267,6 +267,11 @@ def _ListFilesResponse_from_mldev(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -370,7 +375,9 @@ def _list(
     return_value = types.ListFilesResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 
@@ -848,7 +855,9 @@ async def _list(
     return_value = types.ListFilesResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 

google/genai/models.py

@@ -3737,6 +3737,11 @@ def _ListModelsResponse_from_mldev(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -4499,6 +4504,11 @@ def _ListModelsResponse_from_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
+
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -5261,7 +5271,9 @@ def _list(
     return_value = types.ListModelsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 
@@ -6876,7 +6888,9 @@ async def _list(
     return_value = types.ListModelsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
     )
-
+    return_value.sdk_http_response = types.HttpResponse(
+        headers=response.headers
+    )
     self._api_client._verify_response(return_value)
     return return_value
 

google/genai/pagers.py

@@ -18,7 +18,8 @@
 # pylint: disable=protected-access
 
 import copy
-from typing import Any, AsyncIterator,Awaitable, Callable, Generic, Iterator, Literal, TypeVar
+from typing import Any, AsyncIterator,Awaitable, Callable, Generic, Iterator, Literal, TypeVar, Union
+from . import types
 
 T = TypeVar('T')
 
@@ -43,6 +44,8 @@ def _init_page(
     self._page = getattr(response, self._name) or []
     self._idx = 0
 
+    self._sdk_http_response = getattr(response, 'sdk_http_response', None)
+
     if not config:
       request_config = {}
     elif isinstance(config, dict):
@@ -110,6 +113,13 @@ def page_size(self) -> int:
 
     return self._page_size
 
+  @property
+  def sdk_http_response(self) -> Union[types.HttpResponse, None]:
+    """Returns the http response of the API response.
+    """
+
+    return self._sdk_http_response
+
   @property
   def config(self) -> dict[str, Any]:
     """Returns the configuration when making the API request for the next page.

google/genai/tests/batches/test_list.py

@@ -55,6 +55,7 @@
 
 def test_pager(client):
   batch_jobs = client.batches.list(config={'page_size': 10})
+  assert 'content-type' in batch_jobs.sdk_http_response.headers
   assert batch_jobs.name == 'batch_jobs'
   assert batch_jobs.page_size == 10
   assert len(batch_jobs) <= 10
@@ -68,6 +69,7 @@ def test_pager(client):
 async def test_async_pager(client):
   batch_jobs = await client.aio.batches.list(config={'page_size': 10})
 
+  assert 'Content-Type' in batch_jobs.sdk_http_response.headers
   assert batch_jobs.name == 'batch_jobs'
   assert batch_jobs.page_size == 10
   assert len(batch_jobs) <= 10

google/genai/tests/caches/test_list.py

@@ -40,7 +40,7 @@
 
 def test_pager(client):
   cached_contents = client.caches.list(config={'page_size': 2})
-
+  assert 'content-type' in cached_contents.sdk_http_response.headers
   assert cached_contents.name == 'cached_contents'
   assert cached_contents.page_size == 2
   assert len(cached_contents) <= 2
@@ -56,6 +56,7 @@ def test_pager(client):
 async def test_async_pager(client):
   cached_contents = await client.aio.caches.list(config={'page_size': 2})
 
+  assert 'Content-Type' in cached_contents.sdk_http_response.headers
   assert cached_contents.name == 'cached_contents'
   assert cached_contents.page_size == 2
   assert len(cached_contents) <= 2

google/genai/tests/files/test_list.py

@@ -43,7 +43,7 @@
 def test_pager(client):
   with pytest_helper.exception_if_vertex(client, ValueError):
     files = client.files.list(config={'page_size': 2})
-
+    assert 'content-type' in files.sdk_http_response.headers
     assert files.name == 'files'
     assert files.page_size == 2
     assert len(files) <= 2
@@ -60,6 +60,7 @@ async def test_async_pager(client):
   with pytest_helper.exception_if_vertex(client, ValueError):
     files = await client.aio.files.list(config={'page_size': 2})
 
+    assert 'Content-Type' in files.sdk_http_response.headers
     assert files.name == 'files'
     assert files.page_size == 2
     assert len(files) <= 2

google/genai/tests/models/test_list.py

@@ -92,6 +92,8 @@ def test_base_models_pager(client):
 
   # Iterate through all the pages. Then next_page() should raise an exception.
   for _ in pager:
+    assert 'content-type' in pager.sdk_http_response.headers
+    assert 'content-encoding' in pager.sdk_http_response.headers
     pass
   with pytest.raises(IndexError, match='No more pages to fetch.'):
     pager.next_page()
@@ -193,6 +195,8 @@ def test_empty_api_response_empty_dict_headers(mock_api_client, client):
 async def test_tuned_models_async_pager(client):
   pager = await client.aio.models.list(config={'page_size': 10, 'query_base': False})
 
+  assert 'Content-Type' in pager.sdk_http_response.headers
+  assert 'Content-Encoding' in pager.sdk_http_response.headers
   assert pager.name == 'models'
   assert pager.page_size == 10
   assert len(pager) <= 10

google/genai/tests/tunings/test_list.py

@@ -47,7 +47,7 @@
 
 def test_pager(client):
   tuning_jobs = client.tunings.list(config={'page_size': 2})
-
+  assert 'content-type' in tuning_jobs.sdk_http_response.headers
   assert tuning_jobs.name == 'tuning_jobs'
   assert tuning_jobs.page_size == 2
   assert len(tuning_jobs) <= 2
@@ -63,6 +63,7 @@ def test_pager(client):
 async def test_async_pager(client):
   tuning_jobs = await client.aio.tunings.list(config={'page_size': 2})
 
+  assert 'Content-Type' in tuning_jobs.sdk_http_response.headers
   assert tuning_jobs.name == 'tuning_jobs'
   assert tuning_jobs.page_size == 2
   assert len(tuning_jobs) <= 2