google-genai 1.19.0__py3-none-any.whl → 1.21.0__py3-none-any.whl

google/genai/live.py

@@ -79,6 +79,7 @@ _FUNCTION_RESPONSE_REQUIRES_ID = (
     ' response of a ToolCall.FunctionalCalls in Google AI.'
 )
 
+
 class AsyncSession:
   """[Preview] AsyncSession."""
 
@@ -215,11 +216,11 @@ class AsyncSession:
 
     if self._api_client.vertexai:
       client_content_dict = live_converters._LiveClientContent_to_vertex(
-          api_client=self._api_client, from_object=client_content
+          from_object=client_content
       )
     else:
       client_content_dict = live_converters._LiveClientContent_to_mldev(
-          api_client=self._api_client, from_object=client_content
+          from_object=client_content
       )
 
     await self._ws.send(json.dumps({'client_content': client_content_dict}))
@@ -313,13 +314,13 @@ class AsyncSession:
     if self._api_client.vertexai:
       realtime_input_dict = (
           live_converters._LiveSendRealtimeInputParameters_to_vertex(
-              api_client=self._api_client, from_object=realtime_input
+              from_object=realtime_input
           )
       )
     else:
       realtime_input_dict = (
           live_converters._LiveSendRealtimeInputParameters_to_mldev(
-              api_client=self._api_client, from_object=realtime_input
+              from_object=realtime_input
           )
       )
     realtime_input_dict = _common.convert_to_dict(realtime_input_dict)
@@ -399,11 +400,11 @@ class AsyncSession:
     tool_response = t.t_tool_response(function_responses)
     if self._api_client.vertexai:
       tool_response_dict = live_converters._LiveClientToolResponse_to_vertex(
-          api_client=self._api_client, from_object=tool_response
+          from_object=tool_response
       )
     else:
       tool_response_dict = live_converters._LiveClientToolResponse_to_mldev(
-          api_client=self._api_client, from_object=tool_response
+          from_object=tool_response
       )
       for response in tool_response_dict.get('functionResponses', []):
         if response.get('id') is None:
@@ -527,13 +528,9 @@ class AsyncSession:
       response = {}
 
     if self._api_client.vertexai:
-      response_dict = live_converters._LiveServerMessage_from_vertex(
-          self._api_client, response
-      )
+      response_dict = live_converters._LiveServerMessage_from_vertex(response)
     else:
-      response_dict = live_converters._LiveServerMessage_from_mldev(
-          self._api_client, response
-      )
+      response_dict = live_converters._LiveServerMessage_from_mldev(response)
 
     return types.LiveServerMessage._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -647,13 +644,13 @@ class AsyncSession:
           content_input_parts.append(item)
       if self._api_client.vertexai:
         contents = [
-            _Content_to_vertex(self._api_client, item, to_object)
-            for item in t.t_contents(self._api_client, content_input_parts)
+            _Content_to_vertex(item, to_object)
+            for item in t.t_contents(content_input_parts)
         ]
       else:
         contents = [
-            _Content_to_mldev(self._api_client, item, to_object)
-            for item in t.t_contents(self._api_client, content_input_parts)
+            _Content_to_mldev(item, to_object)
+            for item in t.t_contents(content_input_parts)
         ]
 
       content_dict_list: list[types.ContentDict] = []
@@ -1063,7 +1060,7 @@ async def _t_live_connect_config(
   elif isinstance(config, dict):
     if getv(config, ['system_instruction']) is not None:
       converted_system_instruction = t.t_content(
-          api_client, getv(config, ['system_instruction'])
+          getv(config, ['system_instruction'])
       )
     else:
       converted_system_instruction = None
@@ -1073,9 +1070,7 @@ async def _t_live_connect_config(
     if config.system_instruction is None:
       system_instruction = None
     else:
-      system_instruction = t.t_content(
-          api_client, getv(config, ['system_instruction'])
-      )
+      system_instruction = t.t_content(getv(config, ['system_instruction']))
     parameter_model = config
     parameter_model.system_instruction = system_instruction
 

google/genai/live_music.py

@@ -58,7 +58,7 @@ class AsyncMusicSession:
       raise NotImplementedError('Live music generation is not supported in Vertex AI.')
     else:
       client_content_dict = live_converters._LiveMusicClientContent_to_mldev(
-          api_client=self._api_client, from_object={'weighted_prompts': prompts}
+          from_object={'weighted_prompts': prompts}
       )
     await self._ws.send(json.dumps({'clientContent': client_content_dict}))
 
@@ -70,7 +70,7 @@ class AsyncMusicSession:
       raise NotImplementedError('Live music generation is not supported in Vertex AI.')
     else:
       config_dict = live_converters._LiveMusicGenerationConfig_to_mldev(
-          api_client=self._api_client, from_object=config
+          from_object=config
       )
     await self._ws.send(json.dumps({'musicGenerationConfig': config_dict}))
 
@@ -82,7 +82,7 @@ class AsyncMusicSession:
       raise NotImplementedError('Live music generation is not supported in Vertex AI.')
     else:
       playback_control_dict = live_converters._LiveMusicClientMessage_to_mldev(
-          api_client=self._api_client, from_object={'playback_control': playback_control}
+          from_object={'playback_control': playback_control}
       )
       await self._ws.send(json.dumps(playback_control_dict))
 
@@ -135,7 +135,7 @@ class AsyncMusicSession:
       raise NotImplementedError('Live music generation is not supported in Vertex AI.')
     else:
       response_dict = live_converters._LiveMusicServerMessage_from_mldev(
-          self._api_client, response
+          response
       )
 
     return types.LiveMusicServerMessage._from_response(
@@ -173,7 +173,6 @@ class AsyncLiveMusic(_api_module.BaseModule):
       # Only mldev supported
       request_dict = _common.convert_to_dict(
           live_converters._LiveMusicConnectParameters_to_mldev(
-              api_client=self._api_client,
               from_object=types.LiveMusicConnectParameters(
                 model=transformed_model,
               ).model_dump(exclude_none=True)