Skip to content

Commit 9326da9

Browse files
wanlin31copybara-github
authored andcommitted
chore: follow up on enabling headers in tuning
PiperOrigin-RevId: 788151847
1 parent 417e655 commit 9326da9

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

google/genai/tunings.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -799,7 +799,9 @@ def _get(
799799
return_value = types.TuningJob._from_response(
800800
response=response_dict, kwargs=parameter_model.model_dump()
801801
)
802-
802+
return_value.sdk_http_response = types.HttpResponse(
803+
headers=response.headers
804+
)
803805
self._api_client._verify_response(return_value)
804806
return return_value
805807

@@ -1147,7 +1149,9 @@ async def _get(
11471149
return_value = types.TuningJob._from_response(
11481150
response=response_dict, kwargs=parameter_model.model_dump()
11491151
)
1150-
1152+
return_value.sdk_http_response = types.HttpResponse(
1153+
headers=response.headers
1154+
)
11511155
self._api_client._verify_response(return_value)
11521156
return return_value
11531157

0 commit comments

Comments
 (0)