google-genai 1.61.0__py3-none-any.whl → 1.62.0__py3-none-any.whl

google/genai/tests/errors/test_api_error.py

@@ -23,6 +23,7 @@ import pickle
 
 import httpx
 import pytest
+import websockets
 
 from ... import errors
 
@@ -257,6 +258,43 @@ def test_constructor_message_not_present():
   }
 
 
+def test_constructor_with_websocket_connection_closed_error():
+  actual_error = errors.APIError(
+      1007,
+      'At most one response modality can be specified in the setup request.'
+      ' To enable simultaneous transcription and audio output,',
+      None,
+  )
+  assert actual_error.code == 1007
+  assert (
+      actual_error.details
+      == 'At most one response modality can be specified in the setup request.'
+      ' To enable simultaneous transcription and audio output,',
+  )
+  assert actual_error.status == None
+  assert actual_error.message == None
+
+
+def test_raise_for_websocket_connection_closed_error():
+  try:
+    errors.APIError.raise_error(
+        1007,
+        'At most one response modality can be specified in the setup request.'
+        ' To enable simultaneous transcription and audio output,',
+        None,
+    )
+  except errors.APIError as actual_error:
+    assert actual_error.code == 1007
+    assert (
+        actual_error.details
+        == 'At most one response modality can be specified in the setup'
+        ' request.'
+        ' To enable simultaneous transcription and audio output,'
+    )
+    assert actual_error.status == None
+    assert actual_error.message == None
+
+
 def test_raise_for_response_code_exist_json_decoder_error():
   class FakeResponse(httpx.Response):
 

google/genai/tunings.py

@@ -1390,10 +1390,14 @@ class Tunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(response_dict)
+      response_dict = _TuningJob_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     if not self._api_client.vertexai:
-      response_dict = _TuningJob_from_mldev(response_dict)
+      response_dict = _TuningJob_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1452,10 +1456,14 @@ class Tunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _ListTuningJobsResponse_from_vertex(response_dict)
+      response_dict = _ListTuningJobsResponse_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     if not self._api_client.vertexai:
-      response_dict = _ListTuningJobsResponse_from_mldev(response_dict)
+      response_dict = _ListTuningJobsResponse_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.ListTuningJobsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1526,10 +1534,14 @@ class Tunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _CancelTuningJobResponse_from_vertex(response_dict)
+      response_dict = _CancelTuningJobResponse_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     if not self._api_client.vertexai:
-      response_dict = _CancelTuningJobResponse_from_mldev(response_dict)
+      response_dict = _CancelTuningJobResponse_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.CancelTuningJobResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1602,7 +1614,9 @@ class Tunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(response_dict)
+      response_dict = _TuningJob_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1677,7 +1691,9 @@ class Tunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if not self._api_client.vertexai:
-      response_dict = _TuningOperation_from_mldev(response_dict)
+      response_dict = _TuningOperation_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.TuningOperation._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1882,10 +1898,14 @@ class AsyncTunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(response_dict)
+      response_dict = _TuningJob_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     if not self._api_client.vertexai:
-      response_dict = _TuningJob_from_mldev(response_dict)
+      response_dict = _TuningJob_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1946,10 +1966,14 @@ class AsyncTunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _ListTuningJobsResponse_from_vertex(response_dict)
+      response_dict = _ListTuningJobsResponse_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     if not self._api_client.vertexai:
-      response_dict = _ListTuningJobsResponse_from_mldev(response_dict)
+      response_dict = _ListTuningJobsResponse_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.ListTuningJobsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -2020,10 +2044,14 @@ class AsyncTunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _CancelTuningJobResponse_from_vertex(response_dict)
+      response_dict = _CancelTuningJobResponse_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     if not self._api_client.vertexai:
-      response_dict = _CancelTuningJobResponse_from_mldev(response_dict)
+      response_dict = _CancelTuningJobResponse_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.CancelTuningJobResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -2096,7 +2124,9 @@ class AsyncTunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(response_dict)
+      response_dict = _TuningJob_from_vertex(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -2171,7 +2201,9 @@ class AsyncTunings(_api_module.BaseModule):
     response_dict = {} if not response.body else json.loads(response.body)
 
     if not self._api_client.vertexai:
-      response_dict = _TuningOperation_from_mldev(response_dict)
+      response_dict = _TuningOperation_from_mldev(
+          response_dict, None, parameter_model
+      )
 
     return_value = types.TuningOperation._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()

google/genai/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 #
 
-__version__ = '1.61.0' # x-release-please-version
+__version__ = '1.62.0' # x-release-please-version

{google_genai-1.61.0.dist-info → google_genai-1.62.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.61.0
+Version: 1.62.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License-Expression: Apache-2.0

{google_genai-1.61.0.dist-info → google_genai-1.62.0.dist-info}/RECORD

@@ -20,26 +20,26 @@ google/genai/caches.py,sha256=24GhR6Nj-1Bdaey6ff6NNmP5pIjvlO0S1Dv_-CrCtyI,48118
 google/genai/chats.py,sha256=q3_iONHx8l-Y7EG29_f45yjFDdL2VFefUDLMdJtw0sE,16583
 google/genai/client.py,sha256=odJ3erYMO1tkvn_9AOyfLawNX2tZf71fbVzqmmLEjOY,21327
 google/genai/documents.py,sha256=GuYWVgRaYBzHv9aMgcZBrVkFlpBeaAeZsfhgvC_SLTA,16218
-google/genai/errors.py,sha256=2Iux-5iGUJzEB6OWE3R18SckMEvNimOfpNCQIYL6cn8,8118
+google/genai/errors.py,sha256=ch8Vo5LJJFp2fbCQlBGLn7zMk23uPBkXdDvf8b-uYi8,8555
 google/genai/file_search_stores.py,sha256=4EW_hRjkaGYu4gbQyNbiswTWk5nUgooxB5pqMi_WAME,41418
 google/genai/files.py,sha256=iGYDbHkbnYsqZhkHuKMjrABsMhJivtdPM6Qgog5Q08M,39059
 google/genai/interactions.py,sha256=GLpo-rTdxb1JfL3LXE0c52JWaEGloLnPeCI9hoO_M2g,642
-google/genai/live.py,sha256=aBa_JndziEQjxhdpqChmGHaG4m9npOp9Z0HTXnoZdl8,41804
-google/genai/live_music.py,sha256=Y7I7jh5SAKgyjBIMLboH0oTnZJ18uOT2SpRDKURvp94,6783
+google/genai/live.py,sha256=A9nqCw5oTsjtu-7raxavjddhK48jUM1M5G7cqehvzkk,42544
+google/genai/live_music.py,sha256=Tc-dfxvWt7gxcpaCmomtNpFu2RcIaU5xuuwsajkdDHs,7288
 google/genai/local_tokenizer.py,sha256=EKZ72cV2Zfutlo_efMOPnLRNZN4WQe57rD3G80cF340,14109
-google/genai/models.py,sha256=jiQF2S3P5dwQXTNBhG0sxO3yotftpTz60kpu1sxYyA0,239060
+google/genai/models.py,sha256=bruUlkh8A1LnIDaUUrRHL2sBIc0o6sncCmFm-W2ccCs,251648
 google/genai/operations.py,sha256=FXJOnYp2HA5Xqa5mHW67mSa9p1D_EaGGyUoKc09xFPQ,16384
 google/genai/pagers.py,sha256=1i8NXDuKuQznFin5H5-I0hmj6H5YMTYsEJP7S0ITSbY,7158
 google/genai/py.typed,sha256=RsMFoLwBkAvY05t6izop4UHZtqOPLiKp3GkIEizzmQY,40
 google/genai/tokens.py,sha256=4BPW0gGWFeFVk3INkuY2tfREnsrvzQDhouvRI6_F9Q8,12235
-google/genai/tunings.py,sha256=XiFPNyNEOXuc1XHWfFzB58SyRr6xVHf8JdOPM8Ro_BQ,80739
+google/genai/tunings.py,sha256=xXePLiUbXAqxKaQaadQa6JVa0AXX1rb4rj7d2z7pAk8,81395
 google/genai/types.py,sha256=GmIqqe9PHbgUBvtOMvx5YRCzcFarMr_tErPCR3eL4G0,662323
-google/genai/version.py,sha256=rFKjURomIPq57gTDPrrnIFspz7KSUhTGYmdYiRRgC0c,627
+google/genai/version.py,sha256=baXn09WKoxORFH_2FZYVfx2hXcnfVehXLdriWUApne8,627
 google/genai/_interactions/__init__.py,sha256=Zl5wlyZ1be3a1ENIERjj9ocpELL89RdZkHF3MnCkB3M,3571
-google/genai/_interactions/_base_client.py,sha256=9U2GjsnrAEmFEIEWIDylGX9P-c038S9TqzcK2FKoKTw,74460
+google/genai/_interactions/_base_client.py,sha256=eVxa4rIq_ZKWmmVON8Me_SiHpmGr2ugTpDV9zTpZ45w,74709
 google/genai/_interactions/_client.py,sha256=VkBifJIA0IchcNRUUb4E8w02udzmJMedYPXYnqX1iKc,21234
 google/genai/_interactions/_client_adapter.py,sha256=QLQa352-22JgCa72uFo0UGzZr0v-BaTZ4yUVKCQ5LIU,1349
-google/genai/_interactions/_compat.py,sha256=1HosuxO1kuZL6GwDmAoJD5kY999gjys4ifLdhXiJ4KI,7139
+google/genai/_interactions/_compat.py,sha256=eVykYZJNmzZw9idjzze7HdTUYSYhqm9Dl3uI77TiDfo,7204
 google/genai/_interactions/_constants.py,sha256=t7rpey7f7Z65WNYx8OzR-VGkTEHaz_WCLOREzHLjFhk,1039
 google/genai/_interactions/_exceptions.py,sha256=XvtN0Ir69uwzlI9s-2gvXZJyDa3NZ0IqsMapC9nM8Bc,3825
 google/genai/_interactions/_files.py,sha256=K7qgCZWoSaNI4RrdY1cUnNMXNF3Usq1IgCvzqwFgfMQ,4144
@@ -53,6 +53,7 @@ google/genai/_interactions/_version.py,sha256=HYUoiFQBODx2RJNZpIxeUnGFUe8Gc9FEO6
 google/genai/_interactions/_utils/__init__.py,sha256=fNs6SzzZLqVABQ0TgZ3z1LBri3iVIdtvKo0GcTiNXJM,2882
 google/genai/_interactions/_utils/_compat.py,sha256=NfPeqV7YsaW2h40s7NODt-uOngiMLf9KboR0W-f64gU,1794
 google/genai/_interactions/_utils/_datetime_parse.py,sha256=dYUm7Y3pBD4dzQmpr-onIalbkEQH8gSqdSok5wx24wE,4781
+google/genai/_interactions/_utils/_json.py,sha256=miy3kDffBjNQQQyYCVdHiMRU-ymfcXI2nyY7S6_Efzs,1539
 google/genai/_interactions/_utils/_logs.py,sha256=z9QPllBTUdyCkIpb3iHi_NZnM3GTG_OQtD4t1Htpex4,1404
 google/genai/_interactions/_utils/_proxy.py,sha256=V_QfMtbDL4_OFaYynWeyCCkP7KPBuV7GEJspth55rS0,2552
 google/genai/_interactions/_utils/_reflection.py,sha256=F4i9cYP3sGnxZ-dhp3lcFSRiB115znLodOM-gcVh-wA,1941
@@ -221,7 +222,7 @@ google/genai/tests/documents/test_delete.py,sha256=O33JZOTpdhrCRmAudC6YudHpuv46u
 google/genai/tests/documents/test_get.py,sha256=Z-aMd_vxjnxUVDg_4iphCkXJnqDM6ZMaiqa87ip172s,3191
 google/genai/tests/documents/test_list.py,sha256=qgdc__sH88NA8Ptd83BfPsGLaoZntdn0D6eHajr-jl0,2425
 google/genai/tests/errors/__init__.py,sha256=8hdiuOvLIiX_to2O6rc8Af04-85qQuRu62gOhQQyNx8,39
-google/genai/tests/errors/test_api_error.py,sha256=1QgJNSRC-KAIR2Ri7NkKG3uGcU8X-Mr2exL75NvnpME,10954
+google/genai/tests/errors/test_api_error.py,sha256=aeWHvizYE4tXMX_DEaYsH0FTfjQUtkWcNA3jWQ3gF8k,12181
 google/genai/tests/file_search_stores/__init__.py,sha256=IGpNdiCD34aFEuUWM4QyGS-ekZM29FxL1uM2oRP1E4M,644
 google/genai/tests/file_search_stores/test_create.py,sha256=1a90j-1oOYBPF7DPIxgMFblhxLtmo_KBL3eUrCOLpzI,2090
 google/genai/tests/file_search_stores/test_delete.py,sha256=hRfxcMbaB3Q93izLd0JfS4HOu7sttugV0bLmF6GrGIs,2158
@@ -352,8 +353,8 @@ google/genai/tests/types/test_part_type.py,sha256=a_0eE2CgE7Ju6ePmu3DrupsZZCFiyD
 google/genai/tests/types/test_schema_from_json_schema.py,sha256=WiraJx5mRLW8lKHiWsyHTBYge4jVpeCYklbqGd4BkZQ,13894
 google/genai/tests/types/test_schema_json_schema.py,sha256=zn1gjsXcHTztzmajr8YX3Yb6rWnchjgjw6Yj1NZnG-0,15536
 google/genai/tests/types/test_types.py,sha256=Tqc_PpGo9-XH1xfkAjTCM9xngan_4w__hgpA0BkyA8g,84416
-google_genai-1.61.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-1.61.0.dist-info/METADATA,sha256=VvVzYRFsT_8gLRN1hgJmqu5U87Qplzn2ZNy4-7vikDY,53135
-google_genai-1.61.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-google_genai-1.61.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-1.61.0.dist-info/RECORD,,
+google_genai-1.62.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-1.62.0.dist-info/METADATA,sha256=kZdH4KSMC2R7oZBE2eQB6_NqBq6JHaydSeSC4cNKWAM,53135
+google_genai-1.62.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+google_genai-1.62.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-1.62.0.dist-info/RECORD,,