google-genai 1.27.0__py3-none-any.whl → 1.29.0__py3-none-any.whl

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, Union
+from typing import Any, AsyncIterator, Awaitable, Callable, Generic, Iterator, Literal, TypeVar, Union
+from . import _common
 from . import types
 
 T = TypeVar('T')
@@ -68,7 +69,7 @@ class _BasePager(Generic[T]):
 
   @property
   def page(self) -> list[T]:
-    """Returns a subset of the entire list of items. 
+    """Returns a subset of the entire list of items.
 
     For the number of items returned, see `pageSize()`.
 
@@ -115,13 +116,12 @@ class _BasePager(Generic[T]):
 
   @property
   def sdk_http_response(self) -> Union[types.HttpResponse, None]:
-    """Returns the http response of the API response.
-    """
+    """Returns the http response of the API response."""
 
     return self._sdk_http_response
 
   @property
-  def config(self) -> dict[str, Any]:
+  def config(self) -> _common.StringDict:
     """Returns the configuration when making the API request for the next page.
 
     A configuration is a set of optional parameters and arguments that can be

google/genai/tokens.py

@@ -27,7 +27,7 @@ from . import types
 logger = logging.getLogger('google_genai.tokens')
 
 
-def _get_field_masks(setup: Dict[str, Any]) -> str:
+def _get_field_masks(setup: _common.StringDict) -> str:
   """Return field_masks"""
   fields = []
   for k, v in setup.items():
@@ -42,9 +42,9 @@ def _get_field_masks(setup: Dict[str, Any]) -> str:
 
 
 def _convert_bidi_setup_to_token_setup(
-    request_dict: dict[str, Any],
+    request_dict: _common.StringDict,
     config: Optional[types.CreateAuthTokenConfigOrDict] = None,
-) -> Dict[str, Any]:
+) -> _common.StringDict:
   """Converts bidiGenerateContentSetup."""
   bidi_setup = request_dict.get('bidiGenerateContentSetup')
   if bidi_setup and bidi_setup.get('setup'):

google/genai/tunings.py

@@ -423,6 +423,11 @@ def _TuningJob_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, ['name']) is not None:
     setv(to_object, ['name'], getv(from_object, ['name']))
 
@@ -532,6 +537,11 @@ def _TuningOperation_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, ['name']) is not None:
     setv(to_object, ['name'], getv(from_object, ['name']))
 
@@ -596,6 +606,11 @@ def _TuningJob_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, ['name']) is not None:
     setv(to_object, ['name'], getv(from_object, ['name']))
 
@@ -784,7 +799,9 @@ class Tunings(_api_module.BaseModule):
     return_value = types.TuningJob._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
 
@@ -918,7 +935,9 @@ class Tunings(_api_module.BaseModule):
     return_value = types.TuningJob._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
 
@@ -987,7 +1006,9 @@ class Tunings(_api_module.BaseModule):
     return_value = types.TuningOperation._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
 
@@ -1128,7 +1149,9 @@ class AsyncTunings(_api_module.BaseModule):
     return_value = types.TuningJob._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
 
@@ -1264,7 +1287,9 @@ class AsyncTunings(_api_module.BaseModule):
     return_value = types.TuningJob._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
 
@@ -1333,7 +1358,9 @@ class AsyncTunings(_api_module.BaseModule):
     return_value = types.TuningOperation._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