google-genai 1.10.0__py3-none-any.whl → 1.11.0__py3-none-any.whl

google/genai/models.py

@@ -176,6 +176,20 @@ def _Schema_to_mldev(
   return to_object
 
 
+def _ModelSelectionConfig_to_mldev(
+    api_client: BaseApiClient,
+    from_object: Union[dict, object],
+    parent_object: Optional[dict] = None,
+) -> dict:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['feature_selection_preference']) is not None:
+    raise ValueError(
+        'feature_selection_preference parameter is not supported in Gemini API.'
+    )
+
+  return to_object
+
+
 def _SafetySetting_to_mldev(
     api_client: BaseApiClient,
     from_object: Union[dict, object],
@@ -393,6 +407,9 @@ def _SpeechConfig_to_mldev(
         ),
     )
 
+  if getv(from_object, ['language_code']) is not None:
+    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
+
   return to_object
 
 
@@ -497,6 +514,11 @@ def _GenerateContentConfig_to_mldev(
   if getv(from_object, ['routing_config']) is not None:
     raise ValueError('routing_config parameter is not supported in Gemini API.')
 
+  if getv(from_object, ['model_selection_config']) is not None:
+    raise ValueError(
+        'model_selection_config parameter is not supported in Gemini API.'
+    )
+
   if getv(from_object, ['safety_settings']) is not None:
     setv(
         parent_object,
@@ -1270,6 +1292,22 @@ def _Schema_to_vertex(
   return to_object
 
 
+def _ModelSelectionConfig_to_vertex(
+    api_client: BaseApiClient,
+    from_object: Union[dict, object],
+    parent_object: Optional[dict] = None,
+) -> dict:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['feature_selection_preference']) is not None:
+    setv(
+        to_object,
+        ['featureSelectionPreference'],
+        getv(from_object, ['feature_selection_preference']),
+    )
+
+  return to_object
+
+
 def _SafetySetting_to_vertex(
     api_client: BaseApiClient,
     from_object: Union[dict, object],
@@ -1493,6 +1531,9 @@ def _SpeechConfig_to_vertex(
         ),
     )
 
+  if getv(from_object, ['language_code']) is not None:
+    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
+
   return to_object
 
 
@@ -1597,6 +1638,15 @@ def _GenerateContentConfig_to_vertex(
   if getv(from_object, ['routing_config']) is not None:
     setv(to_object, ['routingConfig'], getv(from_object, ['routing_config']))
 
+  if getv(from_object, ['model_selection_config']) is not None:
+    setv(
+        to_object,
+        ['modelConfig'],
+        _ModelSelectionConfig_to_vertex(
+            api_client, getv(from_object, ['model_selection_config']), to_object
+        ),
+    )
+
   if getv(from_object, ['safety_settings']) is not None:
     setv(
         parent_object,
@@ -2659,6 +2709,16 @@ def _GenerateVideosParameters_to_vertex(
   return to_object
 
 
+def _FeatureSelectionPreference_to_mldev_enum_validate(enum_value: Any):
+  if enum_value in set([
+      'FEATURE_SELECTION_PREFERENCE_UNSPECIFIED',
+      'PRIORITIZE_QUALITY',
+      'BALANCED',
+      'PRIORITIZE_COST',
+  ]):
+    raise ValueError(f'{enum_value} enum value is not supported in Gemini API.')
+
+
 def _SafetyFilterLevel_to_mldev_enum_validate(enum_value: Any):
   if enum_value in set(['BLOCK_NONE']):
     raise ValueError(f'{enum_value} enum value is not supported in Gemini API.')
@@ -6433,7 +6493,9 @@ class AsyncModels(_api_module.BaseModule):
       if remaining_remote_calls_afc == 0:
         logger.info('Reached max remote calls for automatic function calling.')
 
-      function_map = _extra_utils.get_function_map(config)
+      function_map = _extra_utils.get_function_map(
+          config, is_caller_method_async=True
+      )
       if not function_map:
         break
       if not response:
@@ -6444,8 +6506,10 @@ class AsyncModels(_api_module.BaseModule):
           or not response.candidates[0].content.parts
       ):
         break
-      func_response_parts = _extra_utils.get_function_response_parts(
-          response, function_map
+      func_response_parts = (
+          await _extra_utils.get_function_response_parts_async(
+              response, function_map
+          )
       )
       if not func_response_parts:
         break
@@ -6565,7 +6629,9 @@ class AsyncModels(_api_module.BaseModule):
               'Reached max remote calls for automatic function calling.'
           )
 
-        function_map = _extra_utils.get_function_map(config)
+        function_map = _extra_utils.get_function_map(
+            config, is_caller_method_async=True
+        )
 
         if i == 1:
           # First request gets a function call.
@@ -6581,8 +6647,10 @@ class AsyncModels(_api_module.BaseModule):
                   or not chunk.candidates[0].content.parts
               ):
                 break
-              func_response_parts = _extra_utils.get_function_response_parts(
-                  chunk, function_map
+              func_response_parts = (
+                  await _extra_utils.get_function_response_parts_async(
+                      chunk, function_map
+                  )
               )
               if not func_response_parts:
                 yield chunk