feat: allow methods in batch to return headers in sdk_http_response by default

PiperOrigin-RevId: 789392436

Author: wanlin31

google/genai/batches.py

@@ -1648,6 +1648,11 @@ def _DeleteResourceJob_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']))
 
@@ -1815,6 +1820,11 @@ def _DeleteResourceJob_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']))
 
@@ -2186,7 +2196,9 @@ def delete(
     return_value = types.DeleteResourceJob._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
 
@@ -2619,7 +2631,9 @@ async def delete(
     return_value = types.DeleteResourceJob._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/types.py

@@ -10771,6 +10771,9 @@ class _DeleteBatchJobParametersDict(TypedDict, total=False):
 class DeleteResourceJob(_common.BaseModel):
   """The return value of delete operation."""
 
+  sdk_http_response: Optional[HttpResponse] = Field(
+      default=None, description="""Used to retain the full HTTP response."""
+  )
   name: Optional[str] = Field(default=None, description="""""")
   done: Optional[bool] = Field(default=None, description="""""")
   error: Optional[JobError] = Field(default=None, description="""""")
@@ -10779,6 +10782,9 @@ class DeleteResourceJob(_common.BaseModel):
 class DeleteResourceJobDict(TypedDict, total=False):
   """The return value of delete operation."""
 
+  sdk_http_response: Optional[HttpResponseDict]
+  """Used to retain the full HTTP response."""
+
   name: Optional[str]
   """"""