feat: Add empty response for tunings.cancel()

PiperOrigin-RevId: 839451352

Author: speedstorm1

google/genai/tests/tunings/test_cancel.py

@@ -23,7 +23,7 @@
     pytest_helper.TestTableItem(
         name="test_cancel_vertex",
         parameters=genai_types._CancelTuningJobParameters(
-            name="projects/801452371447/locations/us-central1/tuningJobs/6871785856412155904"
+            name="projects/801452371447/locations/us-central1/tuningJobs/2470687341547618304"
         ),
         exception_if_mldev="Not Found",
     ),

google/genai/tunings.py

@@ -110,6 +110,34 @@ def _CancelTuningJobParameters_to_vertex(
   return to_object
 
 
+def _CancelTuningJobResponse_from_mldev(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+    root_object: Optional[Union[dict[str, Any], object]] = 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'])
+    )
+
+  return to_object
+
+
+def _CancelTuningJobResponse_from_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+    root_object: Optional[Union[dict[str, Any], object]] = 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'])
+    )
+
+  return to_object
+
+
 def _CreateTuningJobConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -1310,7 +1338,7 @@ def cancel(
       *,
       name: str,
       config: Optional[types.CancelTuningJobConfigOrDict] = None,
-  ) -> None:
+  ) -> types.CancelTuningJobResponse:
     """Cancels a tuning job.
 
     Args:
@@ -1362,6 +1390,23 @@ def cancel(
         'post', path, request_dict, http_options
     )
 
+    response_dict = {} if not response.body else json.loads(response.body)
+
+    if self._api_client.vertexai:
+      response_dict = _CancelTuningJobResponse_from_vertex(response_dict)
+
+    if not self._api_client.vertexai:
+      response_dict = _CancelTuningJobResponse_from_mldev(response_dict)
+
+    return_value = types.CancelTuningJobResponse._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
+
   def _tune(
       self,
       *,
@@ -1787,7 +1832,7 @@ async def cancel(
       *,
       name: str,
       config: Optional[types.CancelTuningJobConfigOrDict] = None,
-  ) -> None:
+  ) -> types.CancelTuningJobResponse:
     """Cancels a tuning job asynchronously.
 
     Args:
@@ -1839,6 +1884,23 @@ async def cancel(
         'post', path, request_dict, http_options
     )
 
+    response_dict = {} if not response.body else json.loads(response.body)
+
+    if self._api_client.vertexai:
+      response_dict = _CancelTuningJobResponse_from_vertex(response_dict)
+
+    if not self._api_client.vertexai:
+      response_dict = _CancelTuningJobResponse_from_mldev(response_dict)
+
+    return_value = types.CancelTuningJobResponse._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
+
   async def _tune(
       self,
       *,

google/genai/types.py

@@ -11869,6 +11869,26 @@ class _CancelTuningJobParametersDict(TypedDict, total=False):
 ]
 
 
+class CancelTuningJobResponse(_common.BaseModel):
+  """Empty response for tunings.cancel method."""
+
+  sdk_http_response: Optional[HttpResponse] = Field(
+      default=None, description="""Used to retain the full HTTP response."""
+  )
+
+
+class CancelTuningJobResponseDict(TypedDict, total=False):
+  """Empty response for tunings.cancel method."""
+
+  sdk_http_response: Optional[HttpResponseDict]
+  """Used to retain the full HTTP response."""
+
+
+CancelTuningJobResponseOrDict = Union[
+    CancelTuningJobResponse, CancelTuningJobResponseDict
+]
+
+
 class TuningExample(_common.BaseModel):
   """A single example for tuning.